observability-toolkit 1.8.0 → 1.8.4

package/dist/test-helpers/test-data-builders.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-data-builders.test.d.ts","sourceRoot":"","sources":["../../src/test-helpers/test-data-builders.test.ts"],"names":[],"mappings":""}

package/dist/test-helpers/test-data-builders.test.js

@@ -0,0 +1,306 @@
+import { describe, it, beforeEach } from 'node:test';
+import assert from 'node:assert';
+import { 
+// Constants
+TEST_TIMESTAMP_BASE, TEST_TIMESTAMP, TEST_TIMESTAMP_ISO, TEST_DURATION, TEST_DURATION_MS, 
+// Core builders
+resetBuilderCounters, createMockSpan, createMockSpans, createMockLog, createMockLogs, createMockMetric, createMockMetrics, createMockLLMEvent, createMockLLMEvents, createMockEvaluation, createMockEvaluations, 
+// Specialized builders
+createHttpSpan, createDbSpan, createErrorLog, createCorrelatedLog, createHistogramMetric, createLLMEventWithTokens, createScoredEvaluation, createLabeledEvaluation, } from './test-data-builders.js';
+describe('test-data-builders', () => {
+    beforeEach(() => {
+        resetBuilderCounters();
+    });
+    describe('Constants', () => {
+        it('should export correct timestamp base', () => {
+            assert.strictEqual(TEST_TIMESTAMP_BASE, 1700000000);
+        });
+        it('should export timestamp as [seconds, nanoseconds]', () => {
+            assert.deepStrictEqual(TEST_TIMESTAMP, [1700000000, 0]);
+        });
+        it('should export ISO timestamp string', () => {
+            assert.strictEqual(TEST_TIMESTAMP_ISO, '2023-11-14T22:13:20.000Z');
+        });
+        it('should export duration as [seconds, nanoseconds]', () => {
+            assert.deepStrictEqual(TEST_DURATION, [2, 500000000]);
+        });
+        it('should export duration in milliseconds', () => {
+            assert.strictEqual(TEST_DURATION_MS, 2500);
+        });
+    });
+    describe('createMockSpan', () => {
+        it('should create span with defaults', () => {
+            const span = createMockSpan();
+            assert.strictEqual(span.traceId, 'trace1');
+            assert.strictEqual(span.spanId, 'span1');
+            assert.strictEqual(span.name, 'test-operation-1');
+            assert.deepStrictEqual(span.startTime, [1700000000, 0]);
+        });
+        it('should increment IDs for each call', () => {
+            const span1 = createMockSpan();
+            const span2 = createMockSpan();
+            assert.strictEqual(span1.traceId, 'trace1');
+            assert.strictEqual(span2.traceId, 'trace2');
+        });
+        it('should allow overrides', () => {
+            const span = createMockSpan({
+                traceId: 'custom-trace',
+                name: 'custom-op',
+                kind: 2,
+                status: { code: 1, message: 'OK' },
+            });
+            assert.strictEqual(span.traceId, 'custom-trace');
+            assert.strictEqual(span.name, 'custom-op');
+            assert.strictEqual(span.kind, 2);
+            assert.deepStrictEqual(span.status, { code: 1, message: 'OK' });
+        });
+        it('should support resource field', () => {
+            const span = createMockSpan({
+                resource: { serviceName: 'test-service', serviceVersion: '1.0.0' },
+            });
+            assert.strictEqual(span.resource?.serviceName, 'test-service');
+        });
+        it('should support attributes field', () => {
+            const span = createMockSpan({
+                attributes: { 'http.method': 'GET', 'http.status_code': 200 },
+            });
+            assert.strictEqual(span.attributes?.['http.method'], 'GET');
+        });
+    });
+    describe('createMockSpans', () => {
+        it('should create multiple spans', () => {
+            const spans = createMockSpans(3);
+            assert.strictEqual(spans.length, 3);
+            assert.strictEqual(spans[0].traceId, 'trace1');
+            assert.strictEqual(spans[1].traceId, 'trace2');
+            assert.strictEqual(spans[2].traceId, 'trace3');
+        });
+        it('should apply static overrides to all spans', () => {
+            const spans = createMockSpans(2, { kind: 1 });
+            assert.strictEqual(spans[0].kind, 1);
+            assert.strictEqual(spans[1].kind, 1);
+        });
+        it('should apply dynamic overrides with index', () => {
+            const spans = createMockSpans(3, (i) => ({
+                name: `op-${i}`,
+                kind: i,
+            }));
+            assert.strictEqual(spans[0].name, 'op-0');
+            assert.strictEqual(spans[0].kind, 0);
+            assert.strictEqual(spans[1].name, 'op-1');
+            assert.strictEqual(spans[1].kind, 1);
+        });
+    });
+    describe('createMockLog', () => {
+        it('should create log with defaults', () => {
+            const log = createMockLog();
+            assert.strictEqual(log.timestamp, TEST_TIMESTAMP_ISO);
+            assert.strictEqual(log.body, 'Test log message 1');
+        });
+        it('should allow overrides', () => {
+            const log = createMockLog({
+                body: 'Custom message',
+                severity: 'ERROR',
+                traceId: 'trace-123',
+            });
+            assert.strictEqual(log.body, 'Custom message');
+            assert.strictEqual(log.severity, 'ERROR');
+            assert.strictEqual(log.traceId, 'trace-123');
+        });
+    });
+    describe('createMockLogs', () => {
+        it('should create multiple logs', () => {
+            const logs = createMockLogs(3);
+            assert.strictEqual(logs.length, 3);
+        });
+        it('should apply dynamic overrides', () => {
+            const severities = ['ERROR', 'WARN', 'INFO'];
+            const logs = createMockLogs(3, (i) => ({
+                severity: severities[i],
+            }));
+            assert.strictEqual(logs[0].severity, 'ERROR');
+            assert.strictEqual(logs[1].severity, 'WARN');
+            assert.strictEqual(logs[2].severity, 'INFO');
+        });
+    });
+    describe('createMockMetric', () => {
+        it('should create metric with defaults', () => {
+            const metric = createMockMetric();
+            assert.strictEqual(metric.timestamp, TEST_TIMESTAMP_ISO);
+            assert.strictEqual(metric.name, 'test.metric.1');
+            assert.strictEqual(metric.value, 100);
+            assert.strictEqual(metric.type, 'counter');
+        });
+        it('should allow overrides', () => {
+            const metric = createMockMetric({
+                name: 'http.requests.total',
+                value: 500,
+                type: 'gauge',
+                unit: 'requests',
+            });
+            assert.strictEqual(metric.name, 'http.requests.total');
+            assert.strictEqual(metric.value, 500);
+            assert.strictEqual(metric.type, 'gauge');
+            assert.strictEqual(metric.unit, 'requests');
+        });
+    });
+    describe('createMockMetrics', () => {
+        it('should create multiple metrics', () => {
+            const metrics = createMockMetrics(3);
+            assert.strictEqual(metrics.length, 3);
+            assert.strictEqual(metrics[0].name, 'test.metric.1');
+            assert.strictEqual(metrics[2].name, 'test.metric.3');
+        });
+    });
+    describe('createMockLLMEvent', () => {
+        it('should create LLM event with defaults', () => {
+            const event = createMockLLMEvent();
+            assert.strictEqual(event.timestamp, TEST_TIMESTAMP_ISO);
+            assert.strictEqual(event.name, 'llm.completion');
+            assert.strictEqual(event.attributes['gen_ai.request.model'], 'claude-3-opus');
+            assert.strictEqual(event.attributes['gen_ai.system'], 'anthropic');
+            assert.strictEqual(event.attributes['gen_ai.usage.input_tokens'], 100);
+            assert.strictEqual(event.attributes['gen_ai.usage.output_tokens'], 50);
+        });
+        it('should merge attribute overrides', () => {
+            const event = createMockLLMEvent({
+                attributes: {
+                    'gen_ai.request.model': 'gpt-4',
+                    'gen_ai.system': 'openai',
+                    'custom.field': 'value',
+                },
+            });
+            assert.strictEqual(event.attributes['gen_ai.request.model'], 'gpt-4');
+            assert.strictEqual(event.attributes['gen_ai.system'], 'openai');
+            assert.strictEqual(event.attributes['custom.field'], 'value');
+        });
+    });
+    describe('createMockLLMEvents', () => {
+        it('should create multiple LLM events', () => {
+            const events = createMockLLMEvents(3);
+            assert.strictEqual(events.length, 3);
+        });
+    });
+    describe('createMockEvaluation', () => {
+        it('should create evaluation with defaults', () => {
+            const evaluation = createMockEvaluation();
+            assert.strictEqual(evaluation.timestamp, TEST_TIMESTAMP_ISO);
+            assert.strictEqual(evaluation.attributes['gen_ai.evaluation.name'], 'Evaluation-1');
+            assert.strictEqual(evaluation.attributes['gen_ai.evaluation.score.value'], 0.85);
+        });
+        it('should merge attribute overrides', () => {
+            const evaluation = createMockEvaluation({
+                traceId: 'trace-abc',
+                attributes: {
+                    'gen_ai.evaluation.name': 'Relevance',
+                    'gen_ai.evaluation.score.value': 0.92,
+                    'gen_ai.evaluation.score.label': 'pass',
+                },
+            });
+            assert.strictEqual(evaluation.traceId, 'trace-abc');
+            assert.strictEqual(evaluation.attributes['gen_ai.evaluation.name'], 'Relevance');
+            assert.strictEqual(evaluation.attributes['gen_ai.evaluation.score.value'], 0.92);
+            assert.strictEqual(evaluation.attributes['gen_ai.evaluation.score.label'], 'pass');
+        });
+    });
+    describe('createMockEvaluations', () => {
+        it('should create multiple evaluations', () => {
+            const evaluations = createMockEvaluations(3);
+            assert.strictEqual(evaluations.length, 3);
+            assert.strictEqual(evaluations[0].attributes['gen_ai.evaluation.name'], 'Evaluation-1');
+            assert.strictEqual(evaluations[2].attributes['gen_ai.evaluation.name'], 'Evaluation-3');
+        });
+    });
+    describe('Specialized builders', () => {
+        describe('createHttpSpan', () => {
+            it('should create HTTP span with method and status', () => {
+                const span = createHttpSpan('GET', 200);
+                assert.strictEqual(span.name, 'http-request');
+                assert.strictEqual(span.kind, 2);
+                assert.strictEqual(span.attributes?.['http.method'], 'GET');
+                assert.strictEqual(span.attributes?.['http.status_code'], 200);
+            });
+            it('should allow additional overrides', () => {
+                const span = createHttpSpan('POST', 201, {
+                    attributes: { 'http.url': '/api/users' },
+                });
+                assert.strictEqual(span.attributes?.['http.method'], 'POST');
+                assert.strictEqual(span.attributes?.['http.url'], '/api/users');
+            });
+        });
+        describe('createDbSpan', () => {
+            it('should create database span with operation and system', () => {
+                const span = createDbSpan('SELECT', 'postgresql');
+                assert.strictEqual(span.name, 'db-select');
+                assert.strictEqual(span.kind, 2);
+                assert.strictEqual(span.attributes?.['db.operation'], 'SELECT');
+                assert.strictEqual(span.attributes?.['db.system'], 'postgresql');
+            });
+        });
+        describe('createErrorLog', () => {
+            it('should create ERROR severity log', () => {
+                const log = createErrorLog('Connection failed');
+                assert.strictEqual(log.body, 'Connection failed');
+                assert.strictEqual(log.severity, 'ERROR');
+            });
+        });
+        describe('createCorrelatedLog', () => {
+            it('should create log with trace correlation', () => {
+                const log = createCorrelatedLog('trace-abc', 'span-def', {
+                    body: 'Correlated message',
+                });
+                assert.strictEqual(log.traceId, 'trace-abc');
+                assert.strictEqual(log.spanId, 'span-def');
+                assert.strictEqual(log.body, 'Correlated message');
+            });
+        });
+        describe('createHistogramMetric', () => {
+            it('should create histogram metric with buckets', () => {
+                const buckets = [
+                    { le: 0.1, count: 10 },
+                    { le: 1.0, count: 25 },
+                    { le: 10.0, count: 100 },
+                ];
+                const metric = createHistogramMetric('request.duration', buckets);
+                assert.strictEqual(metric.name, 'request.duration');
+                assert.strictEqual(metric.type, 'histogram');
+                assert.deepStrictEqual(metric.histogram?.buckets, buckets);
+                assert.strictEqual(metric.histogram?.count, 100);
+            });
+        });
+        describe('createLLMEventWithTokens', () => {
+            it('should create LLM event with specific token counts', () => {
+                const event = createLLMEventWithTokens(500, 200);
+                assert.strictEqual(event.attributes['gen_ai.usage.input_tokens'], 500);
+                assert.strictEqual(event.attributes['gen_ai.usage.output_tokens'], 200);
+            });
+        });
+        describe('createScoredEvaluation', () => {
+            it('should create evaluation with name and score', () => {
+                const evaluation = createScoredEvaluation('Accuracy', 0.95);
+                assert.strictEqual(evaluation.attributes['gen_ai.evaluation.name'], 'Accuracy');
+                assert.strictEqual(evaluation.attributes['gen_ai.evaluation.score.value'], 0.95);
+            });
+        });
+        describe('createLabeledEvaluation', () => {
+            it('should create evaluation with name and label', () => {
+                const evaluation = createLabeledEvaluation('Safety', 'pass');
+                assert.strictEqual(evaluation.attributes['gen_ai.evaluation.name'], 'Safety');
+                assert.strictEqual(evaluation.attributes['gen_ai.evaluation.score.label'], 'pass');
+            });
+        });
+    });
+    describe('resetBuilderCounters', () => {
+        it('should reset all counters', () => {
+            createMockSpan();
+            createMockSpan();
+            createMockLog();
+            resetBuilderCounters();
+            const span = createMockSpan();
+            const log = createMockLog();
+            assert.strictEqual(span.traceId, 'trace1');
+            assert.strictEqual(log.body, 'Test log message 1');
+        });
+    });
+});
+//# sourceMappingURL=test-data-builders.test.js.map

