Skip to content

Commit a4696dc

Browse files
committed
Use FakeImage in server test cases.
We have a class FakeImage to fake one or more images. So use it in test_server.py. Change-Id: I276e4ade5aecefbe66b9722c1dfbac10b3bd14b6 Implements: blueprint improve-image-unittest-framework
1 parent 481b711 commit a4696dc

1 file changed

Lines changed: 16 additions & 24 deletions

File tree

openstackclient/tests/compute/v2/test_server.py

Lines changed: 16 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -107,11 +107,7 @@ def setUp(self):
107107

108108
self.servers_mock.create.return_value = self.new_server
109109

110-
self.image = fakes.FakeResource(
111-
None,
112-
copy.deepcopy(image_fakes.IMAGE),
113-
loaded=True,
114-
)
110+
self.image = image_fakes.FakeImage.create_one_image()
115111
self.cimages_mock.get.return_value = self.image
116112

117113
self.flavor = compute_fakes.FakeFlavor.create_one_flavor()
@@ -537,13 +533,9 @@ def setUp(self):
537533
# This is the return value for utils.find_resource()
538534
self.servers_mock.get.return_value = self.server
539535

540-
self.servers_mock.create_image.return_value = image_fakes.image_id
541-
542-
self.images_mock.get.return_value = fakes.FakeResource(
543-
None,
544-
copy.deepcopy(image_fakes.IMAGE),
545-
loaded=True,
546-
)
536+
self.image = image_fakes.FakeImage.create_one_image()
537+
self.images_mock.get.return_value = self.image
538+
self.servers_mock.create_image.return_value = self.image.id
547539

548540
# Get the command object to test
549541
self.cmd = server.CreateServerImage(self.app, None)
@@ -569,12 +561,12 @@ def test_server_image_create_no_options(self):
569561
collist = ('id', 'name', 'owner', 'protected', 'tags', 'visibility')
570562
self.assertEqual(collist, columns)
571563
datalist = (
572-
image_fakes.image_id,
573-
image_fakes.image_name,
574-
image_fakes.image_owner,
575-
image_fakes.image_protected,
576-
image_fakes.image_tags,
577-
image_fakes.image_visibility,
564+
self.image.id,
565+
self.image.name,
566+
self.image.owner,
567+
self.image.protected,
568+
self.image.tags,
569+
self.image.visibility,
578570
)
579571
self.assertEqual(datalist, data)
580572

@@ -601,12 +593,12 @@ def test_server_image_create_name(self):
601593
collist = ('id', 'name', 'owner', 'protected', 'tags', 'visibility')
602594
self.assertEqual(collist, columns)
603595
datalist = (
604-
image_fakes.image_id,
605-
image_fakes.image_name,
606-
image_fakes.image_owner,
607-
image_fakes.image_protected,
608-
image_fakes.image_tags,
609-
image_fakes.image_visibility,
596+
self.image.id,
597+
self.image.name,
598+
self.image.owner,
599+
self.image.protected,
600+
self.image.tags,
601+
self.image.visibility,
610602
)
611603
self.assertEqual(datalist, data)
612604

0 commit comments

Comments
 (0)