def test_get_object_returns_node_if_owner(self):
owner = factory.make_User()
device = self.make_device_with_ip_address(owner=owner)
handler = DeviceHandler(owner, {})
self.assertEqual(
device.system_id,
handler.get_object({"system_id": device.system_id}).system_id)
def test_get_object_returns_device_if_super_user(self):
admin = factory.make_admin()
owner = factory.make_User()
device = self.make_device_with_ip_address(owner=owner)
handler = DeviceHandler(admin, {}, None)
self.assertEqual(
device.system_id,
handler.get_object({"system_id": device.system_id}).system_id,
)
def test_get_object_raises_exception_if_owner_by_another_user(self):
user = factory.make_User()
device = self.make_device_with_ip_address()
handler = DeviceHandler(user, {})
with ExpectedException(HandlerDoesNotExistError):
handler.get_object({"system_id": device.system_id})
