From 51de3aa628b4211a67b80356d7829e03062aa56b Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 29 Dec 2025 20:17:15 +0000
Subject: [PATCH 1/2] Initial plan


From add04a8fc4d026838f279347fe7c92e78c542ac9 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 29 Dec 2025 20:26:16 +0000
Subject: [PATCH 2/2] Fix FetchDailyIndexTask test to use new JobQueue API

Co-authored-by: sroussey <127349+sroussey@users.noreply.github.com>
---
 src/task/index/FetchDailyIndexTask.test.ts | 43 +++++++++++++++-------
 1 file changed, 29 insertions(+), 14 deletions(-)

diff --git a/src/task/index/FetchDailyIndexTask.test.ts b/src/task/index/FetchDailyIndexTask.test.ts
index b15aa38..ee9e5b5 100644
--- a/src/task/index/FetchDailyIndexTask.test.ts
+++ b/src/task/index/FetchDailyIndexTask.test.ts
@@ -9,12 +9,11 @@ import { afterAll, beforeAll, describe, expect, it, mock } from "bun:test";
 import { FetchDailyIndexTask } from "./FetchDailyIndexTask";
 import { TaskFailedError } from "@workglow/task-graph";
 import { FetchUrlTaskOutput } from "@workglow/tasks";
-import { JobQueue } from "@workglow/job-queue";
+import { JobQueueServer, JobQueueClient, EvenlySpacedRateLimiter } from "@workglow/job-queue";
 import { FetchUrlTaskInput } from "@workglow/tasks";
 import { SecFetchJob } from "../../fetch/SecFetchJob";
 import { SecJobQueueName } from "../../config/Constants";
 import { InMemoryQueueStorage } from "@workglow/storage";
-import { InMemoryRateLimiter } from "@workglow/job-queue";
 import { EnvToDI } from "../../config/EnvToDI";
 import { readFileSync } from "fs";
 import { join } from "path";
@@ -88,26 +87,42 @@ const oldFetch = global.fetch;
 EnvToDI();
 describe("FetchDailyIndexTask", () => {
   let db: any;
+  let server: JobQueueServer<FetchUrlTaskInput, FetchUrlTaskOutput, SecFetchJob>;
 
   beforeAll(() => {
     (global as any).fetch = mockFetch;
-    getTaskQueueRegistry().registerQueue(
-      new JobQueue<FetchUrlTaskInput, FetchUrlTaskOutput, SecFetchJob>(
-        SecJobQueueName,
-        SecFetchJob,
-        {
-          storage: new InMemoryQueueStorage(SecJobQueueName),
-          limiter: new InMemoryRateLimiter({ maxExecutions: 10, windowSizeInSeconds: 1 }),
-          waitDurationInMilliseconds: 1,
-        }
-      )
+    
+    const storage = new InMemoryQueueStorage<FetchUrlTaskInput, FetchUrlTaskOutput>(SecJobQueueName);
+    
+    server = new JobQueueServer<FetchUrlTaskInput, FetchUrlTaskOutput, SecFetchJob>(
+      SecFetchJob,
+      {
+        queueName: SecJobQueueName,
+        storage: storage,
+        limiter: new EvenlySpacedRateLimiter({ maxExecutions: 10, windowSizeInSeconds: 1 }),
+        pollIntervalMs: 1,
+      }
     );
-    getTaskQueueRegistry().startQueues();
+    
+    const client = new JobQueueClient<FetchUrlTaskInput, FetchUrlTaskOutput>({
+      storage: storage,
+      queueName: SecJobQueueName,
+    });
+    
+    client.attach(server);
+    
+    getTaskQueueRegistry().registerQueue({
+      server: server,
+      client: client,
+      storage: storage,
+    });
+    
+    server.start();
   });
 
   afterAll(() => {
     (global as any).fetch = oldFetch;
-    getTaskQueueRegistry().stopQueues();
+    server.stop();
     setTaskQueueRegistry(null);
   });