fix: QueueException on test

kestra-io · Sep 12, 2024 · 27adc76 · 27adc76
1 parent ecfc3c4
commit 27adc76
Show file tree

Hide file tree

Showing 14 changed files with 28 additions and 14 deletions.
diff --git a/src/test/java/io/kestra/plugin/notifications/discord/DiscordExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/discord/DiscordExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -45,7 +46,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/google/GoogleChatExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/google/GoogleChatExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -44,7 +45,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/mail/MailExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/mail/MailExecutionTest.java
@@ -2,6 +2,7 @@
 
 import io.kestra.core.models.executions.Execution;
 import io.kestra.core.models.flows.State;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.core.runners.StandAloneRunner;
@@ -37,7 +38,7 @@ void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void testFlow() throws TimeoutException {
+    void testFlow() throws TimeoutException, QueueException {
         Execution execution = runnerUtils.runOne(null, "io.kestra.tests", "mail");
         assertThat(execution.getTaskRunList(), hasSize(2));
         assertThat(execution.getTaskRunList().get(1).getState().getCurrent(), is(State.Type.FAILED));

diff --git a/src/test/java/io/kestra/plugin/notifications/opsgenie/OpsgenieExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/opsgenie/OpsgenieExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -44,7 +45,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/pagerduty/PagerDutyExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/pagerduty/PagerDutyExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -44,7 +45,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/sendgrid/SendGridMailExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/sendgrid/SendGridMailExecutionTest.java
@@ -2,6 +2,7 @@
 
 import io.kestra.core.models.executions.Execution;
 import io.kestra.core.models.flows.State;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.core.runners.StandAloneRunner;
@@ -37,7 +38,7 @@ void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void testFlow() throws TimeoutException {
+    void testFlow() throws TimeoutException, QueueException {
         Execution execution = runnerUtils.runOne(null, "io.kestra.tests", "sendgrid");
         assertThat(execution.getTaskRunList(), hasSize(2));
         assertThat(execution.getTaskRunList().get(1).getState().getCurrent(), is(State.Type.SUCCESS));

diff --git a/src/test/java/io/kestra/plugin/notifications/sentry/SentryExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/sentry/SentryExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -44,7 +45,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/slack/SlackExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/slack/SlackExecutionTest.java
@@ -1,6 +1,7 @@
 package io.kestra.plugin.notifications.slack;
 
 import com.google.common.collect.ImmutableMap;
+import io.kestra.core.queues.QueueException;
 import io.kestra.plugin.notifications.FakeWebhookController;
 import io.micronaut.context.ApplicationContext;
 import io.micronaut.runtime.server.EmbeddedServer;
@@ -43,7 +44,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/teams/TeamsExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/teams/TeamsExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -43,7 +44,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/telegram/TelegramExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/telegram/TelegramExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.core.runners.StandAloneRunner;
@@ -43,7 +44,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/twilio/TwilioExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/twilio/TwilioExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -44,7 +45,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/whatsapp/WhatsAppExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/whatsapp/WhatsAppExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -45,7 +46,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/zenduty/ZendutyExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/zenduty/ZendutyExecutionTest.java
@@ -2,6 +2,7 @@
 
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.plugin.notifications.FakeWebhookController;
@@ -44,7 +45,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();
 

diff --git a/src/test/java/io/kestra/plugin/notifications/zulip/ZulipExecutionTest.java b/src/test/java/io/kestra/plugin/notifications/zulip/ZulipExecutionTest.java
@@ -3,6 +3,7 @@
 import com.google.common.collect.ImmutableMap;
 import io.kestra.core.junit.annotations.KestraTest;
 import io.kestra.core.models.executions.Execution;
+import io.kestra.core.queues.QueueException;
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.RunnerUtils;
 import io.kestra.core.runners.StandAloneRunner;
@@ -43,7 +44,7 @@ protected void init() throws IOException, URISyntaxException {
     }
 
     @Test
-    void flow() throws TimeoutException {
+    void flow() throws TimeoutException, QueueException {
         EmbeddedServer embeddedServer = applicationContext.getBean(EmbeddedServer.class);
         embeddedServer.start();