gogcli-mcp 2.0.8 → 2.0.9

package/tests/tools/docs.test.ts

@@ -1,22 +1,11 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { registerDocsTools } from '../../src/tools/docs.js';
 import * as runner from '../../src/runner.js';
+import { setupHandlers as setupHandlersBase, type ToolHandler } from '../helpers/test-harness.js';
 
 vi.mock('../../src/runner.js');
 
-type ToolHandler = (args: Record<string, unknown>) => Promise<{ content: Array<{ type: string; text: string }> }>;
-
-function setupHandlers(): Map<string, ToolHandler> {
-  const server = new McpServer({ name: 'test', version: '0.0.0' });
-  const handlers = new Map<string, ToolHandler>();
-  vi.spyOn(server, 'registerTool').mockImplementation((name, _config, cb) => {
-    handlers.set(name, cb as ToolHandler);
-    return undefined as never;
-  });
-  registerDocsTools(server);
-  return handlers;
-}
+const setupHandlers = () => setupHandlersBase(registerDocsTools);
 
 beforeEach(() => vi.clearAllMocks());
 

package/tests/tools/drive.test.ts

@@ -1,22 +1,11 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { registerDriveTools } from '../../src/tools/drive.js';
 import * as runner from '../../src/runner.js';
+import { setupHandlers as setupHandlersBase, type ToolHandler } from '../helpers/test-harness.js';
 
 vi.mock('../../src/runner.js');
 
-type ToolHandler = (args: Record<string, unknown>) => Promise<{ content: Array<{ type: string; text: string }> }>;
-
-function setupHandlers(): Map<string, ToolHandler> {
-  const server = new McpServer({ name: 'test', version: '0.0.0' });
-  const handlers = new Map<string, ToolHandler>();
-  vi.spyOn(server, 'registerTool').mockImplementation((name, _config, cb) => {
-    handlers.set(name, cb as ToolHandler);
-    return undefined as never;
-  });
-  registerDriveTools(server);
-  return handlers;
-}
+const setupHandlers = () => setupHandlersBase(registerDriveTools);
 
 beforeEach(() => vi.clearAllMocks());
 

package/tests/tools/gmail.test.ts

@@ -1,22 +1,11 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { registerGmailTools } from '../../src/tools/gmail.js';
 import * as runner from '../../src/runner.js';
+import { setupHandlers as setupHandlersBase, type ToolHandler } from '../helpers/test-harness.js';
 
 vi.mock('../../src/runner.js');
 
-type ToolHandler = (args: Record<string, unknown>) => Promise<{ content: Array<{ type: string; text: string }> }>;
-
-function setupHandlers(): Map<string, ToolHandler> {
-  const server = new McpServer({ name: 'test', version: '0.0.0' });
-  const handlers = new Map<string, ToolHandler>();
-  vi.spyOn(server, 'registerTool').mockImplementation((name, _config, cb) => {
-    handlers.set(name, cb as ToolHandler);
-    return undefined as never;
-  });
-  registerGmailTools(server);
-  return handlers;
-}
+const setupHandlers = () => setupHandlersBase(registerGmailTools);
 
 beforeEach(() => vi.clearAllMocks());
 

package/tests/tools/sheets.test.ts

@@ -1,22 +1,11 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { registerSheetsTools } from '../../src/tools/sheets.js';
 import * as runner from '../../src/runner.js';
+import { setupHandlers as setupHandlersBase, type ToolHandler } from '../helpers/test-harness.js';
 
 vi.mock('../../src/runner.js');
 
-type ToolHandler = (args: Record<string, unknown>) => Promise<{ content: Array<{ type: string; text: string }> }>;
-
-function setupHandlers(): Map<string, ToolHandler> {
-  const server = new McpServer({ name: 'test', version: '0.0.0' });
-  const handlers = new Map<string, ToolHandler>();
-  vi.spyOn(server, 'registerTool').mockImplementation((name, _config, cb) => {
-    handlers.set(name, cb as ToolHandler);
-    return undefined as never;
-  });
-  registerSheetsTools(server);
-  return handlers;
-}
+const setupHandlers = () => setupHandlersBase(registerSheetsTools);
 
 beforeEach(() => vi.clearAllMocks());
 

package/tests/tools/slides.test.ts

@@ -1,22 +1,11 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { registerSlidesTools } from '../../src/tools/slides.js';
 import * as runner from '../../src/runner.js';
+import { setupHandlers as setupHandlersBase, type ToolHandler } from '../helpers/test-harness.js';
 
 vi.mock('../../src/runner.js');
 
-type ToolHandler = (args: Record<string, unknown>) => Promise<{ content: Array<{ type: string; text: string }> }>;
-
-function setupHandlers(): Map<string, ToolHandler> {
-  const server = new McpServer({ name: 'test', version: '0.0.0' });
-  const handlers = new Map<string, ToolHandler>();
-  vi.spyOn(server, 'registerTool').mockImplementation((name, _config, cb) => {
-    handlers.set(name, cb as ToolHandler);
-    return undefined as never;
-  });
-  registerSlidesTools(server);
-  return handlers;
-}
+const setupHandlers = () => setupHandlersBase(registerSlidesTools);
 
 beforeEach(() => vi.clearAllMocks());
 

package/tests/tools/tasks.test.ts

@@ -1,22 +1,11 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { registerTasksTools } from '../../src/tools/tasks.js';
 import * as runner from '../../src/runner.js';
+import { setupHandlers as setupHandlersBase, type ToolHandler } from '../helpers/test-harness.js';
 
 vi.mock('../../src/runner.js');
 
-type ToolHandler = (args: Record<string, unknown>) => Promise<{ content: Array<{ type: string; text: string }> }>;
-
-function setupHandlers(): Map<string, ToolHandler> {
-  const server = new McpServer({ name: 'test', version: '0.0.0' });
-  const handlers = new Map<string, ToolHandler>();
-  vi.spyOn(server, 'registerTool').mockImplementation((name, _config, cb) => {
-    handlers.set(name, cb as ToolHandler);
-    return undefined as never;
-  });
-  registerTasksTools(server);
-  return handlers;
-}
+const setupHandlers = () => setupHandlersBase(registerTasksTools);
 
 beforeEach(() => vi.clearAllMocks());
 

package/tests/tools/utils.test.ts

@@ -1,11 +1,45 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
 import * as runner from '../../src/runner.js';
-import { runOrDiagnose } from '../../src/tools/utils.js';
+import { runOrDiagnose, pushPaginationFlags } from '../../src/tools/utils.js';
 
 vi.mock('../../src/runner.js');
 
 beforeEach(() => vi.clearAllMocks());
 
+describe('pushPaginationFlags', () => {
+  it('does nothing when no fields are provided', () => {
+    const args: string[] = ['cmd'];
+    pushPaginationFlags(args, {});
+    expect(args).toEqual(['cmd']);
+  });
+
+  it('pushes --max when a number is provided (including 0)', () => {
+    const args: string[] = ['cmd'];
+    pushPaginationFlags(args, { max: 0 });
+    expect(args).toEqual(['cmd', '--max=0']);
+  });
+
+  it('pushes --page when a string is provided', () => {
+    const args: string[] = ['cmd'];
+    pushPaginationFlags(args, { page: 'tok' });
+    expect(args).toEqual(['cmd', '--page=tok']);
+  });
+
+  it('pushes --all only when true', () => {
+    const args: string[] = ['cmd'];
+    pushPaginationFlags(args, { all: false });
+    expect(args).toEqual(['cmd']);
+    pushPaginationFlags(args, { all: true });
+    expect(args).toEqual(['cmd', '--all']);
+  });
+
+  it('pushes all three in canonical order', () => {
+    const args: string[] = ['cmd'];
+    pushPaginationFlags(args, { max: 50, page: 'tok', all: true });
+    expect(args).toEqual(['cmd', '--max=50', '--page=tok', '--all']);
+  });
+});
+
 describe('runOrDiagnose', () => {
   it('returns output text on success', async () => {
     vi.mocked(runner.run).mockResolvedValue('{"ok":true}');