gistfile1.txt

class TestSchedulerRule : TestRule {
    val testScheduler = TestScheduler()

    @Suppress("UNUSED_ANONYMOUS_PARAMETER")
    override fun apply(base: Statement, d: Description): Statement {
        return object : Statement() {

            @Throws(Throwable::class)
            override fun evaluate() {

                RxJavaPlugins.setIoSchedulerHandler { scheduler -> testScheduler }
                RxJavaPlugins.setComputationSchedulerHandler { scheduler -> testScheduler }
                RxJavaPlugins.setNewThreadSchedulerHandler { scheduler -> testScheduler }
                RxAndroidPlugins.setMainThreadSchedulerHandler { scheduler -> Schedulers.trampoline() }
                RxAndroidPlugins.setInitMainThreadSchedulerHandler { scheduler -> Schedulers.trampoline() }

                try {
                    base.evaluate()
                } finally {
                    RxJavaPlugins.reset()
                    RxAndroidPlugins.reset()
                }
            }
        }
    }
}