36 lines
987 B
Python

"""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": [],
}