def test_device_and_device_id_property_methods() -> None: """Unit test for Address.device and Address.device_id methods""" # Test getter device = SpecificLocation(id=UID()) address_with_device = Address( network=SpecificLocation(id=UID()), domain=SpecificLocation(id=UID()), device=device, vm=SpecificLocation(id=UID()), ) # Test device getter assert address_with_device.device == device # Test device setter an_id = UID(value=uuid.UUID(int=333779996850170035686993356951732753684)) new_device = SpecificLocation(id=an_id) address_with_device.device = new_device assert address_with_device.device == new_device # Test domain_id getter address_without_device = Address( network=SpecificLocation(id=UID()), domain=SpecificLocation(id=UID()), vm=SpecificLocation(id=UID()), ) assert address_with_device.device_id == an_id assert address_without_device.device_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