From: Matt Joiner <anacrolix@gmail.com>
Date: Thu, 16 Dec 2021 00:07:19 +0000 (+1100)
Subject: Fix race in TextPexConnState
X-Git-Tag: v1.39.2~1
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=12c77bc4102e5a858e97769c32913d038c91f454;p=btrtrc.git

Fix race in TextPexConnState
---

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)