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}157158159160161fn 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}173174175176177fn test_a_restore_refuses_a_folder_of_ignored_files_in_the_way() {178 dir := temp_repo('dir-in-way')179 defer { os.rmdir_all(dir) or {} }180 init_repo(dir)181 write(dir, 'out', 'a file\n')182 first := run(dir, 'save', 'out is a file', '--json').out.all_after('"id":"')[..12]183 os.rm(os.join_path(dir, 'out')) or { panic(err) }184 write(dir, 'out/build.log', 'ignored\n')185 assert run(dir, 'save', 'out is gone').code == 0186 r := run(dir, 'load', first, '--yes')187 assert r.code == 1188 assert r.err.contains('out')189 assert os.ls(os.join_path(dir, 'out'))! == ['build.log']190}191192193194fn test_doctor_sweeps_only_old_peek_folders() {195 dir := temp_repo('peek-sweep')196 197 198 fresh := os.join_path(os.temp_dir(), 'koh-peek-fresh${os.getpid()}')199 old := os.join_path(os.temp_dir(), 'koh-peek-old${os.getpid()}')200 defer {201 os.rmdir_all(dir) or {}202 os.rmdir_all(fresh) or {}203 os.rmdir_all(old) or {}204 }205 init_repo(dir)206 write(dir, 'a.txt', 'one\n')207 assert run(dir, 'save', 'one').code == 0208 os.mkdir_all(fresh) or { panic(err) }209 os.mkdir_all(old) or { panic(err) }210 two_days_ago := time.now().unix() - 2 * 86400211 os.utime(old, two_days_ago, two_days_ago) or { panic(err) }212 assert run(dir, 'doctor', '--json').code == 0213 assert os.exists(fresh)214 assert !os.exists(old)215}216217218219220fn test_a_restore_never_overwrites_an_ignored_file() {221 dir := temp_repo('ignored-restore')222 defer { os.rmdir_all(dir) or {} }223 init_repo(dir)224 write(dir, '.env', 'KEY=saved long ago\n')225 first := run(dir, 'save', 'with env', '--json').out.all_after('"id":"')[..12]226 write(dir, '.kohignore', '.env\n')227 assert run(dir, 'save', 'ignore env').code == 0228 write(dir, '.env', 'KEY=never saved\n')229 assert run(dir, 'load', first, '--yes').code == 0230 assert os.read_file(os.join_path(dir, '.env'))! == 'KEY=never saved\n'231232 233 other := temp_repo('ignored-restore-missing')234 defer { os.rmdir_all(other) or {} }235 init_repo(other)236 write(other, '.env', 'KEY=saved long ago\n')237 again := run(other, 'save', 'with env', '--json').out.all_after('"id":"')[..12]238 write(other, '.kohignore', '.env\n')239 assert run(other, 'save', 'ignore env').code == 0240 os.rm(os.join_path(other, '.env')) or { panic(err) }241 assert run(other, 'load', again, '--yes').code == 0242 assert os.read_file(os.join_path(other, '.env'))! == 'KEY=saved long ago\n'243}244245246247fn test_a_going_back_preview_does_not_run_diff() {248 dir := temp_repo('preview-nodiff')249 home := temp_repo('preview-nodiff-home')250 defer {251 os.rmdir_all(dir) or {}252 os.rmdir_all(home) or {}253 }254 init_repo(dir)255 write(dir, 'a.txt', 'one\n')256 assert run_home(dir, home, 'save', 'one').code == 0257 write(dir, 'a.txt', 'two\n')258 assert run_home(dir, home, 'save', 'two').code == 0259 fake := os.join_path(home, 'bin')260 os.mkdir_all(fake) or { panic(err) }261 marker := os.join_path(home, 'diff-ran')262 os.write_file(os.join_path(fake, 'diff'), '#!/bin/sh\ntouch ${q(marker)}\nexit 2\n') or {263 panic(err)264 }265 os.chmod(os.join_path(fake, 'diff'), 0o755) or { panic(err) }266 assert run_home_with_path(dir, home, fake, 'undo', '--yes').code == 0267 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'268 assert !os.exists(marker)269}270271272273fn test_a_restore_leaves_unchanged_files_alone() {274 dir := temp_repo('restore-skip')275 defer { os.rmdir_all(dir) or {} }276 init_repo(dir)277 write(dir, 'a.txt', 'one\n')278 write(dir, 'same.txt', 'never changes\n')279 assert run(dir, 'save', 'one').code == 0280 write(dir, 'a.txt', 'two\n')281 assert run(dir, 'save', 'two').code == 0282 long_ago := i64(1700000000)283 os.utime(os.join_path(dir, 'same.txt'), long_ago, long_ago) or { panic(err) }284 assert run(dir, 'undo', '--yes').code == 0285 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'286 assert os.file_last_mod_unix(os.join_path(dir, 'same.txt')) == long_ago287}288289290291fn test_status_sees_an_executable_bit_change() {292 dir := temp_repo('chmod')293 defer { os.rmdir_all(dir) or {} }294 init_repo(dir)295 write(dir, 'run.sh', 'echo hi\n')296 assert run(dir, 'save', 'script').code == 0297 os.chmod(os.join_path(dir, 'run.sh'), 0o755) or { panic(err) }298 status := run(dir, 'status', '--json')299 assert status.out.contains('"dirty":true')300 assert status.out.contains('"modified":["run.sh"]')301}302303fn test_log_and_diff() {304 dir := temp_repo('history')305 defer { os.rmdir_all(dir) or {} }306 init_repo(dir)307 write(dir, 'a.txt', 'one\n')308 first := run(dir, 'save', 'one', '--session', 'alpha', '--json')309 id1 := first.out.all_after('"id":"')[..12]310 write(dir, 'a.txt', 'two\n')311 second := run(dir, 'save', 'two', '--session', 'beta', '--json')312 id2 := second.out.all_after('"id":"')[..12]313 cp := run(dir, 'checkpoint', 'checkpoint', '--json')314 assert cp.code == 0315 log := run(dir, 'log')316 assert log.code == 0317 assert log.out.contains('▸')318 assert log.out.contains('◐')319 lj := run(dir, 'log', '--json')320 assert lj.code == 0321 assert lj.out.contains('"head_id"')322 assert lj.out.contains('"saves"')323 assert run(dir, 'log', '--since', 'today').out.contains('two')324 session := run(dir, 'log', '--session', 'alpha')325 assert session.out.contains('one')326 assert !session.out.contains('two')327 write(dir, 'a.txt', 'three\n')328 d := run(dir, 'diff')329 assert d.code == 0330 assert d.out.contains('--- a/')331 assert d.out.contains('+++ b/')332 assert run(dir, 'diff', id1).code == 0333 assert run(dir, 'diff', id1, id2).code == 0334 dj := run(dir, 'diff', '--json')335 assert dj.code == 0336 assert dj.out.contains('"from"')337 assert dj.out.contains('"to"')338 assert dj.out.contains('"files"')339 assert dj.out.contains('"change":"modified"')340 assert run(dir, 'diff', 'not-a-real-id').code == 1341 assert run(dir, 'diff', id1, id2, 'third').code == 1342 write(dir, 'blob.bin', '\x00\x01\x02')343 diff_binary := run(dir, 'diff', '--json')344 assert diff_binary.out.contains('"binary":true')345 assert diff_binary.out.contains('Binary files differ')346}347348fn test_undo_revert_and_load() {349 dir := temp_repo('restore')350 defer { os.rmdir_all(dir) or {} }351 init_repo(dir)352 write(dir, 'a.txt', 'one\n')353 one := run(dir, 'save', 'one', '--json').out354 id1 := one.all_after('"id":"')[..12]355 write(dir, 'a.txt', 'two\n')356 write(dir, 'gone.txt', 'gone\n')357 second := run(dir, 'save', 'two', '--json').out358 id2 := second.all_after('"id":"')[..12]359 write(dir, 'a.txt', 'three\n')360 third := run(dir, 'save', 'three', '--json').out361 id3 := third.all_after('"id":"')[..12]362 363 364 write(dir, 'never.txt', 'keep me\n')365 log_before := run(dir, 'log').out366 for args in [['undo', '--yes'], ['load', id1, '--yes'], ['pluck', id2, '--yes']] {367 blocked := run(dir, ...args)368 assert blocked.code == 1369 assert blocked.err.contains('unsaved changes')370 }371 assert run(dir, 'log').out == log_before372 os.rm(os.join_path(dir, 'never.txt')) or { panic(err) }373 write(dir, 'a.txt', 'edited, never saved\n')374 assert run(dir, 'undo', '--yes').err.contains('unsaved changes')375 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'edited, never saved\n'376 write(dir, 'a.txt', 'three\n')377 os.rm(os.join_path(dir, 'gone.txt')) or { panic(err) }378 assert run(dir, 'load', id1, '--yes').err.contains('unsaved changes')379 write(dir, 'gone.txt', 'gone\n')380 undo := run(dir, 'undo', '--yes', '--json')381 assert undo.code == 0382 if !undo.out.contains(id2) {383 panic('undo JSON did not name ${id2}: ${undo.out}')384 }385 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'386 assert os.exists(os.join_path(dir, 'gone.txt'))387 assert run(dir, 'load', id3, '--yes').code == 0388 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'389 assert run(dir, 'undo', '2', '--yes').code == 0390 assert run(dir, 'load', id2, '--yes').code == 0391 assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')392 write(dir, 'a.txt', 'dirty\n')393 before := run(dir, 'log')394 refuse := run(dir, 'revert')395 assert refuse.code == 1396 assert refuse.err.contains('needs --yes when not interactive')397 assert run(dir, 'log').out == before.out398 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'399 write(dir, 'untracked.txt', 'remove\n')400 rv := run(dir, 'revert', '--yes')401 assert rv.code == 0402 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'403 assert !os.exists(os.join_path(dir, 'untracked.txt'))404 assert run(dir, 'revert', '--yes').out.contains('working tree already matches')405 assert run(dir, 'load', '--checkpoint', '--yes').code == 0406 assert run(dir, 'load', id2, '--json').code == 1407 assert run(dir, 'load', id2, '--json').err.contains('error')408 assert run(dir, 'load').code == 2409 assert run(dir, 'load', 'unknown-id', '--yes').code == 1410}411412fn test_flag_and_flagged_log() {413 dir := temp_repo('flag')414 defer { os.rmdir_all(dir) or {} }415 init_repo(dir)416 write(dir, 'flag.txt', 'one\n')417 saved := run(dir, 'save', 'flag me', '--json')418 id := saved.out.all_after('"id":"')[..12]419 flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')420 assert flagged.code == 0421 assert flagged.out.contains('"flagged":true')422 assert flagged.out.contains('"reason":"unsafe"')423 list := run(dir, 'flag', '--json')424 assert list.code == 0425 assert list.out.contains(id)426 assert list.out.contains('unsafe')427 assert run(dir, 'log').out.contains('⚑')428 assert run(dir, 'flag', '--clear', id).code == 0429 assert run(dir, 'flag').out.contains('nothing is flagged')430 assert run(dir, 'flag', 'unknown-id').code == 1431 assert run(dir, 'flag').code == 0432}433434fn test_pluck_forward_reverse_dry_run_and_conflict() {435 dir := temp_repo('pluck')436 defer { os.rmdir_all(dir) or {} }437 init_repo(dir)438 write(dir, 'a.txt', 'one\n')439 first := run(dir, 'save', 'one', '--json').out440 id1 := first.all_after('"id":"')[..12]441 write(dir, 'a.txt', 'two\n')442 second := run(dir, 'save', 'two', '--json').out443 id2 := second.all_after('"id":"')[..12]444 assert run(dir, 'load', id1, '--yes').code == 0445 dry := run(dir, 'pluck', id2, '--dry-run')446 assert dry.code == 0447 assert dry.out.contains('a.txt')448 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'449 assert run(dir, 'pluck', id2, '--yes').code == 0450 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'451 assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0452 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'453 write(dir, 'a.txt', 'different\n')454 assert run(dir, 'pluck', id2, '--dry-run').code == 1455 assert run(dir, 'save', 'different').code == 0456 conflict := run(dir, 'pluck', id2, '--yes')457 assert conflict.code == 1458 assert conflict.err.contains('a.txt')459 assert run(dir, 'pluck').code == 2460 assert run(dir, 'pluck', 'unknown-id').code == 1461}462463fn test_peek_and_doctor() {464 dir := temp_repo('peek-doctor')465 defer { os.rmdir_all(dir) or {} }466 init_repo(dir)467 write(dir, 'peek.txt', 'peek me\n')468 assert run(dir, 'save', 'peekable').code == 0469 id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]470 before_log := run(dir, 'log', '--json').out471 peek := run(dir, 'peek', id, '--json')472 assert peek.code == 0473 peek_path := peek.out.all_after('"path":"').before('"')474 assert peek.out.contains('"id":"${id}"')475 assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'476 assert run(dir, 'status').out.contains('no changes')477 assert run(dir, 'log', '--json').out == before_log478 assert run(dir, 'peek').code == 2479 assert run(dir, 'peek', 'unknown-id').code == 1480 backup := os.join_path(dir, 'doctor.backup')481 healthy := run(dir, 'doctor')482 assert healthy.code == 0483 assert healthy.out.contains('is sound')484 report := run(dir, 'doctor', '--json')485 assert report.code == 0486 for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',487 'backup'] {488 read_json_has(report.out, key)489 }490 backed := run(dir, 'doctor', '--backup', backup)491 assert backed.code == 0492 assert os.exists(backup)493 assert backed.out.contains(backup)494 assert run(dir, 'doctor', '--backup', backup).code == 1495 assert run(dir, 'peek', id, '--json').code == 0496 497 kept := run(dir, 'doctor', '--json')498 assert kept.code == 0499 assert os.exists(peek_path)500 os.rmdir_all(peek_path) or {}501}502503fn test_strip_refuses_without_terminal() {504 dir := temp_repo('strip')505 defer { os.rmdir_all(dir) or {} }506 init_repo(dir)507 write(dir, 'secret.txt', 'keep\n')508 assert run(dir, 'save', 'secret').code == 0509 before_log := run(dir, 'log', '--json').out510 refused := run(dir, 'strip', 'secret.txt')511 assert refused.code == 1512 assert refused.err.contains('needs a terminal')513 assert run(dir, 'log', '--json').out == before_log514 assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'515 assert run(dir, 'log', '--json').out.contains('secret')516}517518fn test_saves_and_checkpoints_aliases() {519 dir := temp_repo('aliases')520 defer { os.rmdir_all(dir) or {} }521 init_repo(dir)522 write(dir, 'alias.txt', 'one\n')523 assert run(dir, 'save', 'plain save').code == 0524 assert run(dir, 'checkpoint', 'checkpoint save').code == 0525 saves := run(dir, 'saves')526 assert saves.code == 0527 assert saves.out.contains('plain save')528 assert saves.out.contains('checkpoint save')529 checkpoints := run(dir, 'checkpoints')530 assert checkpoints.code == 0531 assert checkpoints.out.contains('checkpoint save')532 assert !checkpoints.out.contains('plain save')533 json_result := run(dir, 'checkpoints', '--json')534 assert json_result.code == 0535 assert json_result.out.contains('"head_id"')536 assert json_result.out.contains('"saves"')537 assert json_result.out.contains('"is_checkpoint":true')538 assert !json_result.out.contains('"is_checkpoint":false')539}540541fn test_remote_configuration_is_isolated_and_atomic() {542 dir := temp_repo('remote')543 home := temp_repo('remote-home')544 defer {545 os.rmdir_all(dir) or {}546 os.rmdir_all(home) or {}547 }548 empty := run_home(dir, home, 'remote', '--json')549 assert empty.code == 0550 assert empty.out.contains('"configured":false')551 target := os.join_path(home, 'remote-store')552 set := run_home(dir, home, 'remote', 'folder', target)553 assert set.code == 0554 assert os.is_dir(target)555 state := run_home(dir, home, 'remote', '--json')556 assert state.code == 0557 assert state.out.contains('"kind":"folder"')558 assert state.out.contains(target)559 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').code == 1560 assert run_home(dir, home, 'remote', '--json').out.contains(target)561 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').err.contains('KOH_ACCESS_KEY')562 assert run_home(dir, home, 'remote', '--forget').code == 0563 assert run_home(dir, home, 'remote', '--json').out.contains('"configured":false')564}565566567568fn test_a_renamed_clone_keeps_offering_to_its_repo() {569 syncroot := temp_repo('rename-clone')570 home := temp_repo('rename-clone-home')571 defer {572 os.rmdir_all(syncroot) or {}573 os.rmdir_all(home) or {}574 }575 remote := os.join_path(syncroot, 'remote')576 a := os.join_path(syncroot, 'A', 'proj')577 b_parent := os.join_path(syncroot, 'B')578 os.mkdir_all(a) or { panic(err) }579 os.mkdir_all(b_parent) or { panic(err) }580 init_repo(a)581 write(a, 'a.txt', 'from A\n')582 assert run_home(a, home, 'remote', 'folder', remote).code == 0583 assert run_home(a, home, 'save', 'A first').code == 0584 assert run_home(b_parent, home, 'steal', 'proj').code == 0585 renamed := os.join_path(b_parent, 'renamed')586 os.mv(os.join_path(b_parent, 'proj'), renamed) or { panic(err) }587 write(renamed, 'b.txt', 'from B\n')588 assert run_home(renamed, home, 'save', 'B first').code == 0589 assert !os.exists(os.join_path(remote, 'renamed'))590 assert run_home(a, home, 'steal').code == 0591 assert os.read_file(os.join_path(a, 'b.txt'))! == 'from B\n'592}593594fn test_offer_steal_folder_remote() {595 syncroot := temp_repo('sync')596 home := temp_repo('sync-home')597 defer {598 os.rmdir_all(syncroot) or {}599 os.rmdir_all(home) or {}600 }601 remote := os.join_path(syncroot, 'remote')602 a := os.join_path(syncroot, 'A', 'proj')603 b_parent := os.join_path(syncroot, 'B')604 os.mkdir_all(a) or { panic(err) }605 os.mkdir_all(b_parent) or { panic(err) }606 init_repo(a)607 write(a, 'a.txt', 'from A\n')608 assert run_home(a, home, 'remote', 'folder', remote).code == 0609 assert run_home(a, home, 'save', 'A first').code == 0610 clone := run_home(b_parent, home, 'steal', 'proj')611 assert clone.code == 0612 b := os.join_path(b_parent, 'proj')613 assert os.read_file(os.join_path(b, 'a.txt'))! == 'from A\n'614 write(b, 'b.txt', 'from B\n')615 assert run_home(b, home, 'save', 'B first').code == 0616 write(a, 'a2.txt', 'second\n')617 assert run_home(a, home, 'save', 'A second').code == 0618 assert run_home(a, home, 'steal').code == 0619 assert os.exists(os.join_path(a, 'b.txt'))620 assert run_home(b_parent, home, 'steal', '--list').out.contains('proj')621 write(a, 'clash.txt', 'base\n')622 assert run_home(a, home, 'save', 'clash base').code == 0623 assert run_home(b, home, 'steal').code == 0624 write(a, 'clash.txt', 'A version\n')625 assert run_home(a, home, 'save', 'A clash').code == 0626 write(b, 'clash.txt', 'B version\n')627 assert run_home(b, home, 'save', 'B clash').code == 0628 refused := run_home(b, home, 'steal')629 assert refused.code == 1630 assert refused.err.contains('--mine') || refused.err.contains('--theirs')631 assert run_home(b, home, 'steal', '--theirs').code == 0632 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'633 write(a, 'clash.txt', 'A second\n')634 assert run_home(a, home, 'save', 'A second clash').code == 0635 write(b, 'clash.txt', 'B second\n')636 assert run_home(b, home, 'save', 'B second clash').code == 0637 assert run_home(b, home, 'steal', '--mine').code == 0638 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B second\n'639 assert run_home(b, home, 'log').out.contains('combine: kept mine for')640}641642fn test_man_and_koh2_handoff() {643 dir := temp_repo('man')644 home := temp_repo('man-home')645 defer {646 os.rmdir_all(dir) or {}647 os.rmdir_all(home) or {}648 }649 whole := run_home(dir, home, 'man')650 assert whole.code == 0651 assert whole.out.contains('NAME')652 assert whole.out.contains('COMMANDS')653 strip := run_home(dir, home, 'man', 'strip')654 assert strip.code == 0655 assert strip.out.contains('strip')656 assert !strip.out.contains('koh save ')657 assert run_home(dir, home, 'man', 'nonsense').code == 2658 legacy := temp_repo('legacy')659 defer { os.rmdir_all(legacy) or {} }660 os.mkdir_all(os.join_path(legacy, '.koh')) or { panic(err) }661 fake := os.join_path(home, 'bin')662 os.mkdir_all(fake) or { panic(err) }663 fake_koh := os.join_path(fake, 'koh2')664 os.write_file(fake_koh, '#!/bin/sh\nprintf "handoff:%s\\n" "$*"\nexit 7\n') or { panic(err) }665 os.chmod(fake_koh, 0o755) or { panic(err) }666 handoff := run_home_with_path(legacy, home, fake, 'status', '--json')667 assert handoff.code == 7668 assert handoff.out.contains('handoff:status --json')669 670 assert handoff.err.contains('Koh 2 repo')671 assert handoff.err.trim_space().split_into_lines().len == 1672 assert handoff.out.trim_space() == 'handoff:status --json'673}674675fn run_home_with_path(repo string, home string, bin_dir string, args ...string) Result {676 stamp := '${os.getpid()}-${time.now().unix_nano()}'677 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')678 out_path := '${base}.out'679 err_path := '${base}.err'680 code_path := '${base}.code'681 mut command := 'HOME=${q(home)} PATH=${q(bin_dir + ':' + os.getenv('PATH'))} ${q(binary)}'682 for arg in args {683 command += ' ${q(arg)}'684 }685 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" $? >${q(code_path)}'686 result := os.execute(shell)687 if result.exit_code != 0 {688 panic('could not run test command: ${result.output}')689 }690 code := os.read_file(code_path) or { panic(err) }691 out := os.read_file(out_path) or { '' }692 err := os.read_file(err_path) or { '' }693 os.rm(out_path) or {}694 os.rm(err_path) or {}695 os.rm(code_path) or {}696 return Result{697 code: code.int()698 out: out699 err: err700 }701}702703704705fn koh2_binary() string {706 return os.find_abs_path_of_executable('koh') or { '' }707}708709710711fn run_koh2(dir string, command string) {712 os.execute('cd ${q(dir)} && ${command} >/dev/null 2>&1')713}714715fn save_ids(dir string) []string {716 mut ids := []string{}717 for name in os.ls(os.join_path(dir, '.koh', 'snapshots')) or { []string{} } {718 if name.ends_with('.json') {719 ids << name.trim_right('.json')720 }721 }722 ids.sort()723 return ids724}725726fn test_migrate_converts_a_koh2_repo_and_keeps_every_save() {727 koh2 := koh2_binary()728 if koh2 == '' {729 println('skipping the migration tests: koh (Koh 2) is not on PATH')730 return731 }732 dir := temp_repo('migrate')733 defer { os.rmdir_all(dir) or {} }734 home := temp_repo('migrate-home')735 defer { os.rmdir_all(home) or {} }736 run_koh2(dir, '${q(koh2)} init')737 738 write(dir, 'code.txt', 'x'.repeat(700))739 write(dir, 'tiny.txt', 'tiny\n')740 run_koh2(dir, '${q(koh2)} save "first save"')741 run_koh2(dir, '${q(koh2)} checkpoint "before something"')742 write(dir, 'code.txt', 'y'.repeat(700))743 run_koh2(dir, '${q(koh2)} save "second save"')744 original := save_ids(dir)745 assert original.len == 3746747 report := run_home(dir, home, 'migrate')748 assert report.code == 0749 assert report.out.contains('migrated 3 saves')750 751 assert os.is_file(os.join_path(dir, '.koh'))752 assert os.is_dir(os.join_path(dir, '.koh-v2'))753 assert os.is_file(os.join_path(dir, '.kohignore'))754 assert !os.exists(os.join_path(dir, '.koh-migrate'))755756 757 logged := run_home(dir, home, 'log', '--json')758 assert logged.code == 0759 for id in original {760 assert logged.out.contains(id)761 }762 assert run_home(dir, home, 'doctor').code == 0763 assert run_home(dir, home, 'status').code == 0764765 766 assert run_home(dir, home, 'migrate').code == 1767}768769fn test_migrate_refuses_a_tree_that_does_not_match_the_last_save() {770 koh2 := koh2_binary()771 if koh2 == '' {772 return773 }774 dir := temp_repo('migrate-dirty')775 defer { os.rmdir_all(dir) or {} }776 home := temp_repo('migrate-dirty-home')777 defer { os.rmdir_all(home) or {} }778 run_koh2(dir, '${q(koh2)} init')779 write(dir, 'code.txt', 'x'.repeat(700))780 run_koh2(dir, '${q(koh2)} save "only save"')781 write(dir, 'code.txt', 'unsaved change\n')782783 refused := run_home(dir, home, 'migrate')784 assert refused.code == 1785 assert refused.err.contains('save with koh2 first')786 787 assert os.is_dir(os.join_path(dir, '.koh'))788 assert !os.exists(os.join_path(dir, '.koh-v2'))789 assert !os.exists(os.join_path(dir, '.koh-migrate'))790}