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 98 for args in [['unknown-command', '--json'], ['save', '--json'], ['save', 'x', '--nope', '--json']] {99 r2 := run(dir, ...args)100 assert r2.code == 2101 assert r2.out == ''102 assert r2.err.starts_with('{"error":'), r2.err103 }104}105106107fn test_the_help_screen_lists_commands_and_no_flags() {108 dir := temp_repo('help')109 defer { os.rmdir_all(dir) or {} }110 help := run(dir, '--help')111 assert help.code == 0112 for name in ['save', 'checkpoint', 'status', 'log', 'diff', 'undo', 'revert', 'load', 'offer',113 'steal'] {114 assert help.out.contains(' ${name} '), name115 }116 assert help.out.contains('koh man <command>')117 118 assert !help.out.contains('--'), help.out119 for line in help.out.split_into_lines() {120 assert line.runes().len <= 80, line121 }122 123 bare := run(dir)124 assert bare.code == 2125 assert bare.out == ''126 assert bare.err.contains('koh man <command>')127 short := run(dir, '-h')128 assert short.code == 0129 assert short.out == help.out130}131132133fn test_the_help_screen_has_no_colour_when_it_is_off() {134 dir := temp_repo('help_plain')135 defer { os.rmdir_all(dir) or {} }136 home := temp_repo('help_plain_home')137 defer { os.rmdir_all(home) or {} }138 out_path := os.join_path(os.temp_dir(), 'koh3-help-${os.getpid()}-${time.now().unix_nano()}')139 defer { os.rm(out_path) or {} }140 code := os.system('cd ${q(dir)} && HOME=${q(home)} NO_COLOR=1 ${q(binary)} --help > ${q(out_path)}')141 assert code == 0142 text := os.read_file(out_path) or { panic(err) }143 assert !text.contains('\x1b'), text144 assert text.contains(' save snapshot your work')145}146147148fn test_version_matches_the_manifest() {149 dir := temp_repo('version')150 defer { os.rmdir_all(dir) or {} }151 version := run(dir, '--version').out.trim_space().all_after('koh ')152 manifest := os.read_file(os.join_path(root, 'v.mod')) or { panic(err) }153 assert manifest.contains("version: '${version}'"), version154}155156fn test_save_checkpoint_and_status() {157 dir := temp_repo('save')158 defer { os.rmdir_all(dir) or {} }159 init_repo(dir)160 write(dir, 'tracked.txt', 'one\n')161 r := run(dir, 'save', 'first message', '--agent', 'agent-7', '--session', 'session-a', '--json')162 assert r.code == 0163 for key in ['id', 'parent', 'message', 'created_at', 'machine', 'is_checkpoint', 'agent', 'session',164 'files', 'size_bytes'] {165 read_json_has(r.out, key)166 }167 assert r.out.contains('"message":"first message"')168 assert r.out.contains('"agent":"agent-7"')169 assert r.out.contains('"session":"session-a"')170 assert r.out.contains('"is_checkpoint":false')171 assert run(dir, 'save', 'again').code == 1172 assert run(dir, 'save', 'again').err.contains('nothing has changed since the last save')173 assert run(dir, 'save').code == 2174 assert run(dir, 'save', 'x', '--bogus').code == 2175 cp := run(dir, 'checkpoint', 'still first', '--json')176 assert cp.code == 0177 assert cp.out.contains('"is_checkpoint":true')178 assert cp.out.contains('"parent":')179 clean := run(dir, 'status')180 assert clean.code == 0181 assert clean.out.contains('no changes')182 write(dir, 'tracked.txt', 'two\n')183 write(dir, 'new.txt', 'new\n')184 os.rm(os.join_path(dir, 'tracked.txt')) or { panic(err) }185 write(dir, '.kohignore', '*.log\n')186 status := run(dir, 'status')187 assert status.code == 0188 assert status.out.contains('+')189 assert status.out.contains('~') || status.out.contains('-')190 assert status.out.contains('added')191 js := run(dir, 'status', '--json')192 assert js.code == 0193 for key in ['head_id', 'head_msg', 'dirty', 'modified', 'added', 'deleted', 'ignore_changed',194 'saves', 'size_bytes', 'head_flagged'] {195 read_json_has(js.out, key)196 }197 assert js.out.contains('"ignore_changed":true')198 199 assert js.out.contains('"saves":2')200 assert js.out.contains('"size_bytes":${os.file_size(os.join_path(dir, '.koh'))}')201 assert js.out.contains('"head_flagged":false')202 head := js.out.all_after('"head_id":"').all_before('"')203 assert run(dir, 'flag', head, '--reason', 'bad').code == 0204 assert run(dir, 'status', '--json').out.contains('"head_flagged":true')205}206207fn test_status_requires_repo_and_ignore_rules() {208 dir := temp_repo('status')209 defer { os.rmdir_all(dir) or {} }210 assert run(dir, 'status').code == 1211 init_repo(dir)212 write(dir, '.kohignore', '*.log\nbuild/\n')213 write(dir, 'keep.txt', 'keep\n')214 write(dir, 'ignored.log', 'ignore\n')215 write(dir, 'build/output.txt', 'ignore\n')216 assert run(dir, 'save', 'keep').code == 0217 assert run(dir, 'status').out.contains('no changes')218 assert !run(dir, 'status').out.contains('ignored.log')219 assert !run(dir, 'status').out.contains('build/output.txt')220 assert !run(dir, 'status').out.contains('LOG.md')221}222223224225226fn test_a_save_succeeds_when_log_md_cannot_be_written() {227 dir := temp_repo('logmd')228 defer { os.rmdir_all(dir) or {} }229 init_repo(dir)230 os.mkdir_all(os.join_path(dir, 'LOG.md')) or { panic(err) }231 write(dir, 'a.txt', 'one\n')232 r := run(dir, 'save', 'first', '--json')233 assert r.code == 0234 assert r.out.starts_with('{')235 assert r.err.contains('LOG.md')236 assert run(dir, 'log', '--json').out.contains('"message":"first"')237}238239240241242fn test_a_restore_refuses_a_folder_of_ignored_files_in_the_way() {243 dir := temp_repo('dir-in-way')244 defer { os.rmdir_all(dir) or {} }245 init_repo(dir)246 write(dir, 'out', 'a file\n')247 first := run(dir, 'save', 'out is a file', '--json').out.all_after('"id":"')[..12]248 os.rm(os.join_path(dir, 'out')) or { panic(err) }249 write(dir, 'out/build.log', 'ignored\n')250 assert run(dir, 'save', 'out is gone').code == 0251 r := run(dir, 'load', first, '--yes')252 assert r.code == 1253 assert r.err.contains('out')254 assert os.ls(os.join_path(dir, 'out'))! == ['build.log']255}256257258259260fn test_a_restore_never_writes_through_a_symlinked_folder() {261 dir := temp_repo('link-in-way')262 outside := temp_repo('link-target')263 defer {264 os.rmdir_all(dir) or {}265 os.rmdir_all(outside) or {}266 }267 init_repo(dir)268 write(dir, 'docs/a.md', 'saved\n')269 first := run(dir, 'save', 'docs', '--json').out.all_after('"id":"')[..12]270 os.rmdir_all(os.join_path(dir, 'docs')) or { panic(err) }271 write(outside, 'a.md', 'precious\n')272 os.symlink(outside, os.join_path(dir, 'docs')) or { panic(err) }273 assert run(dir, 'save', 'docs is a link now').code == 0274 head := run(dir, 'status', '--json').out275 r := run(dir, 'load', first, '--yes')276 assert r.code == 1277 assert r.err.contains('docs')278 assert os.read_file(os.join_path(outside, 'a.md'))! == 'precious\n'279 assert run(dir, 'status', '--json').out == head280}281282283284285286fn test_a_restore_cut_short_points_at_revert() {287 dir := temp_repo('cut-short')288 defer {289 os.chmod(os.join_path(dir, 'zz'), 0o755) or {}290 os.rmdir_all(dir) or {}291 }292 init_repo(dir)293 write(dir, 'zero', '0\n')294 assert run(dir, 'save', 'zero').code == 0295 write(dir, 'a.txt', 'v1\n')296 write(dir, 'zz/f', 'v1\n')297 assert run(dir, 'save', 'first').code == 0298 write(dir, 'a.txt', 'v2\n')299 write(dir, 'zz/f', 'v2\n')300 assert run(dir, 'save', 'second').code == 0301 302 303 os.chmod(os.join_path(dir, 'zz'), 0o555)!304 stopped := run(dir, 'undo', '--yes')305 assert stopped.code == 1306 assert stopped.err.contains('koh revert')307 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'v1\n'308 assert os.read_file(os.join_path(dir, 'zz/f'))! == 'v2\n'309 310 os.chmod(os.join_path(dir, 'zz'), 0o000)!311 assert run(dir, 'status').err.contains('koh revert')312 os.chmod(os.join_path(dir, 'zz'), 0o755)!313314 assert run(dir, 'status', '--json').out.contains('"restore_unfinished":true')315 for args in [['save', 'half'], ['checkpoint', 'half'], ['undo', '--yes']] {316 r := run(dir, ...args)317 assert r.code == 1318 assert r.err.contains('koh revert')319 }320 assert run(dir, 'revert', '--yes').code == 0321 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'v1\n'322 assert os.read_file(os.join_path(dir, 'zz/f'))! == 'v1\n'323 assert run(dir, 'status', '--json').out.contains('"restore_unfinished":false')324 write(dir, 'a.txt', 'v3\n')325 assert run(dir, 'save', 'carry on').code == 0326}327328329330fn test_doctor_sweeps_only_old_peek_folders() {331 dir := temp_repo('peek-sweep')332 333 334 fresh := os.join_path(os.temp_dir(), 'koh-peek-fresh${os.getpid()}')335 old := os.join_path(os.temp_dir(), 'koh-peek-old${os.getpid()}')336 defer {337 os.rmdir_all(dir) or {}338 os.rmdir_all(fresh) or {}339 os.rmdir_all(old) or {}340 }341 init_repo(dir)342 write(dir, 'a.txt', 'one\n')343 assert run(dir, 'save', 'one').code == 0344 os.mkdir_all(fresh) or { panic(err) }345 os.mkdir_all(old) or { panic(err) }346 two_days_ago := time.now().unix() - 2 * 86400347 os.utime(old, two_days_ago, two_days_ago) or { panic(err) }348 assert run(dir, 'doctor', '--json').code == 0349 assert os.exists(fresh)350 assert !os.exists(old)351}352353354355356fn test_a_restore_never_overwrites_an_ignored_file() {357 dir := temp_repo('ignored-restore')358 defer { os.rmdir_all(dir) or {} }359 init_repo(dir)360 write(dir, '.env', 'KEY=saved long ago\n')361 first := run(dir, 'save', 'with env', '--json').out.all_after('"id":"')[..12]362 write(dir, '.kohignore', '.env\n')363 assert run(dir, 'save', 'ignore env').code == 0364 write(dir, '.env', 'KEY=never saved\n')365 assert run(dir, 'load', first, '--yes').code == 0366 assert os.read_file(os.join_path(dir, '.env'))! == 'KEY=never saved\n'367368 369 other := temp_repo('ignored-restore-missing')370 defer { os.rmdir_all(other) or {} }371 init_repo(other)372 write(other, '.env', 'KEY=saved long ago\n')373 again := run(other, 'save', 'with env', '--json').out.all_after('"id":"')[..12]374 write(other, '.kohignore', '.env\n')375 assert run(other, 'save', 'ignore env').code == 0376 os.rm(os.join_path(other, '.env')) or { panic(err) }377 assert run(other, 'load', again, '--yes').code == 0378 assert os.read_file(os.join_path(other, '.env'))! == 'KEY=saved long ago\n'379}380381382383fn test_a_going_back_preview_does_not_run_diff() {384 dir := temp_repo('preview-nodiff')385 home := temp_repo('preview-nodiff-home')386 defer {387 os.rmdir_all(dir) or {}388 os.rmdir_all(home) or {}389 }390 init_repo(dir)391 write(dir, 'a.txt', 'one\n')392 assert run_home(dir, home, 'save', 'one').code == 0393 write(dir, 'a.txt', 'two\n')394 assert run_home(dir, home, 'save', 'two').code == 0395 fake := os.join_path(home, 'bin')396 os.mkdir_all(fake) or { panic(err) }397 marker := os.join_path(home, 'diff-ran')398 os.write_file(os.join_path(fake, 'diff'), '#!/bin/sh\ntouch ${q(marker)}\nexit 2\n') or {399 panic(err)400 }401 os.chmod(os.join_path(fake, 'diff'), 0o755) or { panic(err) }402 assert run_home_with_path(dir, home, fake, 'undo', '--yes').code == 0403 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'404 assert !os.exists(marker)405}406407408409fn test_a_restore_leaves_unchanged_files_alone() {410 dir := temp_repo('restore-skip')411 defer { os.rmdir_all(dir) or {} }412 init_repo(dir)413 write(dir, 'a.txt', 'one\n')414 write(dir, 'same.txt', 'never changes\n')415 assert run(dir, 'save', 'one').code == 0416 write(dir, 'a.txt', 'two\n')417 assert run(dir, 'save', 'two').code == 0418 long_ago := i64(1700000000)419 os.utime(os.join_path(dir, 'same.txt'), long_ago, long_ago) or { panic(err) }420 assert run(dir, 'undo', '--yes').code == 0421 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'422 assert os.file_last_mod_unix(os.join_path(dir, 'same.txt')) == long_ago423}424425426427fn test_status_sees_an_executable_bit_change() {428 dir := temp_repo('chmod')429 defer { os.rmdir_all(dir) or {} }430 init_repo(dir)431 write(dir, 'run.sh', 'echo hi\n')432 assert run(dir, 'save', 'script').code == 0433 os.chmod(os.join_path(dir, 'run.sh'), 0o755) or { panic(err) }434 status := run(dir, 'status', '--json')435 assert status.out.contains('"dirty":true')436 assert status.out.contains('"modified":["run.sh"]')437}438439fn test_log_and_diff() {440 dir := temp_repo('history')441 defer { os.rmdir_all(dir) or {} }442 init_repo(dir)443 write(dir, 'a.txt', 'one\n')444 first := run(dir, 'save', 'one', '--session', 'alpha', '--json')445 id1 := first.out.all_after('"id":"')[..12]446 write(dir, 'a.txt', 'two\n')447 second := run(dir, 'save', 'two', '--session', 'beta', '--json')448 id2 := second.out.all_after('"id":"')[..12]449 cp := run(dir, 'checkpoint', 'checkpoint', '--json')450 assert cp.code == 0451 log := run(dir, 'log')452 assert log.code == 0453 assert log.out.contains('▸')454 assert log.out.contains('◐')455 lj := run(dir, 'log', '--json')456 assert lj.code == 0457 assert lj.out.contains('"head_id"')458 assert lj.out.contains('"saves"')459 assert run(dir, 'log', '--since', 'today').out.contains('two')460 session := run(dir, 'log', '--session', 'alpha')461 assert session.out.contains('one')462 assert !session.out.contains('two')463 write(dir, 'a.txt', 'three\n')464 d := run(dir, 'diff')465 assert d.code == 0466 assert d.out.contains('--- a/')467 assert d.out.contains('+++ b/')468 assert run(dir, 'diff', id1).code == 0469 assert run(dir, 'diff', id1, id2).code == 0470 dj := run(dir, 'diff', '--json')471 assert dj.code == 0472 assert dj.out.contains('"from"')473 assert dj.out.contains('"to"')474 assert dj.out.contains('"files"')475 assert dj.out.contains('"change":"modified"')476 assert run(dir, 'diff', 'not-a-real-id').code == 1477 assert run(dir, 'diff', id1, id2, 'third').code == 1478 write(dir, 'blob.bin', '\x00\x01\x02')479 diff_binary := run(dir, 'diff', '--json')480 assert diff_binary.out.contains('"binary":true')481 assert diff_binary.out.contains('Binary files differ')482}483484fn test_undo_revert_and_load() {485 dir := temp_repo('restore')486 defer { os.rmdir_all(dir) or {} }487 init_repo(dir)488 write(dir, 'a.txt', 'one\n')489 one := run(dir, 'save', 'one', '--json').out490 id1 := one.all_after('"id":"')[..12]491 write(dir, 'a.txt', 'two\n')492 write(dir, 'gone.txt', 'gone\n')493 second := run(dir, 'save', 'two', '--json').out494 id2 := second.all_after('"id":"')[..12]495 write(dir, 'a.txt', 'three\n')496 third := run(dir, 'save', 'three', '--json').out497 id3 := third.all_after('"id":"')[..12]498 499 500 write(dir, 'never.txt', 'keep me\n')501 log_before := run(dir, 'log').out502 for args in [['undo', '--yes'], ['load', id1, '--yes'], ['pluck', id2, '--yes']] {503 blocked := run(dir, ...args)504 assert blocked.code == 1505 assert blocked.err.contains('unsaved changes')506 }507 assert run(dir, 'log').out == log_before508 os.rm(os.join_path(dir, 'never.txt')) or { panic(err) }509 write(dir, 'a.txt', 'edited, never saved\n')510 assert run(dir, 'undo', '--yes').err.contains('unsaved changes')511 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'edited, never saved\n'512 write(dir, 'a.txt', 'three\n')513 os.rm(os.join_path(dir, 'gone.txt')) or { panic(err) }514 assert run(dir, 'load', id1, '--yes').err.contains('unsaved changes')515 write(dir, 'gone.txt', 'gone\n')516 undo := run(dir, 'undo', '--yes', '--json')517 assert undo.code == 0518 if !undo.out.contains(id2) {519 panic('undo JSON did not name ${id2}: ${undo.out}')520 }521 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'522 assert os.exists(os.join_path(dir, 'gone.txt'))523 assert run(dir, 'load', id3, '--yes').code == 0524 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'525 assert run(dir, 'undo', '2', '--yes').code == 0526 assert run(dir, 'load', id2, '--yes').code == 0527 assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')528 write(dir, 'a.txt', 'dirty\n')529 before := run(dir, 'log')530 refuse := run(dir, 'revert')531 assert refuse.code == 1532 assert refuse.err.contains('needs --yes when not interactive')533 assert run(dir, 'log').out == before.out534 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'535 write(dir, 'untracked.txt', 'remove\n')536 rv := run(dir, 'revert', '--yes', '--json')537 assert rv.code == 0538 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'539 assert !os.exists(os.join_path(dir, 'untracked.txt'))540 541 kept := rv.out.all_after('"checkpoint":"')[..12]542 assert run(dir, 'load', kept, '--yes').code == 0543 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'544 assert os.read_file(os.join_path(dir, 'untracked.txt'))! == 'remove\n'545 write(dir, 'a.txt', 'dirty again\n')546 assert run(dir, 'revert', '--yes').out.contains('koh load ')547 assert run(dir, 'revert', '--yes').out.contains('working tree already matches')548 assert run(dir, 'load', '--checkpoint', '--yes').code == 0549 assert run(dir, 'load', id2, '--json').code == 1550 assert run(dir, 'load', id2, '--json').err.contains('error')551 assert run(dir, 'load').code == 2552 assert run(dir, 'load', 'unknown-id', '--yes').code == 1553}554555fn test_flag_and_flagged_log() {556 dir := temp_repo('flag')557 defer { os.rmdir_all(dir) or {} }558 init_repo(dir)559 write(dir, 'flag.txt', 'one\n')560 saved := run(dir, 'save', 'flag me', '--json')561 id := saved.out.all_after('"id":"')[..12]562 flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')563 assert flagged.code == 0564 assert flagged.out.contains('"flagged":true')565 assert flagged.out.contains('"reason":"unsafe"')566 list := run(dir, 'flag', '--json')567 assert list.code == 0568 assert list.out.contains(id)569 assert list.out.contains('unsafe')570 assert run(dir, 'log').out.contains('⚑')571 assert run(dir, 'flag', '--clear', id).code == 0572 assert run(dir, 'flag').out.contains('nothing is flagged')573 assert run(dir, 'flag', 'unknown-id').code == 1574 assert run(dir, 'flag').code == 0575}576577fn test_pluck_forward_reverse_dry_run_and_conflict() {578 dir := temp_repo('pluck')579 defer { os.rmdir_all(dir) or {} }580 init_repo(dir)581 write(dir, 'a.txt', 'one\n')582 first := run(dir, 'save', 'one', '--json').out583 id1 := first.all_after('"id":"')[..12]584 write(dir, 'a.txt', 'two\n')585 second := run(dir, 'save', 'two', '--json').out586 id2 := second.all_after('"id":"')[..12]587 assert run(dir, 'load', id1, '--yes').code == 0588 dry := run(dir, 'pluck', id2, '--dry-run')589 assert dry.code == 0590 assert dry.out.contains('a.txt')591 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'592 assert run(dir, 'pluck', id2, '--yes').code == 0593 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'594 assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0595 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'596 write(dir, 'a.txt', 'different\n')597 assert run(dir, 'pluck', id2, '--dry-run').code == 1598 assert run(dir, 'save', 'different').code == 0599 conflict := run(dir, 'pluck', id2, '--yes')600 assert conflict.code == 1601 assert conflict.err.contains('a.txt')602 assert run(dir, 'pluck').code == 2603 assert run(dir, 'pluck', 'unknown-id').code == 1604}605606fn test_peek_and_doctor() {607 dir := temp_repo('peek-doctor')608 defer { os.rmdir_all(dir) or {} }609 init_repo(dir)610 write(dir, 'peek.txt', 'peek me\n')611 assert run(dir, 'save', 'peekable').code == 0612 id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]613 before_log := run(dir, 'log', '--json').out614 peek := run(dir, 'peek', id, '--json')615 assert peek.code == 0616 peek_path := peek.out.all_after('"path":"').before('"')617 assert peek.out.contains('"id":"${id}"')618 assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'619 assert run(dir, 'status').out.contains('no changes')620 assert run(dir, 'log', '--json').out == before_log621 assert run(dir, 'peek').code == 2622 assert run(dir, 'peek', 'unknown-id').code == 1623 backup := os.join_path(dir, 'doctor.backup')624 healthy := run(dir, 'doctor')625 assert healthy.code == 0626 assert healthy.out.contains('is sound')627 report := run(dir, 'doctor', '--json')628 assert report.code == 0629 for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',630 'backup'] {631 read_json_has(report.out, key)632 }633 backed := run(dir, 'doctor', '--backup', backup)634 assert backed.code == 0635 assert os.exists(backup)636 assert backed.out.contains(backup)637 assert run(dir, 'doctor', '--backup', backup).code == 1638 assert run(dir, 'peek', id, '--json').code == 0639 640 kept := run(dir, 'doctor', '--json')641 assert kept.code == 0642 assert os.exists(peek_path)643 os.rmdir_all(peek_path) or {}644}645646fn test_strip_refuses_without_terminal() {647 dir := temp_repo('strip')648 defer { os.rmdir_all(dir) or {} }649 init_repo(dir)650 write(dir, 'secret.txt', 'keep\n')651 assert run(dir, 'save', 'secret').code == 0652 before_log := run(dir, 'log', '--json').out653 refused := run(dir, 'strip', 'secret.txt')654 assert refused.code == 1655 assert refused.err.contains('needs a terminal')656 assert run(dir, 'log', '--json').out == before_log657 assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'658 assert run(dir, 'log', '--json').out.contains('secret')659}660661fn test_saves_and_checkpoints_aliases() {662 dir := temp_repo('aliases')663 defer { os.rmdir_all(dir) or {} }664 init_repo(dir)665 write(dir, 'alias.txt', 'one\n')666 assert run(dir, 'save', 'plain save').code == 0667 assert run(dir, 'checkpoint', 'checkpoint save').code == 0668 saves := run(dir, 'saves')669 assert saves.code == 0670 assert saves.out.contains('plain save')671 672 assert !saves.out.contains('checkpoint save')673 checkpoints := run(dir, 'checkpoints')674 assert checkpoints.code == 0675 assert checkpoints.out.contains('checkpoint save')676 assert !checkpoints.out.contains('plain save')677 json_saves := run(dir, 'saves', '--json')678 assert json_saves.code == 0679 assert json_saves.out.contains('"is_checkpoint":false')680 assert !json_saves.out.contains('"is_checkpoint":true')681 json_result := run(dir, 'checkpoints', '--json')682 assert json_result.code == 0683 assert json_result.out.contains('"head_id"')684 assert json_result.out.contains('"saves"')685 assert json_result.out.contains('"is_checkpoint":true')686 assert !json_result.out.contains('"is_checkpoint":false')687 688 assert run(dir, 'log').out.contains('checkpoint save')689}690691fn test_remote_configuration_is_isolated_and_atomic() {692 dir := temp_repo('remote')693 home := temp_repo('remote-home')694 defer {695 os.rmdir_all(dir) or {}696 os.rmdir_all(home) or {}697 }698 empty := run_home(dir, home, 'remote', '--json')699 assert empty.code == 0700 assert empty.out.contains('"configured":false')701 target := os.join_path(home, 'remote-store')702 set := run_home(dir, home, 'remote', 'folder', target)703 assert set.code == 0704 assert os.is_dir(target)705 state := run_home(dir, home, 'remote', '--json')706 assert state.code == 0707 assert state.out.contains('"kind":"folder"')708 assert state.out.contains(target)709 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').code == 1710 assert run_home(dir, home, 'remote', '--json').out.contains(target)711 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').err.contains('KOH_ACCESS_KEY')712 assert run_home(dir, home, 'remote', '--forget').code == 0713 assert run_home(dir, home, 'remote', '--json').out.contains('"configured":false')714}715716717718fn test_remote_written_by_hand() {719 dir := temp_repo('remote-by-hand')720 home := temp_repo('remote-by-hand-home')721 defer {722 os.rmdir_all(dir) or {}723 os.rmdir_all(home) or {}724 }725 config := os.join_path(home, '.config', 'koh')726 os.mkdir_all(config) or { panic(err) }727 target := os.join_path(home, 'remote-store')728 os.mkdir_all(target) or { panic(err) }729 os.write_file(os.join_path(config, 'remote'), 'kind = folder\npath = ${target}\n') or {730 panic(err)731 }732 assert run_home(dir, home, 'init').code == 0733 os.write_file(os.join_path(dir, 'a.txt'), 'a\n') or { panic(err) }734 assert run_home(dir, home, 'save', 'first').code == 0735 offered := run_home(dir, home, 'offer', '--json')736 assert offered.code == 0737 assert offered.out.contains('"head"')738739 740 os.write_file(os.join_path(config, 'remote'), 'kind=s3\nendpoint=http://127.0.0.1:1\nbucket=b\nregion=r\n') or {741 panic(err)742 }743 credentials := os.join_path(config, 'credentials')744 os.write_file(credentials, 'access_key=AK\nsecret_key=SK\n') or { panic(err) }745 os.chmod(credentials, 0o644) or { panic(err) }746 loose := run_home(dir, home, 'offer')747 assert loose.code == 1748 assert loose.err.contains('chmod 600')749750 751 os.write_file(os.join_path(config, 'remote'), 'kind=ftp\n') or { panic(err) }752 unreadable := run_home(dir, home, 'offer')753 assert unreadable.code == 1754 assert unreadable.err.contains('not readable')755 assert run_home(dir, home, 'remote', 'folder', target).code == 0756 assert run_home(dir, home, 'offer').code == 0757}758759760761fn test_a_renamed_clone_keeps_offering_to_its_repo() {762 syncroot := temp_repo('rename-clone')763 home := temp_repo('rename-clone-home')764 defer {765 os.rmdir_all(syncroot) or {}766 os.rmdir_all(home) or {}767 }768 remote := os.join_path(syncroot, 'remote')769 a := os.join_path(syncroot, 'A', 'proj')770 b_parent := os.join_path(syncroot, 'B')771 os.mkdir_all(a) or { panic(err) }772 os.mkdir_all(b_parent) or { panic(err) }773 init_repo(a)774 write(a, 'a.txt', 'from A\n')775 assert run_home(a, home, 'remote', 'folder', remote).code == 0776 assert run_home(a, home, 'save', 'A first').code == 0777 assert run_home(b_parent, home, 'steal', 'proj').code == 0778 renamed := os.join_path(b_parent, 'renamed')779 os.mv(os.join_path(b_parent, 'proj'), renamed) or { panic(err) }780 write(renamed, 'b.txt', 'from B\n')781 assert run_home(renamed, home, 'save', 'B first').code == 0782 assert !os.exists(os.join_path(remote, 'renamed'))783 assert run_home(a, home, 'steal').code == 0784 assert os.read_file(os.join_path(a, 'b.txt'))! == 'from B\n'785}786787788789fn test_clone_of_a_damaged_remote_leaves_nothing() {790 syncroot := temp_repo('damaged')791 home := temp_repo('damaged-home')792 defer {793 os.rmdir_all(syncroot) or {}794 os.rmdir_all(home) or {}795 }796 remote := os.join_path(syncroot, 'remote')797 a := os.join_path(syncroot, 'A', 'proj')798 b_parent := os.join_path(syncroot, 'B')799 os.mkdir_all(a) or { panic(err) }800 os.mkdir_all(b_parent) or { panic(err) }801 init_repo(a)802 write(a, 'a.txt', 'from A\n')803 assert run_home(a, home, 'remote', 'folder', remote).code == 0804 assert run_home(a, home, 'save', 'A first').code == 0805 objects := os.join_path(remote, 'proj', 'objects')806 for name in os.ls(objects) or { panic(err) } {807 os.chmod(os.join_path(objects, name), 0o644) or {}808 os.write_file(os.join_path(objects, name), 'junk') or { panic(err) }809 }810 clone := run_home(b_parent, home, 'steal', 'proj')811 assert clone.code == 1812 assert clone.err.contains('is damaged')813 assert !os.exists(os.join_path(b_parent, 'proj'))814}815816fn test_offer_steal_folder_remote() {817 syncroot := temp_repo('sync')818 home := temp_repo('sync-home')819 defer {820 os.rmdir_all(syncroot) or {}821 os.rmdir_all(home) or {}822 }823 remote := os.join_path(syncroot, 'remote')824 a := os.join_path(syncroot, 'A', 'proj')825 b_parent := os.join_path(syncroot, 'B')826 os.mkdir_all(a) or { panic(err) }827 os.mkdir_all(b_parent) or { panic(err) }828 init_repo(a)829 write(a, 'a.txt', 'from A\n')830 assert run_home(a, home, 'remote', 'folder', remote).code == 0831 assert run_home(a, home, 'save', 'A first').code == 0832 clone := run_home(b_parent, home, 'steal', 'proj')833 assert clone.code == 0834 b := os.join_path(b_parent, 'proj')835 assert os.read_file(os.join_path(b, 'a.txt'))! == 'from A\n'836 write(b, 'b.txt', 'from B\n')837 assert run_home(b, home, 'save', 'B first').code == 0838 write(a, 'a2.txt', 'second\n')839 assert run_home(a, home, 'save', 'A second').code == 0840 assert run_home(a, home, 'steal').code == 0841 assert os.exists(os.join_path(a, 'b.txt'))842 assert run_home(b_parent, home, 'steal', '--list').out.contains('proj')843 write(a, 'clash.txt', 'base\n')844 assert run_home(a, home, 'save', 'clash base').code == 0845 assert run_home(b, home, 'steal').code == 0846 write(a, 'clash.txt', 'A version\n')847 assert run_home(a, home, 'save', 'A clash').code == 0848 write(b, 'clash.txt', 'B version\n')849 assert run_home(b, home, 'save', 'B clash').code == 0850 refused := run_home(b, home, 'steal')851 assert refused.code == 1852 assert refused.err.contains('--mine') || refused.err.contains('--theirs')853 b_head := run_home(b, home, 'status', '--json').out.all_after('"head_id":"')[..12]854 theirs := run_home(b, home, 'steal', '--theirs', '--json')855 assert theirs.code == 0856 857 858 assert theirs.out.contains('"previous_head":"${b_head}"')859 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'860 assert run_home(b, home, 'load', b_head, '--yes').code == 0861 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B version\n'862 assert run_home(b, home, 'load', theirs.out.all_after('"head":"')[..12], '--yes').code == 0863 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'864 write(a, 'clash.txt', 'A second\n')865 assert run_home(a, home, 'save', 'A second clash').code == 0866 write(b, 'clash.txt', 'B second\n')867 assert run_home(b, home, 'save', 'B second clash').code == 0868 assert run_home(b, home, 'steal', '--mine').code == 0869 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B second\n'870 assert run_home(b, home, 'log').out.contains('combine: kept mine for')871}872873874875fn test_a_clone_reports_flagged_saves() {876 syncroot := temp_repo('clone-flags')877 home := temp_repo('clone-flags-home')878 defer {879 os.rmdir_all(syncroot) or {}880 os.rmdir_all(home) or {}881 }882 a := os.join_path(syncroot, 'A', 'proj')883 os.mkdir_all(a) or { panic(err) }884 init_repo(a)885 assert run_home(a, home, 'remote', 'folder', os.join_path(syncroot, 'remote')).code == 0886 write(a, 'a.txt', 'one\n')887 id := run_home(a, home, 'save', 'risky', '--json').out.all_after('"id":"')[..12]888 assert run_home(a, home, 'flag', id, '--reason', 'broke the build').code == 0889 assert run_home(a, home, 'offer').code == 0890 for name in ['B', 'C'] {891 os.mkdir_all(os.join_path(syncroot, name)) or { panic(err) }892 }893 json_clone := run_home(os.join_path(syncroot, 'B'), home, 'steal', 'proj', '--json')894 assert json_clone.code == 0895 assert json_clone.out.contains('"flagged":["${id}"]')896 text_clone := run_home(os.join_path(syncroot, 'C'), home, 'steal', 'proj')897 assert text_clone.code == 0898 assert text_clone.out.contains('[${id}] came in flagged — broke the build')899}900901902903904fn test_an_unmounted_folder_remote_is_not_recreated() {905 syncroot := temp_repo('unmounted')906 home := temp_repo('unmounted-home')907 defer {908 os.rmdir_all(syncroot) or {}909 os.rmdir_all(home) or {}910 }911 mount := os.join_path(syncroot, 'nas')912 remote := os.join_path(mount, 'koh')913 a := os.join_path(syncroot, 'proj')914 os.mkdir_all(a) or { panic(err) }915 init_repo(a)916 assert run_home(a, home, 'remote', 'folder', remote).code == 0917 write(a, 'a.txt', 'one\n')918 assert run_home(a, home, 'save', 'first').code == 0919 os.rename(mount, os.join_path(syncroot, 'nas-unplugged')) or { panic(err) }920 os.mkdir(mount) or { panic(err) }921 offered := run_home(a, home, 'offer')922 assert offered.code == 1923 assert offered.err.contains('mounted')924 assert !os.exists(remote)925 write(a, 'a.txt', 'two\n')926 saved := run_home(a, home, 'save', 'second')927 assert saved.code == 0928 assert saved.out.contains('mounted')929 assert !os.exists(remote)930}931932fn test_man_and_koh2_handoff() {933 dir := temp_repo('man')934 home := temp_repo('man-home')935 defer {936 os.rmdir_all(dir) or {}937 os.rmdir_all(home) or {}938 }939 whole := run_home(dir, home, 'man')940 assert whole.code == 0941 assert whole.out.contains('NAME')942 assert whole.out.contains('COMMANDS')943 assert !whole.out.contains('\x1b') && !whole.out.contains('◆')944 strip := run_home(dir, home, 'man', 'strip')945 assert strip.code == 0946 assert strip.out.contains('strip')947 assert !strip.out.contains('koh save ')948 steal := run_home(dir, home, 'man', 'steal').out949 assert steal.starts_with(' koh steal') && steal.contains('koh steal my-project')950 assert run_home(dir, home, 'man', 'init').out.trim_space() == "koh init\n Create `.koh` here, and a starter `.kohignore` if there isn't one."951 assert run_home(dir, home, 'man', 'nonsense').code == 2952 legacy := temp_repo('legacy')953 defer { os.rmdir_all(legacy) or {} }954 os.mkdir_all(os.join_path(legacy, '.koh')) or { panic(err) }955 fake := os.join_path(home, 'bin')956 os.mkdir_all(fake) or { panic(err) }957 fake_koh := os.join_path(fake, 'koh2')958 os.write_file(fake_koh, '#!/bin/sh\nprintf "handoff:%s\\n" "$*"\nexit 7\n') or { panic(err) }959 os.chmod(fake_koh, 0o755) or { panic(err) }960 handoff := run_home_with_path(legacy, home, fake, 'status', '--json')961 assert handoff.code == 7962 assert handoff.out.contains('handoff:status --json')963 964 assert handoff.err.contains('Koh 2 repo')965 assert handoff.err.trim_space().split_into_lines().len == 1966 assert handoff.out.trim_space() == 'handoff:status --json'967}968969fn run_home_with_path(repo string, home string, bin_dir string, args ...string) Result {970 stamp := '${os.getpid()}-${time.now().unix_nano()}'971 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')972 out_path := '${base}.out'973 err_path := '${base}.err'974 code_path := '${base}.code'975 mut command := 'HOME=${q(home)} PATH=${q(bin_dir + ':' + os.getenv('PATH'))} ${q(binary)}'976 for arg in args {977 command += ' ${q(arg)}'978 }979 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" \$? >${q(code_path)}'980 result := os.execute(shell)981 if result.exit_code != 0 {982 panic('could not run test command: ${result.output}')983 }984 code := os.read_file(code_path) or { panic(err) }985 out := os.read_file(out_path) or { '' }986 err := os.read_file(err_path) or { '' }987 os.rm(out_path) or {}988 os.rm(err_path) or {}989 os.rm(code_path) or {}990 return Result{991 code: code.int()992 out: out993 err: err994 }995}996997998999fn koh2_binary() string {1000 return os.find_abs_path_of_executable('koh') or { '' }1001}1002100310041005fn run_koh2(dir string, command string) {1006 os.execute('cd ${q(dir)} && ${command} >/dev/null 2>&1')1007}10081009fn save_ids(dir string) []string {1010 mut ids := []string{}1011 for name in os.ls(os.join_path(dir, '.koh', 'snapshots')) or { []string{} } {1012 if name.ends_with('.json') {1013 ids << name.trim_right('.json')1014 }1015 }1016 ids.sort()1017 return ids1018}10191020fn test_migrate_converts_a_koh2_repo_and_keeps_every_save() {1021 koh2 := koh2_binary()1022 if koh2 == '' {1023 println('skipping the migration tests: koh (Koh 2) is not on PATH')1024 return1025 }1026 dir := temp_repo('migrate')1027 defer { os.rmdir_all(dir) or {} }1028 home := temp_repo('migrate-home')1029 defer { os.rmdir_all(home) or {} }1030 run_koh2(dir, '${q(koh2)} init')1031 1032 write(dir, 'code.txt', 'x'.repeat(700))1033 write(dir, 'tiny.txt', 'tiny\n')1034 run_koh2(dir, '${q(koh2)} save "first save"')1035 run_koh2(dir, '${q(koh2)} checkpoint "before something"')1036 write(dir, 'code.txt', 'y'.repeat(700))1037 run_koh2(dir, '${q(koh2)} save "second save"')1038 original := save_ids(dir)1039 assert original.len == 310401041 report := run_home(dir, home, 'migrate')1042 assert report.code == 01043 assert report.out.contains('migrated 3 saves')1044 1045 assert os.is_file(os.join_path(dir, '.koh'))1046 assert os.is_dir(os.join_path(dir, '.koh-v2'))1047 assert os.is_file(os.join_path(dir, '.kohignore'))1048 assert !os.exists(os.join_path(dir, '.koh-migrate'))10491050 1051 logged := run_home(dir, home, 'log', '--json')1052 assert logged.code == 01053 for id in original {1054 assert logged.out.contains(id)1055 }1056 assert run_home(dir, home, 'doctor').code == 01057 assert run_home(dir, home, 'status').code == 010581059 1060 assert run_home(dir, home, 'migrate').code == 11061}10621063fn test_migrate_refuses_a_tree_that_does_not_match_the_last_save() {1064 koh2 := koh2_binary()1065 if koh2 == '' {1066 return1067 }1068 dir := temp_repo('migrate-dirty')1069 defer { os.rmdir_all(dir) or {} }1070 home := temp_repo('migrate-dirty-home')1071 defer { os.rmdir_all(home) or {} }1072 run_koh2(dir, '${q(koh2)} init')1073 write(dir, 'code.txt', 'x'.repeat(700))1074 run_koh2(dir, '${q(koh2)} save "only save"')1075 write(dir, 'code.txt', 'unsaved change\n')10761077 refused := run_home(dir, home, 'migrate')1078 assert refused.code == 11079 assert refused.err.contains('save with koh2 first')1080 1081 assert os.is_dir(os.join_path(dir, '.koh'))1082 assert !os.exists(os.join_path(dir, '.koh-v2'))1083 assert !os.exists(os.join_path(dir, '.koh-migrate'))1084}10851086108710881089fn test_migrate_passes_over_the_files_koh2_never_saved() {1090 koh2 := koh2_binary()1091 if koh2 == '' {1092 return1093 }1094 dir := temp_repo('migrate-koh2-ignores')1095 defer { os.rmdir_all(dir) or {} }1096 home := temp_repo('migrate-koh2-ignores-home')1097 defer { os.rmdir_all(home) or {} }1098 run_koh2(dir, '${q(koh2)} init')1099 write(dir, 'code.txt', 'x'.repeat(700))1100 run_koh2(dir, '${q(koh2)} save "only save"')1101 write(dir, '.kohconfig', 'stable_days = 14\n')1102 write(dir, '.kohignore.public', 'private/\n')1103 write(dir, 'lib/app.bundle', 'bundle\n')1104 write(dir, 'ios/Settings.bundle/Root.plist', 'settings\n')1105 1106 1107 1108 write(dir, 'native/t.dSYM/Info.plist', 'plist\n')1109 run_koh2(dir, '${q(koh2)} save --agent test --session test "add the dsym"')1110 1111 ignore_path := os.join_path(dir, '.koh', 'ignore')1112 rules := os.read_file(ignore_path)!.split_into_lines().filter(it != '*.bundle')1113 os.write_file(ignore_path, rules.join('\n') + '\nnative/*.dSYM\n')!1114 1115 write(dir, 'native/t.dSYM/Info.plist', 'changed\n')1116 changed := run_home(dir, home, 'migrate')1117 assert changed.code == 11118 assert changed.err.contains('native/t.dSYM/Info.plist does not match')1119 write(dir, 'native/t.dSYM/Info.plist', 'plist\n')1120 assert os.execute('cd ${q(dir)} && ${q(koh2)} status --json').output.contains('"dirty":false')11211122 migrated := run_home(dir, home, 'migrate')1123 assert migrated.code == 0, migrated.err1124 1125 assert os.read_file(os.join_path(dir, '.kohconfig'))! == 'stable_days = 14\n'1126}