69 lines
2.0 KiB
Kotlin
69 lines
2.0 KiB
Kotlin
package cash.z.wallet.sdk.sample.demoapp.ui.common
|
|
|
|
import androidx.test.filters.FlakyTest
|
|
import androidx.test.filters.SmallTest
|
|
import cash.z.ecc.android.sdk.demoapp.ui.common.throttle
|
|
import kotlinx.coroutines.ExperimentalCoroutinesApi
|
|
import kotlinx.coroutines.flow.asFlow
|
|
import kotlinx.coroutines.flow.flow
|
|
import kotlinx.coroutines.runBlocking
|
|
import kotlinx.coroutines.test.runTest
|
|
import org.junit.Test
|
|
import kotlin.test.assertTrue
|
|
import kotlin.time.Duration
|
|
import kotlin.time.Duration.Companion.seconds
|
|
import kotlin.time.ExperimentalTime
|
|
import kotlin.time.TimeMark
|
|
import kotlin.time.TimeSource
|
|
|
|
class FlowExtTest {
|
|
|
|
@OptIn(ExperimentalTime::class, ExperimentalCoroutinesApi::class)
|
|
@Test
|
|
@SmallTest
|
|
fun throttle_one_sec() = runTest {
|
|
val timer = TimeSource.Monotonic.markNow()
|
|
val flow = flow {
|
|
while (timer.elapsedNow() <= 5.seconds) {
|
|
emit(1)
|
|
}
|
|
}.throttle(1.seconds)
|
|
|
|
var timeMark: TimeMark? = null
|
|
flow.collect {
|
|
if (timeMark == null) {
|
|
timeMark = TimeSource.Monotonic.markNow()
|
|
} else {
|
|
assert(timeMark!!.elapsedNow() >= 1.seconds)
|
|
timeMark = TimeSource.Monotonic.markNow()
|
|
}
|
|
}
|
|
}
|
|
|
|
@OptIn(ExperimentalTime::class)
|
|
private fun raceConditionTest(duration: Duration): Boolean = runBlocking {
|
|
val flow = (0..1000).asFlow().throttle(duration)
|
|
|
|
val values = mutableListOf<Int>()
|
|
flow.collect {
|
|
values.add(it)
|
|
}
|
|
|
|
return@runBlocking values.zipWithNext().all { it.first <= it.second }
|
|
}
|
|
|
|
@FlakyTest
|
|
@Test
|
|
fun stressTest() = runBlocking {
|
|
for (i in 0..10) {
|
|
assertTrue { raceConditionTest(0.001.seconds) }
|
|
}
|
|
for (i in 0..10) {
|
|
assertTrue { raceConditionTest(0.0001.seconds) }
|
|
}
|
|
for (i in 0..10) {
|
|
assertTrue { raceConditionTest(0.00001.seconds) }
|
|
}
|
|
}
|
|
}
|