From d3c209af13acce9e5f54f226e5f9cccbc8efbae1 Mon Sep 17 00:00:00 2001
From: LukeGus <bugattiguy527@gmail.com>
Date: Tue, 22 Sep 2026 20:20:47 -0500
Subject: [PATCH] fix: mock the ai status fetch in its activate test

---
 plugins/ai/tests/frontend/activate.test.tsx | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/plugins/ai/tests/frontend/activate.test.tsx b/plugins/ai/tests/frontend/activate.test.tsx
index 814097280..4c4d894da 100644
--- a/plugins/ai/tests/frontend/activate.test.tsx
+++ b/plugins/ai/tests/frontend/activate.test.tsx
@@ -1,4 +1,10 @@
-import { afterEach, describe, expect, it } from "vitest";
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { waitFor } from "@testing-library/react";
+
+vi.mock("../../src/frontend/ai-api", async (importOriginal) => ({
+  ...(await importOriginal<Record<string, unknown>>()),
+  getAiStatus: vi.fn(async () => ({ globallyEnabled: true, enabled: true })),
+}));
 import {
   renderWithApp,
   type RenderedPluginApp,
@@ -50,4 +56,9 @@ describe(`${manifest.id} activate`, () => {
     expect(app.registered.settingsComponents()).toEqual([]);
     expect(app.registered.actions()).toEqual([]);
   });
+
+  it("adds the assistant tab once AI is on for the user", async () => {
+    rendered = await renderWithApp(plugin, { manifest, locales });
+    await waitFor(() => expect(rendered!.registered.tabs()).toEqual(["ai"]));
+  });
 });