@@ -55,6 +55,7 @@ def setUp(self):
5555 self .new_image = image_fakes .create_one_image ()
5656 self .image_client .create_image .return_value = self .new_image
5757 self .image_client .update_image .return_value = self .new_image
58+ self .image_client .get_image .return_value = self .new_image
5859
5960 self .project_mock .get .return_value = self .project
6061
@@ -89,6 +90,7 @@ def test_image_reserve_no_options(self, raw_input):
8990 container_format = _image .DEFAULT_CONTAINER_FORMAT ,
9091 disk_format = _image .DEFAULT_DISK_FORMAT ,
9192 )
93+ self .image_client .get_image .assert_called_once_with (self .new_image )
9294
9395 self .assertEqual (self .expected_columns , columns )
9496 self .assertCountEqual (self .expected_data , data )
@@ -146,6 +148,7 @@ def test_image_reserve_options(self, raw_input):
146148 is_protected = self .new_image .is_protected ,
147149 visibility = self .new_image .visibility ,
148150 )
151+ self .image_client .get_image .assert_called_once_with (self .new_image )
149152
150153 self .assertEqual (self .expected_columns , columns )
151154 self .assertCountEqual (self .expected_data , data )
@@ -243,6 +246,7 @@ def test_image_create_file(self):
243246 tags = self .new_image .tags ,
244247 filename = imagefile .name ,
245248 )
249+ self .image_client .get_image .assert_called_once_with (self .new_image )
246250
247251 self .assertEqual (self .expected_columns , columns )
248252 self .assertCountEqual (self .expected_data , data )
@@ -275,6 +279,7 @@ def test_image_create__progress_ignore_with_stdin(
275279 data = fake_stdin ,
276280 validate_checksum = False ,
277281 )
282+ self .image_client .get_image .assert_called_once_with (self .new_image )
278283
279284 self .assertEqual (self .expected_columns , columns )
280285 self .assertCountEqual (self .expected_data , data )
@@ -315,6 +320,7 @@ def test_image_create_import(self, raw_input):
315320 disk_format = _image .DEFAULT_DISK_FORMAT ,
316321 use_import = True ,
317322 )
323+ self .image_client .get_image .assert_called_once_with (self .new_image )
318324
319325 @mock .patch ('osc_lib.utils.find_resource' )
320326 @mock .patch ('openstackclient.image.v2.image.get_data_from_stdin' )
0 commit comments