Fix unit tests

Author: MarceloRGonc

packages/server/api/test/unit/ai/openops-tools.test.ts

@@ -1,3 +1,5 @@
+import { accessTokenManager } from '../../../src/app/authentication/context/access-token-manager';
+
 const mockBasePath = '/mock/base/path';
 const mockApiBaseUrl = 'http://test-api-url';
 const mockTools = {
@@ -16,6 +18,7 @@ const mockTools = {
 const systemMock = {
   get: jest.fn(),
   getOrThrow: jest.fn(),
+  getNumber: jest.fn(),
 };
 
 const loggerMock = {
@@ -27,8 +30,18 @@ const networkUtlsMock = {
   getInternalApiUrl: jest.fn(),
 };
 
-const createMcpClientMock = jest.fn();
 
+const generateServiceTokenMock = jest.fn();
+jest.mock(
+  '../../../src/app/authentication/context/access-token-manager',
+  () => ({
+    accessTokenManager: {
+      generateServiceToken: generateServiceTokenMock,
+    },
+  }),
+);
+
+const createMcpClientMock = jest.fn();
 jest.mock('ai', () => ({
   experimental_createMCPClient: createMcpClientMock,
 }));
@@ -205,7 +218,7 @@ describe('getOpenOpsTools', () => {
       'utf-8',
     );
 
-    await getOpenOpsTools(mockApp, 'auth-2');
+    await getOpenOpsTools(mockApp, 'auth-1');
     expect(fs.writeFile).toHaveBeenCalledTimes(1);
   });
 
@@ -214,6 +227,7 @@ describe('getOpenOpsTools', () => {
       tools: jest.fn().mockResolvedValue(mockTools),
     };
     createMcpClientMock.mockResolvedValue(mockClient);
+    generateServiceTokenMock.mockResolvedValue('auth-service-token');
 
     const result = await getOpenOpsTools(mockApp, 'test-auth-token');
 
@@ -229,7 +243,7 @@ describe('getOpenOpsTools', () => {
           args: [`${mockBasePath}/main.py`],
           env: expect.objectContaining({
             OPENAPI_SCHEMA_PATH: expect.any(String),
-            AUTH_TOKEN: 'test-auth-token',
+            AUTH_TOKEN: 'auth-service-token',
             API_BASE_URL: mockApiBaseUrl,
             OPENOPS_MCP_SERVER_PATH: mockBasePath,
             LOGZIO_TOKEN: 'test-logzio-token',