test(mcp): fix tests

Author: taltultc

src/tools/__tests__/changeCustomer.test.ts

@@ -25,6 +25,18 @@ vi.mock("../../utils/util.js", () => ({
   })),
 }));
 
+// Mock the validateUser function
+vi.mock("../validateUser.js", () => ({
+  handleValidateUserRequest: vi.fn(() => ({
+    content: [
+      {
+        type: "text",
+        text: "User validation successful:\nDomain: example.com (the domain of the user, make it bold)\nEmail: [email protected]",
+      },
+    ],
+  })),
+}));
+
 describe("changeCustomer", () => {
   beforeEach(() => {
     vi.clearAllMocks();
@@ -74,7 +86,7 @@ describe("changeCustomer", () => {
 
       expect(updateCallback).toHaveBeenCalledWith(newContext);
       expect(createSuccessResponse).toHaveBeenCalledWith(
-        "Customer context successfully changed from 'old-customer' to 'new-customer-123'"
+        "Customer context successfully changed to 'example.com (the domain of the user, make it bold)\nEmail: [email protected]'"
       );
     });
 
@@ -92,7 +104,7 @@ describe("changeCustomer", () => {
 
       expect(updateCallback).toHaveBeenCalledWith(newContext);
       expect(createSuccessResponse).toHaveBeenCalledWith(
-        "Customer context successfully changed to 'new-customer-123'"
+        "Customer context successfully changed to 'example.com (the domain of the user, make it bold)\nEmail: [email protected]'"
       );
     });
 
@@ -109,7 +121,7 @@ describe("changeCustomer", () => {
       );
 
       expect(createSuccessResponse).toHaveBeenCalledWith(
-        "Customer context successfully changed from 'old-customer' to 'new-customer-123'"
+        "Customer context successfully changed to 'example.com (the domain of the user, make it bold)\nEmail: [email protected]'"
       );
     });
 

src/tools/__tests__/dimensions.test.ts

@@ -67,9 +67,9 @@ Type: fixed
       const response = await handleDimensionsRequest(mockArgs, mockToken);
 
       expect(makeDoitRequest).toHaveBeenCalledWith(
-        "https://api.doit.com/analytics/v1/dimensions?filter=type%3Afixed&pageToken=next-page&maxResults=60",
+        "https://api.doit.com/analytics/v1/dimensions?filter=type%3Afixed&pageToken=next-page&maxResults=300",
         mockToken,
-        { method: "GET" }
+        { method: "GET", customerContext: undefined }
       );
       expect(createSuccessResponse).toHaveBeenCalledWith(
         expect.stringContaining("Found 1 dimensions (filtered by: type:fixed)")
@@ -93,9 +93,9 @@ Type: fixed
       const response = await handleDimensionsRequest(mockArgs, mockToken);
 
       expect(makeDoitRequest).toHaveBeenCalledWith(
-        "https://api.doit.com/analytics/v1/dimensions?filter=type%3Ainvalid&maxResults=60",
+        "https://api.doit.com/analytics/v1/dimensions?filter=type%3Ainvalid&maxResults=300",
         mockToken,
-        { method: "GET" }
+        { method: "GET", customerContext: undefined }
       );
       expect(createErrorResponse).toHaveBeenCalledWith(
         "No dimensions found, please check the filter parameter, try without filter if you don't know the exact value of the key"
@@ -117,9 +117,9 @@ Type: fixed
       const response = await handleDimensionsRequest(mockArgs, mockToken);
 
       expect(makeDoitRequest).toHaveBeenCalledWith(
-        "https://api.doit.com/analytics/v1/dimensions?maxResults=60",
+        "https://api.doit.com/analytics/v1/dimensions?maxResults=300",
         mockToken,
-        { method: "GET" }
+        { method: "GET", customerContext: undefined }
       );
       expect(createErrorResponse).toHaveBeenCalledWith(
         "Failed to retrieve dimensions data"

src/tools/__tests__/validateUser.test.ts

@@ -48,13 +48,13 @@ describe("validateUser", () => {
         { appendParams: true, method: "GET" }
       );
       expect(createSuccessResponse).toHaveBeenCalledWith(
-        "User validation successful:\nDomain: example.com\nEmail: [email protected]"
+        "User validation successful:\nDomain: example.com (the domain of the user, make it bold)\nEmail: [email protected]"
       );
       expect(response).toEqual({
         content: [
           {
             type: "text",
-            text: "User validation successful:\nDomain: example.com\nEmail: [email protected]",
+            text: "User validation successful:\nDomain: example.com (the domain of the user, make it bold)\nEmail: [email protected]",
           },
         ],
       });