Chats | tests

feature-chats/src/test/java/com/anytypeio/anytype/feature_chats/DefaultCoroutineTestRule.kt

@@ -0,0 +1,30 @@
+package com.anytypeio.anytype.feature_chats
+
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.ExperimentalCoroutinesApi
+import kotlinx.coroutines.test.StandardTestDispatcher
+import kotlinx.coroutines.test.TestDispatcher
+import kotlinx.coroutines.test.resetMain
+import kotlinx.coroutines.test.setMain
+import org.junit.rules.TestWatcher
+import org.junit.runner.Description
+
+@ExperimentalCoroutinesApi
+class DefaultCoroutineTestRule(
+    val dispatcher: TestDispatcher = StandardTestDispatcher()
+) : TestWatcher() {
+    override fun starting(description: Description) {
+        Dispatchers.setMain(dispatcher)
+    }
+
+    override fun finished(description: Description) {
+        Dispatchers.resetMain()
+    }
+
+    fun advanceTime(millis: Long = 100L) {
+        dispatcher.scheduler.advanceTimeBy(millis)
+    }
+
+    fun advanceUntilIdle() = dispatcher.scheduler.advanceUntilIdle()
+
+}