@@ -18,7 +18,6 @@ import (
18
18
"go.uber.org/zap"
19
19
"go.uber.org/zap/zaptest/observer"
20
20
"google.golang.org/grpc"
21
- "google.golang.org/grpc/balancer"
22
21
"google.golang.org/grpc/codes"
23
22
"google.golang.org/grpc/metadata"
24
23
"google.golang.org/grpc/peer"
@@ -92,21 +91,6 @@ func TestNewDefaultServerConfig(t *testing.T) {
92
91
assert .Equal (t , expected , result )
93
92
}
94
93
95
- // testBalancerBuilder facilitates testing validateBalancerName().
96
- type testBalancerBuilder struct {}
97
-
98
- func (testBalancerBuilder ) Build (balancer.ClientConn , balancer.BuildOptions ) balancer.Balancer {
99
- return nil
100
- }
101
-
102
- func (testBalancerBuilder ) Name () string {
103
- return "configgrpc_balancer_test"
104
- }
105
-
106
- func init () {
107
- balancer .Register (testBalancerBuilder {})
108
- }
109
-
110
94
var (
111
95
componentID = component .MustNewID ("component" )
112
96
testAuthID = component .MustNewID ("testauth" )
@@ -333,7 +317,7 @@ func TestGrpcServerValidate(t *testing.T) {
333
317
t .Run (tt .err , func (t * testing.T ) {
334
318
err := tt .gss .Validate ()
335
319
require .Error (t , err )
336
- assert .Regexp (t , tt . err , err )
320
+ assert .ErrorContains (t , err , tt . err )
337
321
})
338
322
}
339
323
}
@@ -390,18 +374,37 @@ func TestGrpcServerAuthSettings(t *testing.T) {
390
374
assert .NotNil (t , srv )
391
375
}
392
376
393
- func TestGRPCClientSettingsError (t * testing.T ) {
394
- tt , err := componenttest .SetupTelemetry (componentID )
395
- require .NoError (t , err )
396
- t .Cleanup (func () { require .NoError (t , tt .Shutdown (context .Background ())) })
377
+ func TestGrpcClientConfigInvalidBalancer (t * testing.T ) {
378
+ settings := ClientConfig {
379
+ Headers : map [string ]configopaque.String {
380
+ "test" : "test" ,
381
+ },
382
+ Endpoint : "localhost:1234" ,
383
+ Compression : "gzip" ,
384
+ TLSSetting : configtls.ClientConfig {
385
+ Insecure : false ,
386
+ },
387
+ Keepalive : & KeepaliveClientConfig {
388
+ Time : time .Second ,
389
+ Timeout : time .Second ,
390
+ PermitWithoutStream : true ,
391
+ },
392
+ ReadBufferSize : 1024 ,
393
+ WriteBufferSize : 1024 ,
394
+ WaitForReady : true ,
395
+ BalancerName : "test" ,
396
+ }
397
+ assert .ErrorContains (t , settings .Validate (), "invalid balancer_name: test" )
398
+ }
397
399
400
+ func TestGRPCClientSettingsError (t * testing.T ) {
398
401
tests := []struct {
399
402
settings ClientConfig
400
403
err string
401
404
host component.Host
402
405
}{
403
406
{
404
- err : "^ failed to load TLS config: failed to load CA CertPool File: failed to load cert /doesnt/exist:" ,
407
+ err : "failed to load TLS config: failed to load CA CertPool File: failed to load cert /doesnt/exist:" ,
405
408
settings : ClientConfig {
406
409
Headers : nil ,
407
410
Endpoint : "" ,
@@ -417,7 +420,7 @@ func TestGRPCClientSettingsError(t *testing.T) {
417
420
},
418
421
},
419
422
{
420
- err : "^ failed to load TLS config: failed to load TLS cert and key: for auth via TLS, provide both certificate and key, or neither" ,
423
+ err : "failed to load TLS config: failed to load TLS cert and key: for auth via TLS, provide both certificate and key, or neither" ,
421
424
settings : ClientConfig {
422
425
Headers : nil ,
423
426
Endpoint : "" ,
@@ -432,28 +435,6 @@ func TestGRPCClientSettingsError(t *testing.T) {
432
435
Keepalive : nil ,
433
436
},
434
437
},
435
- {
436
- err : "invalid balancer_name: test" ,
437
- settings : ClientConfig {
438
- Headers : map [string ]configopaque.String {
439
- "test" : "test" ,
440
- },
441
- Endpoint : "localhost:1234" ,
442
- Compression : "gzip" ,
443
- TLSSetting : configtls.ClientConfig {
444
- Insecure : false ,
445
- },
446
- Keepalive : & KeepaliveClientConfig {
447
- Time : time .Second ,
448
- Timeout : time .Second ,
449
- PermitWithoutStream : true ,
450
- },
451
- ReadBufferSize : 1024 ,
452
- WriteBufferSize : 1024 ,
453
- WaitForReady : true ,
454
- BalancerName : "test" ,
455
- },
456
- },
457
438
{
458
439
err : "failed to resolve authenticator \" doesntexist\" : authenticator not found" ,
459
440
settings : ClientConfig {
@@ -506,9 +487,10 @@ func TestGRPCClientSettingsError(t *testing.T) {
506
487
}
507
488
for _ , test := range tests {
508
489
t .Run (test .err , func (t * testing.T ) {
509
- _ , err := test .settings .ToClientConn (context .Background (), test .host , tt .TelemetrySettings ())
490
+ require .NoError (t , test .settings .Validate ())
491
+ _ , err := test .settings .ToClientConn (context .Background (), test .host , componenttest .NewNopTelemetrySettings ())
510
492
require .Error (t , err )
511
- assert .Regexp (t , test . err , err )
493
+ assert .ErrorContains (t , err , test . err )
512
494
})
513
495
}
514
496
}
@@ -587,7 +569,7 @@ func TestGRPCServerSettingsError(t *testing.T) {
587
569
err string
588
570
}{
589
571
{
590
- err : "^ failed to load TLS config: failed to load CA CertPool File: failed to load cert /doesnt/exist:" ,
572
+ err : "failed to load TLS config: failed to load CA CertPool File: failed to load cert /doesnt/exist:" ,
591
573
settings : ServerConfig {
592
574
NetAddr : confignet.AddrConfig {
593
575
Endpoint : "127.0.0.1:1234" ,
@@ -601,7 +583,7 @@ func TestGRPCServerSettingsError(t *testing.T) {
601
583
},
602
584
},
603
585
{
604
- err : "^ failed to load TLS config: failed to load TLS cert and key: for auth via TLS, provide both certificate and key, or neither" ,
586
+ err : "failed to load TLS config: failed to load TLS cert and key: for auth via TLS, provide both certificate and key, or neither" ,
605
587
settings : ServerConfig {
606
588
NetAddr : confignet.AddrConfig {
607
589
Endpoint : "127.0.0.1:1234" ,
@@ -615,7 +597,7 @@ func TestGRPCServerSettingsError(t *testing.T) {
615
597
},
616
598
},
617
599
{
618
- err : "^ failed to load client CA CertPool: failed to load CA /doesnt/exist:" ,
600
+ err : "failed to load client CA CertPool: failed to load CA /doesnt/exist:" ,
619
601
settings : ServerConfig {
620
602
NetAddr : confignet.AddrConfig {
621
603
Endpoint : "127.0.0.1:1234" ,
@@ -630,7 +612,7 @@ func TestGRPCServerSettingsError(t *testing.T) {
630
612
for _ , test := range tests {
631
613
t .Run (test .err , func (t * testing.T ) {
632
614
_ , err := test .settings .ToServer (context .Background (), componenttest .NewNopHost (), componenttest .NewNopTelemetrySettings ())
633
- assert .Regexp (t , test . err , err )
615
+ assert .ErrorContains (t , err , test . err )
634
616
})
635
617
}
636
618
}
0 commit comments