flaky.kt

flakyBehaviorInterceptor.intercept(action = {
    viewInteraction.perform(ViewAction)
}) {
    var cachedException: Throwable
    var startTime = System.currentTimeMillis()

    do {
        try {
            return action.invoke()
       } catch (e: Throwable) {
            Thread.sleep(intervalMs)
            cachedException = e
        }
    } while(System.currentTimeMillis() - startTime <= timeoutMs)

    throw cachedException
}