Skip to content

Commit fccccde

Browse files
committed
fix: 修复单侧失败问题 --story=129476203
# Reviewed, transaction id: 71494
1 parent ab57beb commit fccccde

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

tests/engine/task/test_task_views.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)