From 846167c63e593c07201035a83dfe9fe736abef0a Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Thu, 30 Sep 2021 12:01:34 +1000
Subject: [PATCH] Fix mainReadLoop benchmark

---
 peerconn_test.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/peerconn_test.go b/peerconn_test.go
index 5e8cc1b5..be88bf47 100644
--- a/peerconn_test.go
+++ b/peerconn_test.go
@@ -1,6 +1,7 @@
 package torrent
 
 import (
+	"errors"
 	"io"
 	"net"
 	"sync"
@@ -107,7 +108,7 @@ func BenchmarkConnectionMainReadLoop(b *testing.B) {
 	r, w := net.Pipe()
 	cn := cl.newConnection(r, true, r.RemoteAddr(), r.RemoteAddr().Network(), regularNetConnPeerConnConnString(r))
 	cn.setTorrent(t)
-	mrlErr := make(chan error)
+	mrlErrChan := make(chan error)
 	msg := pp.Message{
 		Type:  pp.Piece,
 		Piece: make([]byte, defaultChunkSize),
@@ -116,9 +117,9 @@ func BenchmarkConnectionMainReadLoop(b *testing.B) {
 		cl.lock()
 		err := cn.mainReadLoop()
 		if err != nil {
-			mrlErr <- err
+			mrlErrChan <- err
 		}
-		close(mrlErr)
+		close(mrlErrChan)
 	}()
 	wb := msg.MustMarshalBinary()
 	b.SetBytes(int64(len(msg.Piece)))
@@ -142,7 +143,10 @@ func BenchmarkConnectionMainReadLoop(b *testing.B) {
 			panic(err)
 		}
 	}()
-	c.Assert([]error{nil, io.EOF}, quicktest.Contains, <-mrlErr)
+	mrlErr := <-mrlErrChan
+	if mrlErr != nil && !errors.Is(mrlErr, io.EOF) {
+		c.Fatal(mrlErr)
+	}
 	c.Assert(cn._stats.ChunksReadUseful.Int64(), quicktest.Equals, int64(b.N))
 }