@@ -36,10 +36,6 @@ def setUp(self):
3636 self .servers_mock = self .app .client_manager .compute .servers
3737 self .servers_mock .reset_mock ()
3838
39- # Get a shortcut to the compute client ImageManager Mock
40- self .cimages_mock = self .app .client_manager .compute .images
41- self .cimages_mock .reset_mock ()
42-
4339 # Get a shortcut to the compute client FlavorManager Mock
4440 self .flavors_mock = self .app .client_manager .compute .flavors
4541 self .flavors_mock .reset_mock ()
@@ -259,7 +255,7 @@ def setUp(self):
259255 self .servers_mock .create .return_value = self .new_server
260256
261257 self .image = image_fakes .FakeImage .create_one_image ()
262- self .cimages_mock .get .return_value = self .image
258+ self .images_mock .get .return_value = self .image
263259
264260 self .flavor = compute_fakes .FakeFlavor .create_one_flavor ()
265261 self .flavors_mock .get .return_value = self .flavor
@@ -859,7 +855,7 @@ def setUp(self):
859855 self .servers_mock .list .return_value = self .servers
860856
861857 self .image = image_fakes .FakeImage .create_one_image ()
862- self .cimages_mock .get .return_value = self .image
858+ self .images_mock .get .return_value = self .image
863859
864860 self .flavor = compute_fakes .FakeFlavor .create_one_flavor ()
865861 self .flavors_mock .get .return_value = self .flavor
@@ -943,7 +939,7 @@ def test_server_list_with_image(self):
943939 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
944940 columns , data = self .cmd .take_action (parsed_args )
945941
946- self .cimages_mock .get .assert_any_call (self .image .id )
942+ self .images_mock .get .assert_any_call (self .image .id )
947943
948944 self .search_opts ['image' ] = self .image .id
949945 self .servers_mock .list .assert_called_with (** self .kwargs )
@@ -1019,7 +1015,7 @@ def setUp(self):
10191015
10201016 # Return value for utils.find_resource for image
10211017 self .image = image_fakes .FakeImage .create_one_image ()
1022- self .cimages_mock .get .return_value = self .image
1018+ self .images_mock .get .return_value = self .image
10231019
10241020 # Fake the rebuilt new server.
10251021 new_server = compute_fakes .FakeServer .create_one_server ()
@@ -1059,7 +1055,7 @@ def test_rebuild_with_current_image(self):
10591055 self .cmd .take_action (parsed_args )
10601056
10611057 self .servers_mock .get .assert_called_with (self .server .id )
1062- self .cimages_mock .get .assert_called_with (self .image .id )
1058+ self .images_mock .get .assert_called_with (self .image .id )
10631059 self .server .rebuild .assert_called_with (self .image , None )
10641060
10651061 def test_rebuild_with_current_image_and_password (self ):
@@ -1078,7 +1074,7 @@ def test_rebuild_with_current_image_and_password(self):
10781074 self .cmd .take_action (parsed_args )
10791075
10801076 self .servers_mock .get .assert_called_with (self .server .id )
1081- self .cimages_mock .get .assert_called_with (self .image .id )
1077+ self .images_mock .get .assert_called_with (self .image .id )
10821078 self .server .rebuild .assert_called_with (self .image , password )
10831079
10841080 @mock .patch .object (common_utils , 'wait_for_status' , return_value = True )
@@ -1106,7 +1102,7 @@ def test_rebuild_with_wait_ok(self, mock_wait_for_status):
11061102 )
11071103
11081104 self .servers_mock .get .assert_called_with (self .server .id )
1109- self .cimages_mock .get .assert_called_with (self .image .id )
1105+ self .images_mock .get .assert_called_with (self .image .id )
11101106 self .server .rebuild .assert_called_with (self .image , None )
11111107
11121108 @mock .patch .object (common_utils , 'wait_for_status' , return_value = False )
@@ -1130,7 +1126,7 @@ def test_rebuild_with_wait_fails(self, mock_wait_for_status):
11301126 )
11311127
11321128 self .servers_mock .get .assert_called_with (self .server .id )
1133- self .cimages_mock .get .assert_called_with (self .image .id )
1129+ self .images_mock .get .assert_called_with (self .image .id )
11341130 self .server .rebuild .assert_called_with (self .image , None )
11351131
11361132
@@ -1628,7 +1624,7 @@ def setUp(self):
16281624
16291625 # This is the return value for utils.find_resource()
16301626 self .servers_mock .get .return_value = self .server
1631- self .cimages_mock .get .return_value = self .image
1627+ self .images_mock .get .return_value = self .image
16321628 self .flavors_mock .get .return_value = self .flavor
16331629
16341630 # Get the command object to test
@@ -1986,6 +1982,7 @@ def test_prep_server_detail(self, find_resource):
19861982 # Call _prep_server_detail().
19871983 server_detail = server ._prep_server_detail (
19881984 self .app .client_manager .compute ,
1985+ self .app .client_manager .image ,
19891986 _server
19901987 )
19911988 # 'networks' is used to create _server. Remove it.
0 commit comments