package/dist/test-helpers/test-data-builders.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-data-builders.test.js","sourceRoot":"","sources":["../../src/test-helpers/test-data-builders.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO;AACL,YAAY;AACZ,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,aAAa,EACb,gBAAgB;AAChB,gBAAgB;AAChB,oBAAoB,EACpB,cAAc,EACd,eAAe,EACf,aAAa,EACb,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,oBAAoB,EACpB,qBAAqB;AACrB,uBAAuB;AACvB,cAAc,EACd,YAAY,EACZ,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,wBAAwB,EACxB,sBAAsB,EACtB,uBAAuB,GACxB,MAAM,yBAAyB,CAAC;AAEjC,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,UAAU,CAAC,GAAG,EAAE;QACd,oBAAoB,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,CAAC,WAAW,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,eAAe,CAAC,cAAc,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,CAAC,WAAW,CAAC,kBAAkB,EAAE,0BAA0B,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,CAAC,WAAW,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,IAAI,GAAG,cAAc,EAAE,CAAC;YAC9B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC3C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACzC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;YAClD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,cAAc,EAAE,CAAC;YAC/B,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,IAAI,GAAG,cAAc,CAAC;gBAC1B,OAAO,EAAE,cAAc;gBACvB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,CAAC;gBACP,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;aACnC,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;YACjD,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;YAC3C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,IAAI,GAAG,cAAc,CAAC;gBAC1B,QAAQ,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE,cAAc,EAAE,OAAO,EAAE;aACnE,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,IAAI,GAAG,cAAc,CAAC;gBAC1B,UAAU,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,kBAAkB,EAAE,GAAG,EAAE;aAC9D,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACvC,IAAI,EAAE,MAAM,CAAC,EAAE;gBACf,IAAI,EAAE,CAAC;aACR,CAAC,CAAC,CAAC;YACJ,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;YAC5B,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;YACtD,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,GAAG,GAAG,aAAa,CAAC;gBACxB,IAAI,EAAE,gBAAgB;gBACtB,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,WAAW;aACrB,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;YAC/C,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,IAAI,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,UAAU,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,IAAI,GAAG,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;aACxB,CAAC,CAAC,CAAC;YACJ,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC9C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;YAClC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;YACzD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YACjD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,MAAM,GAAG,gBAAgB,CAAC;gBAC9B,IAAI,EAAE,qBAAqB;gBAC3B,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,UAAU;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;YACvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACzC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YACrD,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,KAAK,GAAG,kBAAkB,EAAE,CAAC;YACnC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;YACxD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;YACjD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,sBAAsB,CAAC,EAAE,eAAe,CAAC,CAAC;YAC9E,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,WAAW,CAAC,CAAC;YACnE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,2BAA2B,CAAC,EAAE,GAAG,CAAC,CAAC;YACvE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,4BAA4B,CAAC,EAAE,EAAE,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,KAAK,GAAG,kBAAkB,CAAC;gBAC/B,UAAU,EAAE;oBACV,sBAAsB,EAAE,OAAO;oBAC/B,eAAe,EAAE,QAAQ;oBACzB,cAAc,EAAE,OAAO;iBACxB;aACF,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,sBAAsB,CAAC,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,QAAQ,CAAC,CAAC;YAChE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,UAAU,GAAG,oBAAoB,EAAE,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;YAC7D,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,cAAc,CAAC,CAAC;YACpF,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE,IAAI,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,UAAU,GAAG,oBAAoB,CAAC;gBACtC,OAAO,EAAE,WAAW;gBACpB,UAAU,EAAE;oBACV,wBAAwB,EAAE,WAAW;oBACrC,+BAA+B,EAAE,IAAI;oBACrC,+BAA+B,EAAE,MAAM;iBACxC;aACF,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YACpD,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,WAAW,CAAC,CAAC;YACjF,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE,IAAI,CAAC,CAAC;YACjF,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE,MAAM,CAAC,CAAC;QACrF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,WAAW,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,cAAc,CAAC,CAAC;YACxF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,cAAc,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;YAC9B,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;gBACxD,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBACxC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;gBAC9C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBACjC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,CAAC,CAAC;gBAC5D,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,kBAAkB,CAAC,EAAE,GAAG,CAAC,CAAC;YACjE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;gBAC3C,MAAM,IAAI,GAAG,cAAc,CAAC,MAAM,EAAE,GAAG,EAAE;oBACvC,UAAU,EAAE,EAAE,UAAU,EAAE,YAAY,EAAE;iBACzC,CAAC,CAAC;gBACH,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC,CAAC;gBAC7D,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,UAAU,CAAC,EAAE,YAAY,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;YAC5B,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;gBAC/D,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;gBAClD,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBAC3C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBACjC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,cAAc,CAAC,EAAE,QAAQ,CAAC,CAAC;gBAChE,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,WAAW,CAAC,EAAE,YAAY,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;YAC9B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;gBAC1C,MAAM,GAAG,GAAG,cAAc,CAAC,mBAAmB,CAAC,CAAC;gBAChD,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;gBAClD,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;YACnC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;gBAClD,MAAM,GAAG,GAAG,mBAAmB,CAAC,WAAW,EAAE,UAAU,EAAE;oBACvD,IAAI,EAAE,oBAAoB;iBAC3B,CAAC,CAAC;gBACH,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;gBAC7C,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBAC3C,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;YACrC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;gBACrD,MAAM,OAAO,GAAG;oBACd,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE;oBACtB,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE;oBACtB,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE;iBACzB,CAAC;gBACF,MAAM,MAAM,GAAG,qBAAqB,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;gBAClE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;gBACpD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBAC7C,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC3D,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;YACxC,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;gBAC5D,MAAM,KAAK,GAAG,wBAAwB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBACjD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,2BAA2B,CAAC,EAAE,GAAG,CAAC,CAAC;gBACvE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,4BAA4B,CAAC,EAAE,GAAG,CAAC,CAAC;YAC1E,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;YACtC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;gBACtD,MAAM,UAAU,GAAG,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBAC5D,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,UAAU,CAAC,CAAC;gBAChF,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE,IAAI,CAAC,CAAC;YACnF,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACvC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;gBACtD,MAAM,UAAU,GAAG,uBAAuB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;gBAC7D,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,QAAQ,CAAC,CAAC;gBAC9E,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,UAAU,CAAC,+BAA+B,CAAC,EAAE,MAAM,CAAC,CAAC;YACrF,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,cAAc,EAAE,CAAC;YACjB,cAAc,EAAE,CAAC;YACjB,aAAa,EAAE,CAAC;YAChB,oBAAoB,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,cAAc,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,aAAa,EAAE,CAAC;YAC5B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC3C,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test-helpers/tool-validators.d.ts

