Update AT assertion

Consensys · Jan 14, 2025 · 76dbde1 · 76dbde1
1 parent dc13031
commit 76dbde1
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/.../src/test/java/tech/pegasys/web3signer/tests/publickeys/KeyIdentifiersAcceptanceTest.java b/.../src/test/java/tech/pegasys/web3signer/tests/publickeys/KeyIdentifiersAcceptanceTest.java
@@ -164,9 +164,12 @@ public void publicKeysAreRemovedAfterReloadDefault(final KeyType keyType) {
     // reload is async ... assert that the key is removed
     Awaitility.await()
         .atMost(5, SECONDS)
-        .until(
-            () -> signer.callApiPublicKeys(keyType).jsonPath().<List<String>>get("."),
-            containsInAnyOrder(keys[0]));
+        .untilAsserted(
+            () -> {
+              final List<String> publicKeysList =
+                  signer.callApiPublicKeys(keyType).jsonPath().getList(".");
+              assertThat(publicKeysList).containsOnly(keys[0]);
+            });
   }
 
   @ParameterizedTest
@@ -188,9 +191,12 @@ public void publicKeysNotRemovedAfterReloadWithKeepStaleKeysTrue(final KeyType k
     // reload is async ... assert that the keys are not removed
     Awaitility.await()
         .atMost(5, SECONDS)
-        .until(
-            () -> signer.callApiPublicKeys(keyType).jsonPath().<List<String>>get("."),
-            containsInAnyOrder(keys));
+        .untilAsserted(
+            () -> {
+              final List<String> publicKeysList =
+                  signer.callApiPublicKeys(keyType).jsonPath().getList(".");
+              assertThat(publicKeysList).containsExactlyInAnyOrder(keys);
+            });
   }
 
   @ParameterizedTest