def test_vm_and_vm_id_property_methods() -> None: """Unit test for Address.vm and Address.vm_id methods""" # Test getter vm = SpecificLocation(id=UID()) address_with_vm = Address( network=SpecificLocation(id=UID()), domain=SpecificLocation(id=UID()), device=SpecificLocation(id=UID()), vm=vm, ) # Test device getter assert address_with_vm.vm == vm # Test device setter an_id = UID(value=uuid.UUID(int=333779996850170035686993356951732753684)) new_vm = SpecificLocation(id=an_id) address_with_vm.vm = new_vm assert address_with_vm.vm == new_vm # Test domain_id getter address_without_vm = Address( network=SpecificLocation(id=UID()), domain=SpecificLocation(id=UID()), device=SpecificLocation(id=UID()), ) assert address_with_vm.vm_id == an_id assert address_without_vm.vm_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