Merge "tests: Update fake image client in tests"

1 files changed, 16 insertions, 11 deletions

diff --git a/openstackclient/tests/unit/image/v2/fakes.py b/openstackclient/tests/unit/image/v2/fakes.py
index 910bd726..a0eda6d2 100644
--- a/openstackclient/tests/unit/image/v2/fakes.py
+++ b/openstackclient/tests/unit/image/v2/fakes.py
@@ -24,21 +24,26 @@ from openstackclient.tests.unit.identity.v3 import fakes as identity_fakes
 from openstackclient.tests.unit import utils
 
 
-class FakeImagev2Client(object):
+class FakeImagev2Client:
 
     def __init__(self, **kwargs):
         self.images = mock.Mock()
-        self.images.resource_class = fakes.FakeResource(None, {})
-        self.image_members = mock.Mock()
-        self.image_members.resource_class = fakes.FakeResource(None, {})
-        self.image_tags = mock.Mock()
-        self.image_tags.resource_class = fakes.FakeResource(None, {})
-
+        self.create_image = mock.Mock()
+        self.delete_image = mock.Mock()
+        self.update_image = mock.Mock()
         self.find_image = mock.Mock()
-        self.find_image.resource_class = fakes.FakeResource(None, {})
-
         self.get_image = mock.Mock()
-        self.get_image.resource_class = fakes.FakeResource(None, {})
+        self.download_image = mock.Mock()
+        self.reactivate_image = mock.Mock()
+        self.deactivate_image = mock.Mock()
+
+        self.members = mock.Mock()
+        self.add_member = mock.Mock()
+        self.remove_member = mock.Mock()
+        self.update_member = mock.Mock()
+
+        self.remove_tag = mock.Mock()
+
         self.auth_token = kwargs['token']
         self.management_url = kwargs['endpoint']
         self.version = 2.0
@@ -47,7 +52,7 @@ class FakeImagev2Client(object):
 class TestImagev2(utils.TestCommand):
 
     def setUp(self):
-        super(TestImagev2, self).setUp()
+        super().setUp()
 
         self.app.client_manager.image = FakeImagev2Client(
             endpoint=fakes.AUTH_URL,