@@ -88,6 +88,41 @@ def run_method_with_servers(self, method_name, server_count):
8888 self .assertIsNone (result )
8989
9090
91+ class TestServerAddFloatingIP (TestServer ):
92+
93+ def setUp (self ):
94+ super (TestServerAddFloatingIP , self ).setUp ()
95+
96+ # Get a shortcut to the compute client ServerManager Mock
97+ self .networks_mock = self .app .client_manager .compute .networks
98+
99+ # Get the command object to test
100+ self .cmd = server .AddFloatingIP (self .app , None )
101+
102+ # Set add_floating_ip method to be tested.
103+ self .methods = {
104+ 'add_floating_ip' : None ,
105+ }
106+
107+ def test_server_add_floating_ip (self ):
108+ servers = self .setup_servers_mock (count = 1 )
109+
110+ arglist = [
111+ servers [0 ].id ,
112+ '1.2.3.4' ,
113+ ]
114+ verifylist = [
115+ ('server' , servers [0 ].id ),
116+ ('ip_address' , '1.2.3.4' ),
117+ ]
118+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
119+
120+ result = self .cmd .take_action (parsed_args )
121+
122+ servers [0 ].add_floating_ip .assert_called_once_with ('1.2.3.4' )
123+ self .assertIsNone (result )
124+
125+
91126class TestServerCreate (TestServer ):
92127
93128 columns = (
@@ -843,6 +878,38 @@ def test_rebuild_with_wait_fails(self, mock_wait_for_status):
843878 self .server .rebuild .assert_called_with (self .image , None )
844879
845880
881+ class TestServerRemoveFloatingIP (TestServer ):
882+
883+ def setUp (self ):
884+ super (TestServerRemoveFloatingIP , self ).setUp ()
885+
886+ # Get the command object to test
887+ self .cmd = server .RemoveFloatingIP (self .app , None )
888+
889+ # Set unshelve method to be tested.
890+ self .methods = {
891+ 'remove_floating_ip' : None ,
892+ }
893+
894+ def test_server_remove_floating_ip (self ):
895+ servers = self .setup_servers_mock (count = 1 )
896+
897+ arglist = [
898+ servers [0 ].id ,
899+ '1.2.3.4' ,
900+ ]
901+ verifylist = [
902+ ('server' , servers [0 ].id ),
903+ ('ip_address' , '1.2.3.4' ),
904+ ]
905+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
906+
907+ result = self .cmd .take_action (parsed_args )
908+
909+ servers [0 ].remove_floating_ip .assert_called_once_with ('1.2.3.4' )
910+ self .assertIsNone (result )
911+
912+
846913class TestServerResize (TestServer ):
847914
848915 def setUp (self ):
0 commit comments