From 34421451f5a7c41058c21f977afc9ad6b248259c Mon Sep 17 00:00:00 2001 From: Steven Allen Date: Wed, 13 Sep 2017 15:57:12 -0700 Subject: [PATCH] reset on error in reconnect test --- p2p/test/reconnects/reconnect_test.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/p2p/test/reconnects/reconnect_test.go b/p2p/test/reconnects/reconnect_test.go index a79ac71..55b7b1c 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() + } }() } -- GitLab