import { describe, expect, test } from 'bun:test'
import { sourceEventsFromSdkMessage } from './sdkMessageAdapter.js'

describe('sourceEventsFromSdkMessage', () => {
  test('converts QueryEngine result messages to trajectory events', () => {
    const events = sourceEventsFromSdkMessage({
      type: 'result',
      subtype: 'success',
      stop_reason: 'end_turn',
      duration_ms: 1234,
      duration_api_ms: 1000,
      is_error: false,
      usage: { input_tokens: 10, output_tokens: 5 },
      errors: [],
    })

    expect(events).toEqual([
      {
        type: 'agent_result',
        subtype: 'success',
        stopReason: 'end_turn',
        durationMs: 1234,
        durationApiMs: 1000,
        isError: false,
        usage: { input_tokens: 10, output_tokens: 5 },
        errors: [],
        raw: expect.any(Object),
      },
    ])
  })
})