MyEspressoIdlingResourceRule.kt

class MyEspressoIdlingResourceRule: TestWatcher() {
  private val idlingResource = EspressoIdlingResource.countingIdlingResource
  
  override fun starting(description: Description?) {
    IdlingRegistry.getInstance().register(idlingResource)
    super.starting(description)
  }
  
  override fun finished(description: Description?) {
    IdlingRegistry.getInstance().unregister(idlingResource)
    super.starting(description)
  }
  
}