Merge "Add arg to 'server image create' tests"

2 files changed, 10 insertions, 5 deletions

diff --git a/openstackclient/tests/compute/v2/test_server.py b/openstackclient/tests/compute/v2/test_server.py
index 43aa7a70..19c13440 100644
--- a/openstackclient/tests/compute/v2/test_server.py
+++ b/openstackclient/tests/compute/v2/test_server.py
@@ -285,13 +285,14 @@ class TestServerImageCreate(TestServer):
             compute_fakes.server_name,
         )
 
-        collist = ('id', 'is_public', 'name', 'owner')
+        collist = ('id', 'is_public', 'name', 'owner', 'protected')
         self.assertEqual(collist, columns)
         datalist = (
             image_fakes.image_id,
-            False,
+            image_fakes.image_public,
             image_fakes.image_name,
             image_fakes.image_owner,
+            image_fakes.image_protected,
         )
         self.assertEqual(datalist, data)
 
@@ -315,13 +316,14 @@ class TestServerImageCreate(TestServer):
             'img-nam',
         )
 
-        collist = ('id', 'is_public', 'name', 'owner')
+        collist = ('id', 'is_public', 'name', 'owner', 'protected')
         self.assertEqual(collist, columns)
         datalist = (
             image_fakes.image_id,
-            False,
+            image_fakes.image_public,
             image_fakes.image_name,
             image_fakes.image_owner,
+            image_fakes.image_protected,
         )
         self.assertEqual(datalist, data)
 
diff --git a/openstackclient/tests/image/v2/fakes.py b/openstackclient/tests/image/v2/fakes.py
index 96255cd4..1b7edf08 100644
--- a/openstackclient/tests/image/v2/fakes.py
+++ b/openstackclient/tests/image/v2/fakes.py
@@ -22,12 +22,15 @@ from openstackclient.tests import utils
 image_id = 'im1'
 image_name = 'graven'
 image_owner = 'baal'
+image_public = False
+image_protected = False
 
 IMAGE = {
     'id': image_id,
     'name': image_name,
-    'is_public': False,
+    'is_public': image_public,
     'owner': image_owner,
+    'protected': image_protected,
 }