def get_test_registry(context, **kwargs):
"""Return a test registry object with appropriate attributes.
NOTE: The object leaves the attributes marked as changed, such
that a create() could be used to commit it to the DB.
"""
db_registry = db_utils.get_test_registry(**kwargs)
registry = objects.Registry(context)
for key in db_registry:
setattr(registry, key, db_registry[key])
return registry
def test_get_all_registries_with_unknown_parameter(self,
mock_registry_list):
test_registry = utils.get_test_registry()
registries = [objects.Registry(self.context, **test_registry)]
mock_registry_list.return_value = registries
response = self.get('/v1/registries/?unknown=fake-name',
expect_errors=True)
mock_registry_list.assert_not_called()
self.assertEqual(400, response.status_int)
self.assertEqual('application/json', response.content_type)
self.assertEqual("Unknown parameters: unknown",
response.json['errors'][0]['detail'])
def test_get_all_registries_with_filter(self, mock_registry_list):
test_registry = utils.get_test_registry()
registries = [objects.Registry(self.context, **test_registry)]
mock_registry_list.return_value = registries
response = self.get('/v1/registries/?name=fake-name')
mock_registry_list.assert_called_once_with(
mock.ANY, 1000, None, 'id', 'asc', filters={'name': 'fake-name'})
self.assertEqual(200, response.status_int)
actual_registries = response.json['registries']
self.assertEqual(1, len(actual_registries))
self.assertEqual(test_registry['uuid'],
actual_registries[0].get('uuid'))
def test_get_all_registries_all_projects(self, mock_registry_list,
mock_policy):
mock_policy.return_value = True
test_registry = utils.get_test_registry()
registries = [objects.Registry(self.context, **test_registry)]
mock_registry_list.return_value = registries
response = self.get('/v1/registries/?all_projects=1')
mock_registry_list.assert_called_once_with(mock.ANY,
1000,
None,
'id',
'asc',
filters={})
context = mock_registry_list.call_args[0][0]
self.assertIs(True, context.all_projects)
self.assertEqual(200, response.status_int)
actual_registries = response.json['registries']
self.assertEqual(1, len(actual_registries))
self.assertEqual(test_registry['uuid'],
actual_registries[0].get('uuid'))
def setUp(self):
super(TestRegistryObject, self).setUp()
self.fake_registry = utils.get_test_registry()
def setUp(self):
super(TestRegistryObject, self).setUp()
self.fake_cpuset = utils.get_cpuset_dict()
self.fake_registry = utils.get_test_registry(
cpuset=self.fake_cpuset, cpu_policy='dedicated')