Skip to content

Commit 8f07fec

Browse files
author
Huanxuan Ao
committed
Clean up FakeClient classes in volume fakes
Put FakeTransferClient and FakeServiceClient classes into FakeVolumeClient class, I think we need not to create FakeClient classes for every objects, we can use only FakeVolumeClient for all volume objects. This can reduce repetition, unified codes and make codes look cleaner and easier. Change-Id: I2729ef0e80166f4d49ccd9a48b653e2f215e3bfe
1 parent 684412c commit 8f07fec

6 files changed

Lines changed: 10 additions & 78 deletions

File tree

openstackclient/tests/volume/v1/fakes.py

Lines changed: 2 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -129,25 +129,6 @@
129129
}
130130

131131

132-
class FakeTransferClient(object):
133-
134-
def __init__(self, **kwargs):
135-
136-
self.transfers = mock.Mock()
137-
self.transfers.resource_class = fakes.FakeResource(None, {})
138-
139-
140-
class TestTransfer(utils.TestCommand):
141-
142-
def setUp(self):
143-
super(TestTransfer, self).setUp()
144-
145-
self.app.client_manager.volume = FakeTransferClient(
146-
endpoint=fakes.AUTH_URL,
147-
token=fakes.AUTH_TOKEN
148-
)
149-
150-
151132
class FakeTransfer(object):
152133
"""Fake one or more Transfer."""
153134

@@ -180,24 +161,6 @@ def create_one_transfer(attrs=None):
180161
return transfer
181162

182163

183-
class FakeServiceClient(object):
184-
185-
def __init__(self, **kwargs):
186-
self.services = mock.Mock()
187-
self.services.resource_class = fakes.FakeResource(None, {})
188-
189-
190-
class TestService(utils.TestCommand):
191-
192-
def setUp(self):
193-
super(TestService, self).setUp()
194-
195-
self.app.client_manager.volume = FakeServiceClient(
196-
endpoint=fakes.AUTH_URL,
197-
token=fakes.AUTH_TOKEN
198-
)
199-
200-
201164
class FakeService(object):
202165
"""Fake one or more Services."""
203166

@@ -290,6 +253,8 @@ def __init__(self, **kwargs):
290253
self.qos_specs.resource_class = fakes.FakeResource(None, {})
291254
self.volume_types = mock.Mock()
292255
self.volume_types.resource_class = fakes.FakeResource(None, {})
256+
self.transfers = mock.Mock()
257+
self.transfers.resource_class = fakes.FakeResource(None, {})
293258
self.auth_token = kwargs['token']
294259
self.management_url = kwargs['endpoint']
295260

openstackclient/tests/volume/v1/test_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from openstackclient.volume.v1 import service
1818

1919

20-
class TestService(service_fakes.TestService):
20+
class TestService(service_fakes.TestVolumev1):
2121

2222
def setUp(self):
2323
super(TestService, self).setUp()

openstackclient/tests/volume/v1/test_transfer_request.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from openstackclient.volume.v1 import volume_transfer_request
1818

1919

20-
class TestTransfer(transfer_fakes.TestTransfer):
20+
class TestTransfer(transfer_fakes.TestVolumev1):
2121

2222
def setUp(self):
2323
super(TestTransfer, self).setUp()

openstackclient/tests/volume/v2/fakes.py

Lines changed: 4 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -25,25 +25,6 @@
2525
from openstackclient.tests import utils
2626

2727

28-
class FakeTransferClient(object):
29-
30-
def __init__(self, **kwargs):
31-
32-
self.transfers = mock.Mock()
33-
self.transfers.resource_class = fakes.FakeResource(None, {})
34-
35-
36-
class TestTransfer(utils.TestCommand):
37-
38-
def setUp(self):
39-
super(TestTransfer, self).setUp()
40-
41-
self.app.client_manager.volume = FakeTransferClient(
42-
endpoint=fakes.AUTH_URL,
43-
token=fakes.AUTH_TOKEN
44-
)
45-
46-
4728
class FakeTransfer(object):
4829
"""Fake one or more Transfer."""
4930

@@ -108,24 +89,6 @@ def create_one_type_access(attrs=None):
10889
return type_access
10990

11091

111-
class FakeServiceClient(object):
112-
113-
def __init__(self, **kwargs):
114-
self.services = mock.Mock()
115-
self.services.resource_class = fakes.FakeResource(None, {})
116-
117-
118-
class TestService(utils.TestCommand):
119-
120-
def setUp(self):
121-
super(TestService, self).setUp()
122-
123-
self.app.client_manager.volume = FakeServiceClient(
124-
endpoint=fakes.AUTH_URL,
125-
token=fakes.AUTH_TOKEN
126-
)
127-
128-
12992
class FakeService(object):
13093
"""Fake one or more Services."""
13194

@@ -200,6 +163,10 @@ def __init__(self, **kwargs):
200163
self.qos_specs.resource_class = fakes.FakeResource(None, {})
201164
self.availability_zones = mock.Mock()
202165
self.availability_zones.resource_class = fakes.FakeResource(None, {})
166+
self.transfers = mock.Mock()
167+
self.transfers.resource_class = fakes.FakeResource(None, {})
168+
self.services = mock.Mock()
169+
self.services.resource_class = fakes.FakeResource(None, {})
203170
self.auth_token = kwargs['token']
204171
self.management_url = kwargs['endpoint']
205172

openstackclient/tests/volume/v2/test_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from openstackclient.volume.v2 import service
1818

1919

20-
class TestService(service_fakes.TestService):
20+
class TestService(service_fakes.TestVolume):
2121

2222
def setUp(self):
2323
super(TestService, self).setUp()

openstackclient/tests/volume/v2/test_transfer_request.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from openstackclient.volume.v2 import volume_transfer_request
1818

1919

20-
class TestTransfer(transfer_fakes.TestTransfer):
20+
class TestTransfer(transfer_fakes.TestVolume):
2121

2222
def setUp(self):
2323
super(TestTransfer, self).setUp()

0 commit comments

Comments
 (0)