def test_create_resource_class_already_exists(self):
utils.create_test_resource_class(
context=self.context, uuid='123')
with self.assertRaisesRegex(exception.ResourceClassAlreadyExists,
'A resource class with uuid 123.*'):
utils.create_test_resource_class(
context=self.context, uuid='123')
def test_get_resource_class_by_uuid(self, mock_write, mock_read):
mock_read.side_effect = etcd.EtcdKeyNotFound
resource_class = utils.create_test_resource_class(context=self.context)
mock_read.side_effect = lambda *args: FakeEtcdResult(resource_class.
as_dict())
res = dbapi.get_resource_class(self.context, resource_class.uuid)
self.assertEqual(resource_class.uuid, res.uuid)
self.assertEqual(resource_class.name, res.name)
def test_get_resource_class_by_name(self, mock_db_inst, mock_write,
mock_read):
mock_db_inst.return_value = etcd_api.get_backend()
mock_read.side_effect = etcd.EtcdKeyNotFound
rcs = utils.create_test_resource_class(context=self.context)
mock_read.side_effect = lambda *args: FakeEtcdMultipleResult(
[rcs.as_dict()])
res = dbapi.get_resource_class(self.context, rcs.name)
self.assertEqual(rcs.uuid, res.uuid)
def test_update_resource_class(self):
resource = utils.create_test_resource_class(context=self.context)
old_name = resource.name
new_name = 'new-name'
self.assertNotEqual(old_name, new_name)
res = dbapi.update_resource_class(self.context, resource.id,
{'name': new_name})
self.assertEqual(new_name, res.name)
def test_update_resource_class(self):
resource = utils.create_test_resource_class(context=self.context)
old_name = resource.name
new_name = 'new-name'
self.assertNotEqual(old_name, new_name)
res = dbapi.update_resource_class(
self.context, resource.id, {'name': new_name})
self.assertEqual(new_name, res.name)
def test_destroy_resource_class(self, mock_delete,
mock_write, mock_read):
mock_read.side_effect = etcd.EtcdKeyNotFound
resource_class = utils.create_test_resource_class(
context=self.context)
mock_read.side_effect = lambda *args: FakeEtcdResult(
resource_class.as_dict())
dbapi.destroy_resource_class(self.context, resource_class.uuid)
mock_delete.assert_called_once_with(
'/resource_classes/%s' % resource_class.uuid)
def test_list_resource_classes(self):
names = []
for i in range(1, 6):
resource = utils.create_test_resource_class(
context=self.context,
uuid=uuidutils.generate_uuid(),
name='class' + str(i))
names.append(str(resource['name']))
res = dbapi.list_resource_classes(self.context)
res_names = [r.name for r in res]
self.assertEqual(sorted(names), sorted(res_names))
def test_list_resource_classes(self):
names = []
for i in range(1, 6):
resource = utils.create_test_resource_class(
context=self.context,
uuid=uuidutils.generate_uuid(),
name='class' + str(i))
names.append(six.text_type(resource['name']))
res = dbapi.list_resource_classes(self.context)
res_names = [r.name for r in res]
self.assertEqual(sorted(names), sorted(res_names))
def test_update_resource_class(self, mock_update, mock_write, mock_read):
mock_read.side_effect = etcd.EtcdKeyNotFound
resource_class = utils.create_test_resource_class(context=self.context)
old_name = resource_class.name
new_name = 'new-name'
self.assertNotEqual(old_name, new_name)
mock_read.side_effect = lambda *args: FakeEtcdResult(resource_class.
as_dict())
dbapi.update_resource_class(self.context, resource_class.uuid,
{'name': new_name})
self.assertEqual(
new_name,
json.loads(mock_update.call_args_list[0][0][0].value)['name'])
def test_list_resource_classes_sorted(self, mock_write, mock_read):
names = []
resource_classes = []
mock_read.side_effect = etcd.EtcdKeyNotFound
for i in range(1, 6):
res_class = utils.create_test_resource_class(
context=self.context, name='class'+str(i))
resource_classes.append(res_class.as_dict())
names.append(six.text_type(res_class['name']))
mock_read.side_effect = lambda *args: FakeEtcdMultipleResult(
resource_classes)
res = dbapi.list_resource_classes(self.context, sort_key='name')
res_names = [r.name for r in res]
self.assertEqual(sorted(names), res_names)
def test_list_resource_classes_sorted(self):
names = []
for i in range(5):
resource = utils.create_test_resource_class(
context=self.context,
uuid=uuidutils.generate_uuid(),
name='class' + str(i))
names.append(str(resource.name))
res = dbapi.list_resource_classes(self.context, sort_key='name')
res_names = [r.name for r in res]
self.assertEqual(sorted(names), res_names)
self.assertRaises(exception.InvalidParameterValue,
dbapi.list_resource_classes,
self.context,
sort_key='foo')
def test_list_resource_classes_sorted(self):
names = []
for i in range(5):
resource = utils.create_test_resource_class(
context=self.context,
uuid=uuidutils.generate_uuid(),
name='class' + str(i))
names.append(six.text_type(resource.name))
res = dbapi.list_resource_classes(self.context, sort_key='name')
res_names = [r.name for r in res]
self.assertEqual(sorted(names), res_names)
self.assertRaises(exception.InvalidParameterValue,
dbapi.list_resource_classes,
self.context,
sort_key='foo')
def test_get_resource_class_by_name(self):
resource = utils.create_test_resource_class(context=self.context)
res = dbapi.get_resource_class(self.context, resource.name)
self.assertEqual(resource.id, res.id)
self.assertEqual(resource.name, res.name)
def test_create_resource_class_already_exists(self):
utils.create_test_resource_class(context=self.context, uuid='123')
with self.assertRaisesRegex(exception.ResourceClassAlreadyExists,
'A resource class with uuid 123.*'):
utils.create_test_resource_class(context=self.context, uuid='123')
def test_create_resource_class(self):
utils.create_test_resource_class(context=self.context)
def test_destroy_resource_class(self):
resource = utils.create_test_resource_class(context=self.context)
dbapi.destroy_resource_class(self.context, resource.id)
self.assertRaises(exception.ResourceClassNotFound,
dbapi.get_resource_class,
self.context, resource.id)
def test_update_resource_class_uuid(self, mock_write, mock_read):
mock_read.side_effect = etcd.EtcdKeyNotFound
resource_class = utils.create_test_resource_class(context=self.context)
self.assertRaises(exception.InvalidParameterValue,
dbapi.update_resource_class, self.context,
resource_class.uuid, {'uuid': ''})
def test_create_resource_class(self):
utils.create_test_resource_class(context=self.context)
def test_destroy_resource_class(self):
resource = utils.create_test_resource_class(context=self.context)
dbapi.destroy_resource_class(self.context, resource.id)
self.assertRaises(exception.ResourceClassNotFound,
dbapi.get_resource_class, self.context, resource.id)
def test_get_resource_class_by_name(self):
resource = utils.create_test_resource_class(context=self.context)
res = dbapi.get_resource_class(self.context, resource.name)
self.assertEqual(resource.id, res.id)
self.assertEqual(resource.name, res.name)
def test_create_resource_class(self, mock_write, mock_read):
mock_read.side_effect = etcd.EtcdKeyNotFound
utils.create_test_resource_class(context=self.context)