@@ -0,0 +1,28 @@
+/**
+ * Shared test utilities for validating MCP tool definitions
+ */
+/**
+ * Tool definition interface for validation
+ */
+export interface ToolDefinition {
+    name: string;
+    description: string;
+    inputSchema: unknown;
+    handler: unknown;
+}
+/**
+ * Generate standard tool definition tests
+ * @param tool - The tool to validate
+ * @param expectedName - Expected tool name (e.g., 'obs_query_traces')
+ * @param schema - Expected input schema reference
+ * @param handler - Expected handler function reference
+ * @param requiredDescriptionTerms - Optional terms that must appear in description
+ */
+export declare function validateToolDefinition(tool: ToolDefinition, expectedName: string, schema: unknown, handler: unknown, requiredDescriptionTerms?: string[]): void;
+/**
+ * Generate handler export validation tests
+ * @param handler - The handler function to validate
+ * @param handlerName - Name for test descriptions
+ */
+export declare function validateHandlerExport(handler: unknown, handlerName: string): void;
+//# sourceMappingURL=tool-validators.d.ts.map

package/dist/test-helpers/tool-validators.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-validators.d.ts","sourceRoot":"","sources":["../../src/test-helpers/tool-validators.ts"],"names":[],"mappings":"AAAA;;GAEG;AAKH;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED;;;;;;;GAOG;AACH,wBAAgB,sBAAsB,CACpC,IAAI,EAAE,cAAc,EACpB,YAAY,EAAE,MAAM,EACpB,MAAM,EAAE,OAAO,EACf,OAAO,EAAE,OAAO,EAChB,wBAAwB,CAAC,EAAE,MAAM,EAAE,GAClC,IAAI,CAmCN;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,WAAW,EAAE,MAAM,GAClB,IAAI,CAON"}

