[Rust Frontend] Fix mock engine test shutdown race (#48738)
Signed-off-by: reidliu41 <reid201711@gmail.com>
Reid committed
Jul 16, 2026 at 15:27 UTC
dc9f845ddc54c1df38fdbce5afe03f9fd15813bd
1 file changed
+1
-1
rust/src/mock-engine/src/tests.rs
+1
-1
@@ -90,8 +90,8 @@ async fn shutdown_mock(
90
shutdown: CancellationToken,
91
task: tokio::task::JoinHandle<Result<()>>,
92
) {
93
- client.shutdown().await.expect("client shutdown");
93
shutdown.cancel();
94
+ client.shutdown().await.expect("client shutdown");
95
task.await.expect("mock join").expect("mock run");
96
}
97