from backend.ecmk_fetcher.xml_processor import parse_rdsap


SAMPLE_XML = """<RdSAP-Report xmlns="https://epbr.digital.communities.gov.uk/xsd/rdsap">
<Report-Header>
    <Property>
        <Address>
            <Address-Line-1>1</Address-Line-1>
            <Address-Line-2>Fake Avenue</Address-Line-2>
            <Post-Town>Random</Post-Town>
            <Postcode>AB24 5CD</Postcode>
        </Address>
    </Property>
</Report-Header>

<SAP-Data>
    <SAP-Property-Details>
        <Property-Type>0</Property-Type>

        <SAP-Building-Parts>
            <SAP-Building-Part>
                <Identifier>Main Dwelling</Identifier>

                <SAP-Floor-Dimensions>
                    <SAP-Floor-Dimension>
                        <Heat-Loss-Perimeter>25.31</Heat-Loss-Perimeter>
                        <Room-Height>2.46</Room-Height>
                        <Total-Floor-Area>43.61</Total-Floor-Area>
                        <Party-Wall-Length>0</Party-Wall-Length>
                    </SAP-Floor-Dimension>

                    <SAP-Floor-Dimension>
                        <Heat-Loss-Perimeter>26.16</Heat-Loss-Perimeter>
                        <Room-Height>2.44</Room-Height>
                        <Total-Floor-Area>42.33</Total-Floor-Area>
                        <Party-Wall-Length>0</Party-Wall-Length>
                    </SAP-Floor-Dimension>
                </SAP-Floor-Dimensions>

                <Roof-Construction>4</Roof-Construction>
                <Roof-Insulation-Location>2</Roof-Insulation-Location>
                <Roof-Insulation-Thickness>100mm</Roof-Insulation-Thickness>
            </SAP-Building-Part>
        </SAP-Building-Parts>

    </SAP-Property-Details>
</SAP-Data>
</RdSAP-Report>
"""


def test_parse_rdsap_contract():
    # arrange + act
    result = parse_rdsap(SAMPLE_XML)

    # assert
    assert result == {
        "address": "1, Fake Avenue, Random, AB24 5CD",
        "property_type": "House",
        "building_parts": [
            {
                "identifier": "Main Dwelling",
                "floors": [
                    {
                        "area_m2": 43.61,
                        "height_m": 2.46,
                        "heat_loss_perimeter_m": 25.31,
                        "party_wall_length_m": 0.0,
                    },
                    {
                        "area_m2": 42.33,
                        "height_m": 2.44,
                        "heat_loss_perimeter_m": 26.16,
                        "party_wall_length_m": 0.0,
                    },
                ],
                "roof": {
                    "construction": 4,
                    "insulation_location": 2,
                    "insulation_thickness_mm": 100.0,
                },
            }
        ],
    }