@@ -7,6 +7,8 @@ import com.powersync.bucket.BucketPriority
7
7
import com.powersync.bucket.Checkpoint
8
8
import com.powersync.bucket.OpType
9
9
import com.powersync.bucket.OplogEntry
10
+ import com.powersync.testutils.ActiveDatabaseTest
11
+ import com.powersync.testutils.databaseTest
10
12
import com.powersync.testutils.waitFor
11
13
import kotlinx.coroutines.channels.Channel
12
14
import kotlinx.coroutines.test.runTest
@@ -17,7 +19,7 @@ import kotlin.test.assertFalse
17
19
import kotlin.test.assertNull
18
20
import kotlin.test.assertTrue
19
21
20
- class SyncProgressTest : BaseInMemorySyncTest () {
22
+ class SyncProgressTest {
21
23
private var lastOpId = 0
22
24
23
25
@BeforeTest
@@ -37,7 +39,7 @@ class SyncProgressTest : BaseInMemorySyncTest() {
37
39
count = count,
38
40
)
39
41
40
- private suspend fun addDataLine (
42
+ private suspend fun ActiveDatabaseTest. addDataLine (
41
43
bucket : String ,
42
44
amount : Int ,
43
45
) {
@@ -61,7 +63,7 @@ class SyncProgressTest : BaseInMemorySyncTest() {
61
63
)
62
64
}
63
65
64
- private suspend fun addCheckpointComplete (priority : BucketPriority ? = null) {
66
+ private suspend fun ActiveDatabaseTest. addCheckpointComplete (priority : BucketPriority ? = null) {
65
67
if (priority != null ) {
66
68
syncLines.send(
67
69
SyncLine .CheckpointPartiallyComplete (
@@ -102,9 +104,8 @@ class SyncProgressTest : BaseInMemorySyncTest() {
102
104
103
105
@Test
104
106
fun withoutPriorities () =
105
- runTest {
106
- val syncStream = syncStream()
107
- database.connectInternal(syncStream, 1000L )
107
+ databaseTest {
108
+ database.connect(connector)
108
109
109
110
turbineScope {
110
111
val turbine = database.currentStatus.asFlow().testIn(this )
@@ -152,9 +153,8 @@ class SyncProgressTest : BaseInMemorySyncTest() {
152
153
153
154
@Test
154
155
fun interruptedSync () =
155
- runTest {
156
- var syncStream = syncStream()
157
- database.connectInternal(syncStream, 1000L )
156
+ databaseTest {
157
+ database.connect(connector)
158
158
159
159
turbineScope {
160
160
val turbine = database.currentStatus.asFlow().testIn(this )
@@ -182,10 +182,9 @@ class SyncProgressTest : BaseInMemorySyncTest() {
182
182
syncLines.close()
183
183
184
184
// And reconnecting
185
- database = openDb ()
185
+ database = openDatabase ()
186
186
syncLines = Channel ()
187
- syncStream = syncStream()
188
- database.connectInternal(syncStream, 1000L )
187
+ database.connect(connector)
189
188
190
189
turbineScope {
191
190
val turbine = database.currentStatus.asFlow().testIn(this )
@@ -218,9 +217,8 @@ class SyncProgressTest : BaseInMemorySyncTest() {
218
217
219
218
@Test
220
219
fun interruptedSyncWithNewCheckpoint () =
221
- runTest {
222
- var syncStream = syncStream()
223
- database.connectInternal(syncStream, 1000L )
220
+ databaseTest {
221
+ database.connect(connector)
224
222
225
223
turbineScope {
226
224
val turbine = database.currentStatus.asFlow().testIn(this )
@@ -244,10 +242,9 @@ class SyncProgressTest : BaseInMemorySyncTest() {
244
242
// Close and re-connect
245
243
database.close()
246
244
syncLines.close()
247
- database = openDb ()
245
+ database = openDatabase ()
248
246
syncLines = Channel ()
249
- syncStream = syncStream()
250
- database.connectInternal(syncStream, 1000L )
247
+ database.connect(connector)
251
248
252
249
turbineScope {
253
250
val turbine = database.currentStatus.asFlow().testIn(this )
@@ -279,9 +276,8 @@ class SyncProgressTest : BaseInMemorySyncTest() {
279
276
280
277
@Test
281
278
fun differentPriorities () =
282
- runTest {
283
- val syncStream = syncStream()
284
- database.connectInternal(syncStream, 1000L )
279
+ databaseTest {
280
+ database.connect(connector)
285
281
286
282
turbineScope {
287
283
val turbine = database.currentStatus.asFlow().testIn(this )
0 commit comments