fix(jdbc): forEachItem tests that are flaky with SQLServer

kestra-io · Sep 11, 2024 · d40c5b3 · d40c5b3
1 parent ab1f73f
commit d40c5b3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/jdbc-postgres/src/main/java/io/kestra/runner/postgres/PostgresQueue.java b/jdbc-postgres/src/main/java/io/kestra/runner/postgres/PostgresQueue.java
@@ -2,6 +2,7 @@
 
 import com.fasterxml.jackson.core.JsonProcessingException;
 import io.kestra.core.exceptions.DeserializationException;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.utils.Either;
 import io.kestra.jdbc.repository.AbstractJdbcRepository;
 import io.kestra.jdbc.runner.JdbcQueue;

diff --git a/jdbc/src/test/java/io/kestra/jdbc/runner/JdbcRunnerTest.java b/jdbc/src/test/java/io/kestra/jdbc/runner/JdbcRunnerTest.java
@@ -279,27 +279,27 @@ void skipExecution() throws TimeoutException, QueueException, InterruptedExcepti
     }
 
     @Test
-    void forEachItem() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
+    protected void forEachItem() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
         forEachItemCaseTest.forEachItem();
     }
 
     @Test
-    void forEachItemEmptyItems() throws URISyntaxException, IOException, TimeoutException {
+    protected void forEachItemEmptyItems() throws URISyntaxException, IOException, TimeoutException, QueueException {
         forEachItemCaseTest.forEachItemEmptyItems();
     }
 
     @Test
-    void forEachItemNoWait() throws URISyntaxException, IOException, InterruptedException, TimeoutException {
+    protected void forEachItemNoWait() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
         forEachItemCaseTest.forEachItemNoWait();
     }
 
     @Test
-    void forEachItemFailed() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
+    protected void forEachItemFailed() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
         forEachItemCaseTest.forEachItemFailed();
     }
 
     @Test
-    void forEachItemSubflowOutputs() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
+    protected void forEachItemSubflowOutputs() throws URISyntaxException, IOException, InterruptedException, TimeoutException, QueueException {
         forEachItemCaseTest.forEachItemWithSubflowOutputs();
     }