Make CloseSessionTest more stable with waitUntil

vaadin · Mar 23, 2017 · b983e2e · b983e2e
1 parent d26317d
commit b983e2e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/uitest/src/test/java/com/vaadin/tests/applicationcontext/CloseSessionTest.java b/uitest/src/test/java/com/vaadin/tests/applicationcontext/CloseSessionTest.java
@@ -6,6 +6,7 @@
 import com.vaadin.testbench.By;
 import com.vaadin.testbench.elements.ButtonElement;
 import com.vaadin.testbench.elements.NotificationElement;
+import com.vaadin.testbench.elements.UIElement;
 import com.vaadin.tests.tb3.MultiBrowserTest;
 
 public class CloseSessionTest extends MultiBrowserTest {
@@ -23,6 +24,7 @@ public void setup() throws Exception {
     @Test
     public void testCloseAndReopen() {
         clickButton("Close VaadinServiceSession and reopen page");
+        waitUntil(driver -> isElementPresent(UIElement.class));
         assertLogText(2, "4. Same hash as current? false");
         assertLogText(0, "6. Same WrappedSession id? true");
     }
@@ -34,6 +36,7 @@ public void testCloseAndReopen() {
     @Test
     public void testInvalidateHttpSessionAndReopen() {
         clickButton("Invalidate HttpSession and reopen page");
+        waitUntil(driver -> isElementPresent(UIElement.class));
         assertLogText(2, "4. Same hash as current? false");
         assertLogText(0, "6. Same WrappedSession id? false");
     }