def test_get_resource_class_by_name(self, mock_write, mock_read):
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_get_resource_class_by_uuid(self, mock_db_inst, mock_write,
mock_read):
mock_db_inst.return_value = etcd_api.get_backend()
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 get_by_name(cls, context, name):
"""Find a resource class based on name.
:param name: the name of a resource class.
:param context: Security context
:returns: a :class:`ResourceClass` object.
"""
db_resource = dbapi.get_resource_class(context, name)
resource = ResourceClass._from_db_object(cls(context), db_resource)
return resource
def get_by_uuid(cls, context, uuid):
"""Find a resource class based on uuid.
:param uuid: the uuid of a resource class.
:param context: Security context
:returns: a :class:`ResourceClass` object.
"""
db_resource = dbapi.get_resource_class(context, uuid)
resource = ResourceClass._from_db_object(cls(context), db_resource)
return resource
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_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)
