fix: logger mock and error message

Author: RomneyDa

extensions/cli/src/integration/rule-duplication.test.ts

@@ -31,7 +31,9 @@ vi.mock("../auth/workos.js", () => ({
 vi.mock("../util/logger.js", () => ({
   logger: {
     debug: vi.fn(),
+    info: vi.fn(),
     warn: vi.fn(),
+    error: vi.fn(),
   },
 }));
 

extensions/cli/src/services/ChatHistoryService.test.ts

@@ -1,5 +1,5 @@
 import type { ChatHistoryItem } from "core/index.js";
-import { describe, it, expect, beforeEach, vi } from "vitest";
+import { beforeEach, describe, expect, it, vi } from "vitest";
 
 import { ChatHistoryService } from "./ChatHistoryService.js";
 
@@ -18,6 +18,7 @@ vi.mock("../session.js", () => ({
 vi.mock("../util/logger.js", () => ({
   logger: {
     debug: vi.fn(),
+    info: vi.fn(),
     warn: vi.fn(),
     error: vi.fn(),
   },

extensions/cli/src/services/ModelService.workflow-priority.test.ts

@@ -23,6 +23,8 @@ vi.mock("../config.js", () => ({
 vi.mock("../util/logger.js", () => ({
   logger: {
     debug: vi.fn(),
+    info: vi.fn(),
+    warn: vi.fn(),
     error: vi.fn(),
   },
 }));

extensions/cli/src/services/agent-file-integration.test.ts

@@ -25,8 +25,9 @@ vi.mock("../hubLoader.js", () => ({
 vi.mock("../util/logger.js", () => ({
   logger: {
     debug: vi.fn(),
-    error: vi.fn(),
+    info: vi.fn(),
     warn: vi.fn(),
+    error: vi.fn(),
   },
 }));
 
@@ -417,7 +418,7 @@ describe("Agent file Integration Tests", () => {
           authServiceState,
           apiClientState,
         ),
-      ).rejects.toThrow("Network error");
+      ).rejects.toThrow("Failed to load agent from owner/agent");
 
       const agentFileState = agentFileService.getState();
       expect(agentFileState.agentFile).toBeNull();

extensions/cli/src/util/apiClient.test.ts

@@ -25,6 +25,8 @@ vi.mock("../env.js", () => ({
 vi.mock("./logger.js", () => ({
   logger: {
     debug: vi.fn(),
+    info: vi.fn(),
+    warn: vi.fn(),
     error: vi.fn(),
   },
 }));

extensions/cli/src/util/clipboard.test.ts

@@ -42,6 +42,9 @@ vi.mock("path", () => ({
 vi.mock("./logger.js", () => ({
   logger: {
     debug: vi.fn(),
+    info: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
   },
 }));
 

extensions/cli/src/util/tokenizer.test.ts

@@ -17,6 +17,7 @@ import {
 vi.mock("./logger.js", () => ({
   logger: {
     debug: vi.fn(),
+    info: vi.fn(),
     warn: vi.fn(),
     error: vi.fn(),
   },