@@ -15,7 +15,7 @@ import (
15
15
)
16
16
17
17
func TestNewClient (t * testing.T ) {
18
- t .Run ("client option only" , func (t * testing.T ) {
18
+ t .Run ("client option only (no ctx) " , func (t * testing.T ) {
19
19
c , err := NewClient (rueidis.ClientOption {
20
20
InitAddress : []string {"127.0.0.1:6379" },
21
21
DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
@@ -28,14 +28,27 @@ func TestNewClient(t *testing.T) {
28
28
defer c .Close ()
29
29
})
30
30
31
+ t .Run ("client option only" , func (t * testing.T ) {
32
+ c , err := NewClient (rueidis.ClientOption {
33
+ InitAddress : []string {"127.0.0.1:6379" },
34
+ DialCtxFn : func (ctx context.Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
35
+ return dialer .DialContext (ctx , "tcp" , dst )
36
+ },
37
+ })
38
+ if err != nil {
39
+ t .Fatal (err )
40
+ }
41
+ defer c .Close ()
42
+ })
43
+
31
44
t .Run ("meter provider" , func (t * testing.T ) {
32
45
mr := metric .NewManualReader ()
33
46
meterProvider := metric .NewMeterProvider (metric .WithReader (mr ))
34
47
c , err := NewClient (
35
48
rueidis.ClientOption {
36
49
InitAddress : []string {"127.0.0.1:6379" },
37
- DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
38
- return dialer .Dial ( "tcp" , dst )
50
+ DialCtxFn : func (ctx context. Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
51
+ return dialer .DialContext ( ctx , "tcp" , dst )
39
52
},
40
53
},
41
54
WithMeterProvider (meterProvider ),
@@ -50,8 +63,8 @@ func TestNewClient(t *testing.T) {
50
63
c , err := NewClient (
51
64
rueidis.ClientOption {
52
65
InitAddress : []string {"127.0.0.1:6379" },
53
- DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
54
- return dialer .Dial ( "tcp" , dst )
66
+ DialCtxFn : func (ctx context. Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
67
+ return dialer .DialContext ( ctx , "tcp" , dst )
55
68
},
56
69
},
57
70
WithHistogramOption (HistogramOption {
@@ -79,8 +92,8 @@ func TestNewClientError(t *testing.T) {
79
92
t .Run ("invalid client option" , func (t * testing.T ) {
80
93
_ , err := NewClient (rueidis.ClientOption {
81
94
InitAddress : []string {"" },
82
- DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
83
- return dialer .Dial ( "tcp" , dst )
95
+ DialCtxFn : func (ctx context. Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
96
+ return dialer .DialContext ( ctx , "tcp" , dst )
84
97
},
85
98
})
86
99
if err == nil {
@@ -120,8 +133,8 @@ func TestTrackDialing(t *testing.T) {
120
133
c , err := NewClient (
121
134
rueidis.ClientOption {
122
135
InitAddress : []string {"127.0.0.1:6379" },
123
- DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
124
- return dialer .Dial ( "tcp" , dst )
136
+ DialCtxFn : func (ctx context. Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
137
+ return dialer .DialContext ( ctx , "tcp" , dst )
125
138
},
126
139
},
127
140
WithMeterProvider (meterProvider ),
@@ -169,8 +182,8 @@ func TestTrackDialing(t *testing.T) {
169
182
c , err := NewClient (
170
183
rueidis.ClientOption {
171
184
InitAddress : []string {"127.0.0.1:6379" },
172
- DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
173
- return dialer .Dial ( "tcp" , dst )
185
+ DialCtxFn : func (ctx context. Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
186
+ return dialer .DialContext ( ctx , "tcp" , dst )
174
187
},
175
188
},
176
189
WithMeterProvider (meterProvider ),
@@ -198,8 +211,8 @@ func TestTrackDialing(t *testing.T) {
198
211
_ , err := NewClient (
199
212
rueidis.ClientOption {
200
213
InitAddress : []string {"" },
201
- DialFn : func (dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
202
- return dialer .Dial ( "tcp" , dst )
214
+ DialCtxFn : func (ctx context. Context , dst string , dialer * net.Dialer , _ * tls.Config ) (conn net.Conn , err error ) {
215
+ return dialer .DialContext ( ctx , "tcp" , dst )
203
216
},
204
217
},
205
218
WithMeterProvider (meterProvider ),
0 commit comments