diff --git a/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/AbstractCdiIT.java b/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/AbstractCdiIT.java index 79dbab7..ad14dfd 100644 --- a/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/AbstractCdiIT.java +++ b/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/AbstractCdiIT.java @@ -41,6 +41,17 @@ protected String getText(String id) { return findElement(By.id(id)).getText(); } + protected void waitForCount(int expectedCount, String counter) { + waitUntil(driver -> { + try { + assertCountEquals(expectedCount, counter); + return true; + } catch (Exception ex) { + return false; + } + }, 1); + } + protected void assertCountEquals(int expectedCount, String counter) throws IOException { Assert.assertEquals(expectedCount, getCount(counter)); diff --git a/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextCloseIT.java b/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextCloseIT.java index e2d1a2f..1fb6978 100644 --- a/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextCloseIT.java +++ b/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextCloseIT.java @@ -48,6 +48,6 @@ public void vaadinSessionCloseDestroysSessionContext() throws Exception { private void assertDestroyCountEquals(int expectedCount) throws IOException { - assertCountEquals(expectedCount, DESTROY_COUNT); + waitForCount(expectedCount, DESTROY_COUNT); } } diff --git a/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextIT.java b/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextIT.java index 643a0d0..9df150b 100644 --- a/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextIT.java +++ b/integration-tests/common-test-code/src/test/java/com/vaadin/flow/quarkus/it/SessionContextIT.java @@ -83,7 +83,7 @@ private void assertLabelEquals(String expected) { private void assertDestroyCountEquals(int expectedCount) throws IOException { - assertCountEquals(expectedCount, DESTROY_COUNT); + waitForCount(expectedCount, DESTROY_COUNT); } private static Logger getLogger() {