sh-mcp/packages/knowledge-base/test/dev-client.test.ts
Adam Moussa a60a5a5794 Add security-weighted test suite + coverage gate; wire tooling
Add tests for the highest-risk surface (build-plan §5, design.md §7.3):
tool-hiding, server-side scope enforcement (incl. forced hidden calls),
audience binding, input-schema validation, finance redaction on egress, audit
emission with hashed args, prompt-injection regression (tool output is data),
rate limiting, MCP conformance (in-memory transport round-trip), OpenAPI 3.1
validity, and local-auth safety. Add HTTP integration tests (supertest) for both
servers and per-package dev-client tests. 405 tests pass.

Wire the coverage gate into vitest.config.ts: 80% overall, with per-file
thresholds on the auth + dispatch crown jewels; exclude deferred real client
stubs, entrypoints, cdk apps, and aws-only config from the gate (documented).
Extend eslint flat config + add .prettierignore to cover servers/. Commit the
updated package-lock.json.
2026-06-26 12:48:26 -04:00

28 lines
1.1 KiB
TypeScript

import { describe, it, expect } from 'vitest';
import { InMemoryKnowledgeBaseClient } from '../src/dev-client.js';
describe('InMemoryKnowledgeBaseClient', () => {
describe('retrieve', () => {
it('returns the seeded passages', async () => {
const client = new InMemoryKnowledgeBaseClient();
const results = await client.retrieve({ query: 'vendor onboarding' });
expect(results).toHaveLength(3);
expect(results[0].source).toBe('s3://sh-mcp-kb/handbook/vendor-onboarding.md');
expect(results[0].score).toBe(0.92);
expect(results[0].passage).toContain('W-9 intake form');
});
it('respects maxResults', async () => {
const client = new InMemoryKnowledgeBaseClient();
const results = await client.retrieve({ query: 'anything', maxResults: 2 });
expect(results).toHaveLength(2);
});
it('defaults to up to 5 results when maxResults is omitted', async () => {
const client = new InMemoryKnowledgeBaseClient();
const results = await client.retrieve({ query: 'anything' });
expect(results.length).toBeLessThanOrEqual(5);
expect(results).toHaveLength(3);
});
});
});