john
48 lines
2.1 KiB
JavaScript
48 lines
2.1 KiB
JavaScript
import test from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import { createMemoryV2LifecycleService, resolveMemoryV2LifecyclePolicy } from './memory-v2-lifecycle.mjs';
|
|
|
|
test('lifecycle policy defaults to safe disabled workers', () => {
|
|
const policy = resolveMemoryV2LifecyclePolicy({});
|
|
assert.equal(policy.enabled, false);
|
|
assert.equal(policy.promotionEnabled, false);
|
|
assert.equal(policy.compactionEnabled, false);
|
|
assert.equal(policy.reflectionEnabled, false);
|
|
assert.equal(policy.rolloutMode, 'off');
|
|
});
|
|
|
|
test('lifecycle canary only runs for configured users', async () => {
|
|
const lifecycle = createMemoryV2LifecycleService({
|
|
env: {
|
|
MEMORY_LIFECYCLE_ENABLED: '1',
|
|
MEMORY_LIFECYCLE_ROLLOUT_MODE: 'canary',
|
|
MEMORY_LIFECYCLE_ROLLOUT_USER_IDS: 'u-1',
|
|
MEMORY_PROMOTION_ENABLED: '1',
|
|
},
|
|
});
|
|
assert.equal(lifecycle.canRun('u-1'), true);
|
|
assert.equal(lifecycle.canRun('u-2'), false);
|
|
assert.equal((await lifecycle.promote({ userId: 'u-2' })).skipped, true);
|
|
});
|
|
|
|
test('forget is fail-safe when lifecycle is disabled', async () => {
|
|
const lifecycle = createMemoryV2LifecycleService({ env: { MEMORY_LIFECYCLE_ENABLED: '0' } });
|
|
assert.deepEqual(await lifecycle.forgetMemory({ userId: 'u-1', memoryId: 'm-1' }), {
|
|
ok: true, updated: false, skipped: true, reason: 'disabled',
|
|
});
|
|
});
|
|
|
|
test('list and forget use user-scoped SQL', async () => {
|
|
const calls = [];
|
|
const pool = { query: async (sql, params) => {
|
|
calls.push({ sql, params });
|
|
if (sql.startsWith('SELECT')) return [[{ id: 'm-1', user_id: 'u-1', label: 'fact', memory_text: 'x', status: 'active', confidence: 0.9, created_at: 1, updated_at: 2 }]];
|
|
return [{ affectedRows: 1 }];
|
|
} };
|
|
const lifecycle = createMemoryV2LifecycleService({ pool, env: { MEMORY_LIFECYCLE_ENABLED: '1', MEMORY_LIFECYCLE_FORGETTING_ENABLED: '1' } });
|
|
assert.equal((await lifecycle.listMemories({ userId: 'u-1' }))[0].id, 'm-1');
|
|
assert.equal((await lifecycle.forgetMemory({ userId: 'u-1', memoryId: 'm-1' })).updated, true);
|
|
assert.match(calls[0].sql, /user_id = \?/);
|
|
assert.match(calls[1].sql, /user_id = \?/);
|
|
});
|