@@ -216,7 +216,8 @@ def test_operate_task_start(self, mock_task_operation):
216216 assert response .status_code == status .HTTP_200_OK
217217
218218 @patch ("bkflow.task.views.TaskOperation" )
219- def test_operate_task_pause (self , mock_task_operation ):
219+ @patch ("bkflow.task.views.InterfaceModuleClient" )
220+ def test_operate_task_pause (self , mock_interface_client , mock_task_operation ):
220221 """测试任务操作 - pause"""
221222 task_instance = TaskInstance .objects .create_instance (space_id = 1 , pipeline_tree = build_default_pipeline_tree ())
222223 # 启动任务
@@ -226,14 +227,18 @@ def test_operate_task_pause(self, mock_task_operation):
226227 mock_operation .pause .return_value = MagicMock (result = True , data = {}, message = "success" )
227228 mock_task_operation .return_value = mock_operation
228229
230+ # Mock InterfaceModuleClient
231+ mock_interface_client .return_value .broadcast_task_events .return_value = MagicMock ()
232+
229233 view = TaskInstanceViewSet .as_view ({"post" : "operate" })
230234 request = self ._create_request_with_auth ("post" , f"/task/{ task_instance .id } /operate/pause/" , {})
231235
232236 response = view (request , pk = task_instance .id , operation = "pause" )
233237 assert response .status_code == status .HTTP_200_OK
234238
235239 @patch ("bkflow.task.views.TaskOperation" )
236- def test_operate_task_resume (self , mock_task_operation ):
240+ @patch ("bkflow.task.views.InterfaceModuleClient" )
241+ def test_operate_task_resume (self , mock_interface_client , mock_task_operation ):
237242 """测试任务操作 - resume"""
238243 task_instance = TaskInstance .objects .create_instance (space_id = 1 , pipeline_tree = build_default_pipeline_tree ())
239244 # 启动任务
@@ -243,14 +248,18 @@ def test_operate_task_resume(self, mock_task_operation):
243248 mock_operation .resume .return_value = MagicMock (result = True , data = {}, message = "success" )
244249 mock_task_operation .return_value = mock_operation
245250
251+ # Mock InterfaceModuleClient
252+ mock_interface_client .return_value .broadcast_task_events .return_value = MagicMock ()
253+
246254 view = TaskInstanceViewSet .as_view ({"post" : "operate" })
247255 request = self ._create_request_with_auth ("post" , f"/task/{ task_instance .id } /operate/resume/" , {})
248256
249257 response = view (request , pk = task_instance .id , operation = "resume" )
250258 assert response .status_code == status .HTTP_200_OK
251259
252260 @patch ("bkflow.task.views.TaskOperation" )
253- def test_operate_task_revoke (self , mock_task_operation ):
261+ @patch ("bkflow.task.views.InterfaceModuleClient" )
262+ def test_operate_task_revoke (self , mock_interface_client , mock_task_operation ):
254263 """测试任务操作 - revoke"""
255264 task_instance = TaskInstance .objects .create_instance (space_id = 1 , pipeline_tree = build_default_pipeline_tree ())
256265 # 启动任务
@@ -260,6 +269,9 @@ def test_operate_task_revoke(self, mock_task_operation):
260269 mock_operation .revoke .return_value = MagicMock (result = True , data = {}, message = "success" )
261270 mock_task_operation .return_value = mock_operation
262271
272+ # Mock InterfaceModuleClient
273+ mock_interface_client .return_value .broadcast_task_events .return_value = MagicMock ()
274+
263275 view = TaskInstanceViewSet .as_view ({"post" : "operate" })
264276 request = self ._create_request_with_auth ("post" , f"/task/{ task_instance .id } /operate/revoke/" , {})
265277
0 commit comments