Try fix flaky TriggerClusterRoutingTest

neo4j-contrib · Nov 22, 2022 · c412a74 · c412a74
1 parent 006daff
commit c412a74
Showing 1 changed file with 17 additions and 16 deletions.
diff --git a/core/src/test/java/apoc/trigger/TriggerClusterRoutingTest.java b/core/src/test/java/apoc/trigger/TriggerClusterRoutingTest.java
@@ -4,6 +4,7 @@
 import apoc.util.TestContainerUtil;
 import apoc.util.TestcontainersCausalCluster;
 import org.junit.AfterClass;
+import org.junit.Assert;
 import org.junit.BeforeClass;
 import org.junit.Test;
 import org.neo4j.configuration.GraphDatabaseSettings;
@@ -84,26 +85,26 @@ private static void triggerInSysLeaderMemberCommon(String query, String triggerN
                 continue;
             }
             Session session = driver.session(SessionConfig.forDatabase(dbName));
-            session.writeTransaction(tx -> {
-                if (sysIsLeader(tx)) {
-                    tx.run(query, Map.of("name", UUID.randomUUID().toString())).consume();
-                } else {
-                    try {
-                        tx.run(query, Map.of("name", UUID.randomUUID().toString())).consume();
-                        fail("Should fail because of non leader trigger addition");
-                    } catch (Exception e) {
-                        String errorMsg = e.getMessage();
-                        assertTrue("The actual message is: " + errorMsg, errorMsg.contains(triggerNotRoutedError));
-                    }
+            if (sysIsLeader(session)) {
+                final String name = UUID.randomUUID().toString();
+                TestContainerUtil.testCall( session, query,
+                        Map.of("name", name),
+                        row -> Assert.assertEquals(name, row.get("name")) );
+            } else {
+                try {
+                    TestContainerUtil.testCall(session, query,
+                            Map.of("name", UUID.randomUUID().toString()),
+                            row -> fail("Should fail because of non leader trigger addition"));
+                } catch (Exception e) {
+                    String errorMsg = e.getMessage();
+                    assertTrue("The actual message is: " + errorMsg, errorMsg.contains(triggerNotRoutedError));
                 }
-                return null;
-            });
+            }
         }
     }
 
-    private static boolean sysIsLeader(Transaction tx) {
-        final String systemRole = tx.run("CALL dbms.cluster.role('system')")
-                .single().get("role").asString();
+    private static boolean sysIsLeader(Session session) {
+        final String systemRole = TestContainerUtil.singleResultFirstColumn(session, "CALL dbms.cluster.role('system')");
         return "LEADER".equals(systemRole);
     }
 }