package io.oxygenhub.android

import kotlinx.coroutines.Dispatchers

import kotlinx.coroutines.ExperimentalCoroutinesApi

import kotlinx.coroutines.test.TestDispatcher

import kotlinx.coroutines.test.UnconfinedTestDispatcher

import kotlinx.coroutines.test.resetMain

import kotlinx.coroutines.test.setMain

import org.junit.rules.TestWatcher

import org.junit.runner.Description

/** viewModelScope runs on Main: a test dispatcher stands in for it. */

@OptIn(ExperimentalCoroutinesApi::class)

class MainDispatcherRule(val dispatcher: TestDispatcher = UnconfinedTestDispatcher()) : TestWatcher() {

    override fun starting(description: Description) = Dispatchers.setMain(dispatcher)

    override fun finished(description: Description) = Dispatchers.resetMain()

}