SchedulerTests.kt

import com.quizlet.android.logging.ConsoleLogger
import com.quizlet.android.logging.ConsoleLoggingTree
import io.reactivex.Completable
import io.reactivex.Scheduler
import io.reactivex.schedulers.Schedulers
import io.reactivex.schedulers.TestScheduler
import org.junit.After
import org.junit.Before
import org.junit.Test
import timber.log.Timber

class SchedulerTests {

    @Before
    fun setUp() {
        Timber.plant(ConsoleLoggingTree(ConsoleLogger()))
    }

    @After
    fun tearDown() {
        Timber.uprootAll()
    }

    @Test
    fun trampoline() {
        test(Schedulers.trampoline())
    }

    @Test
    fun io() {
        test(Schedulers.io())
    }

    @Test
    fun testScheduler() {
        test(TestScheduler())
    }

    private fun test(scheduler: Scheduler) {
        Completable.defer {
            Timber.d("[1]")
            Thread.sleep(1000)
            Timber.d("[2]")
            Completable.complete()
        }
                .doOnComplete { Timber.d("[3]") }
                .subscribeOn(scheduler)
                .observeOn(scheduler)
                .test()

        Timber.d("[4]")
    }
}