Include count of peer conns in status

[btrtrc.git] / reuse_test.go

diff --git a/reuse_test.go b/reuse_test.go
index 3adf002f6fbeb07d1edd4c2504ca2c9f8f870e4b..40df9a3e4778bbda4f0df70f99e996a610cb6089 100644 (file)
--- a/reuse_test.go
+++ b/reuse_test.go
@@ -42,7 +42,9 @@ func TestUtpLocalPortIsReusable(t *testing.T) {
        c.Assert(err, qt.IsNil)
        defer remote.Close()
        var remoteAccepts int32
+       doneAccepting := make(chan struct{})
        go func() {
+               defer close(doneAccepting)
                for {
                        c, err := remote.Accept()
                        if err != nil {
@@ -66,5 +68,6 @@ func TestUtpLocalPortIsReusable(t *testing.T) {
        c.Assert(err, qt.IsNil)
        defer second.Close()
        remote.Close()
+       <-doneAccepting
        c.Assert(atomic.LoadInt32(&remoteAccepts), qt.Equals, int32(2))
 }