Defer closing stream for reading

libp2p · May 1, 2021 · f2de233 · f2de233
1 parent dccf408
commit f2de233
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/suites/transport/stream_suite.go b/suites/transport/stream_suite.go
@@ -55,7 +55,7 @@ type Options struct {
 }
 
 func fullClose(t *testing.T, s mux.MuxedStream) {
- if err := s.Close(); err != nil {
+ if err := s.CloseWrite(); err != nil {
  t.Error(err)
  s.Reset()
  return
@@ -67,6 +67,9 @@ func fullClose(t *testing.T, s mux.MuxedStream) {
  if len(b) != 0 {
  t.Error("expected to be done reading")
  }
+ if err := s.Close(); err != nil {
+ t.Error(err)
+ }
 }
 
 func randBuf(size int) []byte {

diff --git a/suites/transport/transport_suite.go b/suites/transport/transport_suite.go
@@ -141,7 +141,7 @@ func SubtestBasic(t *testing.T, ta, tb transport.Transport, maddr ma.Multiaddr,
  t.Fatalf("failed to write enough data (a->b)")
  return
  }
- err = s.Close()
+ err = s.CloseWrite()
  if err != nil {
  t.Fatal(err)
  return
@@ -155,6 +155,12 @@ func SubtestBasic(t *testing.T, ta, tb transport.Transport, maddr ma.Multiaddr,
  if !bytes.Equal(testData, buf) {
  t.Errorf("expected %s, got %s", testData, buf)
  }
+
+ err = s.Close()
+ if err != nil {
+ t.Fatal(err)
+ return
+ }
 }
 
 func SubtestPingPong(t *testing.T, ta, tb transport.Transport, maddr ma.Multiaddr, peerA peer.ID) {