Fixed #535: Kafka connector Neo4jSourceTaskTest.'should throw excepti…

…on' is flaky (#579)
neo4j-contrib · Aug 22, 2023 · 7248f2c · 7248f2c
1 parent 6ab213e
commit 7248f2c
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/kafka-connect-neo4j/src/test/kotlin/streams/kafka/connect/source/Neo4jSourceTaskTest.kt b/kafka-connect-neo4j/src/test/kotlin/streams/kafka/connect/source/Neo4jSourceTaskTest.kt
@@ -290,7 +290,17 @@ class Neo4jSourceTaskTest {
         val totalRecords = 10
         insertRecords(totalRecords)
 
-        task.poll()
+        var exception: ConnectException? = null
+        Assert.assertEventually(ThrowingSupplier<Boolean, Exception> {
+            try {
+                task.poll()
+                false
+            } catch (e: ConnectException) {
+                exception = e
+                true
+            }
+        }, Matchers.equalTo(true), 30, TimeUnit.SECONDS)
+        if (exception != null) throw exception as ConnectException
     }
 
     @Test