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}157158fn test_log_and_diff() {159 dir := temp_repo('history')160 defer { os.rmdir_all(dir) or {} }161 init_repo(dir)162 write(dir, 'a.txt', 'one\n')163 first := run(dir, 'save', 'one', '--session', 'alpha', '--json')164 id1 := first.out.all_after('"id":"')[..12]165 write(dir, 'a.txt', 'two\n')166 second := run(dir, 'save', 'two', '--session', 'beta', '--json')167 id2 := second.out.all_after('"id":"')[..12]168 cp := run(dir, 'checkpoint', 'checkpoint', '--json')169 assert cp.code == 0170 log := run(dir, 'log')171 assert log.code == 0172 assert log.out.contains('▸')173 assert log.out.contains('◐')174 lj := run(dir, 'log', '--json')175 assert lj.code == 0176 assert lj.out.contains('"head_id"')177 assert lj.out.contains('"saves"')178 assert run(dir, 'log', '--since', 'today').out.contains('two')179 session := run(dir, 'log', '--session', 'alpha')180 assert session.out.contains('one')181 assert !session.out.contains('two')182 write(dir, 'a.txt', 'three\n')183 d := run(dir, 'diff')184 assert d.code == 0185 assert d.out.contains('--- a/')186 assert d.out.contains('+++ b/')187 assert run(dir, 'diff', id1).code == 0188 assert run(dir, 'diff', id1, id2).code == 0189 dj := run(dir, 'diff', '--json')190 assert dj.code == 0191 assert dj.out.contains('"from"')192 assert dj.out.contains('"to"')193 assert dj.out.contains('"files"')194 assert dj.out.contains('"change":"modified"')195 assert run(dir, 'diff', 'not-a-real-id').code == 1196 assert run(dir, 'diff', id1, id2, 'third').code == 1197 write(dir, 'blob.bin', '\x00\x01\x02')198 diff_binary := run(dir, 'diff', '--json')199 assert diff_binary.out.contains('"binary":true')200 assert diff_binary.out.contains('Binary files differ')201}202203fn test_undo_revert_and_load() {204 dir := temp_repo('restore')205 defer { os.rmdir_all(dir) or {} }206 init_repo(dir)207 write(dir, 'a.txt', 'one\n')208 one := run(dir, 'save', 'one', '--json').out209 id1 := one.all_after('"id":"')[..12]210 write(dir, 'a.txt', 'two\n')211 write(dir, 'gone.txt', 'gone\n')212 second := run(dir, 'save', 'two', '--json').out213 id2 := second.all_after('"id":"')[..12]214 write(dir, 'a.txt', 'three\n')215 third := run(dir, 'save', 'three', '--json').out216 id3 := third.all_after('"id":"')[..12]217 write(dir, 'never.txt', 'keep me\n')218 undo := run(dir, 'undo', '--yes', '--json')219 assert undo.code == 0220 if !undo.out.contains(id2) {221 panic('undo JSON did not name ${id2}: ${undo.out}')222 }223 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'224 assert os.exists(os.join_path(dir, 'gone.txt'))225 assert os.exists(os.join_path(dir, 'never.txt'))226 assert run(dir, 'load', id3, '--yes').code == 0227 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'228 assert run(dir, 'undo', '2', '--yes').code == 0229 assert run(dir, 'load', id2, '--yes').code == 0230 assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')231 write(dir, 'a.txt', 'dirty\n')232 before := run(dir, 'log')233 refuse := run(dir, 'revert')234 assert refuse.code == 1235 assert refuse.err.contains('needs --yes when not interactive')236 assert run(dir, 'log').out == before.out237 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'238 write(dir, 'untracked.txt', 'remove\n')239 rv := run(dir, 'revert', '--yes')240 assert rv.code == 0241 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'242 assert !os.exists(os.join_path(dir, 'untracked.txt'))243 assert run(dir, 'revert', '--yes').out.contains('working tree already matches')244 assert run(dir, 'load', '--checkpoint', '--yes').code == 0245 assert run(dir, 'load', id2, '--json').code == 1246 assert run(dir, 'load', id2, '--json').err.contains('error')247 assert run(dir, 'load').code == 2248 assert run(dir, 'load', 'unknown-id', '--yes').code == 1249}250251fn test_flag_and_flagged_log() {252 dir := temp_repo('flag')253 defer { os.rmdir_all(dir) or {} }254 init_repo(dir)255 write(dir, 'flag.txt', 'one\n')256 saved := run(dir, 'save', 'flag me', '--json')257 id := saved.out.all_after('"id":"')[..12]258 flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')259 assert flagged.code == 0260 assert flagged.out.contains('"flagged":true')261 assert flagged.out.contains('"reason":"unsafe"')262 list := run(dir, 'flag', '--json')263 assert list.code == 0264 assert list.out.contains(id)265 assert list.out.contains('unsafe')266 assert run(dir, 'log').out.contains('⚑')267 assert run(dir, 'flag', '--clear', id).code == 0268 assert run(dir, 'flag').out.contains('nothing is flagged')269 assert run(dir, 'flag', 'unknown-id').code == 1270 assert run(dir, 'flag').code == 0271}272273fn test_pluck_forward_reverse_dry_run_and_conflict() {274 dir := temp_repo('pluck')275 defer { os.rmdir_all(dir) or {} }276 init_repo(dir)277 write(dir, 'a.txt', 'one\n')278 first := run(dir, 'save', 'one', '--json').out279 id1 := first.all_after('"id":"')[..12]280 write(dir, 'a.txt', 'two\n')281 second := run(dir, 'save', 'two', '--json').out282 id2 := second.all_after('"id":"')[..12]283 assert run(dir, 'load', id1, '--yes').code == 0284 dry := run(dir, 'pluck', id2, '--dry-run')285 assert dry.code == 0286 assert dry.out.contains('a.txt')287 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'288 assert run(dir, 'pluck', id2, '--yes').code == 0289 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'290 assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0291 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'292 write(dir, 'a.txt', 'different\n')293 conflict := run(dir, 'pluck', id2, '--yes')294 assert conflict.code == 1295 assert conflict.err.contains('a.txt')296 assert run(dir, 'pluck').code == 2297 assert run(dir, 'pluck', 'unknown-id').code == 1298}299300fn test_peek_and_doctor() {301 dir := temp_repo('peek-doctor')302 defer { os.rmdir_all(dir) or {} }303 init_repo(dir)304 write(dir, 'peek.txt', 'peek me\n')305 assert run(dir, 'save', 'peekable').code == 0306 id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]307 before_log := run(dir, 'log', '--json').out308 peek := run(dir, 'peek', id, '--json')309 assert peek.code == 0310 peek_path := peek.out.all_after('"path":"').before('"')311 assert peek.out.contains('"id":"${id}"')312 assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'313 assert run(dir, 'status').out.contains('no changes')314 assert run(dir, 'log', '--json').out == before_log315 assert run(dir, 'peek').code == 2316 assert run(dir, 'peek', 'unknown-id').code == 1317 backup := os.join_path(dir, 'doctor.backup')318 healthy := run(dir, 'doctor')319 assert healthy.code == 0320 assert healthy.out.contains('is sound')321 report := run(dir, 'doctor', '--json')322 assert report.code == 0323 for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',324 'backup'] {325 read_json_has(report.out, key)326 }327 backed := run(dir, 'doctor', '--backup', backup)328 assert backed.code == 0329 assert os.exists(backup)330 assert backed.out.contains(backup)331 assert run(dir, 'doctor', '--backup', backup).code == 1332 assert run(dir, 'peek', id, '--json').code == 0333 removed := run(dir, 'doctor', '--json')334 assert removed.code == 0335 assert removed.out.contains('"peek_folders_removed":1')336 assert !os.exists(peek_path)337}338339fn test_strip_refuses_without_terminal() {340 dir := temp_repo('strip')341 defer { os.rmdir_all(dir) or {} }342 init_repo(dir)343 write(dir, 'secret.txt', 'keep\n')344 assert run(dir, 'save', 'secret').code == 0345 before_log := run(dir, 'log', '--json').out346 refused := run(dir, 'strip', 'secret.txt')347 assert refused.code == 1348 assert refused.err.contains('needs a terminal')349 assert run(dir, 'log', '--json').out == before_log350 assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'351 assert run(dir, 'log', '--json').out.contains('secret')352}353354fn test_saves_and_checkpoints_aliases() {355 dir := temp_repo('aliases')356 defer { os.rmdir_all(dir) or {} }357 init_repo(dir)358 write(dir, 'alias.txt', 'one\n')359 assert run(dir, 'save', 'plain save').code == 0360 assert run(dir, 'checkpoint', 'checkpoint save').code == 0361 saves := run(dir, 'saves')362 assert saves.code == 0363 assert saves.out.contains('plain save')364 assert saves.out.contains('checkpoint save')365 checkpoints := run(dir, 'checkpoints')366 assert checkpoints.code == 0367 assert checkpoints.out.contains('checkpoint save')368 assert !checkpoints.out.contains('plain save')369 json_result := run(dir, 'checkpoints', '--json')370 assert json_result.code == 0371 assert json_result.out.contains('"head_id"')372 assert json_result.out.contains('"saves"')373 assert json_result.out.contains('"is_checkpoint":true')374 assert !json_result.out.contains('"is_checkpoint":false')375}376377378fn test_remote_configuration_is_isolated_and_atomic() {379 dir := temp_repo('remote')380 home := temp_repo('remote-home')381 defer {382 os.rmdir_all(dir) or {}383 os.rmdir_all(home) or {}384 }385 empty := run_home(dir, home, 'remote', '--json')386 assert empty.code == 0387 assert empty.out.contains('"configured":false')388 target := os.join_path(home, 'remote-store')389 set := run_home(dir, home, 'remote', 'folder', target)390 assert set.code == 0391 assert os.is_dir(target)392 state := run_home(dir, home, 'remote', '--json')393 assert state.code == 0394 assert state.out.contains('"kind":"folder"')395 assert state.out.contains(target)396 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').code == 1397 assert run_home(dir, home, 'remote', '--json').out.contains(target)398 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').err.contains('KOH_ACCESS_KEY')399 assert run_home(dir, home, 'remote', '--forget').code == 0400 assert run_home(dir, home, 'remote', '--json').out.contains('"configured":false')401}402403fn test_offer_steal_folder_remote() {404 syncroot := temp_repo('sync')405 home := temp_repo('sync-home')406 defer {407 os.rmdir_all(syncroot) or {}408 os.rmdir_all(home) or {}409 }410 remote := os.join_path(syncroot, 'remote')411 a := os.join_path(syncroot, 'A', 'proj')412 b_parent := os.join_path(syncroot, 'B')413 os.mkdir_all(a) or { panic(err) }414 os.mkdir_all(b_parent) or { panic(err) }415 init_repo(a)416 write(a, 'a.txt', 'from A\n')417 assert run_home(a, home, 'remote', 'folder', remote).code == 0418 assert run_home(a, home, 'save', 'A first').code == 0419 clone := run_home(b_parent, home, 'steal', 'proj')420 assert clone.code == 0421 b := os.join_path(b_parent, 'proj')422 assert os.read_file(os.join_path(b, 'a.txt'))! == 'from A\n'423 write(b, 'b.txt', 'from B\n')424 assert run_home(b, home, 'save', 'B first').code == 0425 write(a, 'a2.txt', 'second\n')426 assert run_home(a, home, 'save', 'A second').code == 0427 assert run_home(a, home, 'steal').code == 0428 assert os.exists(os.join_path(a, 'b.txt'))429 assert run_home(b_parent, home, 'steal', '--list').out.contains('proj')430}431432fn test_man_and_koh2_handoff() {433 dir := temp_repo('man')434 home := temp_repo('man-home')435 defer {436 os.rmdir_all(dir) or {}437 os.rmdir_all(home) or {}438 }439 whole := run_home(dir, home, 'man')440 assert whole.code == 0441 assert whole.out.contains('NAME')442 assert whole.out.contains('COMMANDS')443 strip := run_home(dir, home, 'man', 'strip')444 assert strip.code == 0445 assert strip.out.contains('strip')446 assert !strip.out.contains('koh save ')447 assert run_home(dir, home, 'man', 'nonsense').code == 2448 legacy := temp_repo('legacy')449 defer { os.rmdir_all(legacy) or {} }450 os.mkdir_all(os.join_path(legacy, '.koh')) or { panic(err) }451 fake := os.join_path(home, 'bin')452 os.mkdir_all(fake) or { panic(err) }453 fake_koh := os.join_path(fake, 'koh2')454 os.write_file(fake_koh, '#!/bin/sh\nprintf "handoff:%s\\n" "$*"\nexit 7\n') or { panic(err) }455 os.chmod(fake_koh, 0o755) or { panic(err) }456 handoff := run_home_with_path(legacy, home, fake, 'status', '--json')457 assert handoff.code == 7458 assert handoff.out.contains('handoff:status --json')459}460461fn run_home_with_path(repo string, home string, bin_dir string, args ...string) Result {462 stamp := '${os.getpid()}-${time.now().unix_nano()}'463 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')464 out_path := '${base}.out'465 err_path := '${base}.err'466 code_path := '${base}.code'467 mut command := 'HOME=${q(home)} PATH=${q(bin_dir + ":" + os.getenv("PATH"))} ${q(binary)}'468 for arg in args {469 command += ' ${q(arg)}'470 }471 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" $? >${q(code_path)}'472 result := os.execute(shell)473 if result.exit_code != 0 {474 panic('could not run test command: ${result.output}')475 }476 code := os.read_file(code_path) or { panic(err) }477 out := os.read_file(out_path) or { '' }478 err := os.read_file(err_path) or { '' }479 os.rm(out_path) or {}480 os.rm(err_path) or {}481 os.rm(code_path) or {}482 return Result{code: code.int(), out: out, err: err}483}