Merge "tests: Simplify mocking in image tests"
This commit is contained in:
@@ -54,9 +54,9 @@ class TestImageCreate(image_fakes.TestImagev1):
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
|
||||
self.image_client.create_image = mock.Mock(return_value=self.new_image)
|
||||
self.image_client.find_image = mock.Mock(return_value=self.new_image)
|
||||
self.image_client.update_image = mock.Mock(return_image=self.new_image)
|
||||
self.image_client.create_image.return_value = self.new_image
|
||||
self.image_client.find_image.return_value = self.new_image
|
||||
self.image_client.update_image.return_value = self.new_image
|
||||
|
||||
# Get the command object to test
|
||||
self.cmd = image.CreateImage(self.app, None)
|
||||
@@ -212,8 +212,8 @@ class TestImageDelete(image_fakes.TestImagev1):
|
||||
super().setUp()
|
||||
|
||||
# This is the return value for utils.find_resource()
|
||||
self.image_client.find_image = mock.Mock(return_value=self._image)
|
||||
self.image_client.delete_image = mock.Mock(return_value=None)
|
||||
self.image_client.find_image.return_value = self._image
|
||||
self.image_client.delete_image.return_value = None
|
||||
|
||||
# Get the command object to test
|
||||
self.cmd = image.DeleteImage(self.app, None)
|
||||
@@ -261,7 +261,6 @@ class TestImageList(image_fakes.TestImagev1):
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
|
||||
self.image_client.images = mock.Mock()
|
||||
self.image_client.images.side_effect = [
|
||||
[self._image],
|
||||
[],
|
||||
@@ -441,8 +440,8 @@ class TestImageSet(image_fakes.TestImagev1):
|
||||
super().setUp()
|
||||
|
||||
# This is the return value for utils.find_resource()
|
||||
self.image_client.find_image = mock.Mock(return_value=self._image)
|
||||
self.image_client.update_image = mock.Mock(return_value=self._image)
|
||||
self.image_client.find_image.return_value = self._image
|
||||
self.image_client.update_image.return_value = self._image
|
||||
|
||||
# Get the command object to test
|
||||
self.cmd = image.SetImage(self.app, None)
|
||||
@@ -712,7 +711,7 @@ class TestImageShow(image_fakes.TestImagev1):
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
|
||||
self.image_client.find_image = mock.Mock(return_value=self._image)
|
||||
self.image_client.find_image.return_value = self._image
|
||||
|
||||
# Get the command object to test
|
||||
self.cmd = image.ShowImage(self.app, None)
|
||||
|
@@ -914,7 +914,7 @@ class TestImageList(TestImage):
|
||||
self.assertEqual(ret_limit, len(tuple(data)))
|
||||
|
||||
def test_image_list_project_option(self):
|
||||
self.image_client.find_image = mock.Mock(return_value=self._image)
|
||||
self.image_client.find_image.return_value = self._image
|
||||
arglist = [
|
||||
'--project',
|
||||
'nova',
|
||||
@@ -931,7 +931,7 @@ class TestImageList(TestImage):
|
||||
|
||||
@mock.patch('osc_lib.utils.find_resource')
|
||||
def test_image_list_marker_option(self, fr_mock):
|
||||
self.image_client.find_image = mock.Mock(return_value=self._image)
|
||||
self.image_client.find_image.return_value = self._image
|
||||
|
||||
arglist = [
|
||||
'--marker',
|
||||
@@ -1721,7 +1721,7 @@ class TestImageShow(TestImage):
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
|
||||
self.image_client.find_image = mock.Mock(return_value=self._data)
|
||||
self.image_client.find_image.return_value = self._data
|
||||
|
||||
# Get the command object to test
|
||||
self.cmd = _image.ShowImage(self.app, None)
|
||||
|
Reference in New Issue
Block a user