@@ -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 )
0 commit comments