package/dist/test-helpers/tool-validators.js

@@ -0,0 +1,56 @@
+/**
+ * Shared test utilities for validating MCP tool definitions
+ */
+import { describe, it } from 'node:test';
+import * as assert from 'node:assert';
+/**
+ * Generate standard tool definition tests
+ * @param tool - The tool to validate
+ * @param expectedName - Expected tool name (e.g., 'obs_query_traces')
+ * @param schema - Expected input schema reference
+ * @param handler - Expected handler function reference
+ * @param requiredDescriptionTerms - Optional terms that must appear in description
+ */
+export function validateToolDefinition(tool, expectedName, schema, handler, requiredDescriptionTerms) {
+    describe('tool definition', () => {
+        it('should have correct tool name', () => {
+            assert.strictEqual(tool.name, expectedName);
+        });
+        it('should have description', () => {
+            assert.ok(tool.description);
+            assert.strictEqual(typeof tool.description, 'string');
+            assert.ok(tool.description.length > 0);
+        });
+        it('should have inputSchema', () => {
+            assert.ok(tool.inputSchema);
+            assert.strictEqual(tool.inputSchema, schema);
+        });
+        it('should have handler function', () => {
+            assert.ok(tool.handler);
+            assert.strictEqual(typeof tool.handler, 'function');
+            assert.strictEqual(tool.handler, handler);
+        });
+        if (requiredDescriptionTerms && requiredDescriptionTerms.length > 0) {
+            it('should have description mentioning key features', () => {
+                const desc = tool.description.toLowerCase();
+                for (const term of requiredDescriptionTerms) {
+                    assert.ok(desc.includes(term.toLowerCase()), `Description should include "${term}"`);
+                }
+            });
+        }
+    });
+}
+/**
+ * Generate handler export validation tests
+ * @param handler - The handler function to validate
+ * @param handlerName - Name for test descriptions
+ */
+export function validateHandlerExport(handler, handlerName) {
+    describe('handler function', () => {
+        it('should be exported', () => {
+            assert.ok(handler);
+            assert.strictEqual(typeof handler, 'function');
+        });
+    });
+}
+//# sourceMappingURL=tool-validators.js.map

