Skip to content

Commit 5fb4559

Browse files
Zuulopenstack-gerrit
authored andcommitted
Merge "tests: Simplify mocking in image tests"
2 parents 50dd542 + 444a1df commit 5fb4559

File tree

2 files changed

+11
-12
lines changed

2 files changed

+11
-12
lines changed

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

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,9 @@ class TestImageCreate(image_fakes.TestImagev1):
5454
def setUp(self):
5555
super().setUp()
5656

57-
self.image_client.create_image = mock.Mock(return_value=self.new_image)
58-
self.image_client.find_image = mock.Mock(return_value=self.new_image)
59-
self.image_client.update_image = mock.Mock(return_image=self.new_image)
57+
self.image_client.create_image.return_value = self.new_image
58+
self.image_client.find_image.return_value = self.new_image
59+
self.image_client.update_image.return_value = self.new_image
6060

6161
# Get the command object to test
6262
self.cmd = image.CreateImage(self.app, None)
@@ -212,8 +212,8 @@ def setUp(self):
212212
super().setUp()
213213

214214
# This is the return value for utils.find_resource()
215-
self.image_client.find_image = mock.Mock(return_value=self._image)
216-
self.image_client.delete_image = mock.Mock(return_value=None)
215+
self.image_client.find_image.return_value = self._image
216+
self.image_client.delete_image.return_value = None
217217

218218
# Get the command object to test
219219
self.cmd = image.DeleteImage(self.app, None)
@@ -261,7 +261,6 @@ class TestImageList(image_fakes.TestImagev1):
261261
def setUp(self):
262262
super().setUp()
263263

264-
self.image_client.images = mock.Mock()
265264
self.image_client.images.side_effect = [
266265
[self._image],
267266
[],
@@ -441,8 +440,8 @@ def setUp(self):
441440
super().setUp()
442441

443442
# This is the return value for utils.find_resource()
444-
self.image_client.find_image = mock.Mock(return_value=self._image)
445-
self.image_client.update_image = mock.Mock(return_value=self._image)
443+
self.image_client.find_image.return_value = self._image
444+
self.image_client.update_image.return_value = self._image
446445

447446
# Get the command object to test
448447
self.cmd = image.SetImage(self.app, None)
@@ -712,7 +711,7 @@ class TestImageShow(image_fakes.TestImagev1):
712711
def setUp(self):
713712
super().setUp()
714713

715-
self.image_client.find_image = mock.Mock(return_value=self._image)
714+
self.image_client.find_image.return_value = self._image
716715

717716
# Get the command object to test
718717
self.cmd = image.ShowImage(self.app, None)

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -914,7 +914,7 @@ def test_image_list_limit_option(self):
914914
self.assertEqual(ret_limit, len(tuple(data)))
915915

916916
def test_image_list_project_option(self):
917-
self.image_client.find_image = mock.Mock(return_value=self._image)
917+
self.image_client.find_image.return_value = self._image
918918
arglist = [
919919
'--project',
920920
'nova',
@@ -931,7 +931,7 @@ def test_image_list_project_option(self):
931931

932932
@mock.patch('osc_lib.utils.find_resource')
933933
def test_image_list_marker_option(self, fr_mock):
934-
self.image_client.find_image = mock.Mock(return_value=self._image)
934+
self.image_client.find_image.return_value = self._image
935935

936936
arglist = [
937937
'--marker',
@@ -1721,7 +1721,7 @@ class TestImageShow(TestImage):
17211721
def setUp(self):
17221722
super().setUp()
17231723

1724-
self.image_client.find_image = mock.Mock(return_value=self._data)
1724+
self.image_client.find_image.return_value = self._data
17251725

17261726
# Get the command object to test
17271727
self.cmd = _image.ShowImage(self.app, None)

0 commit comments

Comments
 (0)