Refactor network to a top-level module

network/default_test.go

@@ -0,0 +1,77 @@
+package network_test
+
+import (
+	"context"
+	"net/http"
+	"testing"
+
+	"github.com/9seconds/mtg/v2/network"
+	"github.com/stretchr/testify/suite"
+)
+
+type DefaultDialerTestSuite struct {
+	suite.Suite
+	HTTPServerTestSuite
+
+	d network.Dialer
+}
+
+func (suite *DefaultDialerTestSuite) SetupSuite() {
+	suite.HTTPServerTestSuite.SetupSuite()
+
+	d, err := network.NewDefaultDialer(0, 0)
+	suite.NoError(err)
+
+	suite.d = d
+}
+
+func (suite *DefaultDialerTestSuite) TestNegativeTimeout() {
+	_, err := network.NewDefaultDialer(-1, 0)
+	suite.Error(err)
+}
+
+func (suite *DefaultDialerTestSuite) TestNegativeBufferSize() {
+	_, err := network.NewDefaultDialer(0, -1)
+	suite.Error(err)
+}
+
+func (suite *DefaultDialerTestSuite) TestUnsupportedProtocol() {
+	_, err := suite.d.DialContext(context.Background(),
+		"udp",
+		suite.HTTPServerAddress())
+	suite.Error(err)
+}
+
+func (suite *DefaultDialerTestSuite) TestCannotDial() {
+	_, err := suite.d.DialContext(context.Background(),
+		"tcp",
+		suite.HTTPServerAddress()+suite.HTTPServerAddress())
+	suite.Error(err)
+}
+
+func (suite *DefaultDialerTestSuite) TestConnectOk() {
+	conn, err := suite.d.DialContext(context.Background(),
+		"tcp",
+		suite.HTTPServerAddress())
+	suite.NoError(err)
+	suite.NotNil(conn)
+
+	conn.Close()
+}
+
+func (suite *DefaultDialerTestSuite) TestHTTPRequest() {
+	httpClient := suite.MakeHTTPClient(suite.d)
+
+	resp, err := httpClient.Get(suite.MakeURL("/get")) // nolint: noctx
+	if err == nil {
+		defer resp.Body.Close()
+	}
+
+	suite.NoError(err)
+	suite.Equal(http.StatusOK, resp.StatusCode)
+}
+
+func TestDefaultDialer(t *testing.T) {
+	t.Parallel()
+	suite.Run(t, &DefaultDialerTestSuite{})
+}