78 lines
3.6 KiB
JavaScript
78 lines
3.6 KiB
JavaScript
import test from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import fs from 'node:fs/promises';
|
|
import os from 'node:os';
|
|
import path from 'node:path';
|
|
import { parseArgs, run } from './stage-legacy-knowledge.mjs';
|
|
|
|
test('dry-run is the default and does not call the mutation endpoint', async () => {
|
|
const directory = await fs.mkdtemp(path.join(os.tmpdir(), 'aihr-legacy-dry-'));
|
|
const options = parseArgs(['--manifest', path.join(directory, 'manifest.json')]);
|
|
const result = await run(options, { fetchImpl: async () => assert.fail('fetch must not be called') });
|
|
assert.equal(result.mode, 'DRY_RUN');
|
|
assert.equal(result.preview, null);
|
|
await assert.rejects(fs.access(options.manifest));
|
|
});
|
|
|
|
test('authenticated dry-run calls only the read-only preview endpoint', async () => {
|
|
const directory = await fs.mkdtemp(path.join(os.tmpdir(), 'aihr-legacy-preview-'));
|
|
const options = parseArgs([
|
|
'--token', 'preview-secret', '--manifest', path.join(directory, 'manifest.json'), '--batch-size', '20'
|
|
]);
|
|
let requestedUrl;
|
|
const result = await run(options, { fetchImpl: async (url, request) => {
|
|
requestedUrl = String(url);
|
|
assert.equal(request.method, 'GET');
|
|
return { ok: true, status: 200, json: async () => ({ code: 200, data: {
|
|
discovered: 20, sourceAvailable: 18, sourceUnavailable: 2, nextCursor: 42, moreAvailable: true
|
|
} }) };
|
|
} });
|
|
assert.match(requestedUrl, /\/api\/knowledge\/quality\/legacy\/preview/);
|
|
assert.equal(result.preview.sourceUnavailable, 2);
|
|
await assert.rejects(fs.access(options.manifest));
|
|
});
|
|
|
|
test('execute writes a resumable manifest without credentials', async () => {
|
|
const directory = await fs.mkdtemp(path.join(os.tmpdir(), 'aihr-legacy-run-'));
|
|
const options = parseArgs([
|
|
'--execute', '--token', 'top-secret', '--manifest', path.join(directory, 'manifest.json'),
|
|
'--max-batches', '2', '--batch-size', '20'
|
|
]);
|
|
const replies = [
|
|
{ discovered: 20, staged: 20, reparsed: 18, quarantined: 2, stagedAssetIds: [11, 12],
|
|
failedAttachmentIds: [], nextCursor: 100, moreAvailable: true },
|
|
{ discovered: 3, staged: 3, reparsed: 2, quarantined: 1, stagedAssetIds: [13],
|
|
failedAttachmentIds: [], nextCursor: 103, moreAvailable: false }
|
|
];
|
|
let calls = 0;
|
|
const result = await run(options, {
|
|
fetchImpl: async (_url, request) => {
|
|
assert.match(request.headers.Authorization, /top-secret/);
|
|
return { ok: true, status: 200, json: async () => ({ code: 200, data: replies[calls++] }) };
|
|
},
|
|
now: () => new Date('2026-08-01T00:00:00Z')
|
|
});
|
|
const manifestText = await fs.readFile(options.manifest, 'utf8');
|
|
const manifest = JSON.parse(manifestText);
|
|
assert.equal(result.status, 'COMPLETE');
|
|
assert.equal(manifest.cursor, 103);
|
|
assert.equal(manifest.batches.length, 2);
|
|
assert.doesNotMatch(manifestText, /top-secret/);
|
|
});
|
|
|
|
test('failed attachment does not advance the recovery cursor', async () => {
|
|
const directory = await fs.mkdtemp(path.join(os.tmpdir(), 'aihr-legacy-fail-'));
|
|
const options = parseArgs([
|
|
'--execute', '--token', 'secret', '--manifest', path.join(directory, 'manifest.json'), '--batch-size', '20'
|
|
]);
|
|
await assert.rejects(run(options, {
|
|
fetchImpl: async () => ({ ok: true, status: 200, json: async () => ({ code: 200, data: {
|
|
discovered: 2, staged: 1, reparsed: 1, quarantined: 0, stagedAssetIds: [11],
|
|
failedAttachmentIds: [42], nextCursor: 42, moreAvailable: true
|
|
} }) })
|
|
}), /cursor was not advanced/);
|
|
const manifest = JSON.parse(await fs.readFile(options.manifest, 'utf8'));
|
|
assert.equal(manifest.status, 'FAILED');
|
|
assert.equal(manifest.cursor, 0);
|
|
});
|