fun ComposeTestRule.waitUntilDisplayed(
  matcher: SemanticsMatcher,
  timeoutMillis: Long = DEFAULT_WAIT_TIMEOUT
): SemanticsNodeInteraction {
  val deadline = System.currentTimeMillis() + timeoutMillis

  while (true) {
    try {
      return waitUntilExists(matcher).assertIsDisplayed()
    } catch (error: AssertionError) {
      if (System.currentTimeMillis() < deadline) {
        android.os.SystemClock.sleep(50)
      } else {
        throw error
      }
    }
  }
}

checkIsDisplayed