"""Test Models.company."""


from aki_prj23_transparenzregister.models.company import Company, CompanyID, Location


def test_to_dict() -> None:
    """Tests if the version tag is entered."""
    company_id = CompanyID("The Shire", "420")
    location = Location(
        city="Insmouth", house_number="19", street="Harbor", zip_code="1890"
    )
    company = Company(
        id=company_id,
        last_update="Tomorrow",
        location=location,
        name="BLANK GmbH",
        relationships=[],
    )

    assert company.to_dict() == {
        "id": {
            "district_court": company_id.district_court,
            "hr_number": company_id.hr_number,
        },
        "last_update": company.last_update,
        "location": {
            "city": location.city,
            "house_number": location.house_number,
            "street": location.street,
            "zip_code": location.zip_code,
        },
        "name": "BLANK GmbH",
        "relationships": [],
    }