claude-code/tests/integration/message-pipeline.test.ts

import { describe, expect, test } from 'bun:test'
import {
  createUserMessage,
  createAssistantMessage,
  normalizeMessages,
  extractTag,
} from '../../src/utils/messages'

// ─── Message Structure ────────────────────────────────────────────────

describe('Message pipeline: message structure', () => {
  test("createUserMessage returns a Message with type 'user'", () => {
    const msg = createUserMessage('hello')
    expect(msg.type).toBe('user')
    expect(msg.message.role).toBe('user')
    expect(msg.uuid).toBeTruthy()
    expect(msg.timestamp).toBeTruthy()
  })

  test("createAssistantMessage returns a Message with type 'assistant'", () => {
    const msg = createAssistantMessage('response')
    expect(msg.type).toBe('assistant')
    expect(msg.message.role).toBe('assistant')
    expect(msg.uuid).toBeTruthy()
  })

  test('user and assistant messages have different UUIDs', () => {
    const user = createUserMessage('hello')
    const assistant = createAssistantMessage('response')
    expect(user.uuid).not.toBe(assistant.uuid)
  })
})

// ─── Tag Extraction ───────────────────────────────────────────────────

describe('Message pipeline: tag extraction', () => {
  test('extractTag returns null for non-matching tag', () => {
    expect(extractTag('no tags here', 'think')).toBeNull()
  })

  test('extractTag returns null for empty string', () => {
    expect(extractTag('', 'think')).toBeNull()
  })

  test('extractTag requires tagName parameter', () => {
    // Calling without tagName throws
    expect(() => (extractTag as any)('hello')).toThrow()
  })
})

// ─── Normalization ────────────────────────────────────────────────────

describe('Message pipeline: normalization', () => {
  test('normalizeMessages returns an array', () => {
    const msg = createUserMessage('hello')
    const result = normalizeMessages([msg])
    expect(Array.isArray(result)).toBe(true)
  })

  test('normalizeMessages preserves at least one message for simple input', () => {
    const msg = createUserMessage('hello')
    const result = normalizeMessages([msg])
    expect(result.length).toBeGreaterThanOrEqual(1)
  })
})