diff --git a/p2p/test/reconnects/reconnect_test.go b/p2p/test/reconnects/reconnect_test.go index a79ac71e7e5babbb677898284b615cb0103b2772..55b7b1ce1e9b35f5a5a328aa28452b37926ba889 100644 --- a/p2p/test/reconnects/reconnect_test.go +++ b/p2p/test/reconnects/reconnect_test.go @@ -31,8 +31,12 @@ func EchoStreamHandler(stream inet.Stream) { c := stream.Conn() log.Debugf("%s echoing %s", c.LocalPeer(), c.RemotePeer()) go func() { - defer stream.Close() - io.Copy(stream, stream) + _, err := io.Copy(stream, stream) + if err == nil { + stream.Close() + } else { + stream.Reset() + } }() }