| """Smoke tests for the shared contracts. If these break, the whole team is blocked.""" |
|
|
| from __future__ import annotations |
|
|
| import pytest |
|
|
| from app.agents.appointment.mocks import fake_booking, fake_communication |
| from app.agents.clerical.mocks import fake_post_discharge_followup, fake_referral_letter |
| from app.agents.discharge.mocks import fake_discharge |
| from app.agents.documentation.mocks import fake_soap_draft |
| from app.agents.handover.mocks import fake_handover |
| from app.agents.roster.mocks import fake_roster |
| from app.contracts import ( |
| BookingResult, |
| CommunicationDraft, |
| DischargeCoordination, |
| HandoverBrief, |
| Priority, |
| RosterResult, |
| SOAPNoteDraft, |
| ) |
|
|
|
|
| def test_soap_mock_validates() -> None: |
| d = fake_soap_draft() |
| assert isinstance(d, SOAPNoteDraft) |
| assert d.soap.subjective |
| assert d.icd10_suggestions |
| assert all(0.0 <= s.confidence <= 1.0 for s in d.icd10_suggestions) |
| assert d.sources, "every SOAP draft must cite at least one source" |
|
|
|
|
| def test_booking_mock_validates() -> None: |
| b = fake_booking() |
| assert isinstance(b, BookingResult) |
| assert b.reminder_jobs, "booking should schedule at least one reminder" |
|
|
|
|
| def test_communication_mock_validates() -> None: |
| c = fake_communication() |
| assert isinstance(c, CommunicationDraft) |
| assert c.body |
|
|
|
|
| def test_discharge_mock_has_five_subtasks() -> None: |
| d = fake_discharge() |
| assert isinstance(d, DischargeCoordination) |
| assert len(d.subtasks) == 5 |
| names = {s.name for s in d.subtasks} |
| assert names == { |
| "discharge_summary", "pharmacy", "family_notification", |
| "follow_up_appointment", "billing_trigger", |
| } |
|
|
|
|
| def test_roster_mock_validates() -> None: |
| r = fake_roster() |
| assert isinstance(r, RosterResult) |
| assert 0 <= r.fairness_score <= 1 |
| assert r.assignments |
|
|
|
|
| def test_handover_mock_ordered_by_priority() -> None: |
| h = fake_handover() |
| assert isinstance(h, HandoverBrief) |
| rank = {Priority.URGENT: 0, Priority.WATCH: 1, Priority.STABLE: 2} |
| ranks = [rank[p.priority] for p in h.patients] |
| assert ranks == sorted(ranks), "handover must surface urgent patients first" |
|
|
|
|
| def test_clerical_mocks_validate() -> None: |
| assert isinstance(fake_referral_letter(), CommunicationDraft) |
| assert isinstance(fake_post_discharge_followup(), CommunicationDraft) |
|
|
|
|
| def test_contracts_forbid_extra_fields() -> None: |
| with pytest.raises(Exception): |
| |
| SOAPNoteDraft( |
| draft_id="x", patient_id="p", encounter_ts=fake_soap_draft().encounter_ts, |
| soap=fake_soap_draft().soap, model="x", extra_field="nope", |
| ) |
|
|
