Skip to content

M
mybee
Commit 77eead48 authored by Petar Radovic's avatar Petar Radovic Committed by GitHub
Browse files
Options

Multistream test fix (#389) 

* multistream test fix
parent f408b582
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
pkg/p2p/libp2p/protocols_test.go
View file @ 77eead48
......@@ -54,11 +54,13 @@ func TestNewStreamMulti(t *testing.T) {
s1, overlay1 := newService(t, 1, libp2p.Options{})
var (
h1calls, h2calls int32
h1 = func(_ context.Context, _ p2p.Peer, _ p2p.Stream) error {
h1 = func(_ context.Context, _ p2p.Peer, s p2p.Stream) error {
defer s.Close()
_ = atomic.AddInt32(&h1calls, 1)
return nil
}
h2 = func(_ context.Context, _ p2p.Peer, _ p2p.Stream) error {
h2 = func(_ context.Context, _ p2p.Peer, s p2p.Stream) error {
defer s.Close()
_ = atomic.AddInt32(&h2calls, 1)
return nil
}
......@@ -79,7 +81,7 @@ func TestNewStreamMulti(t *testing.T) {
if err != nil {
t.Fatal(err)
}
if err := stream.Close(); err != nil {
if err := stream.FullClose(); err != nil {
t.Fatal(err)
}
if atomic.LoadInt32(&h1calls) != 1 {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment