@@ -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