refactor(core): remove Worker constructor

kestra-io · Jun 23, 2024 · 51ec983 · 51ec983
1 parent 0fc39b9
commit 51ec983
Showing 1 changed file with 23 additions and 18 deletions.
diff --git a/src/test/java/io/kestra/plugin/couchbase/TriggerTest.java b/src/test/java/io/kestra/plugin/couchbase/TriggerTest.java
@@ -6,6 +6,7 @@
 import io.kestra.core.repositories.LocalFlowRepositoryLoader;
 import io.kestra.core.runners.Worker;
 import io.kestra.core.schedulers.AbstractScheduler;
+import io.kestra.core.utils.IdUtils;
 import io.kestra.core.utils.TestsUtils;
 import io.kestra.jdbc.runner.JdbcScheduler;
 import io.kestra.core.services.FlowListenersInterface;
@@ -39,6 +40,7 @@ class TriggerTest extends CouchbaseTest {
     @Inject
     private LocalFlowRepositoryLoader localFlowRepositoryLoader;
 
+    @SuppressWarnings("unchecked")
     @Test
     void simpleQueryTrigger() throws Exception {
         Execution execution = triggerFlow();
@@ -52,28 +54,31 @@ protected Execution triggerFlow() throws Exception {
         CountDownLatch queueCount = new CountDownLatch(1);
 
         // scheduler
-        Worker worker = new Worker(applicationContext, 8, null);
-        try (
-            AbstractScheduler scheduler = new JdbcScheduler(
-                this.applicationContext,
-                this.flowListenersService
-            );
-        ) {
-            // wait for execution
-            Flux<Execution> receive = TestsUtils.receive(executionQueue, execution -> {
-                queueCount.countDown();
-                assertThat(execution.getLeft().getFlowId(), is("couchbase-listen"));
-            });
+        try (Worker worker = applicationContext.createBean(Worker.class, IdUtils.create(), 8, null)) {
+            try (
+                AbstractScheduler scheduler = new JdbcScheduler(
+                    this.applicationContext,
+                    this.flowListenersService
+                );
+            ) {
+                // wait for execution
+                Flux<Execution> receive = TestsUtils.receive(executionQueue, execution -> {
+                    queueCount.countDown();
+                    assertThat(execution.getLeft().getFlowId(), is("couchbase-listen"));
+                });
 
-            worker.run();
-            scheduler.run();
+                worker.run();
+                scheduler.run();
 
-            localFlowRepositoryLoader.load(this.getClass().getClassLoader().getResource("flows/couchbase-listen.yml"));
+                localFlowRepositoryLoader.load(this.getClass()
+                    .getClassLoader()
+                    .getResource("flows/couchbase-listen.yml"));
 
-            boolean await = queueCount.await(1, TimeUnit.MINUTES);
-            assertThat(await, is(true));
+                boolean await = queueCount.await(1, TimeUnit.MINUTES);
+                assertThat(await, is(true));
 
-            return receive.blockLast();
+                return receive.blockLast();
+            }
         }
     }
 }