@@ -8216,28 +8216,19 @@ def test_server_ssh_deprecated_opts(self, mock_exec):
82168216class TestServerStart (TestServer ):
82178217
82188218 def setUp (self ):
8219- super (TestServerStart , self ).setUp ()
8219+ super ().setUp ()
82208220
82218221 # Get the command object to test
82228222 self .cmd = server .StartServer (self .app , None )
82238223
8224- # Set methods to be tested.
8225- self .methods = {
8226- 'start' : None ,
8227- }
8228-
82298224 def test_server_start_one_server (self ):
8230- self .run_method_with_servers ( 'start ' , 1 )
8225+ self .run_method_with_sdk_servers ( 'start_server ' , 1 )
82318226
82328227 def test_server_start_multi_servers (self ):
8233- self .run_method_with_servers ( 'start ' , 3 )
8228+ self .run_method_with_sdk_servers ( 'start_server ' , 3 )
82348229
8235- @mock .patch .object (common_utils , 'find_resource' )
8236- def test_server_start_with_all_projects (self , mock_find_resource ):
8237- servers = self .setup_servers_mock (count = 1 )
8238- mock_find_resource .side_effect = compute_fakes .FakeServer .get_servers (
8239- servers , 0 ,
8240- )
8230+ def test_server_start_with_all_projects (self ):
8231+ servers = self .setup_sdk_servers_mock (count = 1 )
82418232
82428233 arglist = [
82438234 servers [0 ].id ,
@@ -8250,36 +8241,30 @@ def test_server_start_with_all_projects(self, mock_find_resource):
82508241
82518242 self .cmd .take_action (parsed_args )
82528243
8253- mock_find_resource .assert_called_once_with (
8254- mock .ANY , servers [0 ].id , all_tenants = True ,
8244+ self .sdk_client .find_server .assert_called_once_with (
8245+ name = servers [0 ].id ,
8246+ details = False ,
8247+ all_projects = True ,
8248+ ignore_missing = False ,
82558249 )
82568250
82578251
82588252class TestServerStop (TestServer ):
82598253
82608254 def setUp (self ):
8261- super (TestServerStop , self ).setUp ()
8255+ super ().setUp ()
82628256
82638257 # Get the command object to test
82648258 self .cmd = server .StopServer (self .app , None )
82658259
8266- # Set methods to be tested.
8267- self .methods = {
8268- 'stop' : None ,
8269- }
8270-
82718260 def test_server_stop_one_server (self ):
8272- self .run_method_with_servers ( 'stop ' , 1 )
8261+ self .run_method_with_sdk_servers ( 'stop_server ' , 1 )
82738262
82748263 def test_server_stop_multi_servers (self ):
8275- self .run_method_with_servers ( 'stop ' , 3 )
8264+ self .run_method_with_sdk_servers ( 'stop_server ' , 3 )
82768265
8277- @mock .patch .object (common_utils , 'find_resource' )
8278- def test_server_start_with_all_projects (self , mock_find_resource ):
8266+ def test_server_start_with_all_projects (self ):
82798267 servers = self .setup_servers_mock (count = 1 )
8280- mock_find_resource .side_effect = compute_fakes .FakeServer .get_servers (
8281- servers , 0 ,
8282- )
82838268
82848269 arglist = [
82858270 servers [0 ].id ,
@@ -8292,8 +8277,11 @@ def test_server_start_with_all_projects(self, mock_find_resource):
82928277
82938278 self .cmd .take_action (parsed_args )
82948279
8295- mock_find_resource .assert_called_once_with (
8296- mock .ANY , servers [0 ].id , all_tenants = True ,
8280+ self .sdk_client .find_server .assert_called_once_with (
8281+ name = servers [0 ].id ,
8282+ details = False ,
8283+ all_projects = True ,
8284+ ignore_missing = False ,
82978285 )
82988286
82998287
0 commit comments