在分页库 v2 中使用 RxJava 时设置 Espresso 空闲资源

Espresso Idling Resource setup while using RxJava in Paging library v2

我正在尝试在使用 Paging 库 v2 和 RxJava 时编写 Espresso 测试:

class PageKeyedItemDataSource<T>(
        private val schedulerProvider: BaseSchedulerProvider,
        private val compositeDisposable: CompositeDisposable,
        private val context : Context
) : PageKeyedDataSource<Int, Character>() {

    private var isNext = true

    private val isNetworkAvailable: Observable<Boolean> =
        Observable.fromCallable { context.isNetworkAvailable() }

    override fun fetchItems(page: Int): Observable<PeopleWrapper> =
        wrapEspressoIdlingResource {
            composeObservable { useCase(query, page) }
        }

    override fun loadAfter(params: LoadParams<Int>, callback: LoadCallback<Int, Character>) {
        if (isNext) {
            _networkState.postValue(NetworkState.LOADING)

            isNetworkAvailable.flatMap { fetchItems(it, params.key) }
                .subscribe({
                    _networkState.postValue(NetworkState.LOADED)
                    //clear retry since last request succeeded
                    retry = null
                    if (it.next == null) {
                        isNext = false
                    }
                    callback.onResult(it.wrapper, params.key + 1)
                }) {
                    retry = {
                        loadAfter(params, callback)
                    }
                    initError(it)
                }.also { compositeDisposable.add(it) }
        }
    }

    override fun loadInitial(
        params: LoadInitialParams<Int>, callback: LoadInitialCallback<Int, Character>,
    ) {
        _networkState.postValue(NetworkState.LOADING)
        isNetworkAvailable.flatMap { fetchItems(it, 1) }
            .subscribe({
                _networkState.postValue(NetworkState.LOADED)
                if (it.next == null) {
                    isNext = false
                }
                callback.onResult(it.wrapper, null, 2)
            }) {
                retry = {
                    loadInitial(params, callback)
                }
                initError(it)
            }.also { compositeDisposable.add(it) }
   }
}

这是我的 wrapEspressoIdlingResource :

inline fun <T> wrapEspressoIdlingResource(task: () -> Observable<T>): Observable<T> = task()
    .doOnSubscribe { EspressoIdlingResource.increment() } // App is busy until further notice
    .doFinally { EspressoIdlingResource.decrement() } // Set app as idle.

但它不会等到数据从网络传来。当我Thread.Sleep在数据传递之前,Espresso测试会通过,所以这与我的Idling Resource设置有关。

我相信它可能与 Paging 库有关,因为当我在没有 Paging 库的其他示例中使用 Observable 类型时,此方法非常适用于它们。

完整的源代码可在以下位置获得:https://github.com/AliRezaeiii/StarWarsSearch-Paging

我错过了什么?

我需要覆盖构建器上的 fetchDispatcher :

class BasePageKeyRepository<T>(
        private val scheduler: BaseSchedulerProvider,
) : PageKeyRepository<T> {

      @MainThread
      override fun getItems(): Listing<T> {

         val sourceFactory = getSourceFactory()

         val rxPagedList = RxPagedListBuilder(sourceFactory, PAGE_SIZE)
            .setFetchScheduler(scheduler.io()).buildObservable()

         ...
     }
}