From 12c77bc4102e5a858e97769c32913d038c91f454 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Thu, 16 Dec 2021 11:07:19 +1100
Subject: [PATCH] Fix race in TextPexConnState

---
 pexconn_test.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pexconn_test.go b/pexconn_test.go
index caee9d84..42c62cff 100644
--- a/pexconn_test.go
+++ b/pexconn_test.go
@@ -26,6 +26,7 @@ func TestPexConnState(t *testing.T) {
 		t.Log(err)
 	}
 
+	connWriteCond := c.messageWriter.writeCond.Signaled()
 	c.pex.Init(c)
 	require.True(t, c.pex.IsEnabled(), "should get enabled")
 	defer c.pex.Close()
@@ -37,7 +38,7 @@ func TestPexConnState(t *testing.T) {
 		out = m
 		return true
 	}
-	<-c.messageWriter.writeCond.Signaled()
+	<-connWriteCond
 	c.pex.Share(testWriter)
 	require.True(t, writerCalled)
 	require.EqualValues(t, pp.Extended, out.Type)
-- 
2.51.0