feat: core data models (Signature, CapturedFlow, EndpointRecord)

2026-05-31 23:51:47 +08:00
parent 422990bc4e
commit d7b4a44673
2 changed files with 98 additions and 0 deletions
@@ -0,0 +1,36 @@
+from auto_reverse.models import CapturedFlow, Signature, status_class
+
+
+def test_status_class_buckets():
+    assert status_class(200) == "2xx"
+    assert status_class(201) == "2xx"
+    assert status_class(404) == "4xx"
+    assert status_class(503) == "5xx"
+
+
+def test_signature_is_hashable_and_equal():
+    a = Signature("GET", "ex.com", "/api/users/{id}", "2xx")
+    b = Signature("GET", "ex.com", "/api/users/{id}", "2xx")
+    assert a == b
+    assert {a, b} == {a}
+
+
+def test_captured_flow_json_body_parsing():
+    flow = CapturedFlow(
+        method="POST", host="ex.com", path="/api/x", query={},
+        req_headers={"content-type": "application/json"}, req_body=b'{"a": 1}',
+        status=201, resp_headers={"content-type": "application/json"},
+        resp_body=b'{"ok": true}', timestamp=0.0,
+    )
+    assert flow.request_json() == {"a": 1}
+    assert flow.response_json() == {"ok": True}
+
+
+def test_captured_flow_non_json_body_returns_none():
+    flow = CapturedFlow(
+        method="GET", host="ex.com", path="/x", query={},
+        req_headers={}, req_body=None, status=200,
+        resp_headers={"content-type": "text/html"}, resp_body=b"<html>",
+        timestamp=0.0,
+    )
+    assert flow.response_json() is None