RxScheduler

RxScheduler.kt

class RxSchedulersOverrideRule(private val scheduler: Scheduler = TestScheduler()) : TestRule {

    private val handler = Function<Scheduler, Scheduler> { scheduler }

    override fun apply(base: Statement, description: Description): Statement =
        object : Statement() {
            override fun evaluate() {
                RxJavaPlugins.reset()
                RxAndroidPlugins.reset()
                RxJavaPlugins.setIoSchedulerHandler(handler)
                RxJavaPlugins.setNewThreadSchedulerHandler(handler)
                RxJavaPlugins.setComputationSchedulerHandler(handler)
                RxJavaPlugins.setSingleSchedulerHandler(handler)
                RxAndroidPlugins.setMainThreadSchedulerHandler(handler)
                try {
                    base.evaluate()
                } finally {
                    RxJavaPlugins.reset()
                    RxAndroidPlugins.reset()
                }
            }
        }
}