Skip to content

Commit f448a97

Browse files
cleanup: replace dial with newclient (#8602)
Fixes: #7049 RELEASE NOTES: None
1 parent 0037c61 commit f448a97

File tree

4 files changed

+54
-24
lines changed

4 files changed

+54
-24
lines changed

clientconn_authority_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -109,9 +109,9 @@ func (s) TestClientConnAuthority(t *testing.T) {
109109

110110
for _, test := range tests {
111111
t.Run(test.name, func(t *testing.T) {
112-
cc, err := Dial(test.target, test.opts...)
112+
cc, err := NewClient(test.target, test.opts...)
113113
if err != nil {
114-
t.Fatalf("Dial(%q) failed: %v", test.target, err)
114+
t.Fatalf("NewClient(%q) failed: %v", test.target, err)
115115
}
116116
defer cc.Close()
117117
if cc.authority != test.wantAuthority {
@@ -130,6 +130,6 @@ func (s) TestClientConnAuthority_CredsAndDialOptionMismatch(t *testing.T) {
130130
opts := []DialOption{WithTransportCredentials(creds), WithAuthority("authority-override")}
131131
if cc, err := NewClient("Non-Existent.Server:8000", opts...); err == nil {
132132
cc.Close()
133-
t.Fatal("grpc.NewClient() succeeded when expected to fail")
133+
t.Fatal("NewClient() succeeded when expected to fail")
134134
}
135135
}

clientconn_parsed_target_test.go

Lines changed: 32 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ import (
2323
"errors"
2424
"fmt"
2525
"net"
26+
"strings"
2627
"testing"
2728
"time"
2829

@@ -203,6 +204,34 @@ func (s) TestParsedTarget_Failure_WithoutCustomDialer(t *testing.T) {
203204
}
204205
}
205206

207+
func (s) TestParsedTarget_Failure_WithoutCustomDialer_WithNewClient(t *testing.T) {
208+
targets := []string{
209+
"",
210+
"unix://a/b/c",
211+
"unix://authority",
212+
"unix-abstract://authority/a/b/c",
213+
"unix-abstract://authority",
214+
}
215+
216+
for _, target := range targets {
217+
t.Run(target, func(t *testing.T) {
218+
ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
219+
defer cancel()
220+
cc, err := NewClient(target, WithTransportCredentials(insecure.NewCredentials()))
221+
if err != nil {
222+
t.Fatalf("NewClient(%q) failed: %v", target, err)
223+
}
224+
defer cc.Close()
225+
const wantErrSubstr = "failed to exit idle mode"
226+
if _, err := cc.NewStream(ctx, &StreamDesc{}, "/my.service.v1.MyService/UnaryCall"); err == nil {
227+
t.Fatalf("NewStream() succeeded with target = %q, cc.parsedTarget = %+v, expected to fail", target, cc.parsedTarget)
228+
} else if !strings.Contains(err.Error(), wantErrSubstr) {
229+
t.Fatalf("NewStream() with target = %q returned unexpected error: got %v, want substring %q", target, err, wantErrSubstr)
230+
}
231+
})
232+
}
233+
}
234+
206235
func (s) TestParsedTarget_WithCustomDialer(t *testing.T) {
207236
resetInitialResolverState()
208237
defScheme := resolver.GetDefaultScheme()
@@ -273,11 +302,12 @@ func (s) TestParsedTarget_WithCustomDialer(t *testing.T) {
273302
return nil, errors.New("dialer error")
274303
}
275304

276-
cc, err := Dial(test.target, WithTransportCredentials(insecure.NewCredentials()), WithContextDialer(dialer))
305+
cc, err := NewClient(test.target, WithTransportCredentials(insecure.NewCredentials()), withDefaultScheme(defScheme), WithContextDialer(dialer))
277306
if err != nil {
278-
t.Fatalf("Dial(%q) failed: %v", test.target, err)
307+
t.Fatalf("grpc.NewClient(%q) failed: %v", test.target, err)
279308
}
280309
defer cc.Close()
310+
cc.Connect()
281311

282312
select {
283313
case addr := <-addrCh:

default_dial_option_server_option_test.go

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -29,12 +29,12 @@ import (
2929
)
3030

3131
func (s) TestAddGlobalDialOptions(t *testing.T) {
32-
// Ensure the Dial fails without credentials
33-
if _, err := Dial("fake"); err == nil {
34-
t.Fatalf("Dialing without a credential did not fail")
32+
// Ensure the NewClient fails without credentials
33+
if _, err := NewClient("fake"); err == nil {
34+
t.Fatalf("NewClient without a credential did not fail")
3535
} else {
3636
if !strings.Contains(err.Error(), "no transport security set") {
37-
t.Fatalf("Dialing failed with unexpected error: %v", err)
37+
t.Fatalf("NewClient failed with unexpected error: %v", err)
3838
}
3939
}
4040

@@ -48,9 +48,9 @@ func (s) TestAddGlobalDialOptions(t *testing.T) {
4848
}
4949
}
5050

51-
// Ensure the Dial passes with the extra dial options
52-
if cc, err := Dial("fake"); err != nil {
53-
t.Fatalf("Dialing with insecure credential failed: %v", err)
51+
// Ensure the NewClient passes with the extra dial options
52+
if cc, err := NewClient("fake"); err != nil {
53+
t.Fatalf("NewClient with insecure credential failed: %v", err)
5454
} else {
5555
cc.Close()
5656
}
@@ -71,8 +71,8 @@ func (s) TestDisableGlobalOptions(t *testing.T) {
7171
// due to the global dial options with credentials not being picked up due
7272
// to global options being disabled.
7373
noTSecStr := "no transport security set"
74-
if _, err := Dial("fake", internal.DisableGlobalDialOptions.(func() DialOption)()); !strings.Contains(fmt.Sprint(err), noTSecStr) {
75-
t.Fatalf("Dialing received unexpected error: %v, want error containing \"%v\"", err, noTSecStr)
74+
if _, err := NewClient("fake", internal.DisableGlobalDialOptions.(func() DialOption)()); !strings.Contains(fmt.Sprint(err), noTSecStr) {
75+
t.Fatalf("NewClient received unexpected error: %v, want error containing %q", err, noTSecStr)
7676
}
7777
}
7878

@@ -95,11 +95,11 @@ func (s) TestGlobalPerTargetDialOption(t *testing.T) {
9595
defer internal.ClearGlobalPerTargetDialOptions()
9696
noTSecStr := "no transport security set"
9797
if _, err := NewClient("dns:///fake"); !strings.Contains(fmt.Sprint(err), noTSecStr) {
98-
t.Fatalf("Dialing received unexpected error: %v, want error containing \"%v\"", err, noTSecStr)
98+
t.Fatalf("NewClient received unexpected error: %v, want error containing %q", err, noTSecStr)
9999
}
100100
cc, err := NewClient("passthrough:///nice")
101101
if err != nil {
102-
t.Fatalf("Dialing with insecure credentials failed: %v", err)
102+
t.Fatalf("NewClient with insecure credentials failed: %v", err)
103103
}
104104
cc.Close()
105105
}
@@ -135,9 +135,9 @@ func (s) TestJoinDialOption(t *testing.T) {
135135
const maxRecvSize = 998765
136136
const initialWindowSize = 100
137137
jdo := newJoinDialOption(WithTransportCredentials(insecure.NewCredentials()), WithReadBufferSize(maxRecvSize), WithInitialWindowSize(initialWindowSize))
138-
cc, err := Dial("fake", jdo)
138+
cc, err := NewClient("fake", jdo)
139139
if err != nil {
140-
t.Fatalf("Dialing with insecure credentials failed: %v", err)
140+
t.Fatalf("NewClient with insecure credentials failed: %v", err)
141141
}
142142
defer cc.Close()
143143
if cc.dopts.copts.ReadBufferSize != maxRecvSize {

resolver_test.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,9 @@ func (s) TestResolverCaseSensitivity(t *testing.T) {
6363
return nil, fmt.Errorf("not dialing with custom dialer")
6464
}
6565

66-
cc, err := Dial(target, WithContextDialer(customDialer), WithTransportCredentials(insecure.NewCredentials()))
66+
cc, err := NewClient(target, WithContextDialer(customDialer), WithTransportCredentials(insecure.NewCredentials()))
6767
if err != nil {
68-
t.Fatalf("Unexpected Dial(%q) error: %v", target, err)
68+
t.Fatalf("Unexpected NewClient(%q) error: %v", target, err)
6969
}
7070
cc.Connect()
7171
if got, want := <-addrCh, "localhost:1234"; got != want {
@@ -85,9 +85,9 @@ func (s) TestResolverCaseSensitivity(t *testing.T) {
8585
// This results in "passthrough" being used with the address as the whole
8686
// target.
8787
target = "caseTest2:///localhost:1234"
88-
cc, err = Dial(target, WithContextDialer(customDialer), WithResolvers(res), WithTransportCredentials(insecure.NewCredentials()))
88+
cc, err = NewClient(target, WithContextDialer(customDialer), withDefaultScheme("passthrough"), WithResolvers(res), WithTransportCredentials(insecure.NewCredentials()))
8989
if err != nil {
90-
t.Fatalf("Unexpected Dial(%q) error: %v", target, err)
90+
t.Fatalf("Unexpected NewClient(%q) error: %v", target, err)
9191
}
9292
cc.Connect()
9393
if got, want := <-addrCh, target; got != want {
@@ -125,7 +125,7 @@ func (s) TestResolverAddressesToEndpoints(t *testing.T) {
125125
WithResolvers(r),
126126
WithDefaultServiceConfig(fmt.Sprintf(`{"loadBalancingConfig": [{"%s":{}}]}`, balancerName)))
127127
if err != nil {
128-
t.Fatalf("grpc.NewClient() failed: %v", err)
128+
t.Fatalf("NewClient() failed: %v", err)
129129
}
130130
cc.Connect()
131131
defer cc.Close()
@@ -176,7 +176,7 @@ func (s) TestResolverAddressesToEndpointsUsingNewAddresses(t *testing.T) {
176176
WithResolvers(r),
177177
WithDefaultServiceConfig(fmt.Sprintf(`{"loadBalancingConfig": [{"%s":{}}]}`, balancerName)))
178178
if err != nil {
179-
t.Fatalf("grpc.NewClient() failed: %v", err)
179+
t.Fatalf("NewClient() failed: %v", err)
180180
}
181181
cc.Connect()
182182
defer cc.Close()

0 commit comments

Comments
 (0)