john
91 lines
2.8 KiB
JavaScript
91 lines
2.8 KiB
JavaScript
import assert from 'node:assert/strict';
|
|
import path from 'node:path';
|
|
import test from 'node:test';
|
|
import { parseMemoryV2StackArgs, runMemoryV2StackCli } from './check-memory-v2-stack.mjs';
|
|
|
|
function writableBuffer() {
|
|
let value = '';
|
|
return {
|
|
write(chunk) {
|
|
value += String(chunk);
|
|
},
|
|
value() {
|
|
return value;
|
|
},
|
|
};
|
|
}
|
|
|
|
test('parseMemoryV2StackArgs maps aggregate options', () => {
|
|
assert.deepEqual(
|
|
parseMemoryV2StackArgs([
|
|
'--base-url',
|
|
'http://127.0.0.1:18081/',
|
|
'--skip-session-flow',
|
|
'--skip-backend-smokes',
|
|
'--expect-backend',
|
|
'pgvector',
|
|
'--expect-selected-backend',
|
|
'pgvector',
|
|
'--timeout-ms',
|
|
'5000',
|
|
], {}),
|
|
{
|
|
baseUrl: 'http://127.0.0.1:18081',
|
|
envFile: path.join(process.cwd(), '.env'),
|
|
runAppCanary: true,
|
|
runSessionFlow: false,
|
|
runBackendSmokes: false,
|
|
expectedBackend: 'pgvector',
|
|
expectedSelectedBackend: 'pgvector',
|
|
timeoutMs: 5000,
|
|
help: false,
|
|
},
|
|
);
|
|
});
|
|
|
|
test('runMemoryV2StackCli aggregates configured and skipped backend checks', async () => {
|
|
const stdout = writableBuffer();
|
|
const code = await runMemoryV2StackCli({
|
|
argv: ['--skip-app-canary', '--skip-session-flow'],
|
|
env: {
|
|
MEMORY_BACKEND: 'pgvector',
|
|
MEMORY_VECTOR_ENABLED: '1',
|
|
MEMORY_PGVECTOR_DATABASE_URL: 'postgresql://db',
|
|
MEMORY_PGVECTOR_EMBEDDING_MODULE: './embed.mjs',
|
|
MEMORY_QDRANT_ENABLED: '1',
|
|
MEMORY_QDRANT_URL: 'http://qdrant.local',
|
|
MEMORY_QDRANT_EMBEDDING_MODULE: './embed.mjs',
|
|
},
|
|
stdout,
|
|
runners: {
|
|
async runContractCli({ stdout: buffer }) {
|
|
buffer.write(JSON.stringify({ ok: true, summary: { backendCount: 9 } }));
|
|
return 0;
|
|
},
|
|
async runHealthCli({ stdout: buffer }) {
|
|
buffer.write(JSON.stringify({ ok: true, summary: { backend: 'pgvector', selectedBackend: 'pgvector' } }));
|
|
return 0;
|
|
},
|
|
async runPgvectorSmokeCli() {
|
|
return { ok: true, source: 'pgvector-smoke' };
|
|
},
|
|
async runQdrantSmokeCli() {
|
|
return { ok: true, source: 'qdrant-raw' };
|
|
},
|
|
async runExternalSmokeCli({ argv: cliArgv }) {
|
|
return { ok: true, backend: cliArgv[1], source: 'runtime-smoke' };
|
|
},
|
|
},
|
|
});
|
|
const report = JSON.parse(stdout.value());
|
|
|
|
assert.equal(code, 0);
|
|
assert.equal(report.ok, true);
|
|
assert.deepEqual(report.summary.configuredBackends, ['pgvector', 'qdrant']);
|
|
assert.ok(report.summary.skippedBackends.includes('weaviate'));
|
|
assert.equal(report.checks.find((item) => item.name === 'contracts').ok, true);
|
|
assert.equal(report.checks.find((item) => item.name === 'health').ok, true);
|
|
assert.equal(report.checks.find((item) => item.name === 'backend_pgvector').ok, true);
|
|
assert.equal(report.checks.find((item) => item.name === 'backend_qdrant').ok, true);
|
|
});
|