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_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()