Custom JUnit Test Rule which allows to override the Main Thread and oder Threads used by RxJava

RxJavaRule.java

import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

import io.reactivex.android.plugins.RxAndroidPlugins;
import io.reactivex.plugins.RxJavaPlugins;
import io.reactivex.schedulers.Schedulers;


public class RxJavaRule implements TestRule {
    @Override
    public Statement apply(Statement base, Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                try {
                    RxAndroidPlugins.setInitMainThreadSchedulerHandler(e -> Schedulers.trampoline());
                    RxJavaPlugins.setIoSchedulerHandler(e -> Schedulers.trampoline());
                    RxJavaPlugins.setInitComputationSchedulerHandler(e -> Schedulers.trampoline());
                    RxJavaPlugins.setInitNewThreadSchedulerHandler(e -> Schedulers.trampoline());
                    RxJavaPlugins.setInitSingleSchedulerHandler(e -> Schedulers.trampoline());
                } catch (Exception e) {
                    throw new RuntimeException("Failed to apply RxSchedulers", e);
                }
            }
        };
    }
}