Merge pull request #31 from libp2p/fix/reset-test-race

fix race in reset test
libp2p · Mar 11, 2018 · 3f15e2c · 3f15e2c
2 parents 2ba1319 + b78f145
commit 3f15e2c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/p2p/protocol/internal/circuitv1-deprecated/relay_test.go b/p2p/protocol/internal/circuitv1-deprecated/relay_test.go
@@ -137,6 +137,8 @@ func TestRelayReset(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	ready := make(chan struct{})
+
 	msg := []byte("relay works!")
 	go func() {
 		list := r3.Listener()
@@ -152,6 +154,8 @@ func TestRelayReset(t *testing.T) {
 			t.Error(err)
 			return
 		}
+		<-ready
+
 		hosts[2].Close()
 	}()
 
@@ -166,6 +170,8 @@ func TestRelayReset(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	close(ready)
+
 	_, err = ioutil.ReadAll(con)
 	if err == nil {
 		t.Fatal("expected error for reset relayed connection")