1module tests23import net4import net.http5import os6import time78const root = @VMODROOT9const binary = os.join_path(root, 'build', 'koh3')1011struct Result {12mut:13 code int14 out string15 err string16}1718fn testsuite_begin() {19 os.mkdir_all(os.join_path(root, 'build')) or { panic(err) }20 build := os.execute('v -o ${q(binary)} ${q(os.join_path(root, 'src', 'koh3.v'))}')21 if build.exit_code != 0 {22 panic('could not build koh3: ${build.output}')23 }24}2526fn q(s string) string {27 return "'${s.replace("'", "'\\''")}'"28}2930fn temp_repo(name string) string {31 dir := os.join_path(os.temp_dir(), 'koh3-cli-${name}-${time.now().unix_nano()}')32 os.mkdir_all(dir) or { panic(err) }33 return dir34}3536fn run(repo string, args ...string) Result {37 home := temp_repo('home')38 defer { os.rmdir_all(home) or {} }39 return run_home(repo, home, ...args)40}4142fn run_home(repo string, home string, args ...string) Result {43 stamp := '${os.getpid()}-${time.now().unix_nano()}'44 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')45 out_path := '${base}.out'46 err_path := '${base}.err'47 code_path := '${base}.code'48 mut command := 'HOME=${q(home)} ${q(binary)}'49 for arg in args {50 command += ' ${q(arg)}'51 }52 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" \$? >${q(code_path)}'53 result := os.execute(shell)54 if result.exit_code != 0 {55 panic('could not run test command: ${result.output}')56 }57 code := os.read_file(code_path) or { panic(err) }58 out := os.read_file(out_path) or { '' }59 err := os.read_file(err_path) or { '' }60 os.rm(out_path) or {}61 os.rm(err_path) or {}62 os.rm(code_path) or {}63 return Result{64 code: code.int()65 out: out66 err: err67 }68}6970fn init_repo(repo string) {71 r := run(repo, 'init')72 assert r.code == 073 assert os.exists(os.join_path(repo, '.koh'))74 assert os.exists(os.join_path(repo, '.kohignore'))75}7677fn write(repo string, path string, text string) {78 full := os.join_path(repo, path)79 os.mkdir_all(os.dir(full)) or { panic(err) }80 os.write_file(full, text) or { panic(err) }81}8283fn read_json_has(text string, key string) {84 if !text.contains(key) {85 panic('missing JSON key ${key}: ${text}')86 }87}8889fn test_init_and_usage() {90 dir := temp_repo('init')91 defer { os.rmdir_all(dir) or {} }92 r := run(dir, 'init', '--json')93 assert r.code == 094 read_json_has(r.out, 'repo')95 read_json_has(r.out, 'ignore_created')96 assert run(dir, 'init').code == 197 assert run(dir, 'unknown-command').code == 298 assert run(dir, 'unknown-command').err.contains('unknown command')99 100 for args in [['unknown-command', '--json'], ['save', '--json'], ['save', 'x', '--nope', '--json']] {101 r2 := run(dir, ...args)102 assert r2.code == 2103 assert r2.out == ''104 assert r2.err.starts_with('{"error":'), r2.err105 }106}107108109110111fn this_platform() string {112 machine := os.uname().machine113 arch := if machine in ['arm64', 'aarch64'] { 'aarch64' } else { 'x86_64' }114 return '${os.user_os()}-${arch}'115}116117118119fn test_build_builds_the_named_platforms() {120 dir := temp_repo('build_ok')121 defer { os.rmdir_all(dir) or {} }122 init_repo(dir)123 host := this_platform()124 write(dir, 'VERSION', '1.2.3\n')125 write(dir, 'koh.build', 'command = mkdir -p out && printf built > out/demo-1.2.3-${host}126output = out/demo-{version}-{platform}127platforms = ${host}128')129 one := run(dir, 'build')130 assert one.code == 0, one.err131 assert one.out.contains('version 1.2.3')132 assert one.out.contains('out/demo-1.2.3-${host}')133 assert os.exists(os.join_path(dir, 'out/demo-1.2.3-${host}'))134135 136 write(dir, 'koh.build', 'command = mkdir -p out && printf m > out/demo-macos-aarch64 && printf l > out/demo-linux-aarch64137output = out/demo-{platform}138platforms = macos-aarch64, linux-aarch64139')140 all := run(dir, 'build', '--all', '--json')141 assert all.code == 0, all.err142 assert all.out.contains('"platform":"macos-aarch64"')143 assert all.out.contains('"platform":"linux-aarch64"')144 assert os.exists(os.join_path(dir, 'out/demo-linux-aarch64'))145 picked := run(dir, 'build', '--platform', 'linux-aarch64')146 assert picked.code == 0, picked.err147 assert picked.out.contains('out/demo-linux-aarch64')148 assert !picked.out.contains('out/demo-macos-aarch64')149}150151152153154155156fn test_update_refuses_a_command_line_it_cannot_answer() {157 dir := temp_repo('update_args')158 defer { os.rmdir_all(dir) or {} }159 init_repo(dir)160 json_without_yes := run(dir, 'update', '--json')161 assert json_without_yes.code == 2162 assert json_without_yes.err.contains('needs --yes')163 assert run(dir, 'update', 'extra').code == 2164}165166fn test_build_builds_koh_itself() {167 dir := temp_repo('selfbuild')168 defer { os.rmdir_all(dir) or {} }169 170 for item in ['src', 'vendor', 'docs', 'v.mod', 'VERSION'] {171 os.cp_all(os.join_path(root, item), os.join_path(dir, item), true) or { panic(err) }172 }173 write(dir, 'koh.build', 'command = mkdir -p zig-out/bin && v -o zig-out/bin/koh3 src/koh3.v174output = zig-out/bin/koh3175platforms = ${this_platform()}176')177 init_repo(dir)178 built := run(dir, 'build')179 assert built.code == 0, built.err180 assert built.out.contains('zig-out/bin/koh3')181 fresh := os.join_path(dir, 'zig-out/bin/koh3')182 assert os.exists(fresh)183 version := os.read_file(os.join_path(dir, 'VERSION'))!.trim_space()184 assert os.execute('${q(fresh)} --version').output.trim_space() == 'koh ${version}'185}186187fn test_build_refuses_what_it_cannot_do() {188 dir := temp_repo('build_bad')189 defer { os.rmdir_all(dir) or {} }190 init_repo(dir)191 missing := run(dir, 'build')192 assert missing.code == 1193 assert missing.err.contains('koh.build')194195 write(dir, 'koh.build', 'command = true196output = out/demo197platforms = macos-aarch64198')199 no_version := run(dir, 'build')200 assert no_version.code == 1201 assert no_version.err.contains('VERSION')202203 write(dir, 'VERSION', '1.2.3\n')204 not_listed := run(dir, 'build', '--platform', 'linux-aarch64')205 assert not_listed.code == 1206 assert not_listed.err.contains('does not build for')207208 unknown := run(dir, 'build', '--platform', 'amiga')209 assert unknown.code == 1210 assert unknown.err.contains('amiga')211212 both := run(dir, 'build', '--all', '--platform', 'macos-aarch64')213 assert both.code == 2214 assert both.err.contains('cannot be combined')215216 write(dir, 'koh.build', 'command = true217output = out/demo218platforms = macos-aarch64, amiga219')220 bad_platform := run(dir, 'build')221 assert bad_platform.code == 1222 assert bad_platform.err.contains('amiga')223224 write(dir, 'koh.build', 'command = sh -c \'echo "no toolchain" >&2; exit 1\'225output = out/demo226platforms = ${this_platform()}227')228 failed := run(dir, 'build')229 assert failed.code == 1230 assert failed.err.contains('no toolchain')231}232233fn test_the_help_screen_lists_commands_and_no_flags() {234 dir := temp_repo('help')235 defer { os.rmdir_all(dir) or {} }236 help := run(dir, '--help')237 assert help.code == 0238 names := ['save', 'checkpoint', 'status', 'view', 'diff', 'undo', 'revert', 'load', 'offer',239 'steal']240 for name in names {241 assert help.out.contains(' ${name} '), name242 }243 assert help.out.contains('koh man <command>')244 245 assert !help.out.contains('--'), help.out246 lines := help.out.split_into_lines()247 for line in lines {248 assert line.runes().len <= 80, line249 }250 251 for name in names {252 for i, line in lines {253 if line.starts_with(' ${name} ') {254 assert i > 0 && lines[i - 1] == '', line255 }256 }257 }258 259 bare := run(dir)260 assert bare.code == 2261 assert bare.out == ''262 assert bare.err.contains('koh man <command>')263 short := run(dir, '-h')264 assert short.code == 0265 assert short.out == help.out266}267268269fn test_the_help_screen_has_no_colour_when_it_is_off() {270 dir := temp_repo('help_plain')271 defer { os.rmdir_all(dir) or {} }272 home := temp_repo('help_plain_home')273 defer { os.rmdir_all(home) or {} }274 out_path := os.join_path(os.temp_dir(), 'koh3-help-${os.getpid()}-${time.now().unix_nano()}')275 defer { os.rm(out_path) or {} }276 code := os.system('cd ${q(dir)} && HOME=${q(home)} NO_COLOR=1 ${q(binary)} --help > ${q(out_path)}')277 assert code == 0278 text := os.read_file(out_path) or { panic(err) }279 assert !text.contains('\x1b'), text280 assert text.contains(' save snapshot your work')281}282283284fn test_version_matches_the_manifest() {285 dir := temp_repo('version')286 defer { os.rmdir_all(dir) or {} }287 version := run(dir, '--version').out.trim_space().all_after('koh ')288 manifest := os.read_file(os.join_path(root, 'v.mod')) or { panic(err) }289 assert manifest.contains("version: '${version}'"), version290}291292fn test_save_checkpoint_and_status() {293 dir := temp_repo('save')294 defer { os.rmdir_all(dir) or {} }295 init_repo(dir)296 write(dir, 'tracked.txt', 'one\n')297 r := run(dir, 'save', 'first message', '--agent', 'agent-7', '--session', 'session-a', '--json')298 assert r.code == 0299 for key in ['id', 'parent', 'message', 'created_at', 'machine', 'is_checkpoint', 'agent', 'session',300 'files', 'size_bytes'] {301 read_json_has(r.out, key)302 }303 assert r.out.contains('"message":"first message"')304 assert r.out.contains('"agent":"agent-7"')305 assert r.out.contains('"session":"session-a"')306 assert r.out.contains('"is_checkpoint":false')307 assert run(dir, 'save', 'again').code == 1308 assert run(dir, 'save', 'again').err.contains('nothing has changed since the last save')309 assert run(dir, 'save').code == 2310 assert run(dir, 'save', 'x', '--bogus').code == 2311 cp := run(dir, 'checkpoint', 'still first', '--json')312 assert cp.code == 0313 assert cp.out.contains('"is_checkpoint":true')314 assert cp.out.contains('"parent":')315 clean := run(dir, 'status')316 assert clean.code == 0317 assert clean.out.contains('no changes')318 write(dir, 'tracked.txt', 'two\n')319 write(dir, 'new.txt', 'new\n')320 os.rm(os.join_path(dir, 'tracked.txt')) or { panic(err) }321 write(dir, '.kohignore', '*.log\n')322 status := run(dir, 'status')323 assert status.code == 0324 assert status.out.contains('+')325 assert status.out.contains('~') || status.out.contains('-')326 assert status.out.contains('added')327 js := run(dir, 'status', '--json')328 assert js.code == 0329 for key in ['head_id', 'head_msg', 'dirty', 'modified', 'added', 'deleted', 'ignore_changed',330 'saves', 'size_bytes', 'head_flagged'] {331 read_json_has(js.out, key)332 }333 assert js.out.contains('"ignore_changed":true')334 335 assert js.out.contains('"saves":2')336 assert js.out.contains('"size_bytes":${os.file_size(os.join_path(dir, '.koh'))}')337 assert js.out.contains('"head_flagged":false')338 head := js.out.all_after('"head_id":"').all_before('"')339 assert run(dir, 'flag', head, '--reason', 'bad').code == 0340 assert run(dir, 'status', '--json').out.contains('"head_flagged":true')341}342343fn test_status_requires_repo_and_ignore_rules() {344 dir := temp_repo('status')345 defer { os.rmdir_all(dir) or {} }346 assert run(dir, 'status').code == 1347 init_repo(dir)348 write(dir, '.kohignore', '*.log\nbuild/\n')349 write(dir, 'keep.txt', 'keep\n')350 write(dir, 'ignored.log', 'ignore\n')351 write(dir, 'build/output.txt', 'ignore\n')352 assert run(dir, 'save', 'keep').code == 0353 assert run(dir, 'status').out.contains('no changes')354 assert !run(dir, 'status').out.contains('ignored.log')355 assert !run(dir, 'status').out.contains('build/output.txt')356 assert !run(dir, 'status').out.contains('LOG.md')357}358359360361362fn test_a_save_succeeds_when_log_md_cannot_be_written() {363 dir := temp_repo('logmd')364 defer { os.rmdir_all(dir) or {} }365 init_repo(dir)366 os.mkdir_all(os.join_path(dir, 'LOG.md')) or { panic(err) }367 write(dir, 'a.txt', 'one\n')368 r := run(dir, 'save', 'first', '--json')369 assert r.code == 0370 assert r.out.starts_with('{')371 assert r.err.contains('LOG.md')372 assert run(dir, 'view', '--json').out.contains('"message":"first"')373}374375376377378fn test_a_restore_refuses_a_folder_of_ignored_files_in_the_way() {379 dir := temp_repo('dir-in-way')380 defer { os.rmdir_all(dir) or {} }381 init_repo(dir)382 write(dir, 'out', 'a file\n')383 first := run(dir, 'save', 'out is a file', '--json').out.all_after('"id":"')[..12]384 os.rm(os.join_path(dir, 'out')) or { panic(err) }385 write(dir, 'out/build.log', 'ignored\n')386 assert run(dir, 'save', 'out is gone').code == 0387 r := run(dir, 'load', first, '--yes')388 assert r.code == 1389 assert r.err.contains('out')390 assert os.ls(os.join_path(dir, 'out'))! == ['build.log']391}392393394395396fn test_a_restore_never_writes_through_a_symlinked_folder() {397 dir := temp_repo('link-in-way')398 outside := temp_repo('link-target')399 defer {400 os.rmdir_all(dir) or {}401 os.rmdir_all(outside) or {}402 }403 init_repo(dir)404 write(dir, 'docs/a.md', 'saved\n')405 first := run(dir, 'save', 'docs', '--json').out.all_after('"id":"')[..12]406 os.rmdir_all(os.join_path(dir, 'docs')) or { panic(err) }407 write(outside, 'a.md', 'precious\n')408 os.symlink(outside, os.join_path(dir, 'docs')) or { panic(err) }409 assert run(dir, 'save', 'docs is a link now').code == 0410 head := run(dir, 'status', '--json').out411 r := run(dir, 'load', first, '--yes')412 assert r.code == 1413 assert r.err.contains('docs')414 assert os.read_file(os.join_path(outside, 'a.md'))! == 'precious\n'415 assert run(dir, 'status', '--json').out == head416}417418419420421422fn test_a_restore_cut_short_points_at_revert() {423 dir := temp_repo('cut-short')424 defer {425 os.chmod(os.join_path(dir, 'zz'), 0o755) or {}426 os.rmdir_all(dir) or {}427 }428 init_repo(dir)429 write(dir, 'zero', '0\n')430 assert run(dir, 'save', 'zero').code == 0431 write(dir, 'a.txt', 'v1\n')432 write(dir, 'zz/f', 'v1\n')433 assert run(dir, 'save', 'first').code == 0434 write(dir, 'a.txt', 'v2\n')435 write(dir, 'zz/f', 'v2\n')436 assert run(dir, 'save', 'second').code == 0437 438 439 os.chmod(os.join_path(dir, 'zz'), 0o555)!440 stopped := run(dir, 'undo', '--yes')441 assert stopped.code == 1442 assert stopped.err.contains('koh revert')443 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'v1\n'444 assert os.read_file(os.join_path(dir, 'zz/f'))! == 'v2\n'445 446 os.chmod(os.join_path(dir, 'zz'), 0o000)!447 assert run(dir, 'status').err.contains('koh revert')448 os.chmod(os.join_path(dir, 'zz'), 0o755)!449450 assert run(dir, 'status', '--json').out.contains('"restore_unfinished":true')451 for args in [['save', 'half'], ['checkpoint', 'half'], ['undo', '--yes']] {452 r := run(dir, ...args)453 assert r.code == 1454 assert r.err.contains('koh revert')455 }456 assert run(dir, 'revert', '--yes').code == 0457 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'v1\n'458 assert os.read_file(os.join_path(dir, 'zz/f'))! == 'v1\n'459 assert run(dir, 'status', '--json').out.contains('"restore_unfinished":false')460 write(dir, 'a.txt', 'v3\n')461 assert run(dir, 'save', 'carry on').code == 0462}463464465466fn test_doctor_sweeps_only_old_peek_folders() {467 dir := temp_repo('peek-sweep')468 469 470 fresh := os.join_path(os.temp_dir(), 'koh-peek-fresh${os.getpid()}')471 old := os.join_path(os.temp_dir(), 'koh-peek-old${os.getpid()}')472 defer {473 os.rmdir_all(dir) or {}474 os.rmdir_all(fresh) or {}475 os.rmdir_all(old) or {}476 }477 init_repo(dir)478 write(dir, 'a.txt', 'one\n')479 assert run(dir, 'save', 'one').code == 0480 os.mkdir_all(fresh) or { panic(err) }481 os.mkdir_all(old) or { panic(err) }482 two_days_ago := time.now().unix() - 2 * 86400483 os.utime(old, two_days_ago, two_days_ago) or { panic(err) }484 assert run(dir, 'doctor', '--json').code == 0485 assert os.exists(fresh)486 assert !os.exists(old)487}488489490491492fn test_a_restore_never_overwrites_an_ignored_file() {493 dir := temp_repo('ignored-restore')494 defer { os.rmdir_all(dir) or {} }495 init_repo(dir)496 write(dir, '.env', 'KEY=saved long ago\n')497 first := run(dir, 'save', 'with env', '--json').out.all_after('"id":"')[..12]498 write(dir, '.kohignore', '.env\n')499 assert run(dir, 'save', 'ignore env').code == 0500 write(dir, '.env', 'KEY=never saved\n')501 assert run(dir, 'load', first, '--yes').code == 0502 assert os.read_file(os.join_path(dir, '.env'))! == 'KEY=never saved\n'503504 505 other := temp_repo('ignored-restore-missing')506 defer { os.rmdir_all(other) or {} }507 init_repo(other)508 write(other, '.env', 'KEY=saved long ago\n')509 again := run(other, 'save', 'with env', '--json').out.all_after('"id":"')[..12]510 write(other, '.kohignore', '.env\n')511 assert run(other, 'save', 'ignore env').code == 0512 os.rm(os.join_path(other, '.env')) or { panic(err) }513 assert run(other, 'load', again, '--yes').code == 0514 assert os.read_file(os.join_path(other, '.env'))! == 'KEY=saved long ago\n'515}516517518519fn test_a_going_back_preview_does_not_run_diff() {520 dir := temp_repo('preview-nodiff')521 home := temp_repo('preview-nodiff-home')522 defer {523 os.rmdir_all(dir) or {}524 os.rmdir_all(home) or {}525 }526 init_repo(dir)527 write(dir, 'a.txt', 'one\n')528 assert run_home(dir, home, 'save', 'one').code == 0529 write(dir, 'a.txt', 'two\n')530 assert run_home(dir, home, 'save', 'two').code == 0531 fake := os.join_path(home, 'bin')532 os.mkdir_all(fake) or { panic(err) }533 marker := os.join_path(home, 'diff-ran')534 os.write_file(os.join_path(fake, 'diff'), '#!/bin/sh\ntouch ${q(marker)}\nexit 2\n') or {535 panic(err)536 }537 os.chmod(os.join_path(fake, 'diff'), 0o755) or { panic(err) }538 assert run_home_with_path(dir, home, fake, 'undo', '--yes').code == 0539 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'540 assert !os.exists(marker)541}542543544545fn test_a_restore_leaves_unchanged_files_alone() {546 dir := temp_repo('restore-skip')547 defer { os.rmdir_all(dir) or {} }548 init_repo(dir)549 write(dir, 'a.txt', 'one\n')550 write(dir, 'same.txt', 'never changes\n')551 assert run(dir, 'save', 'one').code == 0552 write(dir, 'a.txt', 'two\n')553 assert run(dir, 'save', 'two').code == 0554 long_ago := i64(1700000000)555 os.utime(os.join_path(dir, 'same.txt'), long_ago, long_ago) or { panic(err) }556 assert run(dir, 'undo', '--yes').code == 0557 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'558 assert os.file_last_mod_unix(os.join_path(dir, 'same.txt')) == long_ago559}560561562563fn test_status_sees_an_executable_bit_change() {564 dir := temp_repo('chmod')565 defer { os.rmdir_all(dir) or {} }566 init_repo(dir)567 write(dir, 'run.sh', 'echo hi\n')568 assert run(dir, 'save', 'script').code == 0569 os.chmod(os.join_path(dir, 'run.sh'), 0o755) or { panic(err) }570 status := run(dir, 'status', '--json')571 assert status.out.contains('"dirty":true')572 assert status.out.contains('"modified":["run.sh"]')573}574575fn test_view_and_diff() {576 dir := temp_repo('history')577 defer { os.rmdir_all(dir) or {} }578 init_repo(dir)579 write(dir, 'a.txt', 'one\n')580 first := run(dir, 'save', 'one', '--session', 'alpha', '--json')581 id1 := first.out.all_after('"id":"')[..12]582 write(dir, 'a.txt', 'two\n')583 second := run(dir, 'save', 'two', '--session', 'beta', '--json')584 id2 := second.out.all_after('"id":"')[..12]585 cp := run(dir, 'checkpoint', 'checkpoint', '--json')586 assert cp.code == 0587 printed := run(dir, 'view')588 assert printed.code == 0589 assert printed.out.contains('▸')590 assert printed.out.contains('◐')591 lj := run(dir, 'view', '--json')592 assert lj.code == 0593 assert lj.out.contains('"head_id"')594 assert lj.out.contains('"saves"')595 assert run(dir, 'view', '--since', 'today').out.contains('two')596 session := run(dir, 'view', '--session', 'alpha')597 assert session.out.contains('one')598 assert !session.out.contains('two')599 assert run(dir, 'view', '--since', '9999-01-01').out.contains('no saves in that range')600 write(dir, 'a.txt', 'three\n')601 d := run(dir, 'diff')602 assert d.code == 0603 assert d.out.contains('--- a/')604 assert d.out.contains('+++ b/')605 assert run(dir, 'diff', id1).code == 0606 assert run(dir, 'diff', id1, id2).code == 0607 dj := run(dir, 'diff', '--json')608 assert dj.code == 0609 assert dj.out.contains('"from"')610 assert dj.out.contains('"to"')611 assert dj.out.contains('"files"')612 assert dj.out.contains('"change":"modified"')613 assert run(dir, 'diff', 'not-a-real-id').code == 1614 assert run(dir, 'diff', id1, id2, 'third').code == 1615 write(dir, 'blob.bin', '\x00\x01\x02')616 diff_binary := run(dir, 'diff', '--json')617 assert diff_binary.out.contains('"binary":true')618 assert diff_binary.out.contains('Binary files differ')619}620621fn test_undo_revert_and_load() {622 dir := temp_repo('restore')623 defer { os.rmdir_all(dir) or {} }624 init_repo(dir)625 write(dir, 'a.txt', 'one\n')626 one := run(dir, 'save', 'one', '--json').out627 id1 := one.all_after('"id":"')[..12]628 write(dir, 'a.txt', 'two\n')629 write(dir, 'gone.txt', 'gone\n')630 second := run(dir, 'save', 'two', '--json').out631 id2 := second.all_after('"id":"')[..12]632 write(dir, 'a.txt', 'three\n')633 third := run(dir, 'save', 'three', '--json').out634 id3 := third.all_after('"id":"')[..12]635 636 637 write(dir, 'never.txt', 'keep me\n')638 view_before := run(dir, 'view').out639 for args in [['undo', '--yes'], ['load', id1, '--yes'], ['pluck', id2, '--yes']] {640 blocked := run(dir, ...args)641 assert blocked.code == 1642 assert blocked.err.contains('unsaved changes')643 }644 assert run(dir, 'view').out == view_before645 os.rm(os.join_path(dir, 'never.txt')) or { panic(err) }646 write(dir, 'a.txt', 'edited, never saved\n')647 assert run(dir, 'undo', '--yes').err.contains('unsaved changes')648 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'edited, never saved\n'649 write(dir, 'a.txt', 'three\n')650 os.rm(os.join_path(dir, 'gone.txt')) or { panic(err) }651 assert run(dir, 'load', id1, '--yes').err.contains('unsaved changes')652 write(dir, 'gone.txt', 'gone\n')653 undo := run(dir, 'undo', '--yes', '--json')654 assert undo.code == 0655 if !undo.out.contains(id2) {656 panic('undo JSON did not name ${id2}: ${undo.out}')657 }658 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'659 assert os.exists(os.join_path(dir, 'gone.txt'))660 assert run(dir, 'load', id3, '--yes').code == 0661 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'three\n'662 assert run(dir, 'undo', '2', '--yes').code == 0663 assert run(dir, 'load', id2, '--yes').code == 0664 assert run(dir, 'undo', '99', '--yes').err.contains('nothing before')665 write(dir, 'a.txt', 'dirty\n')666 before := run(dir, 'view')667 refuse := run(dir, 'revert')668 assert refuse.code == 1669 assert refuse.err.contains('needs --yes when not interactive')670 assert run(dir, 'view').out == before.out671 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'672 write(dir, 'untracked.txt', 'remove\n')673 rv := run(dir, 'revert', '--yes', '--json')674 assert rv.code == 0675 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'676 assert !os.exists(os.join_path(dir, 'untracked.txt'))677 678 kept := rv.out.all_after('"checkpoint":"')[..12]679 assert run(dir, 'load', kept, '--yes').code == 0680 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'dirty\n'681 assert os.read_file(os.join_path(dir, 'untracked.txt'))! == 'remove\n'682 write(dir, 'a.txt', 'dirty again\n')683 assert run(dir, 'revert', '--yes').out.contains('koh load ')684 assert run(dir, 'revert', '--yes').out.contains('working tree already matches')685 assert run(dir, 'load', '--checkpoint', '--yes').code == 0686 assert run(dir, 'load', id2, '--json').code == 1687 assert run(dir, 'load', id2, '--json').err.contains('error')688 assert run(dir, 'load').code == 2689 assert run(dir, 'load', 'unknown-id', '--yes').code == 1690}691692fn test_flag_and_flagged_view() {693 dir := temp_repo('flag')694 defer { os.rmdir_all(dir) or {} }695 init_repo(dir)696 write(dir, 'flag.txt', 'one\n')697 saved := run(dir, 'save', 'flag me', '--json')698 id := saved.out.all_after('"id":"')[..12]699 flagged := run(dir, 'flag', id, '--reason', 'unsafe', '--json')700 assert flagged.code == 0701 assert flagged.out.contains('"flagged":true')702 assert flagged.out.contains('"reason":"unsafe"')703 list := run(dir, 'flag', '--json')704 assert list.code == 0705 assert list.out.contains(id)706 assert list.out.contains('unsafe')707 assert run(dir, 'view').out.contains('⚑')708 assert run(dir, 'flag', '--clear', id).code == 0709 assert run(dir, 'flag').out.contains('nothing is flagged')710 assert run(dir, 'flag', 'unknown-id').code == 1711 assert run(dir, 'flag').code == 0712}713714fn test_pluck_forward_reverse_dry_run_and_conflict() {715 dir := temp_repo('pluck')716 defer { os.rmdir_all(dir) or {} }717 init_repo(dir)718 write(dir, 'a.txt', 'one\n')719 first := run(dir, 'save', 'one', '--json').out720 id1 := first.all_after('"id":"')[..12]721 write(dir, 'a.txt', 'two\n')722 second := run(dir, 'save', 'two', '--json').out723 id2 := second.all_after('"id":"')[..12]724 assert run(dir, 'load', id1, '--yes').code == 0725 dry := run(dir, 'pluck', id2, '--dry-run')726 assert dry.code == 0727 assert dry.out.contains('a.txt')728 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'729 assert run(dir, 'pluck', id2, '--yes').code == 0730 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'two\n'731 assert run(dir, 'pluck', '--reverse', id2, '--yes').code == 0732 assert os.read_file(os.join_path(dir, 'a.txt'))! == 'one\n'733 write(dir, 'a.txt', 'different\n')734 assert run(dir, 'pluck', id2, '--dry-run').code == 1735 assert run(dir, 'save', 'different').code == 0736 conflict := run(dir, 'pluck', id2, '--yes')737 assert conflict.code == 1738 assert conflict.err.contains('a.txt')739 assert run(dir, 'pluck').code == 2740 assert run(dir, 'pluck', 'unknown-id').code == 1741}742743fn test_peek_and_doctor() {744 dir := temp_repo('peek-doctor')745 defer { os.rmdir_all(dir) or {} }746 init_repo(dir)747 write(dir, 'peek.txt', 'peek me\n')748 assert run(dir, 'save', 'peekable').code == 0749 id := run(dir, 'view', '--json').out.all_after('"id":"')[..12]750 before_view := run(dir, 'view', '--json').out751 peek := run(dir, 'peek', id, '--json')752 assert peek.code == 0753 peek_path := peek.out.all_after('"path":"').before('"')754 assert peek.out.contains('"id":"${id}"')755 assert os.read_file(os.join_path(peek_path, 'peek.txt'))! == 'peek me\n'756 assert run(dir, 'status').out.contains('no changes')757 assert run(dir, 'view', '--json').out == before_view758 assert run(dir, 'peek').code == 2759 assert run(dir, 'peek', 'unknown-id').code == 1760 backup := os.join_path(dir, 'doctor.backup')761 healthy := run(dir, 'doctor')762 assert healthy.code == 0763 assert healthy.out.contains('is sound')764 report := run(dir, 'doctor', '--json')765 assert report.code == 0766 for key in ['problems', 'objects', 'saves', 'checkpoints', 'cached_files', 'peek_folders_removed',767 'backup'] {768 read_json_has(report.out, key)769 }770 backed := run(dir, 'doctor', '--backup', backup)771 assert backed.code == 0772 assert os.exists(backup)773 assert backed.out.contains(backup)774 assert run(dir, 'doctor', '--backup', backup).code == 1775 assert run(dir, 'peek', id, '--json').code == 0776 777 kept := run(dir, 'doctor', '--json')778 assert kept.code == 0779 assert os.exists(peek_path)780 os.rmdir_all(peek_path) or {}781}782783fn test_strip_refuses_without_terminal() {784 dir := temp_repo('strip')785 defer { os.rmdir_all(dir) or {} }786 init_repo(dir)787 write(dir, 'secret.txt', 'keep\n')788 assert run(dir, 'save', 'secret').code == 0789 before_view := run(dir, 'view', '--json').out790 refused := run(dir, 'strip', 'secret.txt')791 assert refused.code == 1792 assert refused.err.contains('needs a terminal')793 assert run(dir, 'view', '--json').out == before_view794 assert os.read_file(os.join_path(dir, 'secret.txt'))! == 'keep\n'795 assert run(dir, 'view', '--json').out.contains('secret')796}797798799800fn test_view_shows_the_line_and_checkpoints_apart() {801 dir := temp_repo('view')802 defer { os.rmdir_all(dir) or {} }803 init_repo(dir)804 write(dir, 'view.txt', 'one\n')805 assert run(dir, 'save', 'plain save').code == 0806 assert run(dir, 'checkpoint', 'checkpoint save').code == 0807 line := run(dir, 'view')808 assert line.code == 0809 assert line.out.contains('plain save')810 assert line.out.contains('checkpoint save')811 checkpoints := run(dir, 'view', '--checkpoints')812 assert checkpoints.code == 0813 assert checkpoints.out.contains('checkpoint save')814 assert !checkpoints.out.contains('plain save')815 json_line := run(dir, 'view', '--json')816 assert json_line.code == 0817 assert json_line.out.contains('"is_checkpoint":false')818 assert json_line.out.contains('"is_checkpoint":true')819 json_checkpoints := run(dir, 'view', '--checkpoints', '--json')820 assert json_checkpoints.code == 0821 assert json_checkpoints.out.contains('"is_checkpoint":true')822 assert !json_checkpoints.out.contains('"is_checkpoint":false')823 824 825 assert run(dir, 'view', 'extra').code == 2826 for gone in ['log', 'saves', 'checkpoints'] {827 refused := run(dir, gone)828 assert refused.code == 2829 assert refused.err.contains('unknown command "${gone}"')830 }831 832 assert line.out.split_into_lines().len == json_line.out.count('"id":"')833}834835fn test_remote_configuration_is_isolated_and_atomic() {836 dir := temp_repo('remote')837 home := temp_repo('remote-home')838 defer {839 os.rmdir_all(dir) or {}840 os.rmdir_all(home) or {}841 }842 empty := run_home(dir, home, 'remote', '--json')843 assert empty.code == 0844 assert empty.out.contains('"configured":false')845 target := os.join_path(home, 'remote-store')846 set := run_home(dir, home, 'remote', 'folder', target)847 assert set.code == 0848 assert os.is_dir(target)849 state := run_home(dir, home, 'remote', '--json')850 assert state.code == 0851 assert state.out.contains('"kind":"folder"')852 assert state.out.contains(target)853 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').code == 1854 assert run_home(dir, home, 'remote', '--json').out.contains(target)855 assert run_home(dir, home, 'remote', 's3', 'http://127.0.0.1:1', 'bucket', 'region').err.contains('KOH_ACCESS_KEY')856 assert run_home(dir, home, 'remote', '--forget').code == 0857 assert run_home(dir, home, 'remote', '--json').out.contains('"configured":false')858}859860861862fn test_remote_written_by_hand() {863 dir := temp_repo('remote-by-hand')864 home := temp_repo('remote-by-hand-home')865 defer {866 os.rmdir_all(dir) or {}867 os.rmdir_all(home) or {}868 }869 config := os.join_path(home, '.config', 'koh')870 os.mkdir_all(config) or { panic(err) }871 target := os.join_path(home, 'remote-store')872 os.mkdir_all(target) or { panic(err) }873 os.write_file(os.join_path(config, 'remote'), 'kind = folder\npath = ${target}\n') or {874 panic(err)875 }876 assert run_home(dir, home, 'init').code == 0877 os.write_file(os.join_path(dir, 'a.txt'), 'a\n') or { panic(err) }878 assert run_home(dir, home, 'save', 'first').code == 0879 offered := run_home(dir, home, 'offer', '--json')880 assert offered.code == 0881 assert offered.out.contains('"head"')882883 884 os.write_file(os.join_path(config, 'remote'), 'kind=s3\nendpoint=http://127.0.0.1:1\nbucket=b\nregion=r\n') or {885 panic(err)886 }887 credentials := os.join_path(config, 'credentials')888 os.write_file(credentials, 'access_key=AK\nsecret_key=SK\n') or { panic(err) }889 os.chmod(credentials, 0o644) or { panic(err) }890 loose := run_home(dir, home, 'offer')891 assert loose.code == 1892 assert loose.err.contains('chmod 600')893894 895 os.write_file(os.join_path(config, 'remote'), 'kind=ftp\n') or { panic(err) }896 unreadable := run_home(dir, home, 'offer')897 assert unreadable.code == 1898 assert unreadable.err.contains('not readable')899 assert run_home(dir, home, 'remote', 'folder', target).code == 0900 assert run_home(dir, home, 'offer').code == 0901}902903904905fn test_a_renamed_clone_keeps_offering_to_its_repo() {906 syncroot := temp_repo('rename-clone')907 home := temp_repo('rename-clone-home')908 defer {909 os.rmdir_all(syncroot) or {}910 os.rmdir_all(home) or {}911 }912 remote := os.join_path(syncroot, 'remote')913 a := os.join_path(syncroot, 'A', 'proj')914 b_parent := os.join_path(syncroot, 'B')915 os.mkdir_all(a) or { panic(err) }916 os.mkdir_all(b_parent) or { panic(err) }917 init_repo(a)918 write(a, 'a.txt', 'from A\n')919 assert run_home(a, home, 'remote', 'folder', remote).code == 0920 assert run_home(a, home, 'save', 'A first').code == 0921 assert run_home(b_parent, home, 'steal', 'proj').code == 0922 renamed := os.join_path(b_parent, 'renamed')923 os.mv(os.join_path(b_parent, 'proj'), renamed) or { panic(err) }924 write(renamed, 'b.txt', 'from B\n')925 assert run_home(renamed, home, 'save', 'B first').code == 0926 assert !os.exists(os.join_path(remote, 'renamed'))927 assert run_home(a, home, 'steal').code == 0928 assert os.read_file(os.join_path(a, 'b.txt'))! == 'from B\n'929}930931932933fn test_clone_of_a_damaged_remote_leaves_nothing() {934 syncroot := temp_repo('damaged')935 home := temp_repo('damaged-home')936 defer {937 os.rmdir_all(syncroot) or {}938 os.rmdir_all(home) or {}939 }940 remote := os.join_path(syncroot, 'remote')941 a := os.join_path(syncroot, 'A', 'proj')942 b_parent := os.join_path(syncroot, 'B')943 os.mkdir_all(a) or { panic(err) }944 os.mkdir_all(b_parent) or { panic(err) }945 init_repo(a)946 write(a, 'a.txt', 'from A\n')947 assert run_home(a, home, 'remote', 'folder', remote).code == 0948 assert run_home(a, home, 'save', 'A first').code == 0949 objects := os.join_path(remote, 'proj', 'objects')950 for name in os.ls(objects) or { panic(err) } {951 os.chmod(os.join_path(objects, name), 0o644) or {}952 os.write_file(os.join_path(objects, name), 'junk') or { panic(err) }953 }954 clone := run_home(b_parent, home, 'steal', 'proj')955 assert clone.code == 1956 assert clone.err.contains('is damaged')957 assert !os.exists(os.join_path(b_parent, 'proj'))958}959960fn test_offer_steal_folder_remote() {961 syncroot := temp_repo('sync')962 home := temp_repo('sync-home')963 defer {964 os.rmdir_all(syncroot) or {}965 os.rmdir_all(home) or {}966 }967 remote := os.join_path(syncroot, 'remote')968 a := os.join_path(syncroot, 'A', 'proj')969 b_parent := os.join_path(syncroot, 'B')970 os.mkdir_all(a) or { panic(err) }971 os.mkdir_all(b_parent) or { panic(err) }972 init_repo(a)973 write(a, 'a.txt', 'from A\n')974 assert run_home(a, home, 'remote', 'folder', remote).code == 0975 assert run_home(a, home, 'save', 'A first').code == 0976 clone := run_home(b_parent, home, 'steal', 'proj')977 assert clone.code == 0978 b := os.join_path(b_parent, 'proj')979 assert os.read_file(os.join_path(b, 'a.txt'))! == 'from A\n'980 write(b, 'b.txt', 'from B\n')981 assert run_home(b, home, 'save', 'B first').code == 0982 write(a, 'a2.txt', 'second\n')983 assert run_home(a, home, 'save', 'A second').code == 0984 assert run_home(a, home, 'steal').code == 0985 assert os.exists(os.join_path(a, 'b.txt'))986 assert run_home(b_parent, home, 'steal', '--list').out.contains('proj')987 write(a, 'clash.txt', 'base\n')988 assert run_home(a, home, 'save', 'clash base').code == 0989 assert run_home(b, home, 'steal').code == 0990 write(a, 'clash.txt', 'A version\n')991 assert run_home(a, home, 'save', 'A clash').code == 0992 write(b, 'clash.txt', 'B version\n')993 assert run_home(b, home, 'save', 'B clash').code == 0994 refused := run_home(b, home, 'steal')995 assert refused.code == 1996 assert refused.err.contains('--mine') || refused.err.contains('--theirs')997 b_head := run_home(b, home, 'status', '--json').out.all_after('"head_id":"')[..12]998 theirs := run_home(b, home, 'steal', '--theirs', '--json')999 assert theirs.code == 01000 1001 1002 assert theirs.out.contains('"previous_head":"${b_head}"')1003 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'1004 assert run_home(b, home, 'load', b_head, '--yes').code == 01005 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B version\n'1006 assert run_home(b, home, 'load', theirs.out.all_after('"head":"')[..12], '--yes').code == 01007 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'A version\n'1008 write(a, 'clash.txt', 'A second\n')1009 assert run_home(a, home, 'save', 'A second clash').code == 01010 write(b, 'clash.txt', 'B second\n')1011 assert run_home(b, home, 'save', 'B second clash').code == 01012 assert run_home(b, home, 'steal', '--mine').code == 01013 assert os.read_file(os.join_path(b, 'clash.txt'))! == 'B second\n'1014 assert run_home(b, home, 'view').out.contains('combine: kept mine for')1015}1016101710181019fn test_a_clone_reports_flagged_saves() {1020 syncroot := temp_repo('clone-flags')1021 home := temp_repo('clone-flags-home')1022 defer {1023 os.rmdir_all(syncroot) or {}1024 os.rmdir_all(home) or {}1025 }1026 a := os.join_path(syncroot, 'A', 'proj')1027 os.mkdir_all(a) or { panic(err) }1028 init_repo(a)1029 assert run_home(a, home, 'remote', 'folder', os.join_path(syncroot, 'remote')).code == 01030 write(a, 'a.txt', 'one\n')1031 id := run_home(a, home, 'save', 'risky', '--json').out.all_after('"id":"')[..12]1032 assert run_home(a, home, 'flag', id, '--reason', 'broke the build').code == 01033 assert run_home(a, home, 'offer').code == 01034 for name in ['B', 'C'] {1035 os.mkdir_all(os.join_path(syncroot, name)) or { panic(err) }1036 }1037 json_clone := run_home(os.join_path(syncroot, 'B'), home, 'steal', 'proj', '--json')1038 assert json_clone.code == 01039 assert json_clone.out.contains('"flagged":["${id}"]')1040 text_clone := run_home(os.join_path(syncroot, 'C'), home, 'steal', 'proj')1041 assert text_clone.code == 01042 assert text_clone.out.contains('[${id}] came in flagged — broke the build')1043}10441045104610471048fn test_an_unmounted_folder_remote_is_not_recreated() {1049 syncroot := temp_repo('unmounted')1050 home := temp_repo('unmounted-home')1051 defer {1052 os.rmdir_all(syncroot) or {}1053 os.rmdir_all(home) or {}1054 }1055 mount := os.join_path(syncroot, 'nas')1056 remote := os.join_path(mount, 'koh')1057 a := os.join_path(syncroot, 'proj')1058 os.mkdir_all(a) or { panic(err) }1059 init_repo(a)1060 assert run_home(a, home, 'remote', 'folder', remote).code == 01061 write(a, 'a.txt', 'one\n')1062 assert run_home(a, home, 'save', 'first').code == 01063 os.rename(mount, os.join_path(syncroot, 'nas-unplugged')) or { panic(err) }1064 os.mkdir(mount) or { panic(err) }1065 offered := run_home(a, home, 'offer')1066 assert offered.code == 11067 assert offered.err.contains('mounted')1068 assert !os.exists(remote)1069 write(a, 'a.txt', 'two\n')1070 saved := run_home(a, home, 'save', 'second')1071 assert saved.code == 01072 assert saved.out.contains('mounted')1073 assert !os.exists(remote)1074}10751076fn test_man_and_koh2_handoff() {1077 dir := temp_repo('man')1078 home := temp_repo('man-home')1079 defer {1080 os.rmdir_all(dir) or {}1081 os.rmdir_all(home) or {}1082 }1083 whole := run_home(dir, home, 'man')1084 assert whole.code == 01085 assert whole.out.contains('NAME')1086 assert whole.out.contains('COMMANDS')1087 assert !whole.out.contains('\x1b') && !whole.out.contains('◆')1088 strip := run_home(dir, home, 'man', 'strip')1089 assert strip.code == 01090 assert strip.out.contains('strip')1091 assert !strip.out.contains('koh save ')1092 steal := run_home(dir, home, 'man', 'steal').out1093 assert steal.starts_with(' koh steal') && steal.contains('koh steal my-project')1094 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."1095 assert run_home(dir, home, 'man', 'nonsense').code == 21096 legacy := temp_repo('legacy')1097 defer { os.rmdir_all(legacy) or {} }1098 os.mkdir_all(os.join_path(legacy, '.koh')) or { panic(err) }1099 fake := os.join_path(home, 'bin')1100 os.mkdir_all(fake) or { panic(err) }1101 fake_koh := os.join_path(fake, 'koh2')1102 os.write_file(fake_koh, '#!/bin/sh\nprintf "handoff:%s\\n" "$*"\nexit 7\n') or { panic(err) }1103 os.chmod(fake_koh, 0o755) or { panic(err) }1104 handoff := run_home_with_path(legacy, home, fake, 'status', '--json')1105 assert handoff.code == 71106 assert handoff.out.contains('handoff:status --json')1107 1108 assert handoff.err.contains('Koh 2 repo')1109 assert handoff.err.trim_space().split_into_lines().len == 11110 assert handoff.out.trim_space() == 'handoff:status --json'1111}11121113fn run_home_with_path(repo string, home string, bin_dir string, args ...string) Result {1114 stamp := '${os.getpid()}-${time.now().unix_nano()}'1115 base := os.join_path(os.temp_dir(), 'koh3-cli-${stamp}')1116 out_path := '${base}.out'1117 err_path := '${base}.err'1118 code_path := '${base}.code'1119 mut command := 'HOME=${q(home)} PATH=${q(bin_dir + ':' + os.getenv('PATH'))} ${q(binary)}'1120 for arg in args {1121 command += ' ${q(arg)}'1122 }1123 shell := 'cd ${q(repo)} && ${command} </dev/null >${q(out_path)} 2>${q(err_path)}; printf "%s" \$? >${q(code_path)}'1124 result := os.execute(shell)1125 if result.exit_code != 0 {1126 panic('could not run test command: ${result.output}')1127 }1128 code := os.read_file(code_path) or { panic(err) }1129 out := os.read_file(out_path) or { '' }1130 err := os.read_file(err_path) or { '' }1131 os.rm(out_path) or {}1132 os.rm(err_path) or {}1133 os.rm(code_path) or {}1134 return Result{1135 code: code.int()1136 out: out1137 err: err1138 }1139}1140114111421143fn koh2_binary() string {1144 return os.find_abs_path_of_executable('koh') or { '' }1145}1146114711481149fn run_koh2(dir string, command string) {1150 os.execute('cd ${q(dir)} && ${command} >/dev/null 2>&1')1151}11521153fn save_ids(dir string) []string {1154 mut ids := []string{}1155 for name in os.ls(os.join_path(dir, '.koh', 'snapshots')) or { []string{} } {1156 if name.ends_with('.json') {1157 ids << name.trim_right('.json')1158 }1159 }1160 ids.sort()1161 return ids1162}11631164fn test_migrate_converts_a_koh2_repo_and_keeps_every_save() {1165 koh2 := koh2_binary()1166 if koh2 == '' {1167 println('skipping the migration tests: koh (Koh 2) is not on PATH')1168 return1169 }1170 dir := temp_repo('migrate')1171 defer { os.rmdir_all(dir) or {} }1172 home := temp_repo('migrate-home')1173 defer { os.rmdir_all(home) or {} }1174 run_koh2(dir, '${q(koh2)} init')1175 1176 write(dir, 'code.txt', 'x'.repeat(700))1177 write(dir, 'tiny.txt', 'tiny\n')1178 run_koh2(dir, '${q(koh2)} save "first save"')1179 run_koh2(dir, '${q(koh2)} checkpoint "before something"')1180 write(dir, 'code.txt', 'y'.repeat(700))1181 run_koh2(dir, '${q(koh2)} save "second save"')1182 original := save_ids(dir)1183 assert original.len == 311841185 report := run_home(dir, home, 'migrate')1186 assert report.code == 01187 assert report.out.contains('migrated 3 saves')1188 1189 assert os.is_file(os.join_path(dir, '.koh'))1190 assert os.is_dir(os.join_path(dir, '.koh-v2'))1191 assert os.is_file(os.join_path(dir, '.kohignore'))1192 assert !os.exists(os.join_path(dir, '.koh-migrate'))11931194 1195 listed := run_home(dir, home, 'view', '--json')1196 assert listed.code == 01197 for id in original {1198 assert listed.out.contains(id)1199 }1200 assert run_home(dir, home, 'doctor').code == 01201 assert run_home(dir, home, 'status').code == 012021203 1204 assert run_home(dir, home, 'migrate').code == 11205}12061207fn test_migrate_refuses_a_tree_that_does_not_match_the_last_save() {1208 koh2 := koh2_binary()1209 if koh2 == '' {1210 return1211 }1212 dir := temp_repo('migrate-dirty')1213 defer { os.rmdir_all(dir) or {} }1214 home := temp_repo('migrate-dirty-home')1215 defer { os.rmdir_all(home) or {} }1216 run_koh2(dir, '${q(koh2)} init')1217 write(dir, 'code.txt', 'x'.repeat(700))1218 run_koh2(dir, '${q(koh2)} save "only save"')1219 write(dir, 'code.txt', 'unsaved change\n')12201221 refused := run_home(dir, home, 'migrate')1222 assert refused.code == 11223 assert refused.err.contains('save with koh2 first')1224 1225 assert os.is_dir(os.join_path(dir, '.koh'))1226 assert !os.exists(os.join_path(dir, '.koh-v2'))1227 assert !os.exists(os.join_path(dir, '.koh-migrate'))1228}12291230123112321233fn test_migrate_passes_over_the_files_koh2_never_saved() {1234 koh2 := koh2_binary()1235 if koh2 == '' {1236 return1237 }1238 dir := temp_repo('migrate-koh2-ignores')1239 defer { os.rmdir_all(dir) or {} }1240 home := temp_repo('migrate-koh2-ignores-home')1241 defer { os.rmdir_all(home) or {} }1242 run_koh2(dir, '${q(koh2)} init')1243 write(dir, 'code.txt', 'x'.repeat(700))1244 run_koh2(dir, '${q(koh2)} save "only save"')1245 write(dir, '.kohconfig', 'stable_days = 14\n')1246 write(dir, '.kohignore.public', 'private/\n')1247 write(dir, 'lib/app.bundle', 'bundle\n')1248 write(dir, 'ios/Settings.bundle/Root.plist', 'settings\n')1249 1250 1251 1252 write(dir, 'native/t.dSYM/Info.plist', 'plist\n')1253 run_koh2(dir, '${q(koh2)} save --agent test --session test "add the dsym"')1254 1255 ignore_path := os.join_path(dir, '.koh', 'ignore')1256 rules := os.read_file(ignore_path)!.split_into_lines().filter(it != '*.bundle')1257 os.write_file(ignore_path, rules.join('\n') + '\nnative/*.dSYM\n')!1258 1259 write(dir, 'native/t.dSYM/Info.plist', 'changed\n')1260 changed := run_home(dir, home, 'migrate')1261 assert changed.code == 11262 assert changed.err.contains('native/t.dSYM/Info.plist does not match')1263 write(dir, 'native/t.dSYM/Info.plist', 'plist\n')1264 assert os.execute('cd ${q(dir)} && ${q(koh2)} status --json').output.contains('"dirty":false')12651266 migrated := run_home(dir, home, 'migrate')1267 assert migrated.code == 0, migrated.err1268 1269 assert os.read_file(os.join_path(dir, '.kohconfig'))! == 'stable_days = 14\n'1270}12711272struct FakeKepr {1273 no_access bool1274 1275 fail string1276 1277 redirect_to string1278mut:1279 objects map[string][]u81280 1281 seen map[string]string1282}12831284fn (mut h FakeKepr) handle(req http.Request) http.Response {1285 if req.method == .post && req.url == '/api/connect' {1286 h.seen['code'] = req.data1287 if h.redirect_to != '' {1288 return http.Response{1289 status_code: 3071290 status_msg: 'Temporary Redirect'1291 header: http.new_header(1292 key: .location1293 value: h.redirect_to1294 )1295 }1296 }1297 if h.fail == '500' {1298 return http.Response{ status_code: 500, status_msg: 'Internal Server Error' }1299 }1300 if h.fail == 'garbage' {1301 return http.Response{ status_code: 200, status_msg: 'OK', body: 'not json at all' }1302 }1303 if h.fail == 'incomplete' {1304 return http.Response{ status_code: 200, status_msg: 'OK', body: '{"endpoint":"http://x","bucket":"","region":"","access_key_id":"","secret_access_key":""}' }1305 }1306 if req.data == 'code=good' && !h.no_access {1307 return http.Response{ status_code: 200, status_msg: 'OK', body: '{"endpoint":"http://${req.header.get_custom('Host', http.HeaderQueryConfig{}) or { '' }}","bucket":"bucket","region":"test","access_key_id":"access","secret_access_key":"secret"}' }1308 }1309 return http.Response{ status_code: 404, status_msg: 'Not Found', body: '{"error":"no_code","message":"no code"}' }1310 }1311 if req.method == .get && req.url.starts_with('/connect') {1312 h.seen['connect'] = req.url1313 return http.Response{ status_code: if h.no_access { 404 } else { 302 }, status_msg: 'Found' }1314 }1315 key := req.url1316 if key.starts_with('/bucket/') {1317 match req.method {1318 .put {1319 if req.header.get_custom('If-None-Match', http.HeaderQueryConfig{}) or { '' } == '*' && key in h.objects {1320 return http.Response{ status_code: 412, status_msg: 'Precondition Failed' }1321 }1322 h.objects[key] = req.data.bytes()1323 return http.Response{ status_code: 200, status_msg: 'OK' }1324 }1325 .get {1326 if body := h.objects[key] {1327 return http.Response{1328 status_code: 2001329 status_msg: 'OK'1330 body: body.bytestr()1331 header: http.new_header(1332 key: .etag1333 value: '"tag"'1334 )1335 }1336 }1337 return http.Response{ status_code: 404, status_msg: 'Not Found' }1338 }1339 .delete {1340 h.objects.delete(key)1341 return http.Response{ status_code: 204, status_msg: 'No Content' }1342 }1343 else {}1344 }1345 }1346 return http.Response{ status_code: 404, status_msg: 'Not Found' }1347}13481349fn start_fake_kepr(no_access bool, fail string, redirect_to string) !(int, &http.Server, &FakeKepr) {1350 for port in 19401 .. 19411 {1351 listener := net.listen_tcp(.ip, '127.0.0.1:${port}') or { continue }1352 fake := &FakeKepr{1353 no_access: no_access1354 fail: fail1355 redirect_to: redirect_to1356 seen: map[string]string{}1357 }1358 mut server := &http.Server{1359 listener: listener1360 handler: fake1361 show_startup_message: false1362 }1363 spawn server.listen_and_serve()1364 server.wait_till_running()!1365 return port, server, fake1366 }1367 return error('no free fake Kepr port')1368}136913701371fn test_steal_refuses_an_address_that_is_not_one() {1372 outside := temp_repo('steal-address')1373 home := temp_repo('steal-address-home')1374 defer { os.rmdir_all(outside) or {} }1375 defer { os.rmdir_all(home) or {} }1376 for bad in ['kepr.uk/', '/shipped', 'kepr.uk/'] {1377 refused := run_home(outside, home, 'steal', bad)1378 assert refused.code == 1, bad1379 assert refused.err.contains('is not <host>/<repo>'), '${bad}: ${refused.err}'1380 }1381 assert run_home(outside, home, 'steal', 'one', 'two').code == 21382 assert run_home(outside, home, 'steal', '--list', 'extra').code == 21383}13841385fn test_remote_kepr_saves_and_refuses_without_changes() {1386 repo := temp_repo('kepr')1387 home := temp_repo('kepr-home')1388 defer { os.rmdir_all(repo) or {} }1389 defer { os.rmdir_all(home) or {} }1390 init_repo(repo)1391 port, mut server, fake := start_fake_kepr(false, '', '')!1392 defer { server.stop() }1393 host := 'http://127.0.0.1:${port}'1394 1395 good := run_home(repo, home, 'remote', '${host}/', '--code', 'good')1396 assert good.code == 0, good.err1397 remote_path := os.join_path(home, '.config', 'koh', 'remote')1398 credentials_path := os.join_path(home, '.config', 'koh', 'credentials')1399 assert os.read_file(remote_path)! == '# written by koh remote — the remote for every repo on this machine\nkind=s3\nendpoint=${host}\nbucket=bucket\nregion=test\nkepr=${host}\n'1400 assert os.read_file(credentials_path)! == '# written by koh remote — keep this file private\naccess_key=access\nsecret_key=secret\n'1401 1402 assert (fake.seen['code'] or { '' }) == 'code=good'1403 assert (fake.seen['connect'] or { '' }) == ''1404 1405 before := os.read_file(remote_path)!1406 before_keys := os.read_file(credentials_path)!1407 bad := run_home(repo, home, 'remote', host, '--code', 'bad')1408 assert bad.code == 11409 assert bad.err.contains('wrong or has expired')1410 1411 1412 1413 1414 assert (fake.seen['connect'] or { '' }).contains('machine=${os.hostname() or { 'unknown' }}')1415 no_terminal := run_home(repo, home, 'remote', host)1416 assert no_terminal.code == 11417 assert no_terminal.err.contains('needs a terminal')1418 assert os.read_file(remote_path)! == before1419 assert os.read_file(credentials_path)! == before_keys1420}14211422fn test_remote_kepr_refuses_hosts_that_are_not_hosts() {1423 repo := temp_repo('kepr-host')1424 home := temp_repo('kepr-host-home')1425 defer { os.rmdir_all(repo) or {} }1426 defer { os.rmdir_all(home) or {} }1427 init_repo(repo)1428 for host in ['kepr.uk/foo', 'https://kepr.uk?x=1', 'kepr.uk#frag', '[email protected]', 'kepr uk',1429 'http://127.0.0.1:1/foo', 'https://'] {1430 refused := run_home(repo, home, 'remote', host, '--code', 'good')1431 assert refused.code == 1, host1432 assert refused.err.contains('is not a Kepr address'), '${host}: ${refused.err}'1433 }1434 1435 assert run_home(repo, home, 'remote', 's3', 'https://x', 'b', 'r', '--code', 'good').code == 21436 assert run_home(repo, home, 'remote', 'folder', '/tmp', '--code', 'good').code == 21437 assert run_home(repo, home, 'remote', '--code', 'good').code == 21438 assert !os.exists(os.join_path(home, '.config', 'koh', 'remote'))1439}14401441fn test_remote_kepr_refuses_what_is_not_a_code_reply() {1442 repo := temp_repo('kepr-reply')1443 home := temp_repo('kepr-reply-home')1444 defer { os.rmdir_all(repo) or {} }1445 defer { os.rmdir_all(home) or {} }1446 init_repo(repo)1447 remote_path := os.join_path(home, '.config', 'koh', 'remote')1448 for mode in ['500', 'garbage', 'incomplete'] {1449 port, mut server, _ := start_fake_kepr(false, mode, '')!1450 refused := run_home(repo, home, 'remote', 'http://127.0.0.1:${port}', '--code', 'good')1451 server.stop()1452 assert refused.code == 1, mode1453 assert refused.err.contains('nothing was changed'), '${mode}: ${refused.err}'1454 assert !os.exists(remote_path)1455 }1456 1457 target_port, mut target, target_fake := start_fake_kepr(false, '', '')!1458 port, mut server, _ := start_fake_kepr(false, '', 'http://127.0.0.1:${target_port}/api/connect')!1459 redirected := run_home(repo, home, 'remote', 'http://127.0.0.1:${port}', '--code', 'good')1460 server.stop()1461 target.stop()1462 assert redirected.code == 11463 assert redirected.err.contains('refused the code')1464 assert target_fake.seen.len == 0, 'the code was forwarded: ${target_fake.seen}'1465 assert !os.exists(remote_path)1466}146714681469fn test_remote_kepr_that_cannot_be_reached_changes_nothing() {1470 repo := temp_repo('kepr-down')1471 home := temp_repo('kepr-down-home')1472 defer { os.rmdir_all(repo) or {} }1473 defer { os.rmdir_all(home) or {} }1474 init_repo(repo)1475 refused := run_home(repo, home, 'remote', 'http://127.0.0.1:1', '--code', 'good')1476 assert refused.code == 11477 assert refused.err.contains('could not reach Kepr')1478 assert !os.exists(os.join_path(home, '.config', 'koh', 'remote'))1479}14801481fn test_remote_kepr_without_bucket_access_is_explained() {1482 repo := temp_repo('kepr-no-access')1483 home := temp_repo('kepr-no-access-home')1484 defer { os.rmdir_all(repo) or {} }1485 defer { os.rmdir_all(home) or {} }1486 init_repo(repo)1487 port, mut server, _ := start_fake_kepr(true, '', '')!1488 defer { server.stop() }1489 r := run_home(repo, home, 'remote', 'http://127.0.0.1:${port}', '--code', 'bad')1490 assert r.code == 11491 assert r.err.contains("doesn't hand out bucket access")1492 assert !os.exists(os.join_path(home, '.config', 'koh', 'remote'))1493}