def delete(self):
"""Delete this component."""
policy.check('delete_component',
pecan.request.security_context)
handler = component_handler.ComponentHandler(
pecan.request.security_context)
return handler.delete(self._id)
def post(self, data):
"""Create a new component."""
handler = component_handler.ComponentHandler(
pecan.request.security_context)
return component.Component.from_db_model(
handler.create(data.as_dict(objects.registry.Component)),
pecan.request.host_url)
def put(self, data):
"""Modify this component."""
handler = component_handler.ComponentHandler(
pecan.request.security_context)
res = handler.update(self._id,
data.as_dict(objects.registry.Component))
return component.Component.from_db_model(res, pecan.request.host_url)
def test_component_get(self, mock_registry):
mock_registry.component.get_by_uuid.return_value = {'assembly_id': 42}
handler = component_handler.ComponentHandler(self.ctx)
res = handler.get('test_id')
self.assertIsNotNone(res)
get_by_uuid = mock_registry.Component.get_by_uuid
get_by_uuid.assert_called_once_with(self.ctx, 'test_id')
def test_component_get(self, mock_registry):
mock_registry.component.get_by_uuid.return_value = {}
handler = component_handler.ComponentHandler()
res = handler.get('test_id')
self.assertIsNotNone(res)
mock_registry.Component.get_by_uuid.\
assert_called_once_with(None, 'test_id')
def get_all(self):
"""Return all components, based on the query provided."""
handler = component_handler.ComponentHandler(
pecan.request.security_context)
return [
component.Component.from_db_model(ser, pecan.request.host_url)
for ser in handler.get_all()
]
def test_delete(self, mock_registry):
db_obj = fakes.FakeComponent()
mock_registry.Component.get_by_uuid.return_value = db_obj
handler = component_handler.ComponentHandler(self.ctx)
handler.delete('test_id')
db_obj.destroy.assert_called_once_with(self.ctx)
mock_registry.Component.get_by_uuid.assert_called_once_with(
self.ctx, 'test_id')
def get_all(self):
"""Return all components, based on the query provided."""
policy.check('get_components',
pecan.request.security_context)
handler = component_handler.ComponentHandler(
pecan.request.security_context)
host_url = pecan.request.application_url.rstrip('/')
return [component.Component.from_db_model(ser, host_url)
for ser in handler.get_all()]
def post(self, data):
"""Create a new component."""
policy.check('create_component',
pecan.request.security_context)
handler = component_handler.ComponentHandler(
pecan.request.security_context)
host_url = pecan.request.application_url.rstrip('/')
return component.Component.from_db_model(
handler.create(data.as_dict(objects.registry.Component)), host_url)
def get(self):
"""Return this component."""
policy.check('show_component',
pecan.request.security_context)
handler = component_handler.ComponentHandler(
pecan.request.security_context)
host_url = pecan.request.application_url.rstrip('/')
return component.Component.from_db_model(handler.get(self._id),
host_url)
def test_create(self, mock_registry):
data = {'name': 'new_name', 'assembly_id': 'new_assembly_id'}
db_obj = fakes.FakeComponent()
mock_registry.Component.return_value = db_obj
handler = component_handler.ComponentHandler(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(res, db_obj)
def put(self, data):
"""Modify this component."""
policy.check('update_component',
pecan.request.security_context)
handler = component_handler.ComponentHandler(
pecan.request.security_context)
res = handler.update(self._id,
data.as_dict(objects.registry.Component))
host_url = pecan.request.application_url.rstrip('/')
return component.Component.from_db_model(res, host_url)
def test_update(self, mock_registry):
data = {'user_id': 'new_user_id'}
db_obj = fakes.FakeComponent()
mock_registry.Component.get_by_uuid.return_value = db_obj
handler = component_handler.ComponentHandler()
res = handler.update('test_id', data)
self.assertEqual(db_obj.user_id, res.user_id)
db_obj.save.assert_called_once_with(None)
mock_registry.Component.get_by_uuid.assert_called_once_with(
None, 'test_id')
def get(self):
"""Return this component."""
handler = component_handler.ComponentHandler()
return handler.get(self._id)
def __init__(self, component_id):
super(ComponentController, self).__init__()
self._id = component_id
self._handler = component_handler.ComponentHandler()
def test_get_all(self, mock_registry):
mock_registry.ComponentList.get_all.return_value = {}
handler = component_handler.ComponentHandler(self.ctx)
res = handler.get_all()
self.assertIsNotNone(res)
mock_registry.ComponentList.get_all.assert_called_once_with(self.ctx)
def test_update(self, mock_registry):
data = {'user_id': 'new_user_id', 'assembly_id': 'new_assembly_id'}
handler = component_handler.ComponentHandler(self.ctx)
handler.update('test_id', data)
mock_registry.Component.update_and_save.assert_called_once_with(
self.ctx, 'test_id', data)
def test_component_get_all(self):
handler = component.ComponentHandler()
res = handler.get_all()
self.assertIsNotNone(res)
def put(self, data):
"""Modify this component."""
handler = component_handler.ComponentHandler()
return handler.update(self._id, data)
def post(self, data):
"""Create a new component."""
handler = component_handler.ComponentHandler()
return handler.create(data)
def delete(self):
"""Delete this component."""
handler = component_handler.ComponentHandler()
return handler.delete(self._id)
def delete(self):
"""Delete this component."""
handler = component_handler.ComponentHandler(
pecan.request.security_context)
return handler.delete(self._id)
def get_all(self):
"""Return all components, based on the query provided."""
handler = component_handler.ComponentHandler()
return handler.get_all()
def get(self):
"""Return this component."""
handler = component_handler.ComponentHandler(
pecan.request.security_context)
return component.Component.from_db_model(handler.get(self._id),
pecan.request.host_url)
def test_create(self, mock_registry):
data = {'user_id': 'new_user_id'}
handler = component_handler.ComponentHandler()
res = handler.create(data)
self.assertEqual('new_user_id', res.user_id)
def __init__(self):
super(ComponentsController, self).__init__()
self._handler = component_handler.ComponentHandler()
objects.load()
