simstudioai--sim/apps/sim/lib/copilot/request/context/result.test.ts

/**
 * @vitest-environment node
 */
import { describe, expect, it } from 'vitest'
import { MothershipStreamV1ToolOutcome } from '@/lib/copilot/generated/mothership-stream-v1'
import { FunctionExecute } from '@/lib/copilot/generated/tool-catalog-v1'
import { buildToolCallSummaries } from '@/lib/copilot/request/context/result'
import { TraceCollector } from '@/lib/copilot/request/trace'
import type { StreamingContext } from '@/lib/copilot/request/types'

function makeContext(): StreamingContext {
  return {
    chatId: undefined,
    requestId: undefined,
    executionId: undefined,
    runId: undefined,
    messageId: 'msg-1',
    accumulatedContent: '',
    finalAssistantContent: '',
    sawMainToolCall: false,
    contentBlocks: [],
    toolCalls: new Map(),
    pendingToolPromises: new Map(),
    awaitingAsyncContinuation: undefined,
    currentThinkingBlock: null,
    subagentThinkingBlocks: new Map(),
    isInThinkingBlock: false,
    subAgentContent: {},
    subAgentToolCalls: {},
    pendingContent: '',
    streamComplete: false,
    wasAborted: false,
    errors: [],
    trace: new TraceCollector(),
    toolPermissions: {
      enabled: false,
      autoAllowed: new Set(),
    },
  }
}

describe('buildToolCallSummaries', () => {
  it.concurrent('keeps pending tools as pending instead of defaulting to success', () => {
    const context = makeContext()
    context.toolCalls.set('tool-1', {
      id: 'tool-1',
      name: 'download_to_workspace_file',
      status: 'pending',
      startTime: 1,
    })

    const summaries = buildToolCallSummaries(context)

    expect(summaries).toHaveLength(1)
    expect(summaries[0]?.status).toBe('pending')
  })

  it.concurrent('keeps executing tools as executing when no result exists yet', () => {
    const context = makeContext()
    context.toolCalls.set('tool-2', {
      id: 'tool-2',
      name: FunctionExecute.id,
      status: 'executing',
      startTime: 1,
    })

    const summaries = buildToolCallSummaries(context)

    expect(summaries).toHaveLength(1)
    expect(summaries[0]?.status).toBe('executing')
  })

  it.concurrent(
    'preserves canonical terminal statuses instead of deriving them from result success',
    () => {
      const context = makeContext()
      context.toolCalls.set('tool-3', {
        id: 'tool-3',
        name: 'download_to_workspace_file',
        status: MothershipStreamV1ToolOutcome.cancelled,
        result: { success: false },
        error: 'Stopped by user',
        startTime: 1,
        endTime: 2,
      })

      const summaries = buildToolCallSummaries(context)

      expect(summaries).toHaveLength(1)
      expect(summaries[0]).toEqual({
        id: 'tool-3',
        name: 'download_to_workspace_file',
        status: MothershipStreamV1ToolOutcome.cancelled,
        params: undefined,
        result: undefined,
        error: 'Stopped by user',
        durationMs: 1,
      })
    }
  )
})