lib/protocol: Send Close message on read error (#7141)

lib/testutils/testutils.go

@@ -6,17 +6,40 @@
 
 package testutils
 
-// BlockingRW implements io.Reader and Writer but never returns when called
-type BlockingRW struct{ nilChan chan struct{} }
+import (
+	"errors"
+	"sync"
+)
 
-func (rw *BlockingRW) Read(p []byte) (n int, err error) {
-	<-rw.nilChan
-	return
+var ErrClosed = errors.New("closed")
+
+// BlockingRW implements io.Reader, Writer and Closer, but only returns when closed
+type BlockingRW struct {
+	c         chan struct{}
+	closeOnce sync.Once
 }
 
-func (rw *BlockingRW) Write(p []byte) (n int, err error) {
-	<-rw.nilChan
-	return
+func NewBlockingRW() *BlockingRW {
+	return &BlockingRW{
+		c:         make(chan struct{}),
+		closeOnce: sync.Once{},
+	}
+}
+func (rw *BlockingRW) Read(p []byte) (int, error) {
+	<-rw.c
+	return 0, ErrClosed
+}
+
+func (rw *BlockingRW) Write(p []byte) (int, error) {
+	<-rw.c
+	return 0, ErrClosed
+}
+
+func (rw *BlockingRW) Close() error {
+	rw.closeOnce.Do(func() {
+		close(rw.c)
+	})
+	return nil
 }
 
 // NoopRW implements io.Reader and Writer but never returns when called