Fix race in TextPexConnState

diff --git a/pexconn_test.go b/pexconn_test.go
index caee9d849fd4d601661eb38ce8a1cff5b242dc57..42c62cff69168ed788b849692f7626572d225c46 100644 (file)
--- 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)