Skip to content

Commit 2135a9e

Browse files
committed
image: Remove FakeImage test helper
We're no longer creating fake versions of glanceclient's 'Resource' object but rather openstacksdk objects. As such, there's no point nesting things under a fake resource class. Change-Id: I39cd5302622f4542db9eebcccfad0cb90d077441 Signed-off-by: Stephen Finucane <sfinucan@redhat.com>
1 parent 690e9a1 commit 2135a9e

11 files changed

Lines changed: 187 additions & 222 deletions

File tree

openstackclient/tests/unit/common/test_project_purge.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ class TestProjectPurge(TestProjectPurgeInit):
7070

7171
project = identity_fakes.FakeProject.create_one_project()
7272
server = compute_fakes.FakeServer.create_one_server()
73-
image = image_fakes.FakeImage.create_one_image()
73+
image = image_fakes.create_one_image()
7474
volume = volume_fakes.FakeVolume.create_one_volume()
7575
backup = volume_fakes.FakeBackup.create_one_backup()
7676
snapshot = volume_fakes.FakeSnapshot.create_one_snapshot()

openstackclient/tests/unit/compute/v2/test_aggregate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -552,7 +552,7 @@ def test_aggregate_unset_no_option(self):
552552

553553
class TestAggregateCacheImage(TestAggregate):
554554

555-
images = image_fakes.FakeImage.create_images(count=2)
555+
images = image_fakes.create_images(count=2)
556556

557557
def setUp(self):
558558
super(TestAggregateCacheImage, self).setUp()

openstackclient/tests/unit/compute/v2/test_server.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1168,7 +1168,7 @@ def setUp(self):
11681168

11691169
self.servers_mock.create.return_value = self.new_server
11701170

1171-
self.image = image_fakes.FakeImage.create_one_image()
1171+
self.image = image_fakes.create_one_image()
11721172
self.find_image_mock.return_value = self.image
11731173
self.get_image_mock.return_value = self.image
11741174

@@ -2918,7 +2918,7 @@ def test_server_create_image_property(self):
29182918
'hypervisor_type': 'qemu',
29192919
}
29202920

2921-
_image = image_fakes.FakeImage.create_one_image(image_info)
2921+
_image = image_fakes.create_one_image(image_info)
29222922
self.images_mock.return_value = [_image]
29232923

29242924
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
@@ -2974,7 +2974,7 @@ def test_server_create_image_property_multi(self):
29742974
'hypervisor_type': 'qemu',
29752975
'hw_disk_bus': 'ide',
29762976
}
2977-
_image = image_fakes.FakeImage.create_one_image(image_info)
2977+
_image = image_fakes.create_one_image(image_info)
29782978
self.images_mock.return_value = [_image]
29792979

29802980
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
@@ -3031,7 +3031,7 @@ def test_server_create_image_property_missed(self):
30313031
'hw_disk_bus': 'ide',
30323032
}
30333033

3034-
_image = image_fakes.FakeImage.create_one_image(image_info)
3034+
_image = image_fakes.create_one_image(image_info)
30353035
self.images_mock.return_value = [_image]
30363036

30373037
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
@@ -3063,8 +3063,8 @@ def test_server_create_image_property_with_image_list(self):
30633063
}
30643064
}
30653065

3066-
target_image = image_fakes.FakeImage.create_one_image(image_info)
3067-
another_image = image_fakes.FakeImage.create_one_image({})
3066+
target_image = image_fakes.create_one_image(image_info)
3067+
another_image = image_fakes.create_one_image({})
30683068
self.images_mock.return_value = [target_image, another_image]
30693069

30703070
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
@@ -4102,7 +4102,7 @@ def setUp(self):
41024102
self.servers = self.setup_servers_mock(3)
41034103
self.servers_mock.list.return_value = self.servers
41044104

4105-
self.image = image_fakes.FakeImage.create_one_image()
4105+
self.image = image_fakes.create_one_image()
41064106

41074107
# self.images_mock.return_value = [self.image]
41084108
self.find_image_mock.return_value = self.image
@@ -6021,7 +6021,7 @@ def setUp(self):
60216021
super(TestServerRebuild, self).setUp()
60226022

60236023
# Return value for utils.find_resource for image
6024-
self.image = image_fakes.FakeImage.create_one_image()
6024+
self.image = image_fakes.create_one_image()
60256025
self.get_image_mock.return_value = self.image
60266026

60276027
# Fake the rebuilt new server.
@@ -6051,7 +6051,7 @@ def setUp(self):
60516051

60526052
def test_rebuild_with_image_name(self):
60536053
image_name = 'my-custom-image'
6054-
user_image = image_fakes.FakeImage.create_one_image(
6054+
user_image = image_fakes.create_one_image(
60556055
attrs={'name': image_name})
60566056
self.find_image_mock.return_value = user_image
60576057

@@ -6600,7 +6600,7 @@ class TestEvacuateServer(TestServer):
66006600
def setUp(self):
66016601
super(TestEvacuateServer, self).setUp()
66026602
# Return value for utils.find_resource for image
6603-
self.image = image_fakes.FakeImage.create_one_image()
6603+
self.image = image_fakes.create_one_image()
66046604
self.images_mock.get.return_value = self.image
66056605

66066606
# Fake the rebuilt new server.
@@ -6794,7 +6794,7 @@ def setUp(self):
67946794
super(TestServerRescue, self).setUp()
67956795

67966796
# Return value for utils.find_resource for image
6797-
self.image = image_fakes.FakeImage.create_one_image()
6797+
self.image = image_fakes.create_one_image()
67986798
self.get_image_mock.return_value = self.image
67996799

68006800
new_server = compute_fakes.FakeServer.create_one_server()
@@ -6835,7 +6835,7 @@ def test_rescue_with_current_image(self):
68356835
self.server.rescue.assert_called_with(image=None, password=None)
68366836

68376837
def test_rescue_with_new_image(self):
6838-
new_image = image_fakes.FakeImage.create_one_image()
6838+
new_image = image_fakes.create_one_image()
68396839
self.find_image_mock.return_value = new_image
68406840
arglist = [
68416841
'--image', new_image.id,
@@ -7950,7 +7950,7 @@ class TestServerShow(TestServer):
79507950
def setUp(self):
79517951
super(TestServerShow, self).setUp()
79527952

7953-
self.image = image_fakes.FakeImage.create_one_image()
7953+
self.image = image_fakes.create_one_image()
79547954
self.flavor = compute_fakes.FakeFlavor.create_one_flavor()
79557955
self.topology = {
79567956
'nodes': [{'vcpu_set': [0, 1]}, {'vcpu_set': [2, 3]}],
@@ -8540,7 +8540,7 @@ def test_prep_server_detail(self, find_resource):
85408540
# - The first time, return server info.
85418541
# - The second time, return image info.
85428542
# - The third time, return flavor info.
8543-
_image = image_fakes.FakeImage.create_one_image()
8543+
_image = image_fakes.create_one_image()
85448544
_flavor = compute_fakes.FakeFlavor.create_one_flavor()
85458545
server_info = {
85468546
'image': {u'id': _image.id},

openstackclient/tests/unit/compute/v2/test_server_backup.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -91,15 +91,15 @@ def setUp(self):
9191

9292
def setup_images_mock(self, count, servers=None):
9393
if servers:
94-
images = image_fakes.FakeImage.create_images(
94+
images = image_fakes.create_images(
9595
attrs={
9696
'name': servers[0].name,
9797
'status': 'active',
9898
},
9999
count=count,
100100
)
101101
else:
102-
images = image_fakes.FakeImage.create_images(
102+
images = image_fakes.create_images(
103103
attrs={
104104
'status': 'active',
105105
},
@@ -178,15 +178,6 @@ def test_server_backup_create_options(self):
178178
def test_server_backup_wait_fail(self, mock_wait_for_status):
179179
servers = self.setup_servers_mock(count=1)
180180
images = self.setup_images_mock(count=1, servers=servers)
181-
# images = image_fakes.FakeImage.create_images(
182-
# attrs={
183-
# 'name': servers[0].name,
184-
# 'status': 'active',
185-
# },
186-
# count=1,
187-
# )
188-
#
189-
# self.images_mock.find_image.return_value = images[0]
190181
self.images_mock.get_image = mock.Mock(
191182
side_effect=images[0],
192183
)

openstackclient/tests/unit/compute/v2/test_server_image.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,15 +88,15 @@ def setUp(self):
8888

8989
def setup_images_mock(self, count, servers=None):
9090
if servers:
91-
images = image_fakes.FakeImage.create_images(
91+
images = image_fakes.create_images(
9292
attrs={
9393
'name': servers[0].name,
9494
'status': 'active',
9595
},
9696
count=count,
9797
)
9898
else:
99-
images = image_fakes.FakeImage.create_images(
99+
images = image_fakes.create_images(
100100
attrs={
101101
'status': 'active',
102102
},

openstackclient/tests/unit/image/v1/fakes.py

Lines changed: 33 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,6 @@
2222
from openstackclient.tests.unit.volume.v1 import fakes as volume_fakes
2323

2424

25-
image_id = 'im1'
26-
image_name = 'graven'
27-
28-
2925
class FakeImagev1Client(object):
3026

3127
def __init__(self, **kwargs):
@@ -51,40 +47,36 @@ def setUp(self):
5147
)
5248

5349

54-
class FakeImage(object):
55-
"""Fake one or more images."""
56-
57-
@staticmethod
58-
def create_one_image(attrs=None):
59-
"""Create a fake image.
60-
61-
:param Dictionary attrs:
62-
A dictionary with all attrbutes of image
63-
:return:
64-
A FakeResource object with id, name, owner, protected,
65-
visibility and tags attrs
66-
"""
67-
attrs = attrs or {}
68-
69-
# Set default attribute
70-
image_info = {
71-
'id': str(uuid.uuid4()),
72-
'name': 'image-name' + uuid.uuid4().hex,
73-
'owner': 'image-owner' + uuid.uuid4().hex,
74-
'container_format': '',
75-
'disk_format': '',
76-
'min_disk': 0,
77-
'min_ram': 0,
78-
'is_public': True,
79-
'protected': False,
80-
'properties': {
81-
'Alpha': 'a',
82-
'Beta': 'b',
83-
'Gamma': 'g'},
84-
'status': 'status' + uuid.uuid4().hex
85-
}
86-
87-
# Overwrite default attributes if there are some attributes set
88-
image_info.update(attrs)
89-
90-
return image.Image(**image_info)
50+
def create_one_image(attrs=None):
51+
"""Create a fake image.
52+
53+
:param Dictionary attrs:
54+
A dictionary with all attrbutes of image
55+
:return:
56+
A FakeResource object with id, name, owner, protected,
57+
visibility and tags attrs
58+
"""
59+
attrs = attrs or {}
60+
61+
# Set default attribute
62+
image_info = {
63+
'id': str(uuid.uuid4()),
64+
'name': 'image-name' + uuid.uuid4().hex,
65+
'owner': 'image-owner' + uuid.uuid4().hex,
66+
'container_format': '',
67+
'disk_format': '',
68+
'min_disk': 0,
69+
'min_ram': 0,
70+
'is_public': True,
71+
'protected': False,
72+
'properties': {
73+
'Alpha': 'a',
74+
'Beta': 'b',
75+
'Gamma': 'g'},
76+
'status': 'status' + uuid.uuid4().hex
77+
}
78+
79+
# Overwrite default attributes if there are some attributes set
80+
image_info.update(attrs)
81+
82+
return image.Image(**image_info)

openstackclient/tests/unit/image/v1/test_image.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def setUp(self):
3434

3535
class TestImageCreate(TestImage):
3636

37-
new_image = image_fakes.FakeImage.create_one_image()
37+
new_image = image_fakes.create_one_image()
3838
columns = (
3939
'container_format',
4040
'disk_format',
@@ -210,7 +210,7 @@ def test_image_create_file(self, mock_open):
210210

211211
class TestImageDelete(TestImage):
212212

213-
_image = image_fakes.FakeImage.create_one_image()
213+
_image = image_fakes.create_one_image()
214214

215215
def setUp(self):
216216
super(TestImageDelete, self).setUp()
@@ -239,7 +239,7 @@ def test_image_delete_no_options(self):
239239

240240
class TestImageList(TestImage):
241241

242-
_image = image_fakes.FakeImage.create_one_image()
242+
_image = image_fakes.create_one_image()
243243

244244
columns = (
245245
'ID',
@@ -443,7 +443,7 @@ def test_image_list_sort_option(self, si_mock):
443443

444444
class TestImageSet(TestImage):
445445

446-
_image = image_fakes.FakeImage.create_one_image()
446+
_image = image_fakes.create_one_image()
447447

448448
def setUp(self):
449449
super(TestImageSet, self).setUp()
@@ -682,8 +682,7 @@ def test_image_set_numeric_options_to_zero(self):
682682

683683
class TestImageShow(TestImage):
684684

685-
_image = image_fakes.FakeImage.create_one_image(
686-
attrs={'size': 2000})
685+
_image = image_fakes.create_one_image(attrs={'size': 2000})
687686
columns = (
688687
'container_format',
689688
'disk_format',

0 commit comments

Comments
 (0)