pass contexts to OpenStream in tests

libp2p · Dec 17, 2020 · 486206a · 486206a
1 parent 0c577bb
commit 486206a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/listener_test.go b/listener_test.go
@@ -1,6 +1,7 @@
 package stream_test
 
 import (
+	"context"
 	"io"
 	"sync"
 	"testing"
@@ -92,7 +93,7 @@ func TestConnectionsClosedIfNotAccepted(t *testing.T) {
 	errCh := make(chan error)
 	go func() {
 		defer conn.Close()
-		str, err := conn.OpenStream()
+		str, err := conn.OpenStream(context.Background())
 		if err != nil {
 			errCh <- err
 			return

diff --git a/upgrader_test.go b/upgrader_test.go
@@ -78,7 +78,7 @@ func testConn(t *testing.T, clientConn, serverConn transport.CapableConn) {
 	t.Helper()
 	require := require.New(t)
 
-	cstr, err := clientConn.OpenStream()
+	cstr, err := clientConn.OpenStream(context.Background())
 	require.NoError(err)
 
 	_, err = cstr.Write([]byte("foobar"))