claude-code/src/utils/__tests__/sanitization.test.ts

import { describe, expect, test } from 'bun:test'
import {
  partiallySanitizeUnicode,
  recursivelySanitizeUnicode,
} from '../sanitization'

// ─── partiallySanitizeUnicode ───────────────────────────────────────────

describe('partiallySanitizeUnicode', () => {
  test('preserves normal ASCII text', () => {
    expect(partiallySanitizeUnicode('hello world')).toBe('hello world')
  })

  test('preserves CJK characters', () => {
    expect(partiallySanitizeUnicode('你好世界')).toBe('你好世界')
  })

  test('removes zero-width spaces', () => {
    expect(partiallySanitizeUnicode('hello\u200Bworld')).toBe('helloworld')
  })

  test('removes BOM', () => {
    expect(partiallySanitizeUnicode('\uFEFFhello')).toBe('hello')
  })

  test('removes directional formatting', () => {
    expect(partiallySanitizeUnicode('hello\u202Aworld')).toBe('helloworld')
  })

  test('removes private use area characters', () => {
    expect(partiallySanitizeUnicode('hello\uE000world')).toBe('helloworld')
  })

  test('handles empty string', () => {
    expect(partiallySanitizeUnicode('')).toBe('')
  })

  test('handles string with only dangerous characters', () => {
    const result = partiallySanitizeUnicode('\u200B\u200C\u200D\uFEFF')
    expect(result.length).toBeLessThanOrEqual(1) // ZWJ may survive NFKC
  })
})

// ─── recursivelySanitizeUnicode ─────────────────────────────────────────

describe('recursivelySanitizeUnicode', () => {
  test('sanitizes string values', () => {
    expect(recursivelySanitizeUnicode('hello\u200Bworld')).toBe('helloworld')
  })

  test('sanitizes array elements', () => {
    const result = recursivelySanitizeUnicode(['a\u200Bb', 'c\uFEFFd'])
    expect(result).toEqual(['ab', 'cd'])
  })

  test('sanitizes object values recursively', () => {
    const result = recursivelySanitizeUnicode({
      key: 'val\u200Bue',
      nested: { inner: 'te\uFEFFst' },
    })
    expect(result).toEqual({ key: 'value', nested: { inner: 'test' } })
  })

  test('preserves numbers', () => {
    expect(recursivelySanitizeUnicode(42)).toBe(42)
  })

  test('preserves booleans', () => {
    expect(recursivelySanitizeUnicode(true)).toBe(true)
  })

  test('preserves null', () => {
    expect(recursivelySanitizeUnicode(null)).toBeNull()
  })
})