49 lines
2.2 KiB
Python
49 lines
2.2 KiB
Python
|
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
|