RxTestBase.java

import org.junit.After;
import org.junit.Before;
import rx.Scheduler;
import rx.android.plugins.RxAndroidPlugins;
import rx.android.plugins.RxAndroidSchedulersHook;
import rx.schedulers.Schedulers;

/**
 * JUnit Tests.
 *
 * @author <a href="mailto:[email protected]">Jared Burrows</a>
 */
public abstract class RxTestBase extends TestBase {
  @Before public void setUp() throws Exception {
    super.setUp();

    // Android
    RxAndroidPlugins.getInstance().reset();
    RxAndroidPlugins.getInstance().registerSchedulersHook(new RxAndroidSchedulersHook() {
      @Override public Scheduler getMainThreadScheduler() {
        return Schedulers.immediate();
      }
    });
  }

  @After public void tearDown() throws Exception {
    super.tearDown();

    // Android
    RxAndroidPlugins.getInstance().reset();
  }
}