ComputationSchedulerTestRule implementation

ComputationSchedulerTestRule.kt

class ComputationSchedulerTestRule : TestRule {

    val computationScheduler: TestScheduler = TestScheduler()

    override fun apply(statement: Statement?, p1: Description?): Statement {

        return object : Statement() {
            @Throws(Throwable::class)
            override fun evaluate() {
                RxJavaPlugins.setComputationSchedulerHandler { computationScheduler }
                statement!!.evaluate()
                RxJavaPlugins.reset()
            }
        }
    }
}