Add tests for socks dialer

mtglib/network/default_test.go

@@ -3,29 +3,21 @@ package network_test
 import (
 	"context"
 	"net/http"
-	"net/http/httptest"
-	"strings"
 	"testing"
 
 	"github.com/9seconds/mtg/v2/mtglib/network"
-	"github.com/mccutchen/go-httpbin/httpbin"
 	"github.com/stretchr/testify/suite"
 )
 
 type DefaultDialerTestSuite struct {
-	suite.Suite
+	HTTPServerTestSuite
 
-	d          network.Dialer
-	srvAddress string
-	srv        *httptest.Server
+	d network.Dialer
 }
 
 func (suite *DefaultDialerTestSuite) SetupSuite() {
-	suite.srv = httptest.NewServer(httpbin.NewHTTPBin().Handler())
-	suite.srvAddress = strings.TrimPrefix(suite.srv.URL, "http://")
-}
+	suite.HTTPServerTestSuite.SetupSuite()
 
-func (suite *DefaultDialerTestSuite) SetupTest() {
 	d, err := network.NewDefaultDialer(0, 0)
 
 	suite.NoError(err)
@@ -33,10 +25,6 @@ func (suite *DefaultDialerTestSuite) SetupTest() {
 	suite.d = d
 }
 
-func (suite *DefaultDialerTestSuite) TearDownSuite() {
-	suite.srv.Close()
-}
-
 func (suite *DefaultDialerTestSuite) TestNegativeTimeout() {
 	_, err := network.NewDefaultDialer(-1, 0)
 
@@ -50,7 +38,9 @@ func (suite *DefaultDialerTestSuite) TestNegativeBufferSize() {
 }
 
 func (suite *DefaultDialerTestSuite) TestUnsupportedProtocol() {
-	_, err := suite.d.DialContext(context.Background(), "udp", suite.srvAddress)
+	_, err := suite.d.DialContext(context.Background(),
+		"udp",
+		suite.HTTPServerAddress())
 
 	suite.Error(err)
 }
@@ -58,7 +48,7 @@ func (suite *DefaultDialerTestSuite) TestUnsupportedProtocol() {
 func (suite *DefaultDialerTestSuite) TestCannotDial() {
 	_, err := suite.d.DialContext(context.Background(),
 		"tcp",
-		suite.srvAddress+suite.srvAddress)
+		suite.HTTPServerAddress()+suite.HTTPServerAddress())
 
 	suite.Error(err)
 }
@@ -66,7 +56,7 @@ func (suite *DefaultDialerTestSuite) TestCannotDial() {
 func (suite *DefaultDialerTestSuite) TestConnectOk() {
 	conn, err := suite.d.DialContext(context.Background(),
 		"tcp",
-		suite.srvAddress)
+		suite.HTTPServerAddress())
 
 	suite.NoError(err)
 	suite.NotNil(conn)
@@ -74,14 +64,14 @@ func (suite *DefaultDialerTestSuite) TestConnectOk() {
 	conn.Close()
 }
 
-func (suite *DefaultDialerTestSuite) TestRequest() {
+func (suite *DefaultDialerTestSuite) TestHTTPRequest() {
 	httpClient := http.Client{
 		Transport: &http.Transport{
 			DialContext: suite.d.DialContext,
 		},
 	}
 
-	resp, err := httpClient.Get(suite.srv.URL + "/get")
+	resp, err := httpClient.Get(suite.httpServer.URL + "/get")
 
 	suite.NoError(err)
 

mtglib/network/init_test.go

@@ -0,0 +1,27 @@
+package network_test
+
+import (
+	"net/http/httptest"
+	"strings"
+
+	"github.com/mccutchen/go-httpbin/httpbin"
+	"github.com/stretchr/testify/suite"
+)
+
+type HTTPServerTestSuite struct {
+	suite.Suite
+
+	httpServer *httptest.Server
+}
+
+func (suite *HTTPServerTestSuite) SetupSuite() {
+	suite.httpServer = httptest.NewServer(httpbin.NewHTTPBin().Handler())
+}
+
+func (suite *HTTPServerTestSuite) TearDownSuite() {
+	suite.httpServer.Close()
+}
+
+func (suite *HTTPServerTestSuite) HTTPServerAddress() string {
+	return strings.TrimPrefix(suite.httpServer.URL, "http://")
+}

mtglib/network/socks5_test.go

@@ -0,0 +1,84 @@
+package network_test
+
+import (
+	"net"
+	"net/http"
+	"net/url"
+	"testing"
+
+	"github.com/9seconds/mtg/v2/mtglib/network"
+	socks5 "github.com/armon/go-socks5"
+	"github.com/stretchr/testify/suite"
+)
+
+type Socks5TestSuite struct {
+	HTTPServerTestSuite
+
+	socksListener net.Listener
+	socksProxy    *socks5.Server
+}
+
+func (suite *Socks5TestSuite) SetupSuite() {
+	suite.HTTPServerTestSuite.SetupSuite()
+
+	socksConf := socks5.Config{
+		Credentials: socks5.StaticCredentials{
+			"user": "password",
+		},
+	}
+
+	suite.socksProxy, _ = socks5.New(&socksConf)
+	suite.socksListener, _ = net.Listen("tcp", "127.0.0.1:0")
+
+	go suite.socksProxy.Serve(suite.socksListener)
+}
+
+func (suite *Socks5TestSuite) TearDownSuite() {
+	suite.socksListener.Close()
+
+	suite.HTTPServerTestSuite.TearDownSuite()
+}
+
+func (suite *Socks5TestSuite) TestRequestFailed() {
+	proxyURL := &url.URL{
+		Scheme: "socks5",
+		User:   url.UserPassword("user2", "password"),
+		Host:   suite.socksListener.Addr().String(),
+	}
+	dialer, _ := network.NewSocks5Dialer(proxyURL, 0, 0)
+
+	httpClient := http.Client{
+		Transport: &http.Transport{
+			DialContext: dialer.DialContext,
+		},
+	}
+
+	_, err := httpClient.Get(suite.httpServer.URL + "/get")
+
+	suite.Error(err)
+}
+
+func (suite *Socks5TestSuite) TestRequestOk() {
+	proxyURL := &url.URL{
+		Scheme: "socks5",
+		User:   url.UserPassword("user", "password"),
+		Host:   suite.socksListener.Addr().String(),
+	}
+	dialer, _ := network.NewSocks5Dialer(proxyURL, 0, 0)
+
+	httpClient := http.Client{
+		Transport: &http.Transport{
+			DialContext: dialer.DialContext,
+		},
+	}
+
+	resp, err := httpClient.Get(suite.httpServer.URL + "/get")
+
+	suite.NoError(err)
+
+	resp.Body.Close()
+}
+
+func TestSocks5TestSuite(t *testing.T) {
+	suite.Run(t, &Socks5TestSuite{})
+}