cli_test.v

23.7 KB · raw

1module tests23import os4import time56const root = @VMODROOT7const binary = os.join_path(root, 'build', 'koh3')89struct Result {10mut:11	code int12	out  string13	err  string14}1516fn testsuite_begin() {17	os.mkdir_all(os.join_path(root, 'build')) or { panic(err) }18	build := os.execute('v -o ${q(binary)} ${q(os.join_path(root, 'src', 'koh3.v'))}')19	if build.exit_code != 0 {20		panic('could not build koh3: ${build.output}')21	}22}2324fn q(s string) string {25	return "'${s.replace("'", "'\\''")}'"26}2728fn temp_repo(name string) string {29	dir := os.join_path(os.temp_dir(), 'koh3-cli-${name}-${time.now().unix_nano()}')30	os.mkdir_all(dir) or { panic(err) }31	return dir32}3334fn run(repo string, args ...string) Result {35	home := temp_repo('home')36	defer { os.rmdir_all(home) or {} }37	return run_home(repo, home, ...args)38}3940fn run_home(repo string, home string, args ...string) Result {41	stamp := '${os.getpid()}-${time.now().unix_nano()}'42	base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')43	out_path := '${base}.out'44	err_path := '${base}.err'45	code_path := '${base}.code'46	mut command := 'HOME=${q(home)} ${q(binary)}'47	for arg in args {48		command += ' ${q(arg)}'49	}50	shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" $? >${q(code_path)}'51	result := os.execute(shell)52	if result.exit_code != 0 {53		panic('could not run test command: ${result.output}')54	}55	code := os.read_file(code_path) or { panic(err) }56	out := os.read_file(out_path) or { '' }57	err := os.read_file(err_path) or { '' }58	os.rm(out_path) or {}59	os.rm(err_path) or {}60	os.rm(code_path) or {}61	return Result{62		code: code.int()63		out:  out64		err:  err65	}66}6768fn init_repo(repo string) {69	r := run(repo, 'init')70	assert r.code == 071	assert os.exists(os.join_path(repo, '.koh'))72	assert os.exists(os.join_path(repo, '.kohignore'))73}7475fn write(repo string, path string, text string) {76	full := os.join_path(repo, path)77	os.mkdir_all(os.dir(full)) or { panic(err) }78	os.write_file(full, text) or { panic(err) }79}8081fn read_json_has(text string, key string) {82	if !text.contains(key) {83		panic('missing JSON key ${key}: ${text}')84	}85}8687fn test_init_and_usage() {88	dir := temp_repo('init')89	defer { os.rmdir_all(dir) or {} }90	r := run(dir, 'init', '--json')91	assert r.code == 092	read_json_has(r.out, 'repo')93	read_json_has(r.out, 'ignore_created')94	assert run(dir, 'init').code == 195	assert run(dir, 'unknown-command').code == 296	assert run(dir, 'unknown-command').err.contains('unknown command')97}9899fn test_save_checkpoint_and_status() {100	dir := temp_repo('save')101	defer { os.rmdir_all(dir) or {} }102	init_repo(dir)103	write(dir, 'tracked.txt', 'one\n')104	r := run(dir, 'save', 'first message', '--agent', 'agent-7', '--session', 'session-a', '--json')105	assert r.code == 0106	for key in ['id', 'parent', 'message', 'created_at', 'machine', 'is_checkpoint', 'agent',107		'session', 'files', 'size_bytes'] {108		read_json_has(r.out, key)109	}110	assert r.out.contains('"message":"first message"')111	assert r.out.contains('"agent":"agent-7"')112	assert r.out.contains('"session":"session-a"')113	assert r.out.contains('"is_checkpoint":false')114	assert run(dir, 'save', 'again').code == 1115	assert run(dir, 'save', 'again').err.contains('nothing has changed since the last save')116	assert run(dir, 'save').code == 2117	assert run(dir, 'save', 'x', '--bogus').code == 2118	cp := run(dir, 'checkpoint', 'still first', '--json')119	assert cp.code == 0120	assert cp.out.contains('"is_checkpoint":true')121	assert cp.out.contains('"parent":')122	clean := run(dir, 'status')123	assert clean.code == 0124	assert clean.out.contains('no changes')125	write(dir, 'tracked.txt', 'two\n')126	write(dir, 'new.txt', 'new\n')127	os.rm(os.join_path(dir, 'tracked.txt')) or { panic(err) }128	write(dir, '.kohignore', '*.log\n')129	status := run(dir, 'status')130	assert status.code == 0131	assert status.out.contains('+')132	assert status.out.contains('~') || status.out.contains('-')133	assert status.out.contains('added')134	js := run(dir, 'status', '--json')135	assert js.code == 0136	for key in ['head_id', 'head_msg', 'dirty', 'modified', 'added', 'deleted', 'ignore_changed'] {137		read_json_has(js.out, key)138	}139	assert js.out.contains('"ignore_changed":true')140}141142fn test_status_requires_repo_and_ignore_rules() {143	dir := temp_repo('status')144	defer { os.rmdir_all(dir) or {} }145	assert run(dir, 'status').code == 1146	init_repo(dir)147	write(dir, '.kohignore', '*.log\nbuild/\n')148	write(dir, 'keep.txt', 'keep\n')149	write(dir, 'ignored.log', 'ignore\n')150	write(dir, 'build/output.txt', 'ignore\n')151	assert run(dir, 'save', 'keep').code == 0152	assert run(dir, 'status').out.contains('no changes')153	assert !run(dir, 'status').out.contains('ignored.log')154	assert !run(dir, 'status').out.contains('build/output.txt')155	assert !run(dir, 'status').out.contains('LOG.md')156}157158// LOG.md is a convenience written after the save commits. If it cannot be159// written, the save still happened, and an agent that retried on a failure160// would only be told nothing had changed.161fn test_a_save_succeeds_when_log_md_cannot_be_written() {162	dir := temp_repo('logmd')163	defer { os.rmdir_all(dir) or {} }164	init_repo(dir)165	os.mkdir_all(os.join_path(dir, 'LOG.md')) or { panic(err) }166	write(dir, 'a.txt', 'one\n')167	r := run(dir, 'save', 'first', '--json')168	assert r.code == 0169	assert r.out.starts_with('{')170	assert r.err.contains('LOG.md')171	assert run(dir, 'log', '--json').out.contains('"message":"first"')172}173174// `save` records the executable bit, so `status` has to see it too, or a175// chmod would be invisible until it turned up in a save.176fn test_status_sees_an_executable_bit_change() {177	dir := temp_repo('chmod')178	defer { os.rmdir_all(dir) or {} }179	init_repo(dir)180	write(dir, 'run.sh', 'echo hi\n')181	assert run(dir, 'save', 'script').code == 0182	os.chmod(os.join_path(dir, 'run.sh'), 0o755) or { panic(err) }183	status := run(dir, 'status', '--json')184	assert status.out.contains('"dirty":true')185	assert status.out.contains('"modified":["run.sh"]')186}187188fn test_log_and_diff() {189	dir := temp_repo('history')190	defer { os.rmdir_all(dir) or {} }191	init_repo(dir)192	write(dir, 'a.txt', 'one\n')193	first := run(dir, 'save', 'one', '--session', 'alpha', '--json')194	id1 := first.out.all_after('"id":"')[..12]195	write(dir, 'a.txt', 'two\n')196	second := run(dir, 'save', 'two', '--session', 'beta', '--json')197	id2 := second.out.all_after('"id":"')[..12]198	cp := run(dir, 'checkpoint', 'checkpoint', '--json')199	assert cp.code == 0200	log := run(dir, 'log')201	assert log.code == 0202	assert log.out.contains('▸')203	assert log.out.contains('◐')204	lj := run(dir, 'log', '--json')205	assert lj.code == 0206	assert lj.out.contains('"head_id"')207	assert lj.out.contains('"saves"')208	assert run(dir, 'log', '--since', 'today').out.contains('two')209	session := run(dir, 'log', '--session', 'alpha')210	assert session.out.contains('one')211	assert !session.out.contains('two')212	write(dir, 'a.txt', 'three\n')213	d := run(dir, 'diff')214	assert d.code == 0215	assert d.out.contains('--- a/')216	assert d.out.contains('+++ b/')217	assert run(dir, 'diff', id1).code == 0218	assert run(dir, 'diff', id1, id2).code == 0219	dj := run(dir, 'diff', '--json')220	assert dj.code == 0221	assert dj.out.contains('"from"')222	assert dj.out.contains('"to"')223	assert dj.out.contains('"files"')224	assert dj.out.contains('"change":"modified"')225	assert run(dir, 'diff', 'not-a-real-id').code == 1226	assert run(dir, 'diff', id1, id2, 'third').code == 1227	write(dir, 'blob.bin', '\x00\x01\x02')228	diff_binary := run(dir, 'diff', '--json')229	assert diff_binary.out.contains('"binary":true')230	assert diff_binary.out.contains('Binary files differ')231}232233fn test_undo_revert_and_load() {234	dir := temp_repo('restore')235	defer { os.rmdir_all(dir) or {} }236	init_repo(dir)237	write(dir, 'a.txt', 'one\n')238	one := run(dir, 'save', 'one', '--json').out239	id1 := one.all_after('"id":"')[..12]240	write(dir, 'a.txt', 'two\n')241	write(dir, 'gone.txt', 'gone\n')242	second := run(dir, 'save', 'two', '--json').out243	id2 := second.all_after('"id":"')[..12]244	write(dir, 'a.txt', 'three\n')245	third := run(dir, 'save', 'three', '--json').out246	id3 := third.all_after('"id":"')[..12]247	// Unsaved work — here a file never saved — blocks every going-back but248	// revert, as in Koh 2: a restore must never overwrite what only the disk has.249	write(dir, 'never.txt', 'keep me\n')250	log_before := run(dir, 'log').out251	for args in [['undo', '--yes'], ['load', id1, '--yes'], ['pluck', id2, '--yes']] {252		blocked := run(dir, ...args)253		assert blocked.code == 1254		assert blocked.err.contains('unsaved changes')255	}256	assert run(dir, 'log').out == log_before257	os.rm(os.join_path(dir, 'never.txt')) or { panic(err) }258	write(dir, 'a.txt', 'edited, never saved\n')259	assert run(dir, 'undo', '--yes').err.contains('unsaved changes')260	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'edited, never saved\n'261	write(dir, 'a.txt', 'three\n')262	os.rm(os.join_path(dir, 'gone.txt')) or { panic(err) }263	assert run(dir, 'load', id1, '--yes').err.contains('unsaved changes')264	write(dir, 'gone.txt', 'gone\n')265	undo := run(dir, 'undo', '--yes', '--json')266	assert undo.code == 0267	if !undo.out.contains(id2) {268		panic('undo JSON did not name ${id2}: ${undo.out}')269	}270	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'271	assert os.exists(os.join_path(dir, 'gone.txt'))272	assert run(dir, 'load', id3, '--yes').code == 0273	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'274	assert run(dir, 'undo', '2', '--yes').code == 0275	assert run(dir, 'load', id2, '--yes').code == 0276	assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')277	write(dir, 'a.txt', 'dirty\n')278	before := run(dir, 'log')279	refuse := run(dir, 'revert')280	assert refuse.code == 1281	assert refuse.err.contains('needs --yes when not interactive')282	assert run(dir, 'log').out == before.out283	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'284	write(dir, 'untracked.txt', 'remove\n')285	rv := run(dir, 'revert', '--yes')286	assert rv.code == 0287	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'288	assert !os.exists(os.join_path(dir, 'untracked.txt'))289	assert run(dir, 'revert', '--yes').out.contains('working tree already matches')290	assert run(dir, 'load', '--checkpoint', '--yes').code == 0291	assert run(dir, 'load', id2, '--json').code == 1292	assert run(dir, 'load', id2, '--json').err.contains('error')293	assert run(dir, 'load').code == 2294	assert run(dir, 'load', 'unknown-id', '--yes').code == 1295}296297fn test_flag_and_flagged_log() {298	dir := temp_repo('flag')299	defer { os.rmdir_all(dir) or {} }300	init_repo(dir)301	write(dir, 'flag.txt', 'one\n')302	saved := run(dir, 'save', 'flag me', '--json')303	id := saved.out.all_after('"id":"')[..12]304	flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')305	assert flagged.code == 0306	assert flagged.out.contains('"flagged":true')307	assert flagged.out.contains('"reason":"unsafe"')308	list := run(dir, 'flag', '--json')309	assert list.code == 0310	assert list.out.contains(id)311	assert list.out.contains('unsafe')312	assert run(dir, 'log').out.contains('⚑')313	assert run(dir, 'flag', '--clear', id).code == 0314	assert run(dir, 'flag').out.contains('nothing is flagged')315	assert run(dir, 'flag', 'unknown-id').code == 1316	assert run(dir, 'flag').code == 0317}318319fn test_pluck_forward_reverse_dry_run_and_conflict() {320	dir := temp_repo('pluck')321	defer { os.rmdir_all(dir) or {} }322	init_repo(dir)323	write(dir, 'a.txt', 'one\n')324	first := run(dir, 'save', 'one', '--json').out325	id1 := first.all_after('"id":"')[..12]326	write(dir, 'a.txt', 'two\n')327	second := run(dir, 'save', 'two', '--json').out328	id2 := second.all_after('"id":"')[..12]329	assert run(dir, 'load', id1, '--yes').code == 0330	dry := run(dir, 'pluck', id2, '--dry-run')331	assert dry.code == 0332	assert dry.out.contains('a.txt')333	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'334	assert run(dir, 'pluck', id2, '--yes').code == 0335	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'336	assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0337	assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'338	write(dir, 'a.txt', 'different\n')339	assert run(dir, 'pluck', id2, '--dry-run').code == 1340	assert run(dir, 'save', 'different').code == 0341	conflict := run(dir, 'pluck', id2, '--yes')342	assert conflict.code == 1343	assert conflict.err.contains('a.txt')344	assert run(dir, 'pluck').code == 2345	assert run(dir, 'pluck', 'unknown-id').code == 1346}347348fn test_peek_and_doctor() {349	dir := temp_repo('peek-doctor')350	defer { os.rmdir_all(dir) or {} }351	init_repo(dir)352	write(dir, 'peek.txt', 'peek me\n')353	assert run(dir, 'save', 'peekable').code == 0354	id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]355	before_log := run(dir, 'log', '--json').out356	peek := run(dir, 'peek', id, '--json')357	assert peek.code == 0358	peek_path := peek.out.all_after('"path":"').before('"')359	assert peek.out.contains('"id":"${id}"')360	assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'361	assert run(dir, 'status').out.contains('no changes')362	assert run(dir, 'log', '--json').out == before_log363	assert run(dir, 'peek').code == 2364	assert run(dir, 'peek', 'unknown-id').code == 1365	backup := os.join_path(dir, 'doctor.backup')366	healthy := run(dir, 'doctor')367	assert healthy.code == 0368	assert healthy.out.contains('is sound')369	report := run(dir, 'doctor', '--json')370	assert report.code == 0371	for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',372		'backup'] {373		read_json_has(report.out, key)374	}375	backed := run(dir, 'doctor', '--backup', backup)376	assert backed.code == 0377	assert os.exists(backup)378	assert backed.out.contains(backup)379	assert run(dir, 'doctor', '--backup', backup).code == 1380	assert run(dir, 'peek', id, '--json').code == 0381	removed := run(dir, 'doctor', '--json')382	assert removed.code == 0383	assert removed.out.contains('"peek_folders_removed":1')384	assert !os.exists(peek_path)385}386387fn test_strip_refuses_without_terminal() {388	dir := temp_repo('strip')389	defer { os.rmdir_all(dir) or {} }390	init_repo(dir)391	write(dir, 'secret.txt', 'keep\n')392	assert run(dir, 'save', 'secret').code == 0393	before_log := run(dir, 'log', '--json').out394	refused := run(dir, 'strip', 'secret.txt')395	assert refused.code == 1396	assert refused.err.contains('needs a terminal')397	assert run(dir, 'log', '--json').out == before_log398	assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'399	assert run(dir, 'log', '--json').out.contains('secret')400}401402fn test_saves_and_checkpoints_aliases() {403	dir := temp_repo('aliases')404	defer { os.rmdir_all(dir) or {} }405	init_repo(dir)406	write(dir, 'alias.txt', 'one\n')407	assert run(dir, 'save', 'plain save').code == 0408	assert run(dir, 'checkpoint', 'checkpoint save').code == 0409	saves := run(dir, 'saves')410	assert saves.code == 0411	assert saves.out.contains('plain save')412	assert saves.out.contains('checkpoint save')413	checkpoints := run(dir, 'checkpoints')414	assert checkpoints.code == 0415	assert checkpoints.out.contains('checkpoint save')416	assert !checkpoints.out.contains('plain save')417	json_result := run(dir, 'checkpoints', '--json')418	assert json_result.code == 0419	assert json_result.out.contains('"head_id"')420	assert json_result.out.contains('"saves"')421	assert json_result.out.contains('"is_checkpoint":true')422	assert !json_result.out.contains('"is_checkpoint":false')423}424425fn test_remote_configuration_is_isolated_and_atomic() {426	dir := temp_repo('remote')427	home := temp_repo('remote-home')428	defer {429		os.rmdir_all(dir) or {}430		os.rmdir_all(home) or {}431	}432	empty := run_home(dir, home, 'remote', '--json')433	assert empty.code == 0434	assert empty.out.contains('"configured":false')435	target := os.join_path(home, 'remote-store')436	set := run_home(dir, home, 'remote', 'folder', target)437	assert set.code == 0438	assert os.is_dir(target)439	state := run_home(dir, home, 'remote', '--json')440	assert state.code == 0441	assert state.out.contains('"kind":"folder"')442	assert state.out.contains(target)443	assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').code == 1444	assert run_home(dir, home, 'remote', '--json').out.contains(target)445	assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').err.contains('KOH_ACCESS_KEY')446	assert run_home(dir, home, 'remote', '--forget').code == 0447	assert run_home(dir, home, 'remote', '--json').out.contains('"configured":false')448}449450// A clone belongs to the repo it was stolen from, whatever its folder is451// called later: renaming it must not start a second repo on the remote.452fn test_a_renamed_clone_keeps_offering_to_its_repo() {453	syncroot := temp_repo('rename-clone')454	home := temp_repo('rename-clone-home')455	defer {456		os.rmdir_all(syncroot) or {}457		os.rmdir_all(home) or {}458	}459	remote := os.join_path(syncroot, 'remote')460	a := os.join_path(syncroot, 'A', 'proj')461	b_parent := os.join_path(syncroot, 'B')462	os.mkdir_all(a) or { panic(err) }463	os.mkdir_all(b_parent) or { panic(err) }464	init_repo(a)465	write(a, 'a.txt', 'from A\n')466	assert run_home(a, home, 'remote', 'folder', remote).code == 0467	assert run_home(a, home, 'save', 'A first').code == 0468	assert run_home(b_parent, home, 'steal', 'proj').code == 0469	renamed := os.join_path(b_parent, 'renamed')470	os.mv(os.join_path(b_parent, 'proj'), renamed) or { panic(err) }471	write(renamed, 'b.txt', 'from B\n')472	assert run_home(renamed, home, 'save', 'B first').code == 0473	assert !os.exists(os.join_path(remote, 'renamed'))474	assert run_home(a, home, 'steal').code == 0475	assert os.read_file(os.join_path(a, 'b.txt'))! == 'from B\n'476}477478fn test_offer_steal_folder_remote() {479	syncroot := temp_repo('sync')480	home := temp_repo('sync-home')481	defer {482		os.rmdir_all(syncroot) or {}483		os.rmdir_all(home) or {}484	}485	remote := os.join_path(syncroot, 'remote')486	a := os.join_path(syncroot, 'A', 'proj')487	b_parent := os.join_path(syncroot, 'B')488	os.mkdir_all(a) or { panic(err) }489	os.mkdir_all(b_parent) or { panic(err) }490	init_repo(a)491	write(a, 'a.txt', 'from A\n')492	assert run_home(a, home, 'remote', 'folder', remote).code == 0493	assert run_home(a, home, 'save', 'A first').code == 0494	clone := run_home(b_parent, home, 'steal', 'proj')495	assert clone.code == 0496	b := os.join_path(b_parent, 'proj')497	assert os.read_file(os.join_path(b, 'a.txt'))! == 'from A\n'498	write(b, 'b.txt', 'from B\n')499	assert run_home(b, home, 'save', 'B first').code == 0500	write(a, 'a2.txt', 'second\n')501	assert run_home(a, home, 'save', 'A second').code == 0502	assert run_home(a, home, 'steal').code == 0503	assert os.exists(os.join_path(a, 'b.txt'))504	assert run_home(b_parent, home, 'steal', '--list').out.contains('proj')505	write(a, 'clash.txt', 'base\n')506	assert run_home(a, home, 'save', 'clash base').code == 0507	assert run_home(b, home, 'steal').code == 0508	write(a, 'clash.txt', 'A version\n')509	assert run_home(a, home, 'save', 'A clash').code == 0510	write(b, 'clash.txt', 'B version\n')511	assert run_home(b, home, 'save', 'B clash').code == 0512	refused := run_home(b, home, 'steal')513	assert refused.code == 1514	assert refused.err.contains('--mine') || refused.err.contains('--theirs')515	assert run_home(b, home, 'steal', '--theirs').code == 0516	assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'517	write(a, 'clash.txt', 'A second\n')518	assert run_home(a, home, 'save', 'A second clash').code == 0519	write(b, 'clash.txt', 'B second\n')520	assert run_home(b, home, 'save', 'B second clash').code == 0521	assert run_home(b, home, 'steal', '--mine').code == 0522	assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B second\n'523	assert run_home(b, home, 'log').out.contains('combine: kept mine for')524}525526fn test_man_and_koh2_handoff() {527	dir := temp_repo('man')528	home := temp_repo('man-home')529	defer {530		os.rmdir_all(dir) or {}531		os.rmdir_all(home) or {}532	}533	whole := run_home(dir, home, 'man')534	assert whole.code == 0535	assert whole.out.contains('NAME')536	assert whole.out.contains('COMMANDS')537	strip := run_home(dir, home, 'man', 'strip')538	assert strip.code == 0539	assert strip.out.contains('strip')540	assert !strip.out.contains('koh save ')541	assert run_home(dir, home, 'man', 'nonsense').code == 2542	legacy := temp_repo('legacy')543	defer { os.rmdir_all(legacy) or {} }544	os.mkdir_all(os.join_path(legacy, '.koh')) or { panic(err) }545	fake := os.join_path(home, 'bin')546	os.mkdir_all(fake) or { panic(err) }547	fake_koh := os.join_path(fake, 'koh2')548	os.write_file(fake_koh, '#!/bin/sh\nprintf "handoff:%s\\n" "$*"\nexit 7\n') or { panic(err) }549	os.chmod(fake_koh, 0o755) or { panic(err) }550	handoff := run_home_with_path(legacy, home, fake, 'status', '--json')551	assert handoff.code == 7552	assert handoff.out.contains('handoff:status --json')553}554555fn run_home_with_path(repo string, home string, bin_dir string, args ...string) Result {556	stamp := '${os.getpid()}-${time.now().unix_nano()}'557	base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')558	out_path := '${base}.out'559	err_path := '${base}.err'560	code_path := '${base}.code'561	mut command := 'HOME=${q(home)} PATH=${q(bin_dir + ':' + os.getenv('PATH'))} ${q(binary)}'562	for arg in args {563		command += ' ${q(arg)}'564	}565	shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" $? >${q(code_path)}'566	result := os.execute(shell)567	if result.exit_code != 0 {568		panic('could not run test command: ${result.output}')569	}570	code := os.read_file(code_path) or { panic(err) }571	out := os.read_file(out_path) or { '' }572	err := os.read_file(err_path) or { '' }573	os.rm(out_path) or {}574	os.rm(err_path) or {}575	os.rm(code_path) or {}576	return Result{577		code: code.int()578		out:  out579		err:  err580	}581}582583// --- migration -------------------------------------------------------------584585fn koh2_binary() string {586	return os.find_abs_path_of_executable('koh') or { '' }587}588589// run_koh2 runs the Koh 2 binary in a fixture; its own output is not under590// test here.591fn run_koh2(dir string, command string) {592	os.execute('cd ${q(dir)} && ${command} >/dev/null 2>&1')593}594595fn save_ids(dir string) []string {596	mut ids := []string{}597	for name in os.ls(os.join_path(dir, '.koh', 'snapshots')) or { []string{} } {598		if name.ends_with('.json') {599			ids << name.trim_right('.json')600		}601	}602	ids.sort()603	return ids604}605606fn test_migrate_converts_a_koh2_repo_and_keeps_every_save() {607	koh2 := koh2_binary()608	if koh2 == '' {609		println('skipping the migration tests: koh (Koh 2) is not on PATH')610		return611	}612	dir := temp_repo('migrate')613	defer { os.rmdir_all(dir) or {} }614	home := temp_repo('migrate-home')615	defer { os.rmdir_all(home) or {} }616	run_koh2(dir, '${q(koh2)} init')617	// One file large enough to be an object, one small enough to be inlined.618	write(dir, 'code.txt', 'x'.repeat(700))619	write(dir, 'tiny.txt', 'tiny\n')620	run_koh2(dir, '${q(koh2)} save "first save"')621	run_koh2(dir, '${q(koh2)} checkpoint "before something"')622	write(dir, 'code.txt', 'y'.repeat(700))623	run_koh2(dir, '${q(koh2)} save "second save"')624	original := save_ids(dir)625	assert original.len == 3626627	report := run_home(dir, home, 'migrate')628	assert report.code == 0629	assert report.out.contains('migrated 3 saves')630	// The swap: `.koh` is a file, the old data is archived, nothing is staging.631	assert os.is_file(os.join_path(dir, '.koh'))632	assert os.is_dir(os.join_path(dir, '.koh-v2'))633	assert os.is_file(os.join_path(dir, '.kohignore'))634	assert !os.exists(os.join_path(dir, '.koh-migrate'))635636	// Every save kept its original ID, and the new repo is sound.637	logged := run_home(dir, home, 'log', '--json')638	assert logged.code == 0639	for id in original {640		assert logged.out.contains(id)641	}642	assert run_home(dir, home, 'doctor').code == 0643	assert run_home(dir, home, 'status').code == 0644645	// Migrating again has nothing to convert.646	assert run_home(dir, home, 'migrate').code == 1647}648649fn test_migrate_refuses_a_tree_that_does_not_match_the_last_save() {650	koh2 := koh2_binary()651	if koh2 == '' {652		return653	}654	dir := temp_repo('migrate-dirty')655	defer { os.rmdir_all(dir) or {} }656	home := temp_repo('migrate-dirty-home')657	defer { os.rmdir_all(home) or {} }658	run_koh2(dir, '${q(koh2)} init')659	write(dir, 'code.txt', 'x'.repeat(700))660	run_koh2(dir, '${q(koh2)} save "only save"')661	write(dir, 'code.txt', 'unsaved change\n')662663	refused := run_home(dir, home, 'migrate')664	assert refused.code == 1665	assert refused.err.contains('save with koh2 first')666	// Nothing moved: still Koh 2, no archive, no staging file.667	assert os.is_dir(os.join_path(dir, '.koh'))668	assert !os.exists(os.join_path(dir, '.koh-v2'))669	assert !os.exists(os.join_path(dir, '.koh-migrate'))670}