feat: enhance mode management by mocking ModeManager and adding context for custom modes

Author: mechanicmuthu

src/core/tools/__tests__/newTaskTool.spec.ts

@@ -15,6 +15,32 @@ vi.mock("vscode", () => ({
 vi.mock("../../../shared/modes", () => ({
 	getModeBySlug: vi.fn(),
 	defaultModeSlug: "ask",
+	modes: [
+		{
+			slug: "code",
+			name: "Code Mode",
+			roleDefinition: "Test role definition",
+			groups: ["command", "read", "edit"],
+		},
+		{ slug: "ask", name: "Ask Mode", roleDefinition: "Test role definition", groups: ["command", "read", "edit"] },
+	],
+}))
+
+// Mock ModeManager - must be hoisted before newTaskTool import
+const mockValidateModeSwitch = vi.fn().mockResolvedValue({ isValid: true })
+const mockGetEnabledModes = vi.fn().mockResolvedValue([
+	{ slug: "code", name: "Code Mode" },
+	{ slug: "ask", name: "Ask Mode" },
+])
+
+vi.mock("../../services/ModeManager", () => ({
+	ModeManager: vi.fn().mockImplementation((context, customModesManager) => {
+		console.log("Mock ModeManager constructor called with:", { context, customModesManager })
+		return {
+			validateModeSwitch: mockValidateModeSwitch,
+			getEnabledModes: mockGetEnabledModes,
+		}
+	}),
 }))
 
 vi.mock("../../prompts/responses", () => ({
@@ -79,6 +105,13 @@ const mockCline = {
 			getState: vi.fn(() => ({ customModes: [], mode: "ask" })),
 			handleModeSwitch: vi.fn(),
 			createTask: mockCreateTask,
+			context: {
+				globalState: { get: vi.fn(), update: vi.fn() },
+				workspaceState: { get: vi.fn(), update: vi.fn() },
+			},
+			customModesManager: {
+				getCustomModes: vi.fn().mockResolvedValue([]),
+			},
 		})),
 	},
 }