summaryrefslogtreecommitdiffstats
path: root/vio
diff options
context:
space:
mode:
Diffstat (limited to 'vio')
-rw-r--r--vio/vio/tests/test_image_view.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/vio/vio/tests/test_image_view.py b/vio/vio/tests/test_image_view.py
index 989eb33..968826f 100644
--- a/vio/vio/tests/test_image_view.py
+++ b/vio/vio/tests/test_image_view.py
@@ -48,3 +48,24 @@ class TestGetDeleteImageView(unittest.TestCase):
mock.Mock(), "vmware_nova", "tenant1", "image1")
self.assertEqual(204, resp.status_code)
mock_delimg.assert_called_once()
+
+
+class TestCreateListImagesView(unittest.TestCase):
+
+ def setUp(self):
+ self.view = views.CreateListImagesView()
+
+ @mock.patch.object(OperateImage.OperateImage, "get_vim_images")
+ @mock.patch.object(extsys, "get_vim_by_id")
+ def test_get(self, mock_getvim, mock_getimgs):
+ mock_getvim.return_value = {
+ "tenant": "tenant-id"
+ }
+ img = mock.Mock()
+ img.to_dict.return_value = {
+ "id": "image-id"
+ }
+ mock_getimgs.return_value = [img]
+ resp = self.view.get(
+ mock.Mock(query_params=[]), "vmware_nova", "tenant1")
+ self.assertEqual(200, resp.status_code)