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})