diff options
| -rw-r--r-- | uitest/src/test/java/com/vaadin/tests/applicationcontext/CloseSessionTest.java | 3 |
1 files changed, 3 insertions, 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 index 84f6085bb8..5a6657801d 100644 --- 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 org.junit.Test; 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 class CloseSessionTest extends MultiBrowserTest { @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 class CloseSessionTest extends MultiBrowserTest { @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"); } |