1515
1616import mock
1717from mock import call
18-
18+ from novaclient import api_versions
1919from osc_lib import exceptions
2020
2121from openstackclient .compute .v2 import service
@@ -340,6 +340,8 @@ def test_service_set_state_up(self):
340340 ('service' , self .service .binary ),
341341 ]
342342 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
343+ self .app .client_manager .compute .api_version = api_versions .APIVersion (
344+ '2.11' )
343345 result = self .cmd .take_action (parsed_args )
344346 self .service_mock .force_down .assert_called_once_with (
345347 self .service .host , self .service .binary , force_down = False )
@@ -359,6 +361,8 @@ def test_service_set_state_down(self):
359361 ('service' , self .service .binary ),
360362 ]
361363 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
364+ self .app .client_manager .compute .api_version = api_versions .APIVersion (
365+ '2.11' )
362366 result = self .cmd .take_action (parsed_args )
363367 self .service_mock .force_down .assert_called_once_with (
364368 self .service .host , self .service .binary , force_down = True )
@@ -380,6 +384,8 @@ def test_service_set_enable_and_state_down(self):
380384 ('service' , self .service .binary ),
381385 ]
382386 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
387+ self .app .client_manager .compute .api_version = api_versions .APIVersion (
388+ '2.11' )
383389 result = self .cmd .take_action (parsed_args )
384390 self .service_mock .enable .assert_called_once_with (
385391 self .service .host , self .service .binary )
@@ -402,6 +408,8 @@ def test_service_set_enable_and_state_down_with_exception(self):
402408 ]
403409 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
404410
411+ self .app .client_manager .compute .api_version = api_versions .APIVersion (
412+ '2.11' )
405413 with mock .patch .object (self .service_mock , 'enable' ,
406414 side_effect = Exception ()):
407415 self .assertRaises (exceptions .CommandError ,
0 commit comments