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
