Fix handling of wg in the networking dial test (#2132)

ava-labs · Oct 3, 2023 · 0022f37 · 0022f37
1 parent 903a588
commit 0022f37
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/network/network_test.go b/network/network_test.go
@@ -638,7 +638,6 @@ func TestDialDeletesNonValidators(t *testing.T) {
 // causes dial to return immediately.
 func TestDialContext(t *testing.T) {
 	_, networks, wg := newFullyConnectedTestNetwork(t, []router.InboundHandler{nil})
-	defer wg.Done()
 
 	dialer := newTestDialer()
 	network := networks[0]
@@ -689,4 +688,7 @@ func TestDialContext(t *testing.T) {
 		require.FailNow(t, "unexpectedly connected to peer")
 	default:
 	}
+
+	network.StartClose()
+	wg.Wait()
 }