tests/test_parser.py

import pytest

from app.common.exceptions import ParserError
from app.models.policy.parser import parse_decision, parse_decision_soft, parse_decision_strict_xml, parse_decision_with_repair


def test_parser_valid_json() -> None:
    payload = """
    {"mode":"REGIMEN_OPT","action_type":"KEEP_REGIMEN","target_drug":null,"replacement_drug":null,
     "dose_bucket":"NA","taper_days":null,"monitoring_plan":null,"candidate_id":"cand_01","confidence":0.7}
    """
    parsed = parse_decision(payload)
    assert parsed.candidate_id == "cand_01"


def test_parser_invalid_json() -> None:
    with pytest.raises(ParserError):
        parse_decision("not-json")


def test_parser_xml_format() -> None:
    payload = """
    <decision>
      <mode>REGIMEN_OPT</mode>
      <action_type>KEEP_REGIMEN</action_type>
      <target_drug>null</target_drug>
      <replacement_drug>null</replacement_drug>
      <dose_bucket>NA</dose_bucket>
      <taper_days>null</taper_days>
      <monitoring_plan>null</monitoring_plan>
      <candidate_id>cand_01</candidate_id>
      <confidence>0.7</confidence>
    </decision>
    """
    parsed = parse_decision(payload)
    assert parsed.candidate_id == "cand_01"


def test_parser_strict_xml_rejects_unknown_field() -> None:
    payload = """
    <decision>
      <mode>REGIMEN_OPT</mode>
      <action_type>KEEP_REGIMEN</action_type>
      <target_drug>null</target_drug>
      <replacement_drug>null</replacement_drug>
      <dose_bucket>NA</dose_bucket>
      <taper_days>null</taper_days>
      <candidate_id>cand_01</candidate_id>
      <confidence>0.7</confidence>
      <unknown>bad</unknown>
    </decision>
    """
    with pytest.raises(ParserError):
        parse_decision_strict_xml(payload)


def test_parser_soft_recovery_from_buffered_xml() -> None:
    payload = "prefix garbage <decision><mode>REGIMEN_OPT</mode><action_type>KEEP_REGIMEN</action_type><target_drug>null</target_drug><replacement_drug>null</replacement_drug><dose_bucket>NA</dose_bucket><taper_days>null</taper_days><candidate_id>cand_01</candidate_id><confidence>0.7</confidence></decision> suffix"
    parsed = parse_decision_soft(payload)
    assert parsed.action_type.value == "KEEP_REGIMEN"


def test_parser_repair_invalid_candidate_and_confidence() -> None:
    payload = "{\"mode\":\"REGIMEN_OPT\",\"action_type\":\"KEEP_REGIMEN\",\"dose_bucket\":\"NA\",\"candidate_id\":\"bad\",\"confidence\":2.4}"
    parsed = parse_decision_with_repair(payload)
    assert parsed.candidate_id.startswith("cand_")
    assert 0.001 <= parsed.confidence <= 0.999