Create integration_v2.py

tests/integration_v2.py

@@ -0,0 +1,103 @@
+# tests/integration_v2.py
+"""
+Integration sanity check for AI E-Consult V2 core layers.
+Run with: python -m tests.integration_v2
+"""
+
+import sys
+import json
+from pathlib import Path
+from typing import Dict, Any
+
+# Ensure repo root on path
+REPO_ROOT = Path(__file__).resolve().parents[1]
+if str(REPO_ROOT) not in sys.path:
+    sys.path.insert(0, str(REPO_ROOT))
+
+from src import config, billing, modal_templates, store  # noqa: E402
+
+RESULTS = {"passed": 0, "failed": 0}
+
+def _ok(name: str):
+    RESULTS["passed"] += 1
+    print(f"[PASS] {name}")
+
+def _fail(name: str, e: Exception):
+    RESULTS["failed"] += 1
+    print(f"[FAIL] {name}: {e}")
+
+def check(name: str):
+    def deco(fn):
+        def wrapper():
+            try:
+                fn()
+                _ok(name)
+            except Exception as e:
+                _fail(name, e)
+        return wrapper
+    return deco
+
+@check("seed_cases_and_v2_schema")
+def test_seed_cases_and_v2_schema():
+    created = store.seed_cases(reset=True)
+    assert len(created) == 4
+    for cid in created:
+        case = store.read_case(cid)
+        assert case and case["schema_version"] == config.SCHEMA_VERSION
+        for k in ["status", "created_at", "updated_at", "billing", "explainability"]:
+            assert k in case
+
+@check("load_sample_case")
+def test_load_sample_case():
+    items = store.list_cases()
+    assert items, "No cases after seeding"
+    case = store.read_case(items[0]["case_id"])
+    assert case is not None
+    print("  case_id:", case["case_id"], "status:", case["status"])
+
+@check("billing_autosuggest_representative")
+def test_billing_autosuggest_representative():
+    combos = [(5, False), (12, True), (35, True)]
+    for minutes, spoke in combos:
+        sugg = billing.autosuggest_cpt(minutes=minutes, spoke=spoke)
+        assert isinstance(sugg, list) and len(sugg) >= 1
+        codes = [s.code for s in sugg]
+        print(f"  minutes={minutes} spoke={spoke} -> {codes} (eligible={[s.code for s in sugg if s.eligible]})")
+
+@check("modal_previews_safe_without_streamlit")
+def test_modal_previews_safe_without_streamlit():
+    # Build a simple note and claim, then render via modal helpers.
+    cases = store.list_cases()
+    case = store.read_case(cases[0]["case_id"])
+    assert case
+    note_md = f"# Consult Note for {case['patient']['name']}\n\nGenerated for integration preview."
+    pick = billing.autosuggest_cpt(minutes=12, spoke=True)
+    code = next((s.code for s in pick if s.eligible), "99447")
+    rate = next((s.rate for s in pick if s.code == code), 55.0)
+    claim = billing.build_837_claim(case, code=code, rate=rate, minutes=12, spoke=True, attested=True)
+    # These calls are safe even if Streamlit is missing—helpers no-op or use fallbacks
+    modal_templates.show_consult_note_preview(note_md)
+    modal_templates.show_837_claim_preview(claim)
+
+@check("export_filename_determinism")
+def test_export_filename_determinism():
+    p1 = config.make_export_path("INTEG", "consult_note.md")
+    p2 = config.make_export_path("INTEG", "claim_837.json")
+    assert p1.parent.exists() and p2.parent.exists()
+    assert p1.name.startswith("EC-INTEG_") and p1.name.endswith("_consult_note.md")
+    assert p2.name.startswith("EC-INTEG_") and p2.name.endswith("_claim_837.json")
+
+def main():
+    test_seed_cases_and_v2_schema()
+    test_load_sample_case()
+    test_billing_autosuggest_representative()
+    test_modal_previews_safe_without_streamlit()
+    test_export_filename_determinism()
+
+    total = RESULTS["passed"] + RESULTS["failed"]
+    print(f"\n=== Integration Summary: {RESULTS['passed']} passed, {RESULTS['failed']} failed, {total} total ===")
+    if RESULTS["failed"]:
+        raise SystemExit(1)
+
+if __name__ == "__main__":
+    main()