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

【问题标题】：Espresso Idling Resource setup while using RxJava in Paging library v2在分页库 v2 中使用 RxJava 时设置 Espresso 空闲资源
【发布时间】：2021-03-19 11:49:55
【问题描述】：

我在使用 Paging library 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

我错过了什么？

【问题讨论】：

标签： android rx-java rx-java2 android-espresso android-paging

【解决方案1】：

我需要覆盖构建器上的 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()

         ...
     }
}

【讨论】：