test_bundle.py

import pytest
from pydantic import ValidationError

from fhir_types.hl7_fhir_r4_core.base import CodeableConcept
from fhir_types.hl7_fhir_r4_core.bundle import Bundle, BundleEntry
from fhir_types.hl7_fhir_r4_core.observation import Observation
from fhir_types.hl7_fhir_r4_core.patient import Patient


def test_bundle_generic_narrows_entry_resources() -> None:
    patient = Patient(id="p-1")
    observation = Observation(id="obs-1", status="final", code=CodeableConcept())

    bundle: Bundle[Patient | Observation] = Bundle(
        type="transaction",
        entry=[
            BundleEntry(resource=patient),
            BundleEntry(resource=observation),
        ],
    )

    observations = [
        e.resource
        for e in (bundle.entry or [])
        if e.resource and e.resource.resourceType == "Observation"
    ]
    assert len(observations) == 1
    assert observations[0].id == "obs-1"


def test_bundle_entry_generic_narrows_resource() -> None:
    patient = Patient(id="p-1")
    entry: BundleEntry[Patient] = BundleEntry(resource=patient)
    resource = entry.resource
    assert resource is not None
    assert resource.resourceType == "Patient"


def test_bundle_without_type_param_is_backwards_compatible() -> None:
    patient = Patient(id="p-1")
    bundle: Bundle = Bundle(
        type="collection",
        entry=[BundleEntry(resource=patient)],
    )
    entry = bundle.entry
    assert entry is not None
    assert len(entry) == 1


def test_bundle_from_json_raises_on_invalid_resource() -> None:
    # Observation requires `status` and `code` — omitting them causes a runtime ValidationError
    bundle_json = """{
        "resourceType": "Bundle",
        "type": "searchset",
        "entry": [{
            "resource": {
                "resourceType": "Observation",
                "id": "obs-1"
            }
        }]
    }"""
    with pytest.raises(ValidationError):
        Bundle.from_json(bundle_json)


def test_bundle_from_json_raises_on_wrong_typed_resource() -> None:
    bundle_json = """{
        "resourceType": "Bundle",
        "type": "searchset",
        "entry": [{
            "resource": {
                "resourceType": "Patient",
                "id": "pt-1"
            }
        }]
    }"""
    with pytest.raises(ValidationError):
        Bundle[Observation].from_json(bundle_json)