def post(self, data):
"""Create a new sensor."""
policy.check('create_sensor', pecan.request.security_context)
handler = sensor_handler.SensorHandler(pecan.request.security_context)
obj = handler.create(data.as_dict(objects.registry.Sensor))
host_url = pecan.request.application_url.rstrip('/')
return sensor.Sensor.from_db_model(obj, host_url)
def get_all(self):
"""Return all sensors, based on the query provided."""
handler = sensor_handler.SensorHandler(pecan.request.security_context)
return [
sensor.Sensor.from_db_model(obj, pecan.request.host_url)
for obj in handler.get_all()
]
def put(self, data):
"""Modify this sensor."""
policy.check('update_sensor', pecan.request.security_context)
handler = sensor_handler.SensorHandler(pecan.request.security_context)
obj = handler.update(self._id, data.as_dict(objects.registry.Sensor))
host_url = pecan.request.application_url.rstrip('/')
return sensor.Sensor.from_db_model(obj, host_url)
def test_sensor_get(self, mock_registry):
mock_registry.Sensor.get_by_uuid.return_value = {}
handler = sensor.SensorHandler()
res = handler.get('test_id')
self.assertIsNotNone(res)
mock_registry.Sensor.get_by_uuid.assert_called_once_with(
None, 'test_id')
def test_sensor_delete(self, mock_registry):
db_obj = fakes.FakeSensor()
mock_registry.Sensor.get_by_uuid.return_value = db_obj
handler = sensor.SensorHandler()
handler.delete('test_id')
db_obj.delete.assert_called_once_with()
mock_registry.Sensor.get_by_uuid.assert_called_once_with(
None, 'test_id')
def get_all(self):
"""Return all sensors, based on the query provided."""
policy.check('get_sensors', pecan.request.security_context)
handler = sensor_handler.SensorHandler(pecan.request.security_context)
host_url = pecan.request.application_url.rstrip('/')
return [
sensor.Sensor.from_db_model(obj, host_url)
for obj in handler.get_all()
]
def test_sensor_create(self, mock_registry):
data = {'name': 'new_name', 'uuid': 'input_uuid'}
db_obj = fakes.FakeSensor()
mock_registry.Sensor.return_value = db_obj
handler = sensor.SensorHandler(self.ctx)
res = handler.create(data)
db_obj.update.assert_called_once_with(data)
db_obj.create.assert_called_once_with(self.ctx)
self.assertEqual(db_obj, res)
def test_sensor_update(self, mock_registry):
data = {'user_id': 'new_user_id'}
db_obj = fakes.FakeSensor()
mock_registry.Sensor.get_by_uuid.return_value = db_obj
handler = sensor.SensorHandler()
res = handler.update('test_id', data)
self.assertEqual(db_obj.user_id, res.user_id)
self.assertEqual(db_obj.project_id, res.project_id)
self.assertEqual(db_obj.name, res.name)
self.assertEqual(db_obj.value, res.value)
self.assertEqual(db_obj.sensor_type, res.sensor_type)
self.assertEqual(db_obj.target_resource, res.target_resource)
self.assertEqual(db_obj.uuid, res.uuid)
self.assertEqual(db_obj.uri, res.uri)
self.assertEqual(db_obj.type, res.type)
db_obj.save.assert_called_once_with(None)
mock_registry.Sensor.get_by_uuid.assert_called_once_with(
None, 'test_id')
def __init__(self, sensor_id):
self._id = sensor_id
self._handler = sensor_handler.SensorHandler()
def test_sensor_create(self, mock_registry):
data = {'user_id': 'new_user_id'}
handler = sensor.SensorHandler()
res = handler.create(data)
self.assertEqual('new_user_id', res.user_id)
def get(self):
"""Return this sensor."""
policy.check('show_sensor', pecan.request.security_context)
handler = sensor_handler.SensorHandler(pecan.request.security_context)
host_url = pecan.request.application_url.rstrip('/')
return sensor.Sensor.from_db_model(handler.get(self._id), host_url)
def test_sensor_get_all(self, mock_registry):
mock_registry.SensorList.get_all.return_value = {}
handler = sensor.SensorHandler()
res = handler.get_all()
self.assertIsNotNone(res)
mock_registry.SensorList.get_all.assert_called_once_with(None)
def post(self, data):
"""Create a new sensor."""
handler = sensor_handler.SensorHandler(pecan.request.security_context)
obj = handler.create(data.as_dict(objects.registry.Sensor))
return sensor.Sensor.from_db_model(obj, pecan.request.host_url)
def delete(self):
"""Delete this sensor."""
policy.check('delete_sensor', pecan.request.security_context)
handler = sensor_handler.SensorHandler(pecan.request.security_context)
handler.delete(self._id)
def put(self, data):
"""Modify this sensor."""
handler = sensor_handler.SensorHandler(pecan.request.security_context)
obj = handler.update(self._id, data.as_dict(objects.registry.Sensor))
return sensor.Sensor.from_db_model(obj, pecan.request.host_url)
def delete(self):
"""Delete this sensor."""
handler = sensor_handler.SensorHandler(pecan.request.security_context)
handler.delete(self._id)
def get(self):
"""Return this sensor."""
handler = sensor_handler.SensorHandler(pecan.request.security_context)
return sensor.Sensor.from_db_model(handler.get(self._id),
pecan.request.host_url)
def __init__(self):
super(SensorsController, self).__init__()
self._handler = sensor_handler.SensorHandler()
def test_sensor_update(self, mock_registry):
data = {'name': 'new_user_name'}
handler = sensor.SensorHandler(self.ctx)
handler.update('test_id', data)
mock_registry.Sensor.update_and_save.assert_called_once_with(
self.ctx, 'test_id', data)
def test_sensor_get(self):
handler = sensor.SensorHandler()
res = handler.get('test_id')
self.assertIsNotNone(res)
def get(self):
"""Return this sensor."""
handler = sensor_handler.SensorHandler()
return handler.get(self._id)
