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_version_matches_the_manifest() {108 dir := temp_repo('version')109 defer { os.rmdir_all(dir) or {} }110 version := run(dir, '--version').out.trim_space().all_after('koh ')111 manifest := os.read_file(os.join_path(root, 'v.mod')) or { panic(err) }112 assert manifest.contains("version: '${version}'"), version113}114115fn test_save_checkpoint_and_status() {116 dir := temp_repo('save')117 defer { os.rmdir_all(dir) or {} }118 init_repo(dir)119 write(dir, 'tracked.txt', 'one\n')120 r := run(dir, 'save', 'first message', '--agent', 'agent-7', '--session', 'session-a', '--json')121 assert r.code == 0122 for key in ['id', 'parent', 'message', 'created_at', 'machine', 'is_checkpoint', 'agent', 'session',123 'files', 'size_bytes'] {124 read_json_has(r.out, key)125 }126 assert r.out.contains('"message":"first message"')127 assert r.out.contains('"agent":"agent-7"')128 assert r.out.contains('"session":"session-a"')129 assert r.out.contains('"is_checkpoint":false')130 assert run(dir, 'save', 'again').code == 1131 assert run(dir, 'save', 'again').err.contains('nothing has changed since the last save')132 assert run(dir, 'save').code == 2133 assert run(dir, 'save', 'x', '--bogus').code == 2134 cp := run(dir, 'checkpoint', 'still first', '--json')135 assert cp.code == 0136 assert cp.out.contains('"is_checkpoint":true')137 assert cp.out.contains('"parent":')138 clean := run(dir, 'status')139 assert clean.code == 0140 assert clean.out.contains('no changes')141 write(dir, 'tracked.txt', 'two\n')142 write(dir, 'new.txt', 'new\n')143 os.rm(os.join_path(dir, 'tracked.txt')) or { panic(err) }144 write(dir, '.kohignore', '*.log\n')145 status := run(dir, 'status')146 assert status.code == 0147 assert status.out.contains('+')148 assert status.out.contains('~') || status.out.contains('-')149 assert status.out.contains('added')150 js := run(dir, 'status', '--json')151 assert js.code == 0152 for key in ['head_id', 'head_msg', 'dirty', 'modified', 'added', 'deleted', 'ignore_changed',153 'saves', 'size_bytes', 'head_flagged'] {154 read_json_has(js.out, key)155 }156 assert js.out.contains('"ignore_changed":true')157 158 assert js.out.contains('"saves":2')159 assert js.out.contains('"size_bytes":${os.file_size(os.join_path(dir, '.koh'))}')160 assert js.out.contains('"head_flagged":false')161 head := js.out.all_after('"head_id":"').all_before('"')162 assert run(dir, 'flag', head, '--reason', 'bad').code == 0163 assert run(dir, 'status', '--json').out.contains('"head_flagged":true')164}165166fn test_status_requires_repo_and_ignore_rules() {167 dir := temp_repo('status')168 defer { os.rmdir_all(dir) or {} }169 assert run(dir, 'status').code == 1170 init_repo(dir)171 write(dir, '.kohignore', '*.log\nbuild/\n')172 write(dir, 'keep.txt', 'keep\n')173 write(dir, 'ignored.log', 'ignore\n')174 write(dir, 'build/output.txt', 'ignore\n')175 assert run(dir, 'save', 'keep').code == 0176 assert run(dir, 'status').out.contains('no changes')177 assert !run(dir, 'status').out.contains('ignored.log')178 assert !run(dir, 'status').out.contains('build/output.txt')179 assert !run(dir, 'status').out.contains('LOG.md')180}181182183184185fn test_a_save_succeeds_when_log_md_cannot_be_written() {186 dir := temp_repo('logmd')187 defer { os.rmdir_all(dir) or {} }188 init_repo(dir)189 os.mkdir_all(os.join_path(dir, 'LOG.md')) or { panic(err) }190 write(dir, 'a.txt', 'one\n')191 r := run(dir, 'save', 'first', '--json')192 assert r.code == 0193 assert r.out.starts_with('{')194 assert r.err.contains('LOG.md')195 assert run(dir, 'log', '--json').out.contains('"message":"first"')196}197198199200201fn test_a_restore_refuses_a_folder_of_ignored_files_in_the_way() {202 dir := temp_repo('dir-in-way')203 defer { os.rmdir_all(dir) or {} }204 init_repo(dir)205 write(dir, 'out', 'a file\n')206 first := run(dir, 'save', 'out is a file', '--json').out.all_after('"id":"')[..12]207 os.rm(os.join_path(dir, 'out')) or { panic(err) }208 write(dir, 'out/build.log', 'ignored\n')209 assert run(dir, 'save', 'out is gone').code == 0210 r := run(dir, 'load', first, '--yes')211 assert r.code == 1212 assert r.err.contains('out')213 assert os.ls(os.join_path(dir, 'out'))! == ['build.log']214}215216217218219fn test_a_restore_never_writes_through_a_symlinked_folder() {220 dir := temp_repo('link-in-way')221 outside := temp_repo('link-target')222 defer {223 os.rmdir_all(dir) or {}224 os.rmdir_all(outside) or {}225 }226 init_repo(dir)227 write(dir, 'docs/a.md', 'saved\n')228 first := run(dir, 'save', 'docs', '--json').out.all_after('"id":"')[..12]229 os.rmdir_all(os.join_path(dir, 'docs')) or { panic(err) }230 write(outside, 'a.md', 'precious\n')231 os.symlink(outside, os.join_path(dir, 'docs')) or { panic(err) }232 assert run(dir, 'save', 'docs is a link now').code == 0233 head := run(dir, 'status', '--json').out234 r := run(dir, 'load', first, '--yes')235 assert r.code == 1236 assert r.err.contains('docs')237 assert os.read_file(os.join_path(outside, 'a.md'))! == 'precious\n'238 assert run(dir, 'status', '--json').out == head239}240241242243244245fn test_a_restore_cut_short_points_at_revert() {246 dir := temp_repo('cut-short')247 defer {248 os.chmod(os.join_path(dir, 'zz'), 0o755) or {}249 os.rmdir_all(dir) or {}250 }251 init_repo(dir)252 write(dir, 'zero', '0\n')253 assert run(dir, 'save', 'zero').code == 0254 write(dir, 'a.txt', 'v1\n')255 write(dir, 'zz/f', 'v1\n')256 assert run(dir, 'save', 'first').code == 0257 write(dir, 'a.txt', 'v2\n')258 write(dir, 'zz/f', 'v2\n')259 assert run(dir, 'save', 'second').code == 0260 261 262 os.chmod(os.join_path(dir, 'zz'), 0o555)!263 stopped := run(dir, 'undo', '--yes')264 assert stopped.code == 1265 assert stopped.err.contains('koh revert')266 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'v1\n'267 assert os.read_file(os.join_path(dir, 'zz/f'))! == 'v2\n'268 269 os.chmod(os.join_path(dir, 'zz'), 0o000)!270 assert run(dir, 'status').err.contains('koh revert')271 os.chmod(os.join_path(dir, 'zz'), 0o755)!272273 assert run(dir, 'status', '--json').out.contains('"restore_unfinished":true')274 for args in [['save', 'half'], ['checkpoint', 'half'], ['undo', '--yes']] {275 r := run(dir, ...args)276 assert r.code == 1277 assert r.err.contains('koh revert')278 }279 assert run(dir, 'revert', '--yes').code == 0280 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'v1\n'281 assert os.read_file(os.join_path(dir, 'zz/f'))! == 'v1\n'282 assert run(dir, 'status', '--json').out.contains('"restore_unfinished":false')283 write(dir, 'a.txt', 'v3\n')284 assert run(dir, 'save', 'carry on').code == 0285}286287288289fn test_doctor_sweeps_only_old_peek_folders() {290 dir := temp_repo('peek-sweep')291 292 293 fresh := os.join_path(os.temp_dir(), 'koh-peek-fresh${os.getpid()}')294 old := os.join_path(os.temp_dir(), 'koh-peek-old${os.getpid()}')295 defer {296 os.rmdir_all(dir) or {}297 os.rmdir_all(fresh) or {}298 os.rmdir_all(old) or {}299 }300 init_repo(dir)301 write(dir, 'a.txt', 'one\n')302 assert run(dir, 'save', 'one').code == 0303 os.mkdir_all(fresh) or { panic(err) }304 os.mkdir_all(old) or { panic(err) }305 two_days_ago := time.now().unix() - 2 * 86400306 os.utime(old, two_days_ago, two_days_ago) or { panic(err) }307 assert run(dir, 'doctor', '--json').code == 0308 assert os.exists(fresh)309 assert !os.exists(old)310}311312313314315fn test_a_restore_never_overwrites_an_ignored_file() {316 dir := temp_repo('ignored-restore')317 defer { os.rmdir_all(dir) or {} }318 init_repo(dir)319 write(dir, '.env', 'KEY=saved long ago\n')320 first := run(dir, 'save', 'with env', '--json').out.all_after('"id":"')[..12]321 write(dir, '.kohignore', '.env\n')322 assert run(dir, 'save', 'ignore env').code == 0323 write(dir, '.env', 'KEY=never saved\n')324 assert run(dir, 'load', first, '--yes').code == 0325 assert os.read_file(os.join_path(dir, '.env'))! == 'KEY=never saved\n'326327 328 other := temp_repo('ignored-restore-missing')329 defer { os.rmdir_all(other) or {} }330 init_repo(other)331 write(other, '.env', 'KEY=saved long ago\n')332 again := run(other, 'save', 'with env', '--json').out.all_after('"id":"')[..12]333 write(other, '.kohignore', '.env\n')334 assert run(other, 'save', 'ignore env').code == 0335 os.rm(os.join_path(other, '.env')) or { panic(err) }336 assert run(other, 'load', again, '--yes').code == 0337 assert os.read_file(os.join_path(other, '.env'))! == 'KEY=saved long ago\n'338}339340341342fn test_a_going_back_preview_does_not_run_diff() {343 dir := temp_repo('preview-nodiff')344 home := temp_repo('preview-nodiff-home')345 defer {346 os.rmdir_all(dir) or {}347 os.rmdir_all(home) or {}348 }349 init_repo(dir)350 write(dir, 'a.txt', 'one\n')351 assert run_home(dir, home, 'save', 'one').code == 0352 write(dir, 'a.txt', 'two\n')353 assert run_home(dir, home, 'save', 'two').code == 0354 fake := os.join_path(home, 'bin')355 os.mkdir_all(fake) or { panic(err) }356 marker := os.join_path(home, 'diff-ran')357 os.write_file(os.join_path(fake, 'diff'), '#!/bin/sh\ntouch ${q(marker)}\nexit 2\n') or {358 panic(err)359 }360 os.chmod(os.join_path(fake, 'diff'), 0o755) or { panic(err) }361 assert run_home_with_path(dir, home, fake, 'undo', '--yes').code == 0362 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'363 assert !os.exists(marker)364}365366367368fn test_a_restore_leaves_unchanged_files_alone() {369 dir := temp_repo('restore-skip')370 defer { os.rmdir_all(dir) or {} }371 init_repo(dir)372 write(dir, 'a.txt', 'one\n')373 write(dir, 'same.txt', 'never changes\n')374 assert run(dir, 'save', 'one').code == 0375 write(dir, 'a.txt', 'two\n')376 assert run(dir, 'save', 'two').code == 0377 long_ago := i64(1700000000)378 os.utime(os.join_path(dir, 'same.txt'), long_ago, long_ago) or { panic(err) }379 assert run(dir, 'undo', '--yes').code == 0380 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'381 assert os.file_last_mod_unix(os.join_path(dir, 'same.txt')) == long_ago382}383384385386fn test_status_sees_an_executable_bit_change() {387 dir := temp_repo('chmod')388 defer { os.rmdir_all(dir) or {} }389 init_repo(dir)390 write(dir, 'run.sh', 'echo hi\n')391 assert run(dir, 'save', 'script').code == 0392 os.chmod(os.join_path(dir, 'run.sh'), 0o755) or { panic(err) }393 status := run(dir, 'status', '--json')394 assert status.out.contains('"dirty":true')395 assert status.out.contains('"modified":["run.sh"]')396}397398fn test_log_and_diff() {399 dir := temp_repo('history')400 defer { os.rmdir_all(dir) or {} }401 init_repo(dir)402 write(dir, 'a.txt', 'one\n')403 first := run(dir, 'save', 'one', '--session', 'alpha', '--json')404 id1 := first.out.all_after('"id":"')[..12]405 write(dir, 'a.txt', 'two\n')406 second := run(dir, 'save', 'two', '--session', 'beta', '--json')407 id2 := second.out.all_after('"id":"')[..12]408 cp := run(dir, 'checkpoint', 'checkpoint', '--json')409 assert cp.code == 0410 log := run(dir, 'log')411 assert log.code == 0412 assert log.out.contains('▸')413 assert log.out.contains('◐')414 lj := run(dir, 'log', '--json')415 assert lj.code == 0416 assert lj.out.contains('"head_id"')417 assert lj.out.contains('"saves"')418 assert run(dir, 'log', '--since', 'today').out.contains('two')419 session := run(dir, 'log', '--session', 'alpha')420 assert session.out.contains('one')421 assert !session.out.contains('two')422 write(dir, 'a.txt', 'three\n')423 d := run(dir, 'diff')424 assert d.code == 0425 assert d.out.contains('--- a/')426 assert d.out.contains('+++ b/')427 assert run(dir, 'diff', id1).code == 0428 assert run(dir, 'diff', id1, id2).code == 0429 dj := run(dir, 'diff', '--json')430 assert dj.code == 0431 assert dj.out.contains('"from"')432 assert dj.out.contains('"to"')433 assert dj.out.contains('"files"')434 assert dj.out.contains('"change":"modified"')435 assert run(dir, 'diff', 'not-a-real-id').code == 1436 assert run(dir, 'diff', id1, id2, 'third').code == 1437 write(dir, 'blob.bin', '\x00\x01\x02')438 diff_binary := run(dir, 'diff', '--json')439 assert diff_binary.out.contains('"binary":true')440 assert diff_binary.out.contains('Binary files differ')441}442443fn test_undo_revert_and_load() {444 dir := temp_repo('restore')445 defer { os.rmdir_all(dir) or {} }446 init_repo(dir)447 write(dir, 'a.txt', 'one\n')448 one := run(dir, 'save', 'one', '--json').out449 id1 := one.all_after('"id":"')[..12]450 write(dir, 'a.txt', 'two\n')451 write(dir, 'gone.txt', 'gone\n')452 second := run(dir, 'save', 'two', '--json').out453 id2 := second.all_after('"id":"')[..12]454 write(dir, 'a.txt', 'three\n')455 third := run(dir, 'save', 'three', '--json').out456 id3 := third.all_after('"id":"')[..12]457 458 459 write(dir, 'never.txt', 'keep me\n')460 log_before := run(dir, 'log').out461 for args in [['undo', '--yes'], ['load', id1, '--yes'], ['pluck', id2, '--yes']] {462 blocked := run(dir, ...args)463 assert blocked.code == 1464 assert blocked.err.contains('unsaved changes')465 }466 assert run(dir, 'log').out == log_before467 os.rm(os.join_path(dir, 'never.txt')) or { panic(err) }468 write(dir, 'a.txt', 'edited, never saved\n')469 assert run(dir, 'undo', '--yes').err.contains('unsaved changes')470 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'edited, never saved\n'471 write(dir, 'a.txt', 'three\n')472 os.rm(os.join_path(dir, 'gone.txt')) or { panic(err) }473 assert run(dir, 'load', id1, '--yes').err.contains('unsaved changes')474 write(dir, 'gone.txt', 'gone\n')475 undo := run(dir, 'undo', '--yes', '--json')476 assert undo.code == 0477 if !undo.out.contains(id2) {478 panic('undo JSON did not name ${id2}: ${undo.out}')479 }480 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'481 assert os.exists(os.join_path(dir, 'gone.txt'))482 assert run(dir, 'load', id3, '--yes').code == 0483 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'484 assert run(dir, 'undo', '2', '--yes').code == 0485 assert run(dir, 'load', id2, '--yes').code == 0486 assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')487 write(dir, 'a.txt', 'dirty\n')488 before := run(dir, 'log')489 refuse := run(dir, 'revert')490 assert refuse.code == 1491 assert refuse.err.contains('needs --yes when not interactive')492 assert run(dir, 'log').out == before.out493 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'494 write(dir, 'untracked.txt', 'remove\n')495 rv := run(dir, 'revert', '--yes', '--json')496 assert rv.code == 0497 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'498 assert !os.exists(os.join_path(dir, 'untracked.txt'))499 500 kept := rv.out.all_after('"checkpoint":"')[..12]501 assert run(dir, 'load', kept, '--yes').code == 0502 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'503 assert os.read_file(os.join_path(dir, 'untracked.txt'))! == 'remove\n'504 write(dir, 'a.txt', 'dirty again\n')505 assert run(dir, 'revert', '--yes').out.contains('koh load ')506 assert run(dir, 'revert', '--yes').out.contains('working tree already matches')507 assert run(dir, 'load', '--checkpoint', '--yes').code == 0508 assert run(dir, 'load', id2, '--json').code == 1509 assert run(dir, 'load', id2, '--json').err.contains('error')510 assert run(dir, 'load').code == 2511 assert run(dir, 'load', 'unknown-id', '--yes').code == 1512}513514fn test_flag_and_flagged_log() {515 dir := temp_repo('flag')516 defer { os.rmdir_all(dir) or {} }517 init_repo(dir)518 write(dir, 'flag.txt', 'one\n')519 saved := run(dir, 'save', 'flag me', '--json')520 id := saved.out.all_after('"id":"')[..12]521 flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')522 assert flagged.code == 0523 assert flagged.out.contains('"flagged":true')524 assert flagged.out.contains('"reason":"unsafe"')525 list := run(dir, 'flag', '--json')526 assert list.code == 0527 assert list.out.contains(id)528 assert list.out.contains('unsafe')529 assert run(dir, 'log').out.contains('⚑')530 assert run(dir, 'flag', '--clear', id).code == 0531 assert run(dir, 'flag').out.contains('nothing is flagged')532 assert run(dir, 'flag', 'unknown-id').code == 1533 assert run(dir, 'flag').code == 0534}535536fn test_pluck_forward_reverse_dry_run_and_conflict() {537 dir := temp_repo('pluck')538 defer { os.rmdir_all(dir) or {} }539 init_repo(dir)540 write(dir, 'a.txt', 'one\n')541 first := run(dir, 'save', 'one', '--json').out542 id1 := first.all_after('"id":"')[..12]543 write(dir, 'a.txt', 'two\n')544 second := run(dir, 'save', 'two', '--json').out545 id2 := second.all_after('"id":"')[..12]546 assert run(dir, 'load', id1, '--yes').code == 0547 dry := run(dir, 'pluck', id2, '--dry-run')548 assert dry.code == 0549 assert dry.out.contains('a.txt')550 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'551 assert run(dir, 'pluck', id2, '--yes').code == 0552 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'553 assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0554 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'555 write(dir, 'a.txt', 'different\n')556 assert run(dir, 'pluck', id2, '--dry-run').code == 1557 assert run(dir, 'save', 'different').code == 0558 conflict := run(dir, 'pluck', id2, '--yes')559 assert conflict.code == 1560 assert conflict.err.contains('a.txt')561 assert run(dir, 'pluck').code == 2562 assert run(dir, 'pluck', 'unknown-id').code == 1563}564565fn test_peek_and_doctor() {566 dir := temp_repo('peek-doctor')567 defer { os.rmdir_all(dir) or {} }568 init_repo(dir)569 write(dir, 'peek.txt', 'peek me\n')570 assert run(dir, 'save', 'peekable').code == 0571 id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]572 before_log := run(dir, 'log', '--json').out573 peek := run(dir, 'peek', id, '--json')574 assert peek.code == 0575 peek_path := peek.out.all_after('"path":"').before('"')576 assert peek.out.contains('"id":"${id}"')577 assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'578 assert run(dir, 'status').out.contains('no changes')579 assert run(dir, 'log', '--json').out == before_log580 assert run(dir, 'peek').code == 2581 assert run(dir, 'peek', 'unknown-id').code == 1582 backup := os.join_path(dir, 'doctor.backup')583 healthy := run(dir, 'doctor')584 assert healthy.code == 0585 assert healthy.out.contains('is sound')586 report := run(dir, 'doctor', '--json')587 assert report.code == 0588 for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',589 'backup'] {590 read_json_has(report.out, key)591 }592 backed := run(dir, 'doctor', '--backup', backup)593 assert backed.code == 0594 assert os.exists(backup)595 assert backed.out.contains(backup)596 assert run(dir, 'doctor', '--backup', backup).code == 1597 assert run(dir, 'peek', id, '--json').code == 0598 599 kept := run(dir, 'doctor', '--json')600 assert kept.code == 0601 assert os.exists(peek_path)602 os.rmdir_all(peek_path) or {}603}604605fn test_strip_refuses_without_terminal() {606 dir := temp_repo('strip')607 defer { os.rmdir_all(dir) or {} }608 init_repo(dir)609 write(dir, 'secret.txt', 'keep\n')610 assert run(dir, 'save', 'secret').code == 0611 before_log := run(dir, 'log', '--json').out612 refused := run(dir, 'strip', 'secret.txt')613 assert refused.code == 1614 assert refused.err.contains('needs a terminal')615 assert run(dir, 'log', '--json').out == before_log616 assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'617 assert run(dir, 'log', '--json').out.contains('secret')618}619620fn test_saves_and_checkpoints_aliases() {621 dir := temp_repo('aliases')622 defer { os.rmdir_all(dir) or {} }623 init_repo(dir)624 write(dir, 'alias.txt', 'one\n')625 assert run(dir, 'save', 'plain save').code == 0626 assert run(dir, 'checkpoint', 'checkpoint save').code == 0627 saves := run(dir, 'saves')628 assert saves.code == 0629 assert saves.out.contains('plain save')630 assert saves.out.contains('checkpoint save')631 checkpoints := run(dir, 'checkpoints')632 assert checkpoints.code == 0633 assert checkpoints.out.contains('checkpoint save')634 assert !checkpoints.out.contains('plain save')635 json_result := run(dir, 'checkpoints', '--json')636 assert json_result.code == 0637 assert json_result.out.contains('"head_id"')638 assert json_result.out.contains('"saves"')639 assert json_result.out.contains('"is_checkpoint":true')640 assert !json_result.out.contains('"is_checkpoint":false')641}642643fn test_remote_configuration_is_isolated_and_atomic() {644 dir := temp_repo('remote')645 home := temp_repo('remote-home')646 defer {647 os.rmdir_all(dir) or {}648 os.rmdir_all(home) or {}649 }650 empty := run_home(dir, home, 'remote', '--json')651 assert empty.code == 0652 assert empty.out.contains('"configured":false')653 target := os.join_path(home, 'remote-store')654 set := run_home(dir, home, 'remote', 'folder', target)655 assert set.code == 0656 assert os.is_dir(target)657 state := run_home(dir, home, 'remote', '--json')658 assert state.code == 0659 assert state.out.contains('"kind":"folder"')660 assert state.out.contains(target)661 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').code == 1662 assert run_home(dir, home, 'remote', '--json').out.contains(target)663 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').err.contains('KOH_ACCESS_KEY')664 assert run_home(dir, home, 'remote', '--forget').code == 0665 assert run_home(dir, home, 'remote', '--json').out.contains('"configured":false')666}667668669670fn test_remote_written_by_hand() {671 dir := temp_repo('remote-by-hand')672 home := temp_repo('remote-by-hand-home')673 defer {674 os.rmdir_all(dir) or {}675 os.rmdir_all(home) or {}676 }677 config := os.join_path(home, '.config', 'koh')678 os.mkdir_all(config) or { panic(err) }679 target := os.join_path(home, 'remote-store')680 os.mkdir_all(target) or { panic(err) }681 os.write_file(os.join_path(config, 'remote'), 'kind = folder\npath = ${target}\n') or {682 panic(err)683 }684 assert run_home(dir, home, 'init').code == 0685 os.write_file(os.join_path(dir, 'a.txt'), 'a\n') or { panic(err) }686 assert run_home(dir, home, 'save', 'first').code == 0687 offered := run_home(dir, home, 'offer', '--json')688 assert offered.code == 0689 assert offered.out.contains('"head"')690691 692 os.write_file(os.join_path(config, 'remote'), 'kind=s3\nendpoint=http://127.0.0.1:1\nbucket=b\nregion=r\n') or {693 panic(err)694 }695 credentials := os.join_path(config, 'credentials')696 os.write_file(credentials, 'access_key=AK\nsecret_key=SK\n') or { panic(err) }697 os.chmod(credentials, 0o644) or { panic(err) }698 loose := run_home(dir, home, 'offer')699 assert loose.code == 1700 assert loose.err.contains('chmod 600')701702 703 os.write_file(os.join_path(config, 'remote'), 'kind=ftp\n') or { panic(err) }704 unreadable := run_home(dir, home, 'offer')705 assert unreadable.code == 1706 assert unreadable.err.contains('not readable')707 assert run_home(dir, home, 'remote', 'folder', target).code == 0708 assert run_home(dir, home, 'offer').code == 0709}710711712713fn test_a_renamed_clone_keeps_offering_to_its_repo() {714 syncroot := temp_repo('rename-clone')715 home := temp_repo('rename-clone-home')716 defer {717 os.rmdir_all(syncroot) or {}718 os.rmdir_all(home) or {}719 }720 remote := os.join_path(syncroot, 'remote')721 a := os.join_path(syncroot, 'A', 'proj')722 b_parent := os.join_path(syncroot, 'B')723 os.mkdir_all(a) or { panic(err) }724 os.mkdir_all(b_parent) or { panic(err) }725 init_repo(a)726 write(a, 'a.txt', 'from A\n')727 assert run_home(a, home, 'remote', 'folder', remote).code == 0728 assert run_home(a, home, 'save', 'A first').code == 0729 assert run_home(b_parent, home, 'steal', 'proj').code == 0730 renamed := os.join_path(b_parent, 'renamed')731 os.mv(os.join_path(b_parent, 'proj'), renamed) or { panic(err) }732 write(renamed, 'b.txt', 'from B\n')733 assert run_home(renamed, home, 'save', 'B first').code == 0734 assert !os.exists(os.join_path(remote, 'renamed'))735 assert run_home(a, home, 'steal').code == 0736 assert os.read_file(os.join_path(a, 'b.txt'))! == 'from B\n'737}738739740741fn test_clone_of_a_damaged_remote_leaves_nothing() {742 syncroot := temp_repo('damaged')743 home := temp_repo('damaged-home')744 defer {745 os.rmdir_all(syncroot) or {}746 os.rmdir_all(home) or {}747 }748 remote := os.join_path(syncroot, 'remote')749 a := os.join_path(syncroot, 'A', 'proj')750 b_parent := os.join_path(syncroot, 'B')751 os.mkdir_all(a) or { panic(err) }752 os.mkdir_all(b_parent) or { panic(err) }753 init_repo(a)754 write(a, 'a.txt', 'from A\n')755 assert run_home(a, home, 'remote', 'folder', remote).code == 0756 assert run_home(a, home, 'save', 'A first').code == 0757 objects := os.join_path(remote, 'proj', 'objects')758 for name in os.ls(objects) or { panic(err) } {759 os.chmod(os.join_path(objects, name), 0o644) or {}760 os.write_file(os.join_path(objects, name), 'junk') or { panic(err) }761 }762 clone := run_home(b_parent, home, 'steal', 'proj')763 assert clone.code == 1764 assert clone.err.contains('is damaged')765 assert !os.exists(os.join_path(b_parent, 'proj'))766}767768fn test_offer_steal_folder_remote() {769 syncroot := temp_repo('sync')770 home := temp_repo('sync-home')771 defer {772 os.rmdir_all(syncroot) or {}773 os.rmdir_all(home) or {}774 }775 remote := os.join_path(syncroot, 'remote')776 a := os.join_path(syncroot, 'A', 'proj')777 b_parent := os.join_path(syncroot, 'B')778 os.mkdir_all(a) or { panic(err) }779 os.mkdir_all(b_parent) or { panic(err) }780 init_repo(a)781 write(a, 'a.txt', 'from A\n')782 assert run_home(a, home, 'remote', 'folder', remote).code == 0783 assert run_home(a, home, 'save', 'A first').code == 0784 clone := run_home(b_parent, home, 'steal', 'proj')785 assert clone.code == 0786 b := os.join_path(b_parent, 'proj')787 assert os.read_file(os.join_path(b, 'a.txt'))! == 'from A\n'788 write(b, 'b.txt', 'from B\n')789 assert run_home(b, home, 'save', 'B first').code == 0790 write(a, 'a2.txt', 'second\n')791 assert run_home(a, home, 'save', 'A second').code == 0792 assert run_home(a, home, 'steal').code == 0793 assert os.exists(os.join_path(a, 'b.txt'))794 assert run_home(b_parent, home, 'steal', '--list').out.contains('proj')795 write(a, 'clash.txt', 'base\n')796 assert run_home(a, home, 'save', 'clash base').code == 0797 assert run_home(b, home, 'steal').code == 0798 write(a, 'clash.txt', 'A version\n')799 assert run_home(a, home, 'save', 'A clash').code == 0800 write(b, 'clash.txt', 'B version\n')801 assert run_home(b, home, 'save', 'B clash').code == 0802 refused := run_home(b, home, 'steal')803 assert refused.code == 1804 assert refused.err.contains('--mine') || refused.err.contains('--theirs')805 b_head := run_home(b, home, 'status', '--json').out.all_after('"head_id":"')[..12]806 theirs := run_home(b, home, 'steal', '--theirs', '--json')807 assert theirs.code == 0808 809 810 assert theirs.out.contains('"previous_head":"${b_head}"')811 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'812 assert run_home(b, home, 'load', b_head, '--yes').code == 0813 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B version\n'814 assert run_home(b, home, 'load', theirs.out.all_after('"head":"')[..12], '--yes').code == 0815 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'816 write(a, 'clash.txt', 'A second\n')817 assert run_home(a, home, 'save', 'A second clash').code == 0818 write(b, 'clash.txt', 'B second\n')819 assert run_home(b, home, 'save', 'B second clash').code == 0820 assert run_home(b, home, 'steal', '--mine').code == 0821 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B second\n'822 assert run_home(b, home, 'log').out.contains('combine: kept mine for')823}824825826827fn test_a_clone_reports_flagged_saves() {828 syncroot := temp_repo('clone-flags')829 home := temp_repo('clone-flags-home')830 defer {831 os.rmdir_all(syncroot) or {}832 os.rmdir_all(home) or {}833 }834 a := os.join_path(syncroot, 'A', 'proj')835 os.mkdir_all(a) or { panic(err) }836 init_repo(a)837 assert run_home(a, home, 'remote', 'folder', os.join_path(syncroot, 'remote')).code == 0838 write(a, 'a.txt', 'one\n')839 id := run_home(a, home, 'save', 'risky', '--json').out.all_after('"id":"')[..12]840 assert run_home(a, home, 'flag', id, '--reason', 'broke the build').code == 0841 assert run_home(a, home, 'offer').code == 0842 for name in ['B', 'C'] {843 os.mkdir_all(os.join_path(syncroot, name)) or { panic(err) }844 }845 json_clone := run_home(os.join_path(syncroot, 'B'), home, 'steal', 'proj', '--json')846 assert json_clone.code == 0847 assert json_clone.out.contains('"flagged":["${id}"]')848 text_clone := run_home(os.join_path(syncroot, 'C'), home, 'steal', 'proj')849 assert text_clone.code == 0850 assert text_clone.out.contains('[${id}] came in flagged — broke the build')851}852853854855856fn test_an_unmounted_folder_remote_is_not_recreated() {857 syncroot := temp_repo('unmounted')858 home := temp_repo('unmounted-home')859 defer {860 os.rmdir_all(syncroot) or {}861 os.rmdir_all(home) or {}862 }863 mount := os.join_path(syncroot, 'nas')864 remote := os.join_path(mount, 'koh')865 a := os.join_path(syncroot, 'proj')866 os.mkdir_all(a) or { panic(err) }867 init_repo(a)868 assert run_home(a, home, 'remote', 'folder', remote).code == 0869 write(a, 'a.txt', 'one\n')870 assert run_home(a, home, 'save', 'first').code == 0871 os.rename(mount, os.join_path(syncroot, 'nas-unplugged')) or { panic(err) }872 os.mkdir(mount) or { panic(err) }873 offered := run_home(a, home, 'offer')874 assert offered.code == 1875 assert offered.err.contains('mounted')876 assert !os.exists(remote)877 write(a, 'a.txt', 'two\n')878 saved := run_home(a, home, 'save', 'second')879 assert saved.code == 0880 assert saved.out.contains('mounted')881 assert !os.exists(remote)882}883884fn test_man_and_koh2_handoff() {885 dir := temp_repo('man')886 home := temp_repo('man-home')887 defer {888 os.rmdir_all(dir) or {}889 os.rmdir_all(home) or {}890 }891 whole := run_home(dir, home, 'man')892 assert whole.code == 0893 assert whole.out.contains('NAME')894 assert whole.out.contains('COMMANDS')895 assert !whole.out.contains('\x1b') && !whole.out.contains('◆')896 strip := run_home(dir, home, 'man', 'strip')897 assert strip.code == 0898 assert strip.out.contains('strip')899 assert !strip.out.contains('koh save ')900 steal := run_home(dir, home, 'man', 'steal').out901 assert steal.starts_with(' koh steal') && steal.contains('koh steal my-project')902 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."903 assert run_home(dir, home, 'man', 'nonsense').code == 2904 legacy := temp_repo('legacy')905 defer { os.rmdir_all(legacy) or {} }906 os.mkdir_all(os.join_path(legacy, '.koh')) or { panic(err) }907 fake := os.join_path(home, 'bin')908 os.mkdir_all(fake) or { panic(err) }909 fake_koh := os.join_path(fake, 'koh2')910 os.write_file(fake_koh, '#!/bin/sh\nprintf "handoff:%s\\n" "$*"\nexit 7\n') or { panic(err) }911 os.chmod(fake_koh, 0o755) or { panic(err) }912 handoff := run_home_with_path(legacy, home, fake, 'status', '--json')913 assert handoff.code == 7914 assert handoff.out.contains('handoff:status --json')915 916 assert handoff.err.contains('Koh 2 repo')917 assert handoff.err.trim_space().split_into_lines().len == 1918 assert handoff.out.trim_space() == 'handoff:status --json'919}920921fn run_home_with_path(repo string, home string, bin_dir string, args ...string) Result {922 stamp := '${os.getpid()}-${time.now().unix_nano()}'923 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')924 out_path := '${base}.out'925 err_path := '${base}.err'926 code_path := '${base}.code'927 mut command := 'HOME=${q(home)} PATH=${q(bin_dir + ':' + os.getenv('PATH'))} ${q(binary)}'928 for arg in args {929 command += ' ${q(arg)}'930 }931 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" \$? >${q(code_path)}'932 result := os.execute(shell)933 if result.exit_code != 0 {934 panic('could not run test command: ${result.output}')935 }936 code := os.read_file(code_path) or { panic(err) }937 out := os.read_file(out_path) or { '' }938 err := os.read_file(err_path) or { '' }939 os.rm(out_path) or {}940 os.rm(err_path) or {}941 os.rm(code_path) or {}942 return Result{943 code: code.int()944 out: out945 err: err946 }947}948949950951fn koh2_binary() string {952 return os.find_abs_path_of_executable('koh') or { '' }953}954955956957fn run_koh2(dir string, command string) {958 os.execute('cd ${q(dir)} && ${command} >/dev/null 2>&1')959}960961fn save_ids(dir string) []string {962 mut ids := []string{}963 for name in os.ls(os.join_path(dir, '.koh', 'snapshots')) or { []string{} } {964 if name.ends_with('.json') {965 ids << name.trim_right('.json')966 }967 }968 ids.sort()969 return ids970}971972fn test_migrate_converts_a_koh2_repo_and_keeps_every_save() {973 koh2 := koh2_binary()974 if koh2 == '' {975 println('skipping the migration tests: koh (Koh 2) is not on PATH')976 return977 }978 dir := temp_repo('migrate')979 defer { os.rmdir_all(dir) or {} }980 home := temp_repo('migrate-home')981 defer { os.rmdir_all(home) or {} }982 run_koh2(dir, '${q(koh2)} init')983 984 write(dir, 'code.txt', 'x'.repeat(700))985 write(dir, 'tiny.txt', 'tiny\n')986 run_koh2(dir, '${q(koh2)} save "first save"')987 run_koh2(dir, '${q(koh2)} checkpoint "before something"')988 write(dir, 'code.txt', 'y'.repeat(700))989 run_koh2(dir, '${q(koh2)} save "second save"')990 original := save_ids(dir)991 assert original.len == 3992993 report := run_home(dir, home, 'migrate')994 assert report.code == 0995 assert report.out.contains('migrated 3 saves')996 997 assert os.is_file(os.join_path(dir, '.koh'))998 assert os.is_dir(os.join_path(dir, '.koh-v2'))999 assert os.is_file(os.join_path(dir, '.kohignore'))1000 assert !os.exists(os.join_path(dir, '.koh-migrate'))10011002 1003 logged := run_home(dir, home, 'log', '--json')1004 assert logged.code == 01005 for id in original {1006 assert logged.out.contains(id)1007 }1008 assert run_home(dir, home, 'doctor').code == 01009 assert run_home(dir, home, 'status').code == 010101011 1012 assert run_home(dir, home, 'migrate').code == 11013}10141015fn test_migrate_refuses_a_tree_that_does_not_match_the_last_save() {1016 koh2 := koh2_binary()1017 if koh2 == '' {1018 return1019 }1020 dir := temp_repo('migrate-dirty')1021 defer { os.rmdir_all(dir) or {} }1022 home := temp_repo('migrate-dirty-home')1023 defer { os.rmdir_all(home) or {} }1024 run_koh2(dir, '${q(koh2)} init')1025 write(dir, 'code.txt', 'x'.repeat(700))1026 run_koh2(dir, '${q(koh2)} save "only save"')1027 write(dir, 'code.txt', 'unsaved change\n')10281029 refused := run_home(dir, home, 'migrate')1030 assert refused.code == 11031 assert refused.err.contains('save with koh2 first')1032 1033 assert os.is_dir(os.join_path(dir, '.koh'))1034 assert !os.exists(os.join_path(dir, '.koh-v2'))1035 assert !os.exists(os.join_path(dir, '.koh-migrate'))1036}10371038103910401041fn test_migrate_passes_over_the_files_koh2_never_saved() {1042 koh2 := koh2_binary()1043 if koh2 == '' {1044 return1045 }1046 dir := temp_repo('migrate-koh2-ignores')1047 defer { os.rmdir_all(dir) or {} }1048 home := temp_repo('migrate-koh2-ignores-home')1049 defer { os.rmdir_all(home) or {} }1050 run_koh2(dir, '${q(koh2)} init')1051 write(dir, 'code.txt', 'x'.repeat(700))1052 run_koh2(dir, '${q(koh2)} save "only save"')1053 write(dir, '.kohconfig', 'stable_days = 14\n')1054 write(dir, '.kohignore.public', 'private/\n')1055 write(dir, 'lib/app.bundle', 'bundle\n')1056 write(dir, 'ios/Settings.bundle/Root.plist', 'settings\n')1057 1058 1059 1060 write(dir, 'native/t.dSYM/Info.plist', 'plist\n')1061 run_koh2(dir, '${q(koh2)} save --agent test --session test "add the dsym"')1062 1063 ignore_path := os.join_path(dir, '.koh', 'ignore')1064 rules := os.read_file(ignore_path)!.split_into_lines().filter(it != '*.bundle')1065 os.write_file(ignore_path, rules.join('\n') + '\nnative/*.dSYM\n')!1066 1067 write(dir, 'native/t.dSYM/Info.plist', 'changed\n')1068 changed := run_home(dir, home, 'migrate')1069 assert changed.code == 11070 assert changed.err.contains('native/t.dSYM/Info.plist does not match')1071 write(dir, 'native/t.dSYM/Info.plist', 'plist\n')1072 assert os.execute('cd ${q(dir)} && ${q(koh2)} status --json').output.contains('"dirty":false')10731074 migrated := run_home(dir, home, 'migrate')1075 assert migrated.code == 0, migrated.err1076 1077 assert os.read_file(os.join_path(dir, '.kohconfig'))! == 'stable_days = 14\n'1078}