Для этого необходимо реализовать интерфейс IdlingResource, который имеет три метода:

getName() – возвращает имя ресурса;

isIdleNow() – возвращает true, если ресурс idle. При реализации этого метода необходимо задать и проверять критерий перехода в состояние idle кастомного IdlingResource.

registerIdleTransitionCallback(callback: IdlingResource.ResourceCallback) – в этом методе предоставляется callback. Этот callback следует сохранить в поле класса, который реализует IdlingResource.
ResourceCallback используется для того, чтобы оповестить тестовый фреймворк о том, что ресурс перешел в состояние idle, до того как вызовется isIdleNow(). Когда IdlingResource становится idle, следует вызывать callback.onTransitionToIdle().