File size: 5,017 Bytes
d522ffa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | import { describe, expect, it, beforeEach, afterEach } from 'vitest';
import { mkdtemp, mkdir, writeFile, rm } from 'node:fs/promises';
import { tmpdir } from 'node:os';
import { join } from 'node:path';
import { oldMd5BucketName } from '../src/sessions/workdir-bucket.js';
import { detectMigration } from '../src/detect.js';
let src: string;
beforeEach(async () => {
src = await mkdtemp(join(tmpdir(), 'detect-'));
});
afterEach(async () => {
await rm(src, { recursive: true, force: true });
});
describe('detectMigration', () => {
it('returns empty totals when source dir is empty', async () => {
const plan = await detectMigration({ sourcePath: src });
expect(plan.hasConfig).toBe(false);
expect(plan.hasMcp).toBe(false);
expect(plan.totalSessions).toBe(0);
});
it('detects config/mcp/credentials/user-history/plugins/mcp-oauth presence', async () => {
await writeFile(join(src, 'config.toml'), '');
await writeFile(
join(src, 'mcp.json'),
'{"mcpServers":{"server-1":{"url":"https://example.test/mcp","auth":"oauth"},"server-2":{"command":"npx"}}}',
);
await mkdir(join(src, 'credentials'), { recursive: true });
await writeFile(join(src, 'credentials', 'kimi-code.json'), '{}');
await mkdir(join(src, 'user-history'), { recursive: true });
await mkdir(join(src, 'plugins', 'p1'), { recursive: true });
await mkdir(join(src, 'mcp-oauth'), { recursive: true });
await writeFile(join(src, 'mcp-oauth', 'mangled-store-entry'), '');
const plan = await detectMigration({ sourcePath: src });
expect(plan.hasConfig).toBe(true);
expect(plan.hasMcp).toBe(true);
expect(plan.hasUserHistory).toBe(true);
expect(plan.oauthCredentials).toEqual(['kimi-code']);
expect(plan.detectedPlugins).toEqual(['p1']);
expect(plan.detectedMcpOauthServers).toEqual(['server-1']);
});
it('derives OAuth relogin notices from config oauth refs even without credential files', async () => {
await writeFile(
join(src, 'config.toml'),
'[providers."managed:kimi-code"]\ntype = "kimi"\nbase_url = "https://api.example.test/v1"\n\n[providers."managed:kimi-code".oauth]\nstorage = "keyring"\nkey = "oauth/kimi-code"\n',
);
const plan = await detectMigration({ sourcePath: src });
expect(plan.oauthCredentials).toEqual(['kimi-code']);
});
it('treats a config.json-only source as having config', async () => {
await writeFile(join(src, 'config.json'), '{"default_model":"m"}');
const plan = await detectMigration({ sourcePath: src });
expect(plan.hasConfig).toBe(true);
});
it('counts historical flat sessions and title-only sessions', async () => {
const workdir = '/workspace/flat-proj';
const bucket = join(src, 'sessions', oldMd5BucketName(workdir));
await mkdir(join(bucket, 'titled'), { recursive: true });
await writeFile(join(src, 'kimi.json'), JSON.stringify({ work_dirs: [{ path: workdir }] }));
await writeFile(join(bucket, 'flat-1.jsonl'), '{"role":"user","content":"hi"}\n');
await writeFile(join(bucket, 'titled', 'context.jsonl'), '');
await writeFile(
join(bucket, 'titled', 'state.json'),
JSON.stringify({ custom_title: 'Named' }),
);
const plan = await detectMigration({ sourcePath: src });
expect(plan.totalSessions).toBe(2);
expect(plan.sessionScanFailures).toEqual([]);
expect(plan.workdirs[0]?.sessions.map((s) => s.uuid).sort()).toEqual(['flat-1', 'titled']);
});
it('detects a skills-only source', async () => {
await mkdir(join(src, 'skills', 'my-skill'), { recursive: true });
await writeFile(join(src, 'skills', 'my-skill', 'SKILL.md'), '# skill');
const plan = await detectMigration({ sourcePath: src });
expect(plan.hasSkills).toBe(true);
});
it('detects legacy plan files via the injectable plans source dir', async () => {
const plansDir = await mkdtemp(join(tmpdir(), 'detect-plans-'));
try {
await writeFile(join(plansDir, 'hero.md'), '# plan');
const plan = await detectMigration({ sourcePath: src, plansSourcePath: plansDir });
expect(plan.hasPlans).toBe(true);
const empty = await detectMigration({ sourcePath: src, plansSourcePath: join(plansDir, 'nope') });
expect(empty.hasPlans).toBe(false);
} finally {
await rm(plansDir, { recursive: true, force: true });
}
});
it('reports an unknown workdir bucket when kimi.json cannot map it', async () => {
const bucket = join(src, 'sessions', oldMd5BucketName('/workspace/example'));
await mkdir(join(bucket, 'legacy-session'), { recursive: true });
await writeFile(
join(bucket, 'legacy-session', 'context.jsonl'),
'{"role":"user","content":"hello"}\n',
);
const plan = await detectMigration({ sourcePath: src });
expect(plan.totalSessions).toBe(0);
expect(plan.sessionScanFailures).toEqual([
{
sourcePath: bucket,
reason: expect.stringMatching(/workdir.*kimi\.json/i),
},
]);
});
});
|