@@ -192,6 +192,22 @@ def run_method_with_servers(self, method_name, server_count):
192192 method .assert_called_with ()
193193 self .assertIsNone (result )
194194
195+ def run_method_with_sdk_servers (self , method_name , server_count ):
196+ servers = self .setup_sdk_servers_mock (count = server_count )
197+
198+ arglist = [s .id for s in servers ]
199+ verifylist = [
200+ ('server' , arglist ),
201+ ]
202+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
203+
204+ result = self .cmd .take_action (parsed_args )
205+
206+ calls = [call (s .id ) for s in servers ]
207+ method = getattr (self .sdk_client , method_name )
208+ method .assert_has_calls (calls )
209+ self .assertIsNone (result )
210+
195211
196212class TestServerAddFixedIP (TestServer ):
197213
@@ -6062,10 +6078,10 @@ def setUp(self):
60626078 }
60636079
60646080 def test_server_pause_one_server (self ):
6065- self .run_method_with_servers ( 'pause ' , 1 )
6081+ self .run_method_with_sdk_servers ( 'pause_server ' , 1 )
60666082
60676083 def test_server_pause_multi_servers (self ):
6068- self .run_method_with_servers ( 'pause ' , 3 )
6084+ self .run_method_with_sdk_servers ( 'pause_server ' , 3 )
60696085
60706086
60716087class TestServerRebuild (TestServer ):
@@ -8308,10 +8324,10 @@ def setUp(self):
83088324 }
83098325
83108326 def test_server_unpause_one_server (self ):
8311- self .run_method_with_servers ( 'unpause ' , 1 )
8327+ self .run_method_with_sdk_servers ( 'unpause_server ' , 1 )
83128328
83138329 def test_server_unpause_multi_servers (self ):
8314- self .run_method_with_servers ( 'unpause ' , 3 )
8330+ self .run_method_with_sdk_servers ( 'unpause_server ' , 3 )
83158331
83168332
83178333class TestServerUnset (TestServer ):
0 commit comments