diff --git a/request_test.go b/request_test.go index ce1db9d..880a3cd 100644 --- a/request_test.go +++ b/request_test.go @@ -4,7 +4,9 @@ import ( "bytes" "encoding/binary" "io" + "log" "net" + "os" "strings" "testing" ) @@ -50,6 +52,7 @@ func TestRequest_Connect(t *testing.T) { s := &Server{config: &Config{ Rules: PermitAll(), Resolver: DNSResolver{}, + Logger: log.New(os.Stdout, "", log.LstdFlags), }} // Create the connect request @@ -119,6 +122,7 @@ func TestRequest_Connect_RuleFail(t *testing.T) { s := &Server{config: &Config{ Rules: PermitNone(), Resolver: DNSResolver{}, + Logger: log.New(os.Stdout, "", log.LstdFlags), }} // Create the connect request diff --git a/socks5_test.go b/socks5_test.go index f78fc88..c7d6e26 100644 --- a/socks5_test.go +++ b/socks5_test.go @@ -4,7 +4,9 @@ import ( "bytes" "encoding/binary" "io" + "log" "net" + "os" "testing" "time" ) @@ -38,9 +40,10 @@ func TestSOCKS5_Connect(t *testing.T) { creds := StaticCredentials{ "foo": "bar", } - cator := UserPassAuthenticator{Credentials : creds} + cator := UserPassAuthenticator{Credentials: creds} conf := &Config{ - AuthMethods : []Authenticator{cator}, + AuthMethods: []Authenticator{cator}, + Logger: log.New(os.Stdout, "", log.LstdFlags), } serv, err := New(conf) if err != nil {