make this one not racey

Noah-Kennedy · Noah-Kennedy · commit a9570ccc525b · 2025-03-07T13:39:33.000-06:00
diff --git a/tokio/tests/task_hooks.rs b/tokio/tests/task_hooks.rs
@@ -378,9 +378,10 @@ fn run_override(mut builder: runtime::Builder) {
     }
 
     impl TaskHookHarness for TestHooks {
-        fn on_task_terminate(&mut self, _ctx: &mut OnTaskTerminateContext<'_>) {
+        fn before_poll(&mut self, _ctx: &mut BeforeTaskPollContext<'_>) {
             self.counter.fetch_add(1, Ordering::SeqCst);
         }
+
         fn on_child_spawn(
             &mut self,
             _ctx: &mut OnChildTaskSpawnContext<'_>,

Original file line number	Diff line number	Diff line change
`@@ -378,9 +378,10 @@ fn run_override(mut builder: runtime::Builder) {`
`378`	`378`	`}`
`379`	`379`
`380`	`380`	`impl TaskHookHarness for TestHooks {`
`381`		`- fn on_task_terminate(&mut self, _ctx: &mut OnTaskTerminateContext<'_>) {`
	`381`	`+ fn before_poll(&mut self, _ctx: &mut BeforeTaskPollContext<'_>) {`
`382`	`382`	`self.counter.fetch_add(1, Ordering::SeqCst);`
`383`	`383`	`}`
	`384`	`+`
`384`	`385`	`fn on_child_spawn(`
`385`	`386`	`&mut self,`
`386`	`387`	`_ctx: &mut OnChildTaskSpawnContext<'_>,`