Close test's websocket on shutdown

nullstalgia · Sep 24, 2024 · 52c7d83 · 52c7d83
1 parent d0ba664
commit 52c7d83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/websocket_to_txt.rs b/tests/websocket_to_txt.rs
@@ -103,11 +103,11 @@ async fn websocket_to_txt() -> Result<(), iron_heart::errors::AppError> {
         assert_eq!(file_contents.trim().parse::<u16>()?, hr);
     }
 
-    std::fs::remove_file(file_dir)?;
-
     println!("Shutting down, all ok");
 
     parent_token.cancel();
+    std::fs::remove_file(file_dir)?;
+    client.close().await?;
     let _ = app_thread.join();
     Ok(())
 }