Simple JUnit TestRule for do something when Observable schedulers change

RxJavaSchedulersChangedRule.kt

package com.carloseduardo.rules

import org.junit.rules.TestRule
import org.junit.runner.Description
import org.junit.runners.model.Statement
import rx.Scheduler
import rx.android.plugins.RxAndroidPlugins
import rx.android.plugins.RxAndroidSchedulersHook
import rx.plugins.RxJavaHooks
import rx.schedulers.Schedulers

class RxJavaSchedulersChangedRule : TestRule {

    override fun apply(base: Statement?, description: Description?): Statement {
        return object: Statement() {

            override fun evaluate() {

                RxJavaHooks.setOnIOScheduler { Schedulers.immediate() }
                RxJavaHooks.setOnComputationScheduler { Schedulers.immediate() }
                RxJavaHooks.setOnNewThreadScheduler { Schedulers.immediate() }
                RxAndroidPlugins.getInstance().reset()
                RxAndroidPlugins.getInstance().registerSchedulersHook(object: RxAndroidSchedulersHook() {
                    override fun getMainThreadScheduler(): Scheduler {
                        return Schedulers.immediate()
                    }
                })

                try {
                    base?.evaluate()
                } finally {
                    RxJavaHooks.reset()
                    RxAndroidPlugins.getInstance().reset()
                }
            }
        }
    }
}