Tyler
111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
"""Tests for the FastAPI surface in ``cyclone.api`` for the 999 endpoint."""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from cyclone.api import app
|
|
from cyclone.store import store
|
|
|
|
ACCEPTED = Path(__file__).parent / "fixtures" / "minimal_999.txt"
|
|
REJECTED = Path(__file__).parent / "fixtures" / "minimal_999_rejected.txt"
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clear_store():
|
|
"""Reset the module-level store before and after each test."""
|
|
with store._lock:
|
|
store._batches.clear()
|
|
yield
|
|
with store._lock:
|
|
store._batches.clear()
|
|
|
|
|
|
@pytest.fixture
|
|
def client() -> TestClient:
|
|
return TestClient(app)
|
|
|
|
|
|
def test_parse_999_endpoint_happy_path(client: TestClient):
|
|
"""Uploading the minimal accepted 999 returns 200 + an acks row."""
|
|
assert len(store.list_acks()) == 0
|
|
text = ACCEPTED.read_text()
|
|
resp = client.post(
|
|
"/api/parse-999",
|
|
files={"file": ("minimal_999.txt", text, "text/plain")},
|
|
headers={"Accept": "application/json"},
|
|
)
|
|
assert resp.status_code == 200, resp.text
|
|
body = resp.json()
|
|
assert "ack" in body
|
|
assert "parsed" in body
|
|
ack = body["ack"]
|
|
assert ack["accepted_count"] == 1
|
|
assert ack["rejected_count"] == 0
|
|
assert ack["received_count"] == 1
|
|
assert ack["ack_code"] == "A"
|
|
# The raw 999 text round-trips.
|
|
assert "ISA*" in ack["raw_999_text"]
|
|
assert "AK9*A*1*1*0~" in ack["raw_999_text"]
|
|
# One row persisted.
|
|
rows = store.list_acks()
|
|
assert len(rows) == 1
|
|
assert rows[0].ack_code == "A"
|
|
|
|
|
|
def test_parse_999_endpoint_invalid_edi_raises_400(client: TestClient):
|
|
"""Garbage input must surface as 400, never 500."""
|
|
resp = client.post(
|
|
"/api/parse-999",
|
|
files={"file": ("garbage.txt", "not edi", "text/plain")},
|
|
headers={"Accept": "application/json"},
|
|
)
|
|
assert resp.status_code == 400, resp.text
|
|
assert "error" in resp.json()
|
|
|
|
|
|
def test_parse_999_endpoint_empty_file_raises_400(client: TestClient):
|
|
"""An empty upload must be rejected with 400 (matches the 835 path)."""
|
|
resp = client.post(
|
|
"/api/parse-999",
|
|
files={"file": ("empty.txt", "", "text/plain")},
|
|
headers={"Accept": "application/json"},
|
|
)
|
|
assert resp.status_code == 400, resp.text
|
|
|
|
|
|
def test_parse_999_persists_acks_row(client: TestClient):
|
|
"""After a parse-999 call, GET /api/acks returns the row."""
|
|
text = REJECTED.read_text()
|
|
resp = client.post(
|
|
"/api/parse-999",
|
|
files={"file": ("rejected_999.txt", text, "text/plain")},
|
|
headers={"Accept": "application/json"},
|
|
)
|
|
assert resp.status_code == 200, resp.text
|
|
# The new GET endpoint must surface the row.
|
|
list_resp = client.get("/api/acks", headers={"Accept": "application/json"})
|
|
assert list_resp.status_code == 200, list_resp.text
|
|
body = list_resp.json()
|
|
assert body["total"] == 1
|
|
assert body["items"][0]["ack_code"] == "R"
|
|
assert body["items"][0]["accepted_count"] == 0
|
|
assert body["items"][0]["rejected_count"] == 1
|
|
|
|
|
|
def test_parse_999_rejected_fixture_surfaces_r_code(client: TestClient):
|
|
"""The rejected 999 fixture must produce an R ack_code."""
|
|
text = REJECTED.read_text()
|
|
resp = client.post(
|
|
"/api/parse-999",
|
|
files={"file": ("rejected_999.txt", text, "text/plain")},
|
|
headers={"Accept": "application/json"},
|
|
)
|
|
assert resp.status_code == 200
|
|
body = resp.json()
|
|
assert body["ack"]["ack_code"] == "R"
|
|
assert body["ack"]["rejected_count"] == 1
|
|
assert body["ack"]["accepted_count"] == 0
|