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 stamp := '${os.getpid()}-${time.now().unix_nano()}'36 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')37 out_path := '${base}.out'38 err_path := '${base}.err'39 code_path := '${base}.code'40 mut command := '${q(binary)}'41 for arg in args {42 command += ' ${q(arg)}'43 }44 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" $? >${q(code_path)}'45 result := os.execute(shell)46 if result.exit_code != 0 {47 panic('could not run test command: ${result.output}')48 }49 code := os.read_file(code_path) or { panic(err) }50 out := os.read_file(out_path) or { '' }51 err := os.read_file(err_path) or { '' }52 os.rm(out_path) or {}53 os.rm(err_path) or {}54 os.rm(code_path) or {}55 return Result{56 code: code.int()57 out: out58 err: err59 }60}6162fn init_repo(repo string) {63 r := run(repo, 'init')64 assert r.code == 065 assert os.exists(os.join_path(repo, '.koh'))66 assert os.exists(os.join_path(repo, '.kohignore'))67}6869fn write(repo string, path string, text string) {70 full := os.join_path(repo, path)71 os.mkdir_all(os.dir(full)) or { panic(err) }72 os.write_file(full, text) or { panic(err) }73}7475fn read_json_has(text string, key string) {76 if !text.contains(key) {77 panic('missing JSON key ${key}: ${text}')78 }79}8081fn test_init_and_usage() {82 dir := temp_repo('init')83 defer { os.rmdir_all(dir) or {} }84 r := run(dir, 'init', '--json')85 assert r.code == 086 read_json_has(r.out, 'repo')87 read_json_has(r.out, 'ignore_created')88 assert run(dir, 'init').code == 189 assert run(dir, 'unknown-command').code == 290 assert run(dir, 'unknown-command').err.contains('unknown command')91}9293fn test_save_checkpoint_and_status() {94 dir := temp_repo('save')95 defer { os.rmdir_all(dir) or {} }96 init_repo(dir)97 write(dir, 'tracked.txt', 'one\n')98 r := run(dir, 'save', 'first message', '--agent', 'agent-7', '--session', 'session-a', '--json')99 assert r.code == 0100 for key in ['id', 'parent', 'message', 'created_at', 'machine', 'is_checkpoint', 'agent',101 'session', 'files', 'size_bytes'] {102 read_json_has(r.out, key)103 }104 assert r.out.contains('"message":"first message"')105 assert r.out.contains('"agent":"agent-7"')106 assert r.out.contains('"session":"session-a"')107 assert r.out.contains('"is_checkpoint":false')108 assert run(dir, 'save', 'again').code == 1109 assert run(dir, 'save', 'again').err.contains('nothing has changed since the last save')110 assert run(dir, 'save').code == 2111 assert run(dir, 'save', 'x', '--bogus').code == 2112 cp := run(dir, 'checkpoint', 'still first', '--json')113 assert cp.code == 0114 assert cp.out.contains('"is_checkpoint":true')115 assert cp.out.contains('"parent":')116 clean := run(dir, 'status')117 assert clean.code == 0118 assert clean.out.contains('no changes')119 write(dir, 'tracked.txt', 'two\n')120 write(dir, 'new.txt', 'new\n')121 os.rm(os.join_path(dir, 'tracked.txt')) or { panic(err) }122 write(dir, '.kohignore', '*.log\n')123 status := run(dir, 'status')124 assert status.code == 0125 assert status.out.contains('+')126 assert status.out.contains('~') || status.out.contains('-')127 assert status.out.contains('added')128 js := run(dir, 'status', '--json')129 assert js.code == 0130 for key in ['head_id', 'head_msg', 'dirty', 'modified', 'added', 'deleted', 'ignore_changed'] {131 read_json_has(js.out, key)132 }133 assert js.out.contains('"ignore_changed":true')134}135136fn test_status_requires_repo_and_ignore_rules() {137 dir := temp_repo('status')138 defer { os.rmdir_all(dir) or {} }139 assert run(dir, 'status').code == 1140 init_repo(dir)141 write(dir, '.kohignore', '*.log\nbuild/\n')142 write(dir, 'keep.txt', 'keep\n')143 write(dir, 'ignored.log', 'ignore\n')144 write(dir, 'build/output.txt', 'ignore\n')145 assert run(dir, 'save', 'keep').code == 0146 assert run(dir, 'status').out.contains('no changes')147 assert !run(dir, 'status').out.contains('ignored.log')148 assert !run(dir, 'status').out.contains('build/output.txt')149 assert !run(dir, 'status').out.contains('LOG.md')150}151152fn test_log_and_diff() {153 dir := temp_repo('history')154 defer { os.rmdir_all(dir) or {} }155 init_repo(dir)156 write(dir, 'a.txt', 'one\n')157 first := run(dir, 'save', 'one', '--session', 'alpha', '--json')158 id1 := first.out.all_after('"id":"')[..12]159 write(dir, 'a.txt', 'two\n')160 second := run(dir, 'save', 'two', '--session', 'beta', '--json')161 id2 := second.out.all_after('"id":"')[..12]162 cp := run(dir, 'checkpoint', 'checkpoint', '--json')163 assert cp.code == 0164 log := run(dir, 'log')165 assert log.code == 0166 assert log.out.contains('▸')167 assert log.out.contains('◐')168 lj := run(dir, 'log', '--json')169 assert lj.code == 0170 assert lj.out.contains('"head_id"')171 assert lj.out.contains('"saves"')172 assert run(dir, 'log', '--since', 'today').out.contains('two')173 session := run(dir, 'log', '--session', 'alpha')174 assert session.out.contains('one')175 assert !session.out.contains('two')176 write(dir, 'a.txt', 'three\n')177 d := run(dir, 'diff')178 assert d.code == 0179 assert d.out.contains('--- a/')180 assert d.out.contains('+++ b/')181 assert run(dir, 'diff', id1).code == 0182 assert run(dir, 'diff', id1, id2).code == 0183 dj := run(dir, 'diff', '--json')184 assert dj.code == 0185 assert dj.out.contains('"from"')186 assert dj.out.contains('"to"')187 assert dj.out.contains('"files"')188 assert dj.out.contains('"change":"modified"')189 assert run(dir, 'diff', 'not-a-real-id').code == 1190 assert run(dir, 'diff', id1, id2, 'third').code == 1191 write(dir, 'blob.bin', '\x00\x01\x02')192 diff_binary := run(dir, 'diff', '--json')193 assert diff_binary.out.contains('"binary":true')194 assert diff_binary.out.contains('Binary files differ')195}196197fn test_undo_revert_and_load() {198 dir := temp_repo('restore')199 defer { os.rmdir_all(dir) or {} }200 init_repo(dir)201 write(dir, 'a.txt', 'one\n')202 one := run(dir, 'save', 'one', '--json').out203 id1 := one.all_after('"id":"')[..12]204 write(dir, 'a.txt', 'two\n')205 write(dir, 'gone.txt', 'gone\n')206 second := run(dir, 'save', 'two', '--json').out207 id2 := second.all_after('"id":"')[..12]208 write(dir, 'a.txt', 'three\n')209 third := run(dir, 'save', 'three', '--json').out210 id3 := third.all_after('"id":"')[..12]211 write(dir, 'never.txt', 'keep me\n')212 undo := run(dir, 'undo', '--yes', '--json')213 assert undo.code == 0214 if !undo.out.contains(id2) {215 panic('undo JSON did not name ${id2}: ${undo.out}')216 }217 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'218 assert os.exists(os.join_path(dir, 'gone.txt'))219 assert os.exists(os.join_path(dir, 'never.txt'))220 assert run(dir, 'load', id3, '--yes').code == 0221 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'222 assert run(dir, 'undo', '2', '--yes').code == 0223 assert run(dir, 'load', id2, '--yes').code == 0224 assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')225 write(dir, 'a.txt', 'dirty\n')226 before := run(dir, 'log')227 refuse := run(dir, 'revert')228 assert refuse.code == 1229 assert refuse.err.contains('needs --yes when not interactive')230 assert run(dir, 'log').out == before.out231 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'232 write(dir, 'untracked.txt', 'remove\n')233 rv := run(dir, 'revert', '--yes')234 assert rv.code == 0235 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'236 assert !os.exists(os.join_path(dir, 'untracked.txt'))237 assert run(dir, 'revert', '--yes').out.contains('working tree already matches')238 assert run(dir, 'load', '--checkpoint', '--yes').code == 0239 assert run(dir, 'load', id2, '--json').code == 1240 assert run(dir, 'load', id2, '--json').err.contains('error')241 assert run(dir, 'load').code == 2242 assert run(dir, 'load', 'unknown-id', '--yes').code == 1243}244245fn test_flag_and_flagged_log() {246 dir := temp_repo('flag')247 defer { os.rmdir_all(dir) or {} }248 init_repo(dir)249 write(dir, 'flag.txt', 'one\n')250 saved := run(dir, 'save', 'flag me', '--json')251 id := saved.out.all_after('"id":"')[..12]252 flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')253 assert flagged.code == 0254 assert flagged.out.contains('"flagged":true')255 assert flagged.out.contains('"reason":"unsafe"')256 list := run(dir, 'flag', '--json')257 assert list.code == 0258 assert list.out.contains(id)259 assert list.out.contains('unsafe')260 assert run(dir, 'log').out.contains('⚑')261 assert run(dir, 'flag', '--clear', id).code == 0262 assert run(dir, 'flag').out.contains('nothing is flagged')263 assert run(dir, 'flag', 'unknown-id').code == 1264 assert run(dir, 'flag').code == 0265}266267fn test_pluck_forward_reverse_dry_run_and_conflict() {268 dir := temp_repo('pluck')269 defer { os.rmdir_all(dir) or {} }270 init_repo(dir)271 write(dir, 'a.txt', 'one\n')272 first := run(dir, 'save', 'one', '--json').out273 id1 := first.all_after('"id":"')[..12]274 write(dir, 'a.txt', 'two\n')275 second := run(dir, 'save', 'two', '--json').out276 id2 := second.all_after('"id":"')[..12]277 assert run(dir, 'load', id1, '--yes').code == 0278 dry := run(dir, 'pluck', id2, '--dry-run')279 assert dry.code == 0280 assert dry.out.contains('a.txt')281 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'282 assert run(dir, 'pluck', id2, '--yes').code == 0283 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'284 assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0285 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'286 write(dir, 'a.txt', 'different\n')287 conflict := run(dir, 'pluck', id2, '--yes')288 assert conflict.code == 1289 assert conflict.err.contains('a.txt')290 assert run(dir, 'pluck').code == 2291 assert run(dir, 'pluck', 'unknown-id').code == 1292}293294fn test_peek_and_doctor() {295 dir := temp_repo('peek-doctor')296 defer { os.rmdir_all(dir) or {} }297 init_repo(dir)298 write(dir, 'peek.txt', 'peek me\n')299 assert run(dir, 'save', 'peekable').code == 0300 id := run(dir, 'log', '--json').out.all_after('"id":"')[..12]301 before_log := run(dir, 'log', '--json').out302 peek := run(dir, 'peek', id, '--json')303 assert peek.code == 0304 peek_path := peek.out.all_after('"path":"').before('"')305 assert peek.out.contains('"id":"${id}"')306 assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'307 assert run(dir, 'status').out.contains('no changes')308 assert run(dir, 'log', '--json').out == before_log309 assert run(dir, 'peek').code == 2310 assert run(dir, 'peek', 'unknown-id').code == 1311 backup := os.join_path(dir, 'doctor.backup')312 healthy := run(dir, 'doctor')313 assert healthy.code == 0314 assert healthy.out.contains('is sound')315 report := run(dir, 'doctor', '--json')316 assert report.code == 0317 for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',318 'backup'] {319 read_json_has(report.out, key)320 }321 backed := run(dir, 'doctor', '--backup', backup)322 assert backed.code == 0323 assert os.exists(backup)324 assert backed.out.contains(backup)325 assert run(dir, 'doctor', '--backup', backup).code == 1326 assert run(dir, 'peek', id, '--json').code == 0327 removed := run(dir, 'doctor', '--json')328 assert removed.code == 0329 assert removed.out.contains('"peek_folders_removed":1')330 assert !os.exists(peek_path)331}332333fn test_strip_refuses_without_terminal() {334 dir := temp_repo('strip')335 defer { os.rmdir_all(dir) or {} }336 init_repo(dir)337 write(dir, 'secret.txt', 'keep\n')338 assert run(dir, 'save', 'secret').code == 0339 before_log := run(dir, 'log', '--json').out340 refused := run(dir, 'strip', 'secret.txt')341 assert refused.code == 1342 assert refused.err.contains('needs a terminal')343 assert run(dir, 'log', '--json').out == before_log344 assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'345 assert run(dir, 'log', '--json').out.contains('secret')346}347348fn test_saves_and_checkpoints_aliases() {349 dir := temp_repo('aliases')350 defer { os.rmdir_all(dir) or {} }351 init_repo(dir)352 write(dir, 'alias.txt', 'one\n')353 assert run(dir, 'save', 'plain save').code == 0354 assert run(dir, 'checkpoint', 'checkpoint save').code == 0355 saves := run(dir, 'saves')356 assert saves.code == 0357 assert saves.out.contains('plain save')358 assert saves.out.contains('checkpoint save')359 checkpoints := run(dir, 'checkpoints')360 assert checkpoints.code == 0361 assert checkpoints.out.contains('checkpoint save')362 assert !checkpoints.out.contains('plain save')363 json_result := run(dir, 'checkpoints', '--json')364 assert json_result.code == 0365 assert json_result.out.contains('"head_id"')366 assert json_result.out.contains('"saves"')367 assert json_result.out.contains('"is_checkpoint":true')368 assert !json_result.out.contains('"is_checkpoint":false')369}