53 lines
1.9 KiB
TypeScript
53 lines
1.9 KiB
TypeScript
|
import { describe, expect, test } from 'bun:test';
|
||
|
|
import { createRecorder } from '../src/index.ts';
|
||
|
|
|
||
|
|
describe('createRecorder', () => {
|
||
|
|
test('captures attributes and end-state', () => {
|
||
|
|
const rec = createRecorder();
|
||
|
|
const span = rec.startSpan('shade.test', { initial: 'on' });
|
||
|
|
span.setAttribute('extra', 1);
|
||
|
|
span.setAttributes({ batch1: 'a', batch2: 'b' });
|
||
|
|
span.setStatus('ok');
|
||
|
|
span.end();
|
||
|
|
expect(rec.spans).toHaveLength(1);
|
||
|
|
const s = rec.spans[0]!;
|
||
|
|
expect(s.name).toBe('shade.test');
|
||
|
|
expect(s.attributes).toEqual({ initial: 'on', extra: 1, batch1: 'a', batch2: 'b' });
|
||
|
|
expect(s.status).toBe('ok');
|
||
|
|
expect(s.ended).toBe(true);
|
||
|
|
});
|
||
|
|
|
||
|
|
test('records exceptions', () => {
|
||
|
|
const rec = createRecorder();
|
||
|
|
const span = rec.startSpan('shade.test');
|
||
|
|
const err = new Error('boom');
|
||
|
|
span.recordException(err);
|
||
|
|
span.setStatus('error', 'boom');
|
||
|
|
span.end();
|
||
|
|
expect(rec.spans[0]?.exceptions).toEqual([err]);
|
||
|
|
expect(rec.spans[0]?.status).toBe('error');
|
||
|
|
expect(rec.spans[0]?.statusMessage).toBe('boom');
|
||
|
|
});
|
||
|
|
|
||
|
|
test('scanForPII catches forbidden substrings', () => {
|
||
|
|
const rec = createRecorder();
|
||
|
|
const safe = rec.startSpan('shade.upload', { 'shade.peer.hash': 'abc12345' });
|
||
|
|
safe.end();
|
||
|
|
const leaky = rec.startSpan('shade.upload', { 'shade.peer.address': 'alice@example.com' });
|
||
|
|
leaky.end();
|
||
|
|
const hits = rec.scanForPII(['@', 'alice', 'peer.address']);
|
||
|
|
expect(hits.length).toBeGreaterThan(0);
|
||
|
|
// The safe span should not be in the hits.
|
||
|
|
const safeHit = hits.find((h) => h.spanName === 'shade.upload' && h.value === 'abc12345');
|
||
|
|
expect(safeHit).toBeUndefined();
|
||
|
|
});
|
||
|
|
|
||
|
|
test('clear() drops the buffer', () => {
|
||
|
|
const rec = createRecorder();
|
||
|
|
rec.startSpan('a').end();
|
||
|
|
expect(rec.spans).toHaveLength(1);
|
||
|
|
rec.clear();
|
||
|
|
expect(rec.spans).toHaveLength(0);
|
||
|
|
});
|
||
|
|
});
|