@@ -121,10 +121,6 @@ class TestDeleteNetworkAgent(TestNetworkAgent):
121121 def setUp (self ):
122122 super (TestDeleteNetworkAgent , self ).setUp ()
123123 self .network .delete_agent = mock .Mock (return_value = None )
124- self .network .get_agent = (
125- network_fakes .FakeNetworkAgent .get_network_agents (
126- agents = self .network_agents )
127- )
128124
129125 # Get the command object to test
130126 self .cmd = network_agent .DeleteNetworkAgent (self .app , self .namespace )
@@ -140,10 +136,8 @@ def test_network_agent_delete(self):
140136 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
141137
142138 result = self .cmd .take_action (parsed_args )
143- self .network .get_agent .assert_called_once_with (
144- self .network_agents [0 ].id , ignore_missing = False )
145139 self .network .delete_agent .assert_called_once_with (
146- self .network_agents [0 ])
140+ self .network_agents [0 ]. id , ignore_missing = False )
147141 self .assertIsNone (result )
148142
149143 def test_multi_network_agents_delete (self ):
@@ -160,7 +154,7 @@ def test_multi_network_agents_delete(self):
160154
161155 calls = []
162156 for n in self .network_agents :
163- calls .append (call (n ))
157+ calls .append (call (n . id , ignore_missing = False ))
164158 self .network .delete_agent .assert_has_calls (calls )
165159 self .assertIsNone (result )
166160
@@ -175,9 +169,9 @@ def test_multi_network_agents_delete_with_exception(self):
175169 ]
176170 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
177171
178- find_mock_result = [self . network_agents [ 0 ] , exceptions .CommandError ]
179- self .network .get_agent = (
180- mock .Mock (side_effect = find_mock_result )
172+ delete_mock_result = [True , exceptions .CommandError ]
173+ self .network .delete_agent = (
174+ mock .Mock (side_effect = delete_mock_result )
181175 )
182176
183177 try :
@@ -186,13 +180,10 @@ def test_multi_network_agents_delete_with_exception(self):
186180 except exceptions .CommandError as e :
187181 self .assertEqual ('1 of 2 network agents failed to delete.' , str (e ))
188182
189- self .network .get_agent .assert_any_call (
183+ self .network .delete_agent .assert_any_call (
190184 self .network_agents [0 ].id , ignore_missing = False )
191- self .network .get_agent .assert_any_call (
185+ self .network .delete_agent .assert_any_call (
192186 'unexist_network_agent' , ignore_missing = False )
193- self .network .delete_agent .assert_called_once_with (
194- self .network_agents [0 ]
195- )
196187
197188
198189class TestListNetworkAgent (TestNetworkAgent ):
0 commit comments