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

import { describe, expect, test } from 'bun:test'
import { createBufferedWriter } from '../bufferedWriter'

describe('createBufferedWriter', () => {
  test('immediateMode calls writeFn directly', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
      immediateMode: true,
    })
    writer.write('a')
    writer.write('b')
    expect(written).toEqual(['a', 'b'])
  })

  test('buffered mode accumulates until flush', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
    })
    writer.write('hello ')
    writer.write('world')
    expect(written).toEqual([])
    writer.flush()
    expect(written).toEqual(['hello world'])
  })

  test('flush with empty buffer does not call writeFn', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
    })
    writer.flush()
    expect(written).toEqual([])
  })

  test('flush clears the buffer', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
    })
    writer.write('data')
    writer.flush()
    writer.flush() // second flush should be no-op
    expect(written).toEqual(['data'])
  })

  test('overflow triggers deferred flush when maxBufferSize reached', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
      maxBufferSize: 2,
    })
    writer.write('a')
    writer.write('b')
    // 2 writes = maxBufferSize, triggers flushDeferred via setImmediate
    expect(written).toEqual([])
  })

  test('overflow triggers deferred flush when maxBufferBytes reached', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
      maxBufferBytes: 5,
    })
    writer.write('abc')
    writer.write('def')
    // total 6 bytes > 5, triggers flushDeferred
    expect(written).toEqual([])
  })

  test('dispose flushes remaining buffer', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
    })
    writer.write('final')
    writer.dispose()
    expect(written).toEqual(['final'])
  })

  test('dispose flushes pending overflow', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
      maxBufferSize: 1,
    })
    writer.write('overflow-data')
    // overflow triggered but deferred; dispose should flush it synchronously
    writer.dispose()
    expect(written).toEqual(['overflow-data'])
  })

  test('coalesced overflow — multiple overflows merge before write', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
      maxBufferSize: 1,
    })
    writer.write('a') // triggers first overflow (deferred)
    writer.write('b') // pendingOverflow exists, coalesces
    writer.dispose() // flushes coalesced overflow
    expect(written).toEqual(['ab'])
  })

  test('multiple flushes produce concatenated writes', () => {
    const written: string[] = []
    const writer = createBufferedWriter({
      writeFn: c => written.push(c),
    })
    writer.write('batch1')
    writer.flush()
    writer.write('batch2')
    writer.flush()
    expect(written).toEqual(['batch1', 'batch2'])
  })
})