ソースを参照

Rename network_mock to mtglib_netwock_mock

tags/v2.0.0-rc1
9seconds 5年前
コミット
8eb5fed92f
3個のファイルの変更37行の追加37行の削除
  1. 2
    2
      cli/init_test.go
  2. 35
    0
      testlib/mtglib_network_mock.go
  3. 0
    35
      testlib/network_mock.go

+ 2
- 2
cli/init_test.go ファイルの表示

@@ -14,12 +14,12 @@ type CommonTestSuite struct {
14 14
 	suite.Suite
15 15
 
16 16
 	cli         *cli.CLI
17
-	networkMock *testlib.NetworkMock
17
+	networkMock *testlib.MtglibNetworkMock
18 18
 	httpClient  *http.Client
19 19
 }
20 20
 
21 21
 func (suite *CommonTestSuite) SetupTest() {
22
-	suite.networkMock = &testlib.NetworkMock{}
22
+	suite.networkMock = &testlib.MtglibNetworkMock{}
23 23
 	suite.httpClient = &http.Client{}
24 24
 	suite.cli = &cli.CLI{}
25 25
 

+ 35
- 0
testlib/mtglib_network_mock.go ファイルの表示

@@ -0,0 +1,35 @@
1
+package testlib
2
+
3
+import (
4
+	"context"
5
+	"net"
6
+	"net/http"
7
+	"time"
8
+
9
+	"github.com/stretchr/testify/mock"
10
+)
11
+
12
+type MtglibNetworkMock struct {
13
+	mock.Mock
14
+}
15
+
16
+func (n *MtglibNetworkMock) Dial(network, address string) (net.Conn, error) {
17
+	args := n.Called(network, address)
18
+
19
+	return args.Get(0).(net.Conn), args.Error(1)
20
+}
21
+
22
+func (n *MtglibNetworkMock) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
23
+	args := n.Called(ctx, network, address)
24
+
25
+	return args.Get(0).(net.Conn), args.Error(1)
26
+}
27
+
28
+func (m *MtglibNetworkMock) MakeHTTPClient(dialFunc func(ctx context.Context,
29
+	network, address string) (net.Conn, error)) *http.Client {
30
+	return m.Called(dialFunc).Get(0).(*http.Client)
31
+}
32
+
33
+func (m *MtglibNetworkMock) IdleTimeout() time.Duration {
34
+	return m.Called().Get(0).(time.Duration)
35
+}

+ 0
- 35
testlib/network_mock.go ファイルの表示

@@ -1,35 +0,0 @@
1
-package testlib
2
-
3
-import (
4
-	"context"
5
-	"net"
6
-	"net/http"
7
-	"time"
8
-
9
-	"github.com/stretchr/testify/mock"
10
-)
11
-
12
-type NetworkMock struct {
13
-	mock.Mock
14
-}
15
-
16
-func (n *NetworkMock) Dial(network, address string) (net.Conn, error) {
17
-	args := n.Called(network, address)
18
-
19
-	return args.Get(0).(net.Conn), args.Error(1)
20
-}
21
-
22
-func (n *NetworkMock) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
23
-	args := n.Called(ctx, network, address)
24
-
25
-	return args.Get(0).(net.Conn), args.Error(1)
26
-}
27
-
28
-func (n *NetworkMock) MakeHTTPClient(dialFunc func(ctx context.Context,
29
-	network, address string) (net.Conn, error)) *http.Client {
30
-	return n.Called(dialFunc).Get(0).(*http.Client)
31
-}
32
-
33
-func (n *NetworkMock) IdleTimeout() time.Duration {
34
-	return n.Called().Get(0).(time.Duration)
35
-}

読み込み中…
キャンセル
保存