@coldiq/mcp 0.5.2 → 0.5.4

package/tests/tools/extract-post-engagement.test.ts

@@ -26,9 +26,9 @@ describe('extract_post_engagement handler', () => {
     const people = [{ name: 'Alex Vacca', profile_url: 'https://www.linkedin.com/in/alexvacca' }]
     globalThis.fetch = vi.fn(async (url: string | URL) => {
       const u = url.toString()
-      if (u.includes('/jungler/workbooks/wb_1/contacts')) return res(people)
+      if (u.includes('/jungler/workbooks/wb_1/contacts')) return res({ items: people, total: 1, page: 1, pages: 1 })
       if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', workbook_id: 'wb_1' })
-      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING' }, 200, { 'x-coldiq-credits-charged': '10', 'x-coldiq-credits-remaining': '90' })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' }, 200, { 'x-coldiq-credits-charged': '10', 'x-coldiq-credits-remaining': '90' })
       throw new Error(`unexpected url ${u}`)
     }) as typeof fetch
 
@@ -36,15 +36,80 @@ describe('extract_post_engagement handler', () => {
     const parsed = JSON.parse(result.content[0].text)
     expect(result.isError).toBeUndefined()
     expect(parsed.data.people).toEqual(people)
+    expect(parsed.data.count).toBe(1)
     expect(parsed.data.type).toBe('both')
     expect(parsed._meta.credits_charged).toBe(10)
   })
 
+  it('walks every contacts page — a viral post exceeds one page', async () => {
+    const pageOf = (n: number) => Array.from({ length: n }, (_, i) => ({ name: `Person ${i}` }))
+    const seenPages: string[] = []
+    globalThis.fetch = vi.fn(async (url: string | URL) => {
+      const u = url.toString()
+      if (u.includes('/contacts')) {
+        const page = new URL(u).searchParams.get('page') ?? '1'
+        seenPages.push(page)
+        return res({ items: pageOf(page === '1' ? 500 : 103), total: 603, page: Number(page), pages: 2 })
+      }
+      if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', workbook_id: 'wb_1' })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' })
+      throw new Error(`unexpected url ${u}`)
+    }) as typeof fetch
+
+    const result = await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })
+    const parsed = JSON.parse(result.content[0].text)
+    expect(seenPages).toEqual(['1', '2'])
+    expect(parsed.data.count).toBe(603)
+    expect(parsed._meta.truncated_from).toBeUndefined()
+  })
+
+  it('drops the post author, who is collected with zero engagement and never engaged', async () => {
+    // The workbook always collects the post itself, so its author comes back as a
+    // contact with stats 0/0. This tool promises the people who ENGAGED.
+    globalThis.fetch = vi.fn(async (url: string | URL) => {
+      const u = url.toString()
+      if (u.includes('/contacts')) return res({
+        items: [
+          { name: 'Michel Lieben', stats: { comments: 0, reactions: 0 } },
+          { name: 'Jane Smith', stats: { comments: 2, reactions: 1 } },
+          { name: 'No Stats Person' },
+        ],
+        total: 3, page: 1, pages: 1,
+      })
+      if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', workbook_id: 'wb_1' })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' })
+      throw new Error(`unexpected url ${u}`)
+    }) as typeof fetch
+
+    const parsed = JSON.parse((await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })).content[0].text)
+    // Unknown stats are kept — never drop a contact on ambiguity.
+    expect(parsed.data.people.map((p: { name: string }) => p.name)).toEqual(['Jane Smith', 'No Stats Person'])
+  })
+
+  it('reports the task and workbook ids instead of losing the paid run when polling times out', async () => {
+    vi.stubEnv('COLDIQ_ENGAGEMENT_TIMEOUT_MS', '30')
+    globalThis.fetch = vi.fn(async (url: string | URL) => {
+      const u = url.toString()
+      // Never terminal — the caller must still be able to collect what they paid for.
+      if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'started' })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' })
+      throw new Error(`unexpected url ${u}`)
+    }) as typeof fetch
+
+    const result = await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })
+    const parsed = JSON.parse(result.content[0].text)
+    expect(result.isError).toBe(true)
+    expect(parsed.task_id).toBe('task_1')
+    expect(parsed.workbook_id).toBe('wb_1')
+    expect(parsed.error).toMatch(/still running/i)
+    expect(parsed.error).toMatch(/NOT been cancelled/i)
+  })
+
   it('passes activity_filter=commenters when type=comments', async () => {
     let contactsUrl = ''
     globalThis.fetch = vi.fn(async (url: string | URL) => {
       const u = url.toString()
-      if (u.includes('/jungler/workbooks/wb_1/contacts')) { contactsUrl = u; return res([]) }
+      if (u.includes('/jungler/workbooks/wb_1/contacts')) { contactsUrl = u; return res({ items: [], total: 0, pages: 0 }) }
       if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', workbook_id: 'wb_1' })
       if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING' })
       throw new Error(`unexpected url ${u}`)
@@ -67,6 +132,105 @@ describe('extract_post_engagement handler', () => {
     expect(JSON.parse(result.content[0].text).error).toMatch(/extraction failed/i)
   })
 
+  // ── Fallback to the second provider ──────────────────────────────────────
+  //
+  // Only for failures that are OURS (dead run, our provider account out of
+  // credits) — never for a post that genuinely has no engagement, which would
+  // buy an empty second result.
+  const LIMA_PAYLOAD = {
+    items: [{
+      status: 'Completed',
+      data: {
+        post_url: POST_URL,
+        commenters: { comments: [{ actor: { name: 'Jane Smith', profile_url: 'https://linkedin.com/in/jane', headline: 'VP Eng' } }] },
+        reactors: {
+          reactions: [
+            { actor: { name: 'Bob Johnson', profile_url: 'https://linkedin.com/in/bob', headline: 'Sales' } },
+            // Same person also reacted — must collapse to one contact.
+            { actor: { name: 'Jane Smith', profile_url: 'https://linkedin.com/in/jane', headline: 'VP Eng' } },
+          ],
+        },
+      },
+    }],
+  }
+
+  function stubExhaustedThenLima(limaOk = true) {
+    globalThis.fetch = vi.fn(async (url: string | URL) => {
+      const u = url.toString()
+      if (u.includes('/limadata/batch/post-engagements')) {
+        return limaOk ? res({ batch_id: 99 }, 200, { 'x-coldiq-credits-charged': '35' }) : res({ error: 'lima down' }, 502)
+      }
+      if (u.includes('/limadata/batch/results')) return res(LIMA_PAYLOAD)
+      if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', credits_exhausted: true })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' })
+      throw new Error(`unexpected url ${u}`)
+    }) as typeof fetch
+  }
+
+  it('falls back to the second provider when our provider account is out of credits', async () => {
+    stubExhaustedThenLima()
+    const result = await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })
+    const parsed = JSON.parse(result.content[0].text)
+
+    expect(result.isError).toBeUndefined()
+    expect(parsed._meta.provider).toBe('limadata')
+    expect(parsed._meta.credits_charged).toBe(35)
+    // Jane commented AND reacted — one contact, both engagement types.
+    expect(parsed.data.count).toBe(2)
+    const jane = parsed.data.people.find((p: { name: string }) => p.name === 'Jane Smith')
+    expect(jane.engagement_types.sort()).toEqual(['COMMENT', 'REACTION'])
+    expect(jane.profile_url).toBe('https://linkedin.com/in/jane')
+    // The caller must know these records are thinner than the primary's.
+    expect(parsed._meta.note).toMatch(/no work email/i)
+  })
+
+  it('honours type=comments when falling back', async () => {
+    stubExhaustedThenLima()
+    const parsed = JSON.parse((await extractPostEngagementHandler({ post_url: POST_URL, type: 'comments' })).content[0].text)
+    expect(parsed.data.people.map((p: { name: string }) => p.name)).toEqual(['Jane Smith'])
+  })
+
+  it('reports the ORIGINAL failure when the fallback also fails', async () => {
+    stubExhaustedThenLima(false)
+    const result = await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })
+    const parsed = JSON.parse(result.content[0].text)
+    expect(result.isError).toBe(true)
+    expect(parsed.error).toMatch(/out of credits/i)
+    expect(parsed.fallback_error).toBeTruthy()
+  })
+
+  it('does not spend on a fallback when it is switched off', async () => {
+    vi.stubEnv('COLDIQ_ENGAGEMENT_FALLBACK', '0')
+    let limaCalled = false
+    globalThis.fetch = vi.fn(async (url: string | URL) => {
+      const u = url.toString()
+      if (u.includes('/limadata')) { limaCalled = true; return res({}) }
+      if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', credits_exhausted: true })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' })
+      throw new Error(`unexpected url ${u}`)
+    }) as typeof fetch
+
+    const result = await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })
+    expect(result.isError).toBe(true)
+    expect(limaCalled).toBe(false)
+  })
+
+  it('does NOT fall back for a genuinely empty post — that would buy a second empty result', async () => {
+    let limaCalled = false
+    globalThis.fetch = vi.fn(async (url: string | URL) => {
+      const u = url.toString()
+      if (u.includes('/limadata')) { limaCalled = true; return res({}) }
+      if (u.includes('/contacts')) return res({ items: [], total: 0, pages: 0 })
+      if (u.includes('/jungler/tasks/task_1/status')) return res({ task_id: 'task_1', status: 'success', workbook_id: 'wb_1' })
+      if (u.includes('/jungler/workbooks')) return res({ task_id: 'task_1', status: 'PENDING', workbook_id: 'wb_1' })
+      throw new Error(`unexpected url ${u}`)
+    }) as typeof fetch
+
+    const parsed = JSON.parse((await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })).content[0].text)
+    expect(limaCalled).toBe(false)
+    expect(parsed.data.count).toBe(0)
+  })
+
   it('surfaces the create error when the job cannot start', async () => {
     globalThis.fetch = vi.fn(async () => res({ error: 'Invalid LinkedIn post URL' }, 400)) as typeof fetch
     const result = await extractPostEngagementHandler({ post_url: POST_URL, type: 'both' })

package/tests/tools/get-endpoint-details.test.ts

@@ -34,4 +34,16 @@ describe('get_endpoint_details handler', () => {
     expect(parsed.status).toBe(404)
     expect(parsed.error).toBe('Endpoint not found or details unavailable')
   })
+
+  it('forwards the HTTP method for a shared endpoint path', async () => {
+    let capturedBody: unknown
+    globalThis.fetch = vi.fn(async (_url: string | URL | Request, opts?: RequestInit) => {
+      capturedBody = opts?.body ? JSON.parse(opts.body as string) : undefined
+      return new Response(JSON.stringify({ path: '/exa/agent/runs', method: 'POST' }), { status: 200 })
+    }) as typeof fetch
+
+    await getEndpointDetailsHandler({ api: 'exa', path: '/exa/agent/runs', method: 'POST' })
+
+    expect(capturedBody).toEqual({ api: 'exa', path: '/exa/agent/runs', method: 'POST' })
+  })
 })