def allocation_post_data(**kw):
"""Return an Allocation object without internal attributes."""
allocation = db_utils.get_test_allocation(**kw)
return {
key: value
for key, value in allocation.items() if key in _ALLOCATION_POST_FIELDS
}
def allocation_post_data(node=None, **kw):
"""Return an Allocation object without internal attributes."""
allocation = db_utils.get_test_allocation(**kw)
if node:
# This is not a database field, so it has to be handled explicitly
allocation['node'] = node
return remove_other_fields(allocation,
al_controller.ALLOCATION_SCHEMA['properties'])
def allocation_post_data(node=None, **kw):
"""Return an Allocation object without internal attributes."""
allocation = db_utils.get_test_allocation(**kw)
if node:
# This is not a database field, so it has to be handled explicitly
allocation['node'] = node
return {key: value for key, value in allocation.items()
if key in _ALLOCATION_POST_FIELDS}
def test_create(self):
allocation = objects.Allocation(self.context, **self.fake_allocation)
with mock.patch.object(self.dbapi, 'create_allocation',
autospec=True) as mock_create_allocation:
mock_create_allocation.return_value = (
db_utils.get_test_allocation())
allocation.create()
args, _kwargs = mock_create_allocation.call_args
self.assertEqual(objects.Allocation.VERSION, args[0]['version'])
def test_refresh(self):
uuid = self.fake_allocation['uuid']
returns = [self.fake_allocation,
db_utils.get_test_allocation(name='newname')]
expected = [mock.call(uuid), mock.call(uuid)]
with mock.patch.object(self.dbapi, 'get_allocation_by_uuid',
side_effect=returns,
autospec=True) as mock_get_allocation:
p = objects.Allocation.get_by_uuid(self.context, uuid)
self.assertEqual(self.fake_allocation['name'], p.name)
p.refresh()
self.assertEqual('newname', p.name)
self.assertEqual(expected, mock_get_allocation.call_args_list)
self.assertEqual(self.context, p._context)
def test_save(self):
uuid = self.fake_allocation['uuid']
test_time = datetime.datetime(2000, 1, 1, 0, 0)
with mock.patch.object(self.dbapi, 'get_allocation_by_uuid',
autospec=True) as mock_get_allocation:
mock_get_allocation.return_value = self.fake_allocation
with mock.patch.object(self.dbapi, 'update_allocation',
autospec=True) as mock_update_allocation:
mock_update_allocation.return_value = (
db_utils.get_test_allocation(name='newname',
updated_at=test_time))
p = objects.Allocation.get_by_uuid(self.context, uuid)
p.name = 'newname'
p.save()
mock_get_allocation.assert_called_once_with(uuid)
mock_update_allocation.assert_called_once_with(
uuid, {'version': objects.Allocation.VERSION,
'name': 'newname'})
self.assertEqual(self.context, p._context)
res_updated_at = (p.updated_at).replace(tzinfo=None)
self.assertEqual(test_time, res_updated_at)
def allocation_post_data(**kw):
"""Return an Allocation object without internal attributes."""
allocation = db_utils.get_test_allocation(**kw)
return {key: value for key, value in allocation.items()
if key in _ALLOCATION_POST_FIELDS}
def setUp(self):
super(TestAllocationObject, self).setUp()
self.fake_allocation = db_utils.get_test_allocation(name='host1')
def setUp(self):
super(TestConvertToVersion, self).setUp()
self.fake_allocation = db_utils.get_test_allocation()