MockkedDispatcherTest.kt

internal class MockkedDispatcherTest {

  val dispatcher = TestCoroutineDispatcher()
  val scope = TestCoroutineScope(dispatcher)

  @Before
  fun setUp() {

    Dispatchers.setMain(dispatcher)

    mockkStatic(Dispatchers::class)

    every { Dispatchers.Default } returns dispatcher
    every { Dispatchers.Unconfined } returns dispatcher
    every { Dispatchers.IO } returns dispatcher
  }

  @After
  fun tearDown() {
    unmockkStatic(Dispatchers::class)
    Dispatchers.resetMain()
    scope.cleanupTestCoroutines()
  }

  @Test
  fun `Dispatchers should always return local dispatcher`() = runBlocking {
    withContext(Dispatchers.IO) {
      coroutineContext[ContinuationInterceptor] shouldBe dispatcher
    }
  }
}