Fix a failing test

skycoin · Mar 10, 2020 · ecc2051 · ecc2051
1 parent 0f8125b
commit ecc2051
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/internal/skysocks/server_test.go b/internal/skysocks/server_test.go
@@ -11,7 +11,6 @@ import (
 	"time"
 
 	"github.com/SkycoinProject/skycoin/src/util/logging"
-	"github.com/SkycoinProject/yamux"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"golang.org/x/net/nettest"
@@ -54,13 +53,9 @@ func TestProxy(t *testing.T) {
 	conn, err := net.Dial("tcp", l.Addr().String())
 	require.NoError(t, err)
 
-	session, err := yamux.Client(conn, nil)
+	client, err := NewClient(conn)
 	require.NoError(t, err)
 
-	client := &Client{
-		session: session,
-	}
-
 	errChan2 := make(chan error)
 
 	go func() {