From 8eb5fed92fc6af7ee1003c1d6b779b2bfedcd7af Mon Sep 17 00:00:00 2001 From: 9seconds Date: Mon, 15 Mar 2021 10:45:01 +0300 Subject: [PATCH] Rename network_mock to mtglib_netwock_mock --- cli/init_test.go | 4 ++-- testlib/mtglib_network_mock.go | 35 ++++++++++++++++++++++++++++++++++ testlib/network_mock.go | 35 ---------------------------------- 3 files changed, 37 insertions(+), 37 deletions(-) create mode 100644 testlib/mtglib_network_mock.go delete mode 100644 testlib/network_mock.go diff --git a/cli/init_test.go b/cli/init_test.go index 197d6d6..77304e9 100644 --- a/cli/init_test.go +++ b/cli/init_test.go @@ -14,12 +14,12 @@ type CommonTestSuite struct { suite.Suite cli *cli.CLI - networkMock *testlib.NetworkMock + networkMock *testlib.MtglibNetworkMock httpClient *http.Client } func (suite *CommonTestSuite) SetupTest() { - suite.networkMock = &testlib.NetworkMock{} + suite.networkMock = &testlib.MtglibNetworkMock{} suite.httpClient = &http.Client{} suite.cli = &cli.CLI{} diff --git a/testlib/mtglib_network_mock.go b/testlib/mtglib_network_mock.go new file mode 100644 index 0000000..5a8192f --- /dev/null +++ b/testlib/mtglib_network_mock.go @@ -0,0 +1,35 @@ +package testlib + +import ( + "context" + "net" + "net/http" + "time" + + "github.com/stretchr/testify/mock" +) + +type MtglibNetworkMock struct { + mock.Mock +} + +func (n *MtglibNetworkMock) Dial(network, address string) (net.Conn, error) { + args := n.Called(network, address) + + return args.Get(0).(net.Conn), args.Error(1) +} + +func (n *MtglibNetworkMock) DialContext(ctx context.Context, network, address string) (net.Conn, error) { + args := n.Called(ctx, network, address) + + return args.Get(0).(net.Conn), args.Error(1) +} + +func (m *MtglibNetworkMock) MakeHTTPClient(dialFunc func(ctx context.Context, + network, address string) (net.Conn, error)) *http.Client { + return m.Called(dialFunc).Get(0).(*http.Client) +} + +func (m *MtglibNetworkMock) IdleTimeout() time.Duration { + return m.Called().Get(0).(time.Duration) +} diff --git a/testlib/network_mock.go b/testlib/network_mock.go deleted file mode 100644 index 325dfd3..0000000 --- a/testlib/network_mock.go +++ /dev/null @@ -1,35 +0,0 @@ -package testlib - -import ( - "context" - "net" - "net/http" - "time" - - "github.com/stretchr/testify/mock" -) - -type NetworkMock struct { - mock.Mock -} - -func (n *NetworkMock) Dial(network, address string) (net.Conn, error) { - args := n.Called(network, address) - - return args.Get(0).(net.Conn), args.Error(1) -} - -func (n *NetworkMock) DialContext(ctx context.Context, network, address string) (net.Conn, error) { - args := n.Called(ctx, network, address) - - return args.Get(0).(net.Conn), args.Error(1) -} - -func (n *NetworkMock) MakeHTTPClient(dialFunc func(ctx context.Context, - network, address string) (net.Conn, error)) *http.Client { - return n.Called(dialFunc).Get(0).(*http.Client) -} - -func (n *NetworkMock) IdleTimeout() time.Duration { - return n.Called().Get(0).(time.Duration) -}