refactor: Move callsites to liteque to be behind a plugin

1 files changed, 19 insertions, 14 deletions

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));
   };
 }