diff --git a/go.mod b/go.mod index e6dd6ce..79b8059 100644 --- a/go.mod +++ b/go.mod @@ -8,6 +8,7 @@ require ( github.com/libp2p/go-reuseport v0.0.2 github.com/mccutchen/go-httpbin v1.1.1 github.com/pelletier/go-toml v1.8.1 + github.com/stretchr/objx v0.3.0 // indirect github.com/stretchr/testify v1.7.0 golang.org/x/net v0.0.0-20210226172049-e18ecbb05110 ) diff --git a/go.sum b/go.sum index 1563bf5..aea74ac 100644 --- a/go.sum +++ b/go.sum @@ -16,6 +16,9 @@ github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINE github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.3.0 h1:NGXK3lHquSN08v5vWalVI/L8XU9hdzE/G6xsrze47As= +github.com/stretchr/objx v0.3.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= diff --git a/mtglib/network/init_test.go b/mtglib/network/init_test.go index 89aba2b..d44ff5b 100644 --- a/mtglib/network/init_test.go +++ b/mtglib/network/init_test.go @@ -5,12 +5,53 @@ import ( "net" "net/http/httptest" "strings" + "time" "github.com/mccutchen/go-httpbin/httpbin" "github.com/stretchr/testify/mock" "github.com/stretchr/testify/suite" ) +type ConnMock struct { + mock.Mock +} + +func (c *ConnMock) Read(b []byte) (int, error) { + args := c.Called(b) + + return args.Int(0), args.Error(1) +} + +func (c *ConnMock) Write(b []byte) (int, error) { + args := c.Called(b) + + return args.Int(0), args.Error(1) +} + +func (c *ConnMock) Close() error { + return c.Called().Error(0) +} + +func (c *ConnMock) LocalAddr() net.Addr { + return c.Called().Get(0).(net.Addr) +} + +func (c *ConnMock) RemoteAddr() net.Addr { + return c.Called().Get(0).(net.Addr) +} + +func (c *ConnMock) SetDeadline(t time.Time) error { + return c.Called(t).Error(0) +} + +func (c *ConnMock) SetReadDeadline(t time.Time) error { + return c.Called(t).Error(0) +} + +func (c *ConnMock) SetWriteDeadline(t time.Time) error { + return c.Called(t).Error(0) +} + type DialerMock struct { mock.Mock }