From 757b8dfadadd52240c9f1cd584a81fba76eeb8de Mon Sep 17 00:00:00 2001
From: Al Liu <scygliu1@gmail.com>
Date: Mon, 15 Apr 2024 02:24:22 +0800
Subject: [PATCH] WIP

---
 core/src/serf/base/tests/serf.rs        | 8 ++++++++
 core/src/serf/base/tests/serf/remove.rs | 8 ++++++++
 2 files changed, 16 insertions(+)

diff --git a/core/src/serf/base/tests/serf.rs b/core/src/serf/base/tests/serf.rs
index c21415a..b03e79f 100644
--- a/core/src/serf/base/tests/serf.rs
+++ b/core/src/serf/base/tests/serf.rs
@@ -641,6 +641,10 @@ pub async fn serf_coordinates<T>(
       panic!("timed out");
     }
   }
+
+  for s in serfs.iter() {
+    s.shutdown().await.unwrap();
+  }
 }
 
 /// Unit tests for serf name resolution
@@ -721,6 +725,10 @@ pub async fn serf_name_resolution<T>(
       panic!("timed out");
     }
   }
+
+  for s in serfs.iter() {
+    s.shutdown().await.unwrap();
+  }
 }
 
 /// Unit test for serf local member
diff --git a/core/src/serf/base/tests/serf/remove.rs b/core/src/serf/base/tests/serf/remove.rs
index aa4fb16..8a71d33 100644
--- a/core/src/serf/base/tests/serf/remove.rs
+++ b/core/src/serf/base/tests/serf/remove.rs
@@ -86,6 +86,10 @@ pub async fn serf_remove_failed_node<T>(
       panic!("Failed to remove failed node");
     }
   }
+
+  for s in serfs.iter() {
+    s.shutdown().await.unwrap();
+  }
 }
 
 /// Unit test for remove failed node prune
@@ -153,6 +157,10 @@ pub async fn serf_remove_failed_node_prune<T>(
   serfs.swap(1, 2);
 
   wait_until_num_nodes(2, &serfs[..2]).await;
+
+  for s in serfs.iter() {
+    s.shutdown().await.unwrap();
+  }
 }
 
 /// Unit test for remove failed node ourself