package/dist/test-helpers/tool-validators.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-validators.js","sourceRoot":"","sources":["../../src/test-helpers/tool-validators.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAYtC;;;;;;;GAOG;AACH,MAAM,UAAU,sBAAsB,CACpC,IAAoB,EACpB,YAAoB,EACpB,MAAe,EACf,OAAgB,EAChB,wBAAmC;IAEnC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5B,MAAM,CAAC,WAAW,CAAC,OAAO,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACtD,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxB,MAAM,CAAC,WAAW,CAAC,OAAO,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACpD,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAI,wBAAwB,IAAI,wBAAwB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpE,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;gBACzD,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;gBAC5C,KAAK,MAAM,IAAI,IAAI,wBAAwB,EAAE,CAAC;oBAC5C,MAAM,CAAC,EAAE,CACP,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EACjC,+BAA+B,IAAI,GAAG,CACvC,CAAC;gBACJ,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACnC,OAAgB,EAChB,WAAmB;IAEnB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAC5B,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;YACnB,MAAM,CAAC,WAAW,CAAC,OAAO,OAAO,EAAE,UAAU,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/tools/context-stats.d.ts

@@ -37,6 +37,7 @@ export declare function contextStats(rawInput: ContextStatsInput): Promise<{
         input: string;
         output: string;
         total: string;
+        warning?: string;
     };
     breakdown?: {
         systemPrompt: number;

package/dist/tools/context-stats.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"context-stats.d.ts","sourceRoot":"","sources":["../../src/tools/context-stats.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,eAAO,MAAM,kBAAkB;;;;;;;;;;;;EAI7B,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAsJnE,wBAAsB,YAAY,CAAC,QAAQ,EAAE,iBAAiB;eAqC/C,MAAM;aACR,MAAM;YACP,MAAM;aACL;QACP,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,gBAAgB,EAAE,MAAM,CAAC;KAC1B;YACO;QACN,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,aAAa,EAAE,MAAM,CAAC;KACvB;cACS,MAAM;WACT;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE;gBAC3C;QACV,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;KACf;cACS;QACR,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,UAAU,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAC;YAAC,YAAY,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC9E;;;;;;;GA+DJ;AAED,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;CAK5B,CAAC"}
+{"version":3,"file":"context-stats.d.ts","sourceRoot":"","sources":["../../src/tools/context-stats.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,eAAO,MAAM,kBAAkB;;;;;;;;;;;;EAI7B,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAsJnE,wBAAsB,YAAY,CAAC,QAAQ,EAAE,iBAAiB;eAqC/C,MAAM;aACR,MAAM;YACP,MAAM;aACL;QACP,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,gBAAgB,EAAE,MAAM,CAAC;KAC1B;YACO;QACN,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,aAAa,EAAE,MAAM,CAAC;KACvB;cACS,MAAM;WACT;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE;gBAC7D;QACV,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;KACf;cACS;QACR,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,UAAU,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAC;YAAC,YAAY,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC9E;;;;;;;GAmEJ;AAED,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;CAK5B,CAAC"}

package/dist/tools/context-stats.js

@@ -4,12 +4,12 @@
 import { z } from 'zod';
 import { existsSync, readFileSync, readdirSync, statSync } from 'fs';
 import { join } from 'path';
-import { CLAUDE_HOME, CONTEXT_WINDOW_SIZE, MODEL_PRICING } from '../lib/constants.js';
+import { CLAUDE_HOME, CONTEXT_WINDOW_SIZE, MODEL_PRICING, checkPricingStaleness } from '../lib/constants.js';
 import { loadJsonSafe } from '../lib/file-utils.js';
 export const contextStatsSchema = z.object({
-    sessionId: z.string().optional().describe('Specific session ID to analyze (defaults to most recent)'),
-    includeBreakdown: z.boolean().default(true).describe('Include token breakdown by category'),
-    includeCost: z.boolean().default(true).describe('Include cost estimation'),
+    sessionId: z.string().optional().describe('Session ID (default: most recent)'),
+    includeBreakdown: z.boolean().default(true).describe('Include token breakdown'),
+    includeCost: z.boolean().default(true).describe('Include cost estimate'),
 });
 /**
  * Find most recent session file
@@ -169,10 +169,14 @@ export async function contextStats(rawInput) {
     };
     if (input.includeCost) {
         const cost = calculateCost(usage);
+        const staleness = checkPricingStaleness();
         result.cost = {
             input: `$${cost.inputCost.toFixed(4)}`,
             output: `$${cost.outputCost.toFixed(4)}`,
             total: `$${cost.totalCost.toFixed(4)}`,
+            ...(staleness.isStale && {
+                warning: `Pricing data is ${staleness.daysSinceUpdate} days old. Costs may be inaccurate.`,
+            }),
         };
     }
     if (input.includeBreakdown) {
@@ -207,7 +211,7 @@ export async function contextStats(rawInput) {
 }
 export const contextStatsTool = {
     name: 'obs_context_stats',
-    description: 'Get context window utilization statistics for a Claude Code session. Shows token usage, cost estimates, and historical trends.',
+    description: 'Get context window utilization, token usage, and cost estimates.',
     inputSchema: contextStatsSchema,
     handler: contextStats,
 };

package/dist/tools/context-stats.js.map

@@ -1 +1 @@
-{"version":3,"file":"context-stats.js","sourceRoot":"","sources":["../../src/tools/context-stats.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACrE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAmB,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACvG,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpD,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0DAA0D,CAAC;IACrG,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,qCAAqC,CAAC;IAC3F,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,yBAAyB,CAAC;CAC3E,CAAC,CAAC;AAkCH;;GAEG;AACH,SAAS,iBAAiB;IACxB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAClD,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,IAAI,CAAC;IAE1C,IAAI,UAAU,GAAsF,IAAI,CAAC;IAEzG,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC7C,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;YAClC,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAAE,SAAS;YAElC,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC5E,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBAC5C,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBACvC,IAAI,CAAC,UAAU,IAAI,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,KAAK,EAAE,CAAC;oBAClE,UAAU,GAAG;wBACX,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;wBACxC,UAAU,EAAE,OAAO;wBACnB,QAAQ;wBACR,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE;qBACnC,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACnI,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,QAAgB;IACzC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,IAAI,CAAC;IAEvC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEzC,IAAI,WAAW,GAAwB,IAAI,CAAC;QAC5C,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,KAAyB,CAAC;QAE9B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAE/B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACxD,YAAY,EAAE,CAAC;gBACjB,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;oBACzB,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBAC9B,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;oBACzB,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;oBAClC,MAAM,WAAW,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;oBAC5C,MAAM,YAAY,GAAG,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;oBAC9C,MAAM,eAAe,GAAG,KAAK,CAAC,uBAAuB,IAAI,CAAC,CAAC;oBAC3D,MAAM,mBAAmB,GAAG,KAAK,CAAC,2BAA2B,IAAI,CAAC,CAAC;oBAEnE,MAAM,kBAAkB,GAAG,WAAW,GAAG,eAAe,GAAG,mBAAmB,CAAC;oBAC/E,MAAM,kBAAkB,GAAG,CAAC,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,GAAG,CAAC;oBAC5E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,mBAAmB,GAAG,kBAAkB,CAAC,CAAC;oBAExE,WAAW,GAAG;wBACZ,WAAW;wBACX,YAAY;wBACZ,eAAe;wBACf,mBAAmB;wBACnB,kBAAkB;wBAClB,kBAAkB,EAAE,IAAI,CAAC,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC,GAAG,EAAE;wBAC5D,SAAS;wBACT,YAAY;wBACZ,KAAK;qBACN,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,uBAAuB;YACzB,CAAC;QACH,CAAC;QAED,IAAI,WAAW,EAAE,CAAC;YAChB,WAAW,CAAC,YAAY,GAAG,YAAY,CAAC;QAC1C,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,KAAmB;IACxC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,0BAA0B,CAAC;IACxD,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IAErE,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,SAAS,CAAC;IAClE,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;IAErE,OAAO;QACL,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,KAAK;QAChD,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC,GAAG,KAAK;QAClD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,KAAK,CAAC,GAAG,KAAK;KAChE,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAA2B;IAC5D,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACjD,eAAe;IACf,IAAI,WAAW,GAAuE,IAAI,CAAC;IAE3F,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACpB,8BAA8B;QAC9B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QAClD,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YAC7C,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC,SAAS,QAAQ,CAAC,CAAC;gBACxE,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACzB,WAAW,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;oBAC5E,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,WAAW,GAAG,iBAAiB,EAAE,CAAC;IACpC,CAAC;IAED,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,sBAAsB,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,mBAAmB;SACvF,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,iBAAiB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACtD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO;YACL,SAAS,EAAE,WAAW,CAAC,SAAS;YAChC,KAAK,EAAE,oCAAoC;SAC5C,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GA6BR;QACF,SAAS,EAAE,WAAW,CAAC,SAAS;QAChC,OAAO,EAAE,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;QAClD,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,OAAO,EAAE;YACP,WAAW,EAAE,KAAK,CAAC,kBAAkB;YACrC,WAAW,EAAE,GAAG,KAAK,CAAC,kBAAkB,GAAG;YAC3C,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,gBAAgB,EAAE,GAAG,CAAC,GAAG,GAAG,KAAK,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;SACpE;QACD,MAAM,EAAE;YACN,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,MAAM,EAAE,KAAK,CAAC,YAAY;YAC1B,SAAS,EAAE,KAAK,CAAC,eAAe;YAChC,aAAa,EAAE,KAAK,CAAC,mBAAmB;SACzC;QACD,QAAQ,EAAE,KAAK,CAAC,YAAY;KAC7B,CAAC;IAEF,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACtB,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,IAAI,GAAG;YACZ,KAAK,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACtC,MAAM,EAAE,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACxC,KAAK,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;SACvC,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC3B,+CAA+C;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC;QAC1B,MAAM,WAAW,GAAG,KAAK,CAAC;QAC1B,MAAM,QAAQ,GAAG,YAAY,GAAG,WAAW,CAAC;QAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,kBAAkB,GAAG,QAAQ,CAAC,CAAC;QAElE,MAAM,CAAC,SAAS,GAAG;YACjB,YAAY;YACZ,WAAW;YACX,QAAQ;YACR,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,kBAAkB,GAAG,YAAY,GAAG,WAAW,GAAG,QAAQ,CAAC;SACrF,CAAC;IACJ,CAAC;IAED,uBAAuB;IACvB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,sBAAsB,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,YAAY,CAAiB,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAE3G,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;QAEpH,MAAM,CAAC,OAAO,GAAG;YACf,aAAa,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM;YACtC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,EAAE,CAAC,GAAG,EAAE;YACpD,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpD,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;gBAClC,YAAY,EAAE,CAAC,CAAC,YAAY;aAC7B,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,IAAI,EAAE,mBAAmB;IACzB,WAAW,EAAE,gIAAgI;IAC7I,WAAW,EAAE,kBAAkB;IAC/B,OAAO,EAAE,YAAY;CACtB,CAAC"}
+{"version":3,"file":"context-stats.js","sourceRoot":"","sources":["../../src/tools/context-stats.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACrE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,mBAAmB,EAAmB,aAAa,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC9H,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpD,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IAC9E,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IAC/E,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;CACzE,CAAC,CAAC;AAkCH;;GAEG;AACH,SAAS,iBAAiB;IACxB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAClD,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,IAAI,CAAC;IAE1C,IAAI,UAAU,GAAsF,IAAI,CAAC;IAEzG,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC7C,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;YAClC,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAAE,SAAS;YAElC,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC5E,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBAC5C,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBACvC,IAAI,CAAC,UAAU,IAAI,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,KAAK,EAAE,CAAC;oBAClE,UAAU,GAAG;wBACX,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;wBACxC,UAAU,EAAE,OAAO;wBACnB,QAAQ;wBACR,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE;qBACnC,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,UAAU,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACnI,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,QAAgB;IACzC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,IAAI,CAAC;IAEvC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEzC,IAAI,WAAW,GAAwB,IAAI,CAAC;QAC5C,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,KAAyB,CAAC;QAE9B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAE/B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACxD,YAAY,EAAE,CAAC;gBACjB,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;oBACzB,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBAC9B,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;oBACzB,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;oBAClC,MAAM,WAAW,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;oBAC5C,MAAM,YAAY,GAAG,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;oBAC9C,MAAM,eAAe,GAAG,KAAK,CAAC,uBAAuB,IAAI,CAAC,CAAC;oBAC3D,MAAM,mBAAmB,GAAG,KAAK,CAAC,2BAA2B,IAAI,CAAC,CAAC;oBAEnE,MAAM,kBAAkB,GAAG,WAAW,GAAG,eAAe,GAAG,mBAAmB,CAAC;oBAC/E,MAAM,kBAAkB,GAAG,CAAC,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,GAAG,CAAC;oBAC5E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,mBAAmB,GAAG,kBAAkB,CAAC,CAAC;oBAExE,WAAW,GAAG;wBACZ,WAAW;wBACX,YAAY;wBACZ,eAAe;wBACf,mBAAmB;wBACnB,kBAAkB;wBAClB,kBAAkB,EAAE,IAAI,CAAC,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC,GAAG,EAAE;wBAC5D,SAAS;wBACT,YAAY;wBACZ,KAAK;qBACN,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,uBAAuB;YACzB,CAAC;QACH,CAAC;QAED,IAAI,WAAW,EAAE,CAAC;YAChB,WAAW,CAAC,YAAY,GAAG,YAAY,CAAC;QAC1C,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,KAAmB;IACxC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,0BAA0B,CAAC;IACxD,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IAErE,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,SAAS,CAAC;IAClE,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;IAErE,OAAO;QACL,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,KAAK;QAChD,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC,GAAG,KAAK;QAClD,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,KAAK,CAAC,GAAG,KAAK;KAChE,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAA2B;IAC5D,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACjD,eAAe;IACf,IAAI,WAAW,GAAuE,IAAI,CAAC;IAE3F,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACpB,8BAA8B;QAC9B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QAClD,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YAC7C,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC,SAAS,QAAQ,CAAC,CAAC;gBACxE,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACzB,WAAW,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;oBAC5E,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,WAAW,GAAG,iBAAiB,EAAE,CAAC;IACpC,CAAC;IAED,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,sBAAsB,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,mBAAmB;SACvF,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,iBAAiB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACtD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO;YACL,SAAS,EAAE,WAAW,CAAC,SAAS;YAChC,KAAK,EAAE,oCAAoC;SAC5C,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GA6BR;QACF,SAAS,EAAE,WAAW,CAAC,SAAS;QAChC,OAAO,EAAE,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;QAClD,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,OAAO,EAAE;YACP,WAAW,EAAE,KAAK,CAAC,kBAAkB;YACrC,WAAW,EAAE,GAAG,KAAK,CAAC,kBAAkB,GAAG;YAC3C,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,gBAAgB,EAAE,GAAG,CAAC,GAAG,GAAG,KAAK,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;SACpE;QACD,MAAM,EAAE;YACN,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,MAAM,EAAE,KAAK,CAAC,YAAY;YAC1B,SAAS,EAAE,KAAK,CAAC,eAAe;YAChC,aAAa,EAAE,KAAK,CAAC,mBAAmB;SACzC;QACD,QAAQ,EAAE,KAAK,CAAC,YAAY;KAC7B,CAAC;IAEF,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACtB,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;QAC1C,MAAM,CAAC,IAAI,GAAG;YACZ,KAAK,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACtC,MAAM,EAAE,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACxC,KAAK,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACtC,GAAG,CAAC,SAAS,CAAC,OAAO,IAAI;gBACvB,OAAO,EAAE,mBAAmB,SAAS,CAAC,eAAe,qCAAqC;aAC3F,CAAC;SACH,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC3B,+CAA+C;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC;QAC1B,MAAM,WAAW,GAAG,KAAK,CAAC;QAC1B,MAAM,QAAQ,GAAG,YAAY,GAAG,WAAW,CAAC;QAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,kBAAkB,GAAG,QAAQ,CAAC,CAAC;QAElE,MAAM,CAAC,SAAS,GAAG;YACjB,YAAY;YACZ,WAAW;YACX,QAAQ;YACR,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,kBAAkB,GAAG,YAAY,GAAG,WAAW,GAAG,QAAQ,CAAC;SACrF,CAAC;IACJ,CAAC;IAED,uBAAuB;IACvB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,sBAAsB,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,YAAY,CAAiB,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;IAE3G,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;QAEpH,MAAM,CAAC,OAAO,GAAG;YACf,aAAa,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM;YACtC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,EAAE,CAAC,GAAG,EAAE;YACpD,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpD,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;gBAClC,YAAY,EAAE,CAAC,CAAC,YAAY;aAC7B,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,IAAI,EAAE,mBAAmB;IACzB,WAAW,EAAE,kEAAkE;IAC/E,WAAW,EAAE,kBAAkB;IAC/B,OAAO,EAAE,YAAY;CACtB,CAAC"}

package/dist/tools/context-stats.test.js

@@ -1,24 +1,25 @@
 /**
  * Unit tests for context-stats tool
  */
-import { describe, it, after } from 'node:test';
+import { describe, it, beforeEach, afterEach, after } from 'node:test';
 import * as assert from 'node:assert';
 import * as fs from 'fs';
 import * as path from 'path';
 import * as os from 'os';
+import { removeTempDir, resetCleanupFailureCount, assertNoCleanupFailures, } from '../test-helpers/file-utils.js';
+import { setHomeEnv } from '../test-helpers/env-utils.js';
 // Set HOME before importing to ensure constants are set correctly
-const originalHome = process.env.HOME;
 const testHomeBase = fs.mkdtempSync(path.join(os.tmpdir(), 'context-stats-test-'));
-process.env.HOME = testHomeBase;
+const homeScope = setHomeEnv(testHomeBase);
 // Now import tool which imports constants with the correct HOME
 import { contextStats, contextStatsSchema } from './context-stats.js';
 import { CLAUDE_HOME } from '../lib/constants.js';
 // Type helpers for result checking
 function hasError(result) {
-    return 'error' in result && result.error !== undefined;
+    return typeof result === 'object' && result !== null && 'error' in result;
 }
 function isSuccess(result) {
-    return 'sessionId' in result && 'project' in result;
+    return typeof result === 'object' && result !== null && 'sessionId' in result && 'project' in result;
 }
 // Helper to call contextStats with default options
 async function callContextStats(input) {
@@ -45,14 +46,27 @@ function createMockSessionFile(project, sessionId, usage, model = 'claude-sonnet
     return filePath;
 }
 describe('contextStats', () => {
-    after(() => {
-        if (fs.existsSync(testHomeBase)) {
-            fs.rmSync(testHomeBase, { recursive: true });
+    beforeEach(() => {
+        resetCleanupFailureCount();
+        // Clean up projects directory before each test for isolation
+        const projectsDir = path.join(CLAUDE_HOME, 'projects');
+        if (fs.existsSync(projectsDir)) {
+            fs.rmSync(projectsDir, { recursive: true });
         }
-        if (originalHome) {
-            process.env.HOME = originalHome;
+        // Clean up history file
+        const historyFile = path.join(CLAUDE_HOME, 'context-history.json');
+        if (fs.existsSync(historyFile)) {
+            fs.unlinkSync(historyFile);
         }
     });
+    afterEach(() => {
+        assertNoCleanupFailures();
+    });
+    // Final cleanup after all tests
+    after(() => {
+        removeTempDir(testHomeBase);
+        homeScope.restore();
+    });
     it('should validate input schema with defaults', async () => {
         const input = {};
         const result = contextStatsSchema.parse(input);