def test_domain_and_domain_id_property_methods() -> None: """Unit test for Address.domain and Address.domain_id methods""" # Test getter domain = SpecificLocation(id=UID()) address_with_domain = Address( network=SpecificLocation(id=UID()), domain=domain, device=SpecificLocation(id=UID()), vm=SpecificLocation(id=UID()), ) # Test domain getter assert address_with_domain.domain == domain # Test domain setter an_id = UID(value=uuid.UUID(int=333779996850170035686993356951732753684)) new_domain = SpecificLocation(id=an_id) address_with_domain.domain = new_domain assert address_with_domain.domain == new_domain # Test domain_id getter address_without_domain = Address( network=SpecificLocation(id=UID()), device=SpecificLocation(id=UID()), vm=SpecificLocation(id=UID()), ) assert address_with_domain.domain_id == an_id assert address_without_domain.domain_id is None
def test_target_id_property_method_with_a_return() -> None: """Unit test for Address.target_id method""" network = SpecificLocation(id=UID()) domain = SpecificLocation(id=UID()) device = SpecificLocation(id=UID()) vm = SpecificLocation(id=UID()) address = Address( network=network, domain=domain, device=device, vm=vm, ) assert address.target_id == vm address.vm = None assert address.target_id == device address.device = None assert address.target_id == domain address.domain = None assert address.target_id == network