tests/test_response_mapping.py

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent / "api"))

from app.services.ragbot import RagBotService


def test_format_response_uses_synthesizer_payload():
    service = RagBotService()
    workflow_result = {
        "final_response": {
            "biomarker_flags": [
                {
                    "name": "Glucose",
                    "value": 120,
                    "unit": "mg/dL",
                    "status": "HIGH",
                    "reference_range": "70-100 mg/dL",
                    "warning": None,
                }
            ],
            "safety_alerts": [],
            "key_drivers": [],
            "disease_explanation": {"pathophysiology": "", "citations": [], "retrieved_chunks": None},
            "recommendations": {"immediate_actions": [], "lifestyle_changes": [], "monitoring": []},
            "confidence_assessment": {"prediction_reliability": "LOW", "evidence_strength": "WEAK", "limitations": []},
            "patient_summary": {"narrative": ""},
        },
        "biomarker_flags": [],
        "safety_alerts": [],
    }

    response = service._format_response(
        request_id="req_test",
        workflow_result=workflow_result,
        input_biomarkers={"Glucose": 120},
        extracted_biomarkers=None,
        patient_context={},
        model_prediction={"disease": "Diabetes", "confidence": 0.6, "probabilities": {}},
        processing_time_ms=10.0,
    )

    assert response.analysis.biomarker_flags[0].name == "Glucose"