Call `cleanups` after test signal

1 files changed, 9 insertions, 2 deletions

diff --git a/core/testing/runner.odin b/core/testing/runner.odin
index 328186c35..147c6d094 100644
--- a/core/testing/runner.odin
+++ b/core/testing/runner.odin
@@ -604,10 +604,10 @@ runner :: proc(internal_tests: []Internal_Test) -> bool {
 			})
 			fmt.assertf(alloc_error == nil, "Error appending to log messages: %v", alloc_error)
 
-			find_task_data: for &data in task_data_slots {
+			find_task_data_for_timeout: for &data in task_data_slots {
 				if data.it.pkg == it.pkg && data.it.name == it.name {
 					end_t(&data.t)
-					break find_task_data
+					break find_task_data_for_timeout
 				}
 			}
 		}
@@ -655,6 +655,13 @@ runner :: proc(internal_tests: []Internal_Test) -> bool {
 
 			}
 
+			find_task_data_for_stop_signal: for &data in task_data_slots {
+				if data.it.pkg == it.pkg && data.it.name == it.name {
+					end_t(&data.t)
+					break find_task_data_for_stop_signal
+				}
+			}
+
 			when FANCY_OUTPUT {
 				bypass_progress_overwrite = true
 				signals_were_raised = true