def test_get_image(self, mock_image_get_by_uuid):
test_image = db_utils.get_test_image()
test_image_obj = objects.Image(self.context, **test_image)
mock_image_get_by_uuid.return_value = test_image_obj
with mock.patch.object(pecan, 'request'):
image = utils.get_image(test_image['uuid'])
mock_image_get_by_uuid.assert_called_once_with(
mock.ANY, test_image['uuid'])
self.assertEqual(test_image['uuid'], image.uuid)
def test_image_pull(self, mock_pull, mock_save, mock_inspect):
image = Image(self.context, **utils.get_test_image())
ret = {'image': 'repo', 'path': 'out_path', 'driver': 'glance'}
mock_pull.return_value = ret, True
mock_inspect.return_value = {'Id': 'fake-id', 'Size': 512}
self.compute_manager._do_image_pull(self.context, image)
mock_pull.assert_any_call(self.context, image.repo, image.tag)
mock_save.assert_called_once()
mock_inspect.assert_called_once_with(image.repo + ":" + image.tag)
def test_get_image(self, mock_image_get_by_uuid):
test_image = db_utils.get_test_image()
test_image_obj = objects.Image(self.context, **test_image)
mock_image_get_by_uuid.return_value = test_image_obj
with mock.patch.object(pecan, 'request'):
image = utils.get_image(test_image['uuid'])
mock_image_get_by_uuid.assert_called_once_with(
mock.ANY,
test_image['uuid'])
self.assertEqual(test_image['uuid'], image.uuid)
def test_get_one_image_by_uuid(self, mock_image_get_by_uuid, mock_policy):
mock_policy.return_value = True
test_image = utils.get_test_image()
test_image_obj = objects.Image(self.context, **test_image)
mock_image_get_by_uuid.return_value = test_image_obj
response = self.get('/v1/images/%s/' % test_image['uuid'])
mock_image_get_by_uuid.assert_called_once_with(mock.ANY,
test_image['uuid'])
self.assertEqual(200, response.status_int)
self.assertEqual(test_image['uuid'], response.json['uuid'])
def test_get_all_images(self, mock_image_list, mock_policy_enforce):
test_image = utils.get_test_image()
images = [objects.Image(self.context, **test_image)]
mock_image_list.return_value = images
response = self.get('/v1/images/')
mock_image_list.assert_called_once_with(mock.ANY,
1000, None, 'id', 'asc',
filters=None)
self.assertEqual(200, response.status_int)
actual_images = response.json['images']
self.assertEqual(1, len(actual_images))
self.assertEqual(test_image['uuid'],
actual_images[0].get('uuid'))
def test_get_one_image_by_uuid(self, mock_image_get_by_uuid, mock_policy):
mock_policy.return_value = True
test_image = utils.get_test_image()
test_image_obj = objects.Image(self.context, **test_image)
mock_image_get_by_uuid.return_value = test_image_obj
response = self.get('/v1/images/%s/' % test_image['uuid'])
mock_image_get_by_uuid.assert_called_once_with(
mock.ANY,
test_image['uuid'])
self.assertEqual(200, response.status_int)
self.assertEqual(test_image['uuid'],
response.json['uuid'])
self.assertEqual(test_image['host'],
response.json['host'])
def test_get_all_images(self, mock_image_list, mock_policy_enforce):
test_image = utils.get_test_image()
images = [objects.Image(self.context, **test_image)]
mock_image_list.return_value = images
response = self.get('/v1/images/')
mock_image_list.assert_called_once_with(mock.ANY,
1000, None, 'id', 'asc',
filters=None)
self.assertEqual(200, response.status_int)
actual_images = response.json['images']
self.assertEqual(1, len(actual_images))
self.assertEqual(test_image['uuid'],
actual_images[0].get('uuid'))
self.assertEqual(test_image['host'],
actual_images[0].get('host'))
def setUp(self):
super(TestImageObject, self).setUp()
self.fake_image = utils.get_test_image()
