Merge "Implements hide image"

1 files changed, 68 insertions, 0 deletions

diff --git a/openstackclient/tests/unit/image/v2/test_image.py b/openstackclient/tests/unit/image/v2/test_image.py
index 87dfdbea..c44c767b 100644
--- a/openstackclient/tests/unit/image/v2/test_image.py
+++ b/openstackclient/tests/unit/image/v2/test_image.py
@@ -837,6 +837,20 @@ class TestImageList(TestImage):
             status='active'
         )
 
+    def test_image_list_hidden_option(self):
+        arglist = [
+            '--hidden',
+        ]
+        verifylist = [
+            ('hidden', True),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        columns, data = self.cmd.take_action(parsed_args)
+        self.client.images.assert_called_with(
+            is_hidden=True
+        )
+
     def test_image_list_tag_option(self):
         arglist = [
             '--tag', 'abc',
@@ -1439,6 +1453,60 @@ class TestImageSet(TestImage):
         )
         self.assertIsNone(result)
 
+    def test_image_set_hidden(self):
+        arglist = [
+            '--hidden',
+            '--public',
+            image_fakes.image_name,
+        ]
+        verifylist = [
+            ('hidden', True),
+            ('public', True),
+            ('private', False),
+            ('image', image_fakes.image_name),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        kwargs = {
+            'is_hidden': True,
+            'visibility': 'public',
+        }
+        # ImageManager.update(image, **kwargs)
+        self.client.update_image.assert_called_with(
+            self._image.id,
+            **kwargs
+        )
+        self.assertIsNone(result)
+
+    def test_image_set_unhidden(self):
+        arglist = [
+            '--unhidden',
+            '--public',
+            image_fakes.image_name,
+        ]
+        verifylist = [
+            ('hidden', False),
+            ('public', True),
+            ('private', False),
+            ('image', image_fakes.image_name),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        result = self.cmd.take_action(parsed_args)
+
+        kwargs = {
+            'is_hidden': False,
+            'visibility': 'public',
+        }
+        # ImageManager.update(image, **kwargs)
+        self.client.update_image.assert_called_with(
+            self._image.id,
+            **kwargs
+        )
+        self.assertIsNone(result)
+
 
 class TestImageShow(TestImage):