imangazalievm / cleanarchitecturemanifest Goto Github PK

public class RxSchedulersRule implements TestRule {
    private final TestScheduler testScheduler = new TestScheduler();
    private final RxAndroidSchedulersHook androidSchedulersHook = new RxAndroidSchedulersHook() {
        @Override
        public Scheduler getMainThreadScheduler() {
            return Schedulers.trampoline();
        }
    };

    public TestScheduler getTestScheduler() {
        return testScheduler;
    }

    @Override
    public Statement apply(Statement base, Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                RxAndroidPlugins.getInstance().registerSchedulersHook(androidSchedulersHook);
                RxJavaHooks.setOnIOScheduler(scheduler -> testScheduler);
                RxJavaHooks.setOnComputationScheduler(scheduler -> testScheduler);
                RxJavaHooks.setOnNewThreadScheduler(scheduler -> testScheduler);
                base.evaluate();
                RxAndroidPlugins.getInstance().reset();
                RxJavaHooks.reset();
            }
        };
    }
}

public class UseCaseTest {
    @Rule
    public RxSchedulersRule rxSchedulersRule = new RxSchedulersRule();

    private UseCase useCase;

    @Before
    public void setUp() throws Exception {
        useCase = ...
    }

    @Test
    public void some_usecase_test() throws Exception {
        // preparing
        ...

        // testing
        useCase.execute();
        rxSchedulersRule.getTestScheduler().triggerActions();

        // asserting
        ...
    }

}