Skip to content

Commit

Permalink
fix flaky TestResourceManagerAcceptStream test (#1420)
Browse files Browse the repository at this point in the history
  • Loading branch information
marten-seemann authored Apr 22, 2022
1 parent e28b91a commit 37f5850
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions p2p/net/swarm/swarm_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -534,8 +534,11 @@ func TestResourceManagerAcceptStream(t *testing.T) {
rcmgr2.EXPECT().OpenStream(s1.LocalPeer(), network.DirInbound).Return(nil, errors.New("nope"))
str, err := s1.NewStream(context.Background(), s2.LocalPeer())
require.NoError(t, err)
// The peer's resource manager is blocking any new stream.
// Depending on how quickly we receive the stream reset, it surfaces either during the write or the read call.
_, err = str.Write([]byte("foobar"))
require.NoError(t, err)
_, err = str.Read([]byte{0})
if err == nil {
_, err = str.Read([]byte{0})
}
require.EqualError(t, err, "stream reset")
}

0 comments on commit 37f5850

Please sign in to comment.