[bfops/wasm-test]: refactor

Author: bfops

sdks/rust/tests/connect_disconnect_client/src/main.rs

@@ -148,12 +148,9 @@ pub(crate) async fn dispatch() {
 }
 
 async fn wait_for_all(test_counter: &std::sync::Arc<TestCounter>) {
+    // wasm/web callbacks run on the JS event loop, so this wait must stay async.
     #[cfg(target_arch = "wasm32")]
-    {
-        // wasm/web callbacks run on the JS event loop, so this wait must stay async.
-        test_counter.wait_for_all_async().await;
-        return;
-    }
+    test_counter.wait_for_all_async().await;
 
     #[cfg(not(target_arch = "wasm32"))]
     test_counter.wait_for_all();

sdks/rust/tests/test-counter/src/lib.rs

@@ -66,12 +66,10 @@ impl TestCounter {
     }
 
     pub async fn wait_for_all_async(&self) {
+        // wasm/web test clients run callbacks on a single-threaded event loop,
+        // so waiting must be async to allow callback tasks to make progress.
         #[cfg(target_arch = "wasm32")]
-        {
-            // wasm/web test clients run callbacks on a single-threaded event loop,
-            // so waiting must be async to allow callback tasks to make progress.
-            self.wait_for_all_wasm_async().await;
-        }
+        self.wait_for_all_wasm_async().await;
 
         #[cfg(not(target_arch = "wasm32"))]
         self.wait_for_all_native();