add timeout for assertion

getyourguide · Sep 12, 2023 · 60af8ca · 60af8ca
1 parent 5c7684a
commit 60af8ca
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/uiTestGlaze/src/main/java/com/getyourguide/uitestglazesample/AssertionHelper.kt b/uiTestGlaze/src/main/java/com/getyourguide/uitestglazesample/AssertionHelper.kt
@@ -1,9 +1,14 @@
 package com.getyourguide.uitestglazesample
 
 import androidx.test.uiautomator.UiDevice
+import kotlin.time.Duration.Companion.seconds
 
 internal class AssertionHelper(private val findUiElementHelper: FindUiElementHelper) {
 
+    companion object {
+        private val TIMEOUT_TO_FIND_ELEMENT = 3.seconds
+    }
+
     fun assert(
         assertion: Assertion,
         hierarchy: TreeNode,
@@ -16,6 +21,7 @@ internal class AssertionHelper(private val findUiElementHelper: FindUiElementHel
                         hierarchy = hierarchy,
                         optional = true,
                         device = uiDevice,
+                        timeoutToGetAnUiElement = TIMEOUT_TO_FIND_ELEMENT,
                     ) != null
                 ) {
                     throw IllegalStateException("Assertion: $assertion failed. UiElement is visible")
@@ -28,6 +34,7 @@ internal class AssertionHelper(private val findUiElementHelper: FindUiElementHel
                         hierarchy = hierarchy,
                         optional = true,
                         device = uiDevice,
+                        timeoutToGetAnUiElement = TIMEOUT_TO_FIND_ELEMENT,
                     ) == null
                 ) {
                     throw IllegalStateException("Assertion: $assertion failed. UiElement is not visible")
@@ -40,6 +47,7 @@ internal class AssertionHelper(private val findUiElementHelper: FindUiElementHel
                         hierarchy = hierarchy,
                         optional = true,
                         device = uiDevice,
+                        timeoutToGetAnUiElement = TIMEOUT_TO_FIND_ELEMENT,
                     )?.checked != true
                 ) {
                     throw IllegalStateException("Assertion: $assertion failed. UiElement is not checked")
@@ -51,6 +59,7 @@ internal class AssertionHelper(private val findUiElementHelper: FindUiElementHel
                         hierarchy = hierarchy,
                         optional = true,
                         device = uiDevice,
+                        timeoutToGetAnUiElement = TIMEOUT_TO_FIND_ELEMENT,
                     )?.checked != false
                 ) {
                     throw IllegalStateException("Assertion: $assertion failed. UiElement is checked")
@@ -61,6 +70,7 @@ internal class AssertionHelper(private val findUiElementHelper: FindUiElementHel
                     hierarchy = hierarchy,
                     optional = true,
                     device = uiDevice,
+                    timeoutToGetAnUiElement = TIMEOUT_TO_FIND_ELEMENT,
                 )?.enabled != true
             ) {
                 throw IllegalStateException("Assertion: $assertion failed. UiElement is not enabled")
@@ -71,6 +81,7 @@ internal class AssertionHelper(private val findUiElementHelper: FindUiElementHel
                     hierarchy = hierarchy,
                     optional = true,
                     device = uiDevice,
+                    timeoutToGetAnUiElement = TIMEOUT_TO_FIND_ELEMENT,
                 )?.enabled != false
             ) {
                 throw IllegalStateException("Assertion: $assertion failed. UiElement is enabled")