@google-cloud/cloud-run-mcp 1.0.0

package/test/test-tools.js

@@ -0,0 +1,393 @@
+import assert from 'node:assert/strict';
+import { describe, it, mock } from 'node:test';
+import esmock from 'esmock';
+
+describe('registerTools', () => {
+  it('should register all tools', async () => {
+    const server = {
+      registerTool: mock.fn(),
+    };
+
+    const { registerTools } = await esmock('../tools.js', {});
+
+    registerTools(server);
+
+    assert.strictEqual(server.registerTool.mock.callCount(), 9);
+    const toolNames = server.registerTool.mock.calls.map(
+      (call) => call.arguments[0]
+    );
+    assert.deepStrictEqual(
+      toolNames.sort(),
+      [
+        'create_project',
+        'deploy_container_image',
+        'deploy_file_contents',
+        'deploy_local_files',
+        'deploy_local_folder',
+        'get_service',
+        'get_service_log',
+        'list_projects',
+        'list_services',
+      ].sort()
+    );
+  });
+
+  describe('list_projects', () => {
+    it('should list projects', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/gcp-projects.js': {
+          listProjects: () =>
+            Promise.resolve([{ id: 'project1' }, { id: 'project2' }]),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'list_projects'
+      ).arguments[2];
+      const result = await handler({});
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Available GCP Projects:\n- project1\n- project2',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('create_project', () => {
+    it('should create a project with a provided id', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/gcp-projects.js': {
+          createProjectAndAttachBilling: (projectId) =>
+            Promise.resolve({
+              projectId: projectId,
+              billingMessage: 'billing message',
+            }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'create_project'
+      ).arguments[2];
+      const result = await handler({ projectId: 'my-project' });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Successfully created GCP project with ID "my-project". You can now use this project ID for deployments.',
+          },
+        ],
+      });
+    });
+
+    it('should create a project with a generated id', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/gcp-projects.js': {
+          createProjectAndAttachBilling: () =>
+            Promise.resolve({
+              projectId: 'generated-project',
+              billingMessage: 'billing message',
+            }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'create_project'
+      ).arguments[2];
+      const result = await handler({});
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Successfully created GCP project with ID "generated-project". You can now use this project ID for deployments.',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('list_services', () => {
+    it('should list services', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-services.js': {
+          listServices: () =>
+            Promise.resolve([
+              { name: 'service1', uri: 'uri1' },
+              { name: 'service2', uri: 'uri2' },
+            ]),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'list_services'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Services in project my-project (location my-region):\n- service1 (URL: uri1)\n- service2 (URL: uri2)',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('get_service', () => {
+    it('should get a service', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-services.js': {
+          getService: () =>
+            Promise.resolve({
+              name: 'my-service',
+              uri: 'my-uri',
+              lastModifier: 'me',
+            }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'get_service'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+        service: 'my-service',
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Name: my-service\nRegion: my-region\nProject: my-project\nURL: my-uri\nLast deployed by: me',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('get_service_log', () => {
+    it('should get service logs', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      let callCount = 0;
+      const getServiceLogs = () => {
+        callCount++;
+        if (callCount === 1) {
+          return Promise.resolve({
+            logs: 'log1\nlog2',
+            requestOptions: { pageToken: 'nextPage' },
+          });
+        }
+        return Promise.resolve({ logs: 'log3\nlog4', requestOptions: null });
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-services.js': {
+          getServiceLogs: getServiceLogs,
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'get_service_log'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+        service: 'my-service',
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'log1\nlog2\nlog3\nlog4',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('deploy_local_files', () => {
+    it('should deploy local files', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-deploy.js': {
+          deploy: () => Promise.resolve({ uri: 'my-uri' }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'deploy_local_files'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+        service: 'my-service',
+        files: ['file1', 'file2'],
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Cloud Run service my-service deployed in project my-project\nCloud Console: https://console.cloud.google.com/run/detail/my-region/my-service?project=my-project\nService URL: my-uri',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('deploy_local_folder', () => {
+    it('should deploy local folder', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-deploy.js': {
+          deploy: () => Promise.resolve({ uri: 'my-uri' }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'deploy_local_folder'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+        service: 'my-service',
+        folderPath: '/my/folder',
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Cloud Run service my-service deployed from folder /my/folder in project my-project\nCloud Console: https://console.cloud.google.com/run/detail/my-region/my-service?project=my-project\nService URL: my-uri',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('deploy_file_contents', () => {
+    it('should deploy file contents', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-deploy.js': {
+          deploy: () => Promise.resolve({ uri: 'my-uri' }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'deploy_file_contents'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+        service: 'my-service',
+        files: [{ filename: 'file1', content: 'content1' }],
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Cloud Run service my-service deployed in project my-project\nCloud Console: https://console.cloud.google.com/run/detail/my-region/my-service?project=my-project\nService URL: my-uri',
+          },
+        ],
+      });
+    });
+  });
+
+  describe('deploy_container_image', () => {
+    it('should deploy container image', async () => {
+      const server = {
+        registerTool: mock.fn(),
+      };
+
+      const { registerTools } = await esmock('../tools.js', {
+        '../lib/cloud-run-deploy.js': {
+          deployImage: () => Promise.resolve({ uri: 'my-uri' }),
+        },
+      });
+
+      registerTools(server);
+
+      const handler = server.registerTool.mock.calls.find(
+        (call) => call.arguments[0] === 'deploy_container_image'
+      ).arguments[2];
+      const result = await handler({
+        project: 'my-project',
+        region: 'my-region',
+        service: 'my-service',
+        imageUrl: 'gcr.io/my-project/my-image',
+      });
+
+      assert.deepStrictEqual(result, {
+        content: [
+          {
+            type: 'text',
+            text: 'Cloud Run service my-service deployed in project my-project\nCloud Console: https://console.cloud.google.com/run/detail/my-region/my-service?project=my-project\nService URL: my-uri',
+          },
+        ],
+      });
+    });
+  });
+});