@coldiq/mcp 0.1.11 → 0.1.13

package/tests/tools/response-format.test.ts

@@ -0,0 +1,45 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest'
+import { initClient } from '../../src/client.js'
+import { searchCompaniesHandler } from '../../src/tools/search-companies.js'
+import { findEmailHandler } from '../../src/tools/find-email.js'
+
+describe('tool response format', () => {
+  const originalFetch = globalThis.fetch
+
+  beforeEach(() => {
+    initClient('http://test-api.local', 'test-key')
+  })
+
+  afterEach(() => {
+    globalThis.fetch = originalFetch
+    vi.restoreAllMocks()
+  })
+
+  it('success response is compact JSON (no indentation)', async () => {
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ companies: [{ name: 'Acme' }] }), { status: 200 }),
+    ) as typeof fetch
+
+    const result = await searchCompaniesHandler({ keywords: ['SaaS'], limit: 5 })
+    const text = result.content[0].text
+
+    expect(result.isError).toBeFalsy()
+    // Compact JSON has no newlines
+    expect(text).not.toContain('\n')
+    // Must be valid JSON
+    expect(() => JSON.parse(text)).not.toThrow()
+  })
+
+  it('error response is compact JSON (no indentation)', async () => {
+    globalThis.fetch = vi.fn(async () =>
+      new Response(JSON.stringify({ error: 'Bad request' }), { status: 400 }),
+    ) as typeof fetch
+
+    const result = await findEmailHandler({ first_name: 'Michel', last_name: 'Lieben', domain: 'coldiq.com' })
+    const text = result.content[0].text
+
+    expect(result.isError).toBeTruthy()
+    expect(text).not.toContain('\n')
+    expect(() => JSON.parse(text)).not.toThrow()
+  })
+})