迁移主协程规则

Question

我想创建 MainCoroutineRule。
但 TestCoroutineScope 自 1.6.0 起已弃用。
我该如何迁移它？

下面

class MainCoroutineRule(
    val dispatcher: TestCoroutineDispatcher =
        TestCoroutineDispatcher()
): TestWatcher(), TestCoroutineScope by TestCoroutineScope(dispatcher) {
    override fun starting(description: Description) {
        super.starting(description)
        Dispatchers.setMain(dispatcher)
    }
    
    override fun finished(description: Description) {
        super.finished(description)
        cleanupTestCoroutines()
        Dispatchers.resetMain()
    }
}

也是我的 MainCoroutineRule，我想在 JUnit5 上使用它。 我应该怎么做才能迁移 TestWatcher？

Answer 1

这是我到目前为止所拥有的，但我没有 cleanupTestCoroutines 的解决方案，

class MainCoroutineRule(
    private val dispatcher: TestDispatcher = StandardTestDispatcher()
) : TestWatcher() {
    override fun starting(description: Description?) {
        super.starting(description)
        Dispatchers.setMain(dispatcher)
    }

    override fun finished(description: Description?) {
        super.finished(description)
        Dispatchers.resetMain()
    }
}

请随意使用更好的解决方案编辑/修改它

Answer 2

我的解决方案并不完美，但这是我最好的。
根据迁移提示，< code>runTest 不支持替换 cleanupTestCoroutines()。
相反，它们支持更好的结构化并发。
因此您无需担心 cleanupTestCoroutines()。

所以，接下来是我的 MainCoroutineExtension。

@OptIn(ExperimentalCoroutinesApi::class)
class MainCoroutineExtension(
    private val testDispatcher: TestDispatcher = StandardTestDispatcher(),
    val testScope: TestScope = TestScope(testDispatcher),
) : BeforeEachCallback, AfterEachCallback {
    override fun beforeEach(context: ExtensionContext) {
        Dispatchers.setMain(testDispatcher)
    }

    override fun afterEach(context: ExtensionContext) {
        Dispatchers.resetMain()
    }

}

@OptIn(ExperimentalCoroutinesApi::class)
fun MainCoroutineExtension.runTest(
    block: suspend TestScope.() -> Unit
) = this.testScope.runTest {
    block()
}

我使用 JUnit 5 和新的协程。
所以我无法使用 runBlockingTest 因为它很快就会被弃用。
如果您打算迁移您的协程，这将对您有所帮助