| import { describe, it, expect, beforeEach } from 'bun:test' |
| import { MessageBuffer } from '../message-buffer.js' |
|
|
| describe('MessageBuffer', () => { |
| it('accumulates text and flushes on complete', async () => { |
| const flushed: Array<{ text: string; isComplete: boolean }> = [] |
| const buf = new MessageBuffer( |
| (text, isComplete) => { flushed.push({ text, isComplete }) }, |
| 500, |
| 1000, |
| ) |
|
|
| buf.append('Hello ') |
| buf.append('World') |
| await buf.complete() |
|
|
| expect(flushed.length).toBeGreaterThanOrEqual(1) |
| const allText = flushed.map((f) => f.text).join('') |
| expect(allText).toBe('Hello World') |
| |
| expect(flushed[flushed.length - 1]!.isComplete).toBe(true) |
| }) |
|
|
| it('flushes when character threshold is reached', async () => { |
| const flushed: string[] = [] |
| const buf = new MessageBuffer( |
| (text) => { flushed.push(text) }, |
| 10000, |
| 10, |
| ) |
|
|
| buf.append('12345678901') |
|
|
| |
| await new Promise((r) => setTimeout(r, 10)) |
| expect(flushed.length).toBeGreaterThanOrEqual(1) |
|
|
| buf.reset() |
| }) |
|
|
| it('flushes on timer interval', async () => { |
| const flushed: string[] = [] |
| const buf = new MessageBuffer( |
| (text) => { flushed.push(text) }, |
| 50, |
| 1000, |
| ) |
|
|
| buf.append('hi') |
|
|
| |
| await new Promise((r) => setTimeout(r, 80)) |
| expect(flushed).toContain('hi') |
|
|
| buf.reset() |
| }) |
|
|
| it('does not flush empty buffer on complete', async () => { |
| const flushed: string[] = [] |
| const buf = new MessageBuffer( |
| (text) => { flushed.push(text) }, |
| ) |
|
|
| await buf.complete() |
| expect(flushed.length).toBe(0) |
| }) |
|
|
| it('waits for an in-flight flush before complete resolves', async () => { |
| let releaseFlush!: () => void |
| let flushStarted = false |
| const flushed: Array<{ text: string; isComplete: boolean }> = [] |
| const buf = new MessageBuffer( |
| async (text, isComplete) => { |
| flushStarted = true |
| flushed.push({ text, isComplete }) |
| await new Promise<void>((resolve) => { |
| releaseFlush = resolve |
| }) |
| }, |
| 10000, |
| 3, |
| ) |
|
|
| buf.append('abcd') |
| await new Promise((resolve) => setTimeout(resolve, 0)) |
| expect(flushStarted).toBe(true) |
|
|
| let completeResolved = false |
| const completing = buf.complete().then(() => { |
| completeResolved = true |
| }) |
| await new Promise((resolve) => setTimeout(resolve, 0)) |
| expect(completeResolved).toBe(false) |
|
|
| releaseFlush() |
| await completing |
| expect(completeResolved).toBe(true) |
| expect(flushed).toEqual([{ text: 'abcd', isComplete: false }]) |
| }) |
|
|
| it('resets properly between messages', async () => { |
| const flushed: string[] = [] |
| const buf = new MessageBuffer( |
| (text) => { flushed.push(text) }, |
| 500, |
| 1000, |
| ) |
|
|
| buf.append('first') |
| buf.reset() |
| buf.append('second') |
| await buf.complete() |
|
|
| const allText = flushed.map((f) => f).join('') |
| expect(allText).toBe('second') |
| }) |
| }) |
|
|
