mirror of
https://github.com/stablyai/orca.git
synced 2026-09-22 16:02:32 +00:00
Neil
1c41d59203
* perf(relay): drain fragmented frame buffers in linear time * style: follow block-body lint in relay buffer checks
70 lines
2.7 KiB
TypeScript
70 lines
2.7 KiB
TypeScript
import { describe, expect, it, vi } from 'vitest'
|
|
import { RelayFrameBuffer } from './relay-frame-buffer'
|
|
|
|
describe('RelayFrameBuffer', () => {
|
|
it('preserves a byte stream across fragmented peeks, takes, discards and drains', () => {
|
|
const buffer = new RelayFrameBuffer()
|
|
let expected = Buffer.alloc(0)
|
|
for (let step = 0; step < 5000; step += 1) {
|
|
const chunk = Buffer.from([step % 256, (step + 1) % 256, (step + 2) % 256])
|
|
buffer.append(chunk)
|
|
expected = Buffer.concat([expected, chunk])
|
|
if (step % 3 === 0) {
|
|
const count = Math.min(expected.length, 5)
|
|
expect(buffer.peek(count).subarray(0, count)).toEqual(expected.subarray(0, count))
|
|
expect(buffer.take(count)).toEqual(expected.subarray(0, count))
|
|
expected = expected.subarray(count)
|
|
}
|
|
if (step % 7 === 0) {
|
|
const count = Math.min(expected.length, 4)
|
|
buffer.discard(count)
|
|
expected = expected.subarray(count)
|
|
}
|
|
if (step % 101 === 0) {
|
|
expect(buffer.drain()).toEqual(expected)
|
|
expected = Buffer.alloc(0)
|
|
}
|
|
expect(buffer.length).toBe(expected.length)
|
|
}
|
|
expect(buffer.drain()).toEqual(expected)
|
|
expect(buffer.drain()).toEqual(Buffer.alloc(0))
|
|
})
|
|
|
|
it('releases consumed references and amortizes storage compaction in a large backlog', () => {
|
|
const buffer = new RelayFrameBuffer()
|
|
const chunks = Array.from({ length: 32768 }, (_, index) => Buffer.from([index % 256]))
|
|
for (const chunk of chunks) {
|
|
buffer.append(chunk)
|
|
}
|
|
const shifted = vi.spyOn(Array.prototype, 'shift')
|
|
let shiftCount: number
|
|
try {
|
|
buffer.discard(16000)
|
|
shiftCount = shifted.mock.calls.length
|
|
} finally {
|
|
shifted.mockRestore()
|
|
}
|
|
expect(shiftCount).toBe(0)
|
|
const storage = buffer as unknown as { chunks: (Buffer | undefined)[]; head: number }
|
|
expect(storage.chunks.slice(0, storage.head).every((chunk) => chunk === undefined)).toBe(true)
|
|
expect(buffer.take(1000)).toEqual(Buffer.concat(chunks.slice(16000, 17000)))
|
|
expect(storage.chunks.length).toBeLessThan(chunks.length)
|
|
expect(buffer.drain()).toEqual(Buffer.concat(chunks.slice(17000)))
|
|
expect(storage.chunks).toHaveLength(0)
|
|
expect(buffer.length).toBe(0)
|
|
})
|
|
|
|
it('keeps single-chunk views and clears partial data before reuse', () => {
|
|
const buffer = new RelayFrameBuffer()
|
|
const chunk = Buffer.from('abcdef')
|
|
buffer.append(chunk)
|
|
expect(buffer.peek(2)).toBe(chunk)
|
|
const taken = buffer.take(2)
|
|
expect(taken.buffer).toBe(chunk.buffer)
|
|
expect(taken.toString()).toBe('ab')
|
|
buffer.clear()
|
|
buffer.append(Buffer.from('fresh'))
|
|
expect(buffer.drain().toString()).toBe('fresh')
|
|
})
|
|
})
|