@@ -2380,6 +2380,57 @@ def test_server_remove_port_no_neutron(self):
23802380 self .find_port .assert_not_called ()
23812381
23822382
2383+ class TestServerRemoveNetwork (TestServer ):
2384+
2385+ def setUp (self ):
2386+ super (TestServerRemoveNetwork , self ).setUp ()
2387+
2388+ # Get the command object to test
2389+ self .cmd = server .RemoveNetwork (self .app , None )
2390+
2391+ # Set method to be tested.
2392+ self .fake_inf = mock .Mock ()
2393+ self .methods = {
2394+ 'interface_list' : [self .fake_inf ],
2395+ 'interface_detach' : None ,
2396+ }
2397+
2398+ self .find_network = mock .Mock ()
2399+ self .app .client_manager .network .find_network = self .find_network
2400+
2401+ def _test_server_remove_network (self , network_id ):
2402+ self .fake_inf .net_id = network_id
2403+ self .fake_inf .port_id = 'fake-port'
2404+ servers = self .setup_servers_mock (count = 1 )
2405+ network = 'fake-network'
2406+
2407+ arglist = [
2408+ servers [0 ].id ,
2409+ network ,
2410+ ]
2411+ verifylist = [
2412+ ('server' , servers [0 ].id ),
2413+ ('network' , network ),
2414+ ]
2415+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
2416+
2417+ result = self .cmd .take_action (parsed_args )
2418+
2419+ servers [0 ].interface_list .assert_called_once_with ()
2420+ servers [0 ].interface_detach .assert_called_once_with ('fake-port' )
2421+ self .assertIsNone (result )
2422+
2423+ def test_server_remove_network (self ):
2424+ self ._test_server_remove_network (self .find_network .return_value .id )
2425+ self .find_network .assert_called_once_with (
2426+ 'fake-network' , ignore_missing = False )
2427+
2428+ def test_server_remove_network_no_neutron (self ):
2429+ self .app .client_manager .network_endpoint_enabled = False
2430+ self ._test_server_remove_network ('fake-network' )
2431+ self .find_network .assert_not_called ()
2432+
2433+
23832434@mock .patch (
23842435 'openstackclient.api.compute_v2.APIv2.security_group_find'
23852436)
0 commit comments