Skip to content

Commit adde9b3

Browse files
committed
Update test_task_handler_coverage.py
1 parent 1287c4a commit adde9b3

1 file changed

Lines changed: 3 additions & 28 deletions

File tree

tests/unit/automator/test_task_handler_coverage.py

Lines changed: 3 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -406,20 +406,7 @@ def tearDown(self):
406406
# Clean up any started processes
407407
for handler in self.handlers:
408408
try:
409-
# Terminate all task runner processes
410-
for process in handler.task_runner_processes:
411-
if process.is_alive():
412-
process.terminate()
413-
process.join(timeout=1)
414-
if process.is_alive():
415-
process.kill()
416-
# Terminate metrics process if it exists
417-
if hasattr(handler, 'metrics_provider_process') and handler.metrics_provider_process:
418-
if handler.metrics_provider_process.is_alive():
419-
handler.metrics_provider_process.terminate()
420-
handler.metrics_provider_process.join(timeout=1)
421-
if handler.metrics_provider_process.is_alive():
422-
handler.metrics_provider_process.kill()
409+
handler.stop_processes()
423410
except Exception:
424411
pass
425412

@@ -1006,13 +993,7 @@ def tearDown(self):
1006993
# Clean up any started processes
1007994
for handler in self.handlers:
1008995
try:
1009-
# Terminate all task runner processes
1010-
for process in handler.task_runner_processes:
1011-
if process.is_alive():
1012-
process.terminate()
1013-
process.join(timeout=1)
1014-
if process.is_alive():
1015-
process.kill()
996+
handler.stop_processes()
1016997
except Exception:
1017998
pass
1018999
# Restore original environment
@@ -1073,13 +1054,7 @@ def tearDown(self):
10731054
# Clean up any started processes
10741055
for handler in self.handlers:
10751056
try:
1076-
# Terminate all task runner processes
1077-
for process in handler.task_runner_processes:
1078-
if process.is_alive():
1079-
process.terminate()
1080-
process.join(timeout=1)
1081-
if process.is_alive():
1082-
process.kill()
1057+
handler.stop_processes()
10831058
except Exception:
10841059
pass
10851060

0 commit comments

Comments
 (0)