From b5a0f9e854e7ca4fdc901fa98c2434fd1ebf82d6 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 10 Mar 2026 20:50:04 +0000
Subject: [PATCH] Fix tests

---
 .../chat/__tests__/ChatMessage.test.tsx           | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/components/chat/__tests__/ChatMessage.test.tsx b/src/components/chat/__tests__/ChatMessage.test.tsx
index 41b9f190..097e88c1 100644
--- a/src/components/chat/__tests__/ChatMessage.test.tsx
+++ b/src/components/chat/__tests__/ChatMessage.test.tsx
@@ -181,6 +181,21 @@ vi.mock("lucide-react", () => ({
   ChevronDown: () => <div data-testid="chevron-down-icon">ChevronDown</div>,
 }));
 
+// Mock @trustgraph/react-state hooks
+vi.mock("@trustgraph/react-state", () => ({
+  useSettings: () => ({
+    settings: {
+      featureSwitches: { explainability: false },
+    },
+  }),
+  useExplainabilityStore: () => false,
+}));
+
+// Mock ExplainabilityPanel
+vi.mock("../ExplainabilityPanel", () => ({
+  default: () => null,
+}));
+
 describe("ChatMessage", () => {
   it("should render user message with correct styling", () => {
     const message = {