From 8d32055485858210252096483bb20533dc8bdf60 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Sun, 14 Sep 2025 18:16:40 +0000
Subject: refactor: Move callsites to liteque to be behind a plugin

---
 packages/trpc/testUtils.ts | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

(limited to 'packages/trpc/testUtils.ts')

diff --git a/packages/trpc/testUtils.ts b/packages/trpc/testUtils.ts
index 1cc4e727..b8fe3c30 100644
--- a/packages/trpc/testUtils.ts
+++ b/packages/trpc/testUtils.ts
@@ -77,20 +77,25 @@ export async function buildTestContext(
 
 export function defaultBeforeEach(seedDB = true) {
   return async (context: object) => {
-    vi.mock("@karakeep/shared/queues", () => ({
-      LinkCrawlerQueue: {
-        enqueue: vi.fn(),
-      },
-      OpenAIQueue: {
-        enqueue: vi.fn(),
-      },
-      SearchIndexingQueue: {
-        enqueue: vi.fn(),
-      },
-      triggerRuleEngineOnEvent: vi.fn(),
-      triggerSearchReindex: vi.fn(),
-      triggerWebhook: vi.fn(),
-    }));
+    vi.mock("@karakeep/shared-server", async (original) => {
+      const mod =
+        (await original()) as typeof import("@karakeep/shared-server");
+      return {
+        ...mod,
+        LinkCrawlerQueue: {
+          enqueue: vi.fn(),
+        },
+        OpenAIQueue: {
+          enqueue: vi.fn(),
+        },
+        SearchIndexingQueue: {
+          enqueue: vi.fn(),
+        },
+        triggerRuleEngineOnEvent: vi.fn(),
+        triggerSearchReindex: vi.fn(),
+        triggerWebhook: vi.fn(),
+      };
+    });
     Object.assign(context, await buildTestContext(seedDB));
   };
 }
-- 
cgit v1.2.3-70-g09d2