78일차 정답: LAB-A · VIBE-A-TCHECK · answers(정답 워크플로우 · 실행 기록) · 강사가 만든 tcheck

Last3Mon/78/tcheck/tests/test_cli.py

@@ -0,0 +1,49 @@
+import csv
+import json
+
+from tcheck import cli
+from conftest import DATA
+
+EMP = str(DATA / "employees.csv")
+EMP_SCHEMA = str(DATA / "schemas" / "employee.json")
+
+
+def test_check_plain(capsys):
+    assert cli.main(["check", EMP, "--schema-file", EMP_SCHEMA]) == 0
+    out = capsys.readouterr().out
+    assert "9 rows · valid 4 · invalid 5 · score 0.4444" in out and "bad_format:email" in out
+
+
+def test_check_json_like_my_check(capsys, url, n8n):
+    assert cli.main(["check", str(DATA / "tickets-new.json"), "--url", url, "--json"]) == 0   # schema from the file
+    out = json.loads(capsys.readouterr().out)
+    assert {k: out[k] for k in ("schema", "batch", "valid", "rows", "codes")} == {k: n8n["ticket"][k] for k in ("schema", "batch", "valid", "rows", "codes")}
+
+
+def test_out_and_rejects_files(tmp_path):
+    clean, rejects = tmp_path / "clean.csv", tmp_path / "rejects.csv"
+    assert cli.main(["check", EMP, "--schema-file", EMP_SCHEMA, "--out", str(clean), "--rejects", str(rejects)]) == 0
+    assert len(list(csv.DictReader(clean.open()))) == 4
+    bad = list(csv.DictReader(rejects.open()))
+    assert len(bad) == 5 and bad[0]["errors"] == "bad_format:email"
+
+
+def test_gate_exit_codes(capsys):
+    assert cli.main(["gate", EMP, "--schema-file", EMP_SCHEMA]) == 4
+    assert "rejected" in capsys.readouterr().out
+    assert cli.main(["gate", EMP, "--schema-file", EMP_SCHEMA, "--max-invalid", "0.6"]) == 0
+
+
+def test_fix_and_log(capsys, url, api):
+    assert cli.main(["check", EMP, "--schema", "employee", "--url", url, "--batch", "HR-1", "--fix", "--log"]) == 0
+    out = capsys.readouterr().out
+    assert "HR-1: 9줄 중 4줄 통과, 5줄 고칠 것" in out and "email의 형식을 확인해 주세요" in out
+    assert api.log == [{"batch": "HR-1", "dataset": "employee", "total": 9, "valid": 4, "quarantined": 5, "rejected": False,
+                        "score": 0.4444, "codes": api.log[0]["codes"]}]
+
+
+def test_exit_codes_read_and_usage(tmp_path, url, capsys):
+    assert cli.main(["check", str(tmp_path / "none.csv"), "--schema-file", EMP_SCHEMA]) == 1
+    assert cli.main(["check", EMP, "--schema", "nothing", "--url", url]) == 1
+    assert cli.main(["check", EMP]) == 2                                        # which schema?
+    assert cli.main(["nonsense"]) == 2