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
Hide 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) { ...@@ -54,11 +54,13 @@ func TestNewStreamMulti(t *testing.T) {
s1, overlay1 := newService(t, 1, libp2p.Options{}) s1, overlay1 := newService(t, 1, libp2p.Options{})
var ( var (
h1calls, h2calls int32 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) _ = atomic.AddInt32(&h1calls, 1)
return nil 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) _ = atomic.AddInt32(&h2calls, 1)
return nil return nil
} }
...@@ -79,7 +81,7 @@ func TestNewStreamMulti(t *testing.T) { ...@@ -79,7 +81,7 @@ func TestNewStreamMulti(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if err := stream.Close(); err != nil { if err := stream.FullClose(); err != nil {
t.Fatal(err) t.Fatal(err)
} }
if atomic.LoadInt32(&h1calls) != 1 { 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