package kotlinx.coroutines

import kotlinx.coroutines.testing.*
import org.jetbrains.kotlinx.lincheck.*
import org.jetbrains.kotlinx.lincheck.strategy.managed.modelchecking.*
import org.jetbrains.kotlinx.lincheck.strategy.stress.*
import org.junit.*

abstract class AbstractLincheckTest {
    open fun <O: Options<O, *>> O.customize(isStressTest: Boolean): O = this
    open fun ModelCheckingOptions.customize(isStressTest: Boolean): ModelCheckingOptions = this
    open fun StressOptions.customize(isStressTest: Boolean): StressOptions = this

    @Test
    fun modelCheckingTest() = ModelCheckingOptions()
        .iterations(20 * stressTestMultiplierSqrt)
        .invocationsPerIteration(1_000 * stressTestMultiplierSqrt)
        .commonConfiguration()
        .customize(isStressTest)
        .check(this::class)

    @Test
    fun stressTest() = StressOptions()
        .iterations(20 * stressTestMultiplierSqrt)
        .invocationsPerIteration(1_000 * stressTestMultiplierSqrt)
        .commonConfiguration()
        .customize(isStressTest)
        .check(this::class)

    private fun <O : Options<O, *>> O.commonConfiguration(): O = this
        .actorsBefore(if (isStressTest) 3 else 1)
        // All the bugs we have discovered so far
        // were reproducible on at most 3 threads
        .threads(3)
        // 3 operations per thread is sufficient,
        // while increasing this number declines
        // the model checking coverage.
        .actorsPerThread(if (isStressTest) 3 else 2)
        .actorsAfter(if (isStressTest) 3 else 0)
        .customize(isStressTest)
}