@@ -34,6 +34,7 @@ import (
34
34
35
35
// V1 balancer tests use passthrough resolver instead of dns.
36
36
// TODO(bar) remove this when removing v1 balaner entirely.
37
+ "google.golang.org/grpc/internal"
37
38
_ "google.golang.org/grpc/resolver/passthrough"
38
39
)
39
40
@@ -128,7 +129,7 @@ func TestNameDiscovery(t *testing.T) {
128
129
defer cc .Close ()
129
130
req := "port"
130
131
var reply string
131
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [0 ].port {
132
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [0 ].port {
132
133
t .Fatalf ("grpc.Invoke(_, _, _, _, _) = %v, want %s" , err , servers [0 ].port )
133
134
}
134
135
// Inject the name resolution change to remove servers[0] and add servers[1].
@@ -144,7 +145,7 @@ func TestNameDiscovery(t *testing.T) {
144
145
r .w .inject (updates )
145
146
// Loop until the rpcs in flight talks to servers[1].
146
147
for {
147
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [1 ].port {
148
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [1 ].port {
148
149
break
149
150
}
150
151
time .Sleep (10 * time .Millisecond )
@@ -204,7 +205,7 @@ func TestRoundRobin(t *testing.T) {
204
205
var reply string
205
206
// Loop until servers[1] is up
206
207
for {
207
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [1 ].port {
208
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [1 ].port {
208
209
break
209
210
}
210
211
time .Sleep (10 * time .Millisecond )
@@ -217,14 +218,14 @@ func TestRoundRobin(t *testing.T) {
217
218
r .w .inject ([]* naming.Update {u })
218
219
// Loop until both servers[2] are up.
219
220
for {
220
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [2 ].port {
221
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [2 ].port {
221
222
break
222
223
}
223
224
time .Sleep (10 * time .Millisecond )
224
225
}
225
226
// Check the incoming RPCs served in a round-robin manner.
226
227
for i := 0 ; i < 10 ; i ++ {
227
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [i % numServers ].port {
228
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [i % numServers ].port {
228
229
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , i , err , servers [i % numServers ].port )
229
230
}
230
231
}
@@ -252,7 +253,7 @@ func TestCloseWithPendingRPC(t *testing.T) {
252
253
// Loop until the above update applies.
253
254
for {
254
255
ctx , cancel := context .WithTimeout (context .Background (), 10 * time .Millisecond )
255
- if err := Invoke (ctx , "/foo/bar" , & expectedRequest , & reply , cc , FailFast (false )); Code (err ) == codes .DeadlineExceeded {
256
+ if err := Invoke (ctx , "/foo/bar" , & expectedRequest , & reply , cc , FailFast (false )); internal . Code (err ) == codes .DeadlineExceeded {
256
257
cancel ()
257
258
break
258
259
}
@@ -300,7 +301,7 @@ func TestGetOnWaitChannel(t *testing.T) {
300
301
for {
301
302
var reply string
302
303
ctx , cancel := context .WithTimeout (context .Background (), 10 * time .Millisecond )
303
- if err := Invoke (ctx , "/foo/bar" , & expectedRequest , & reply , cc , FailFast (false )); Code (err ) == codes .DeadlineExceeded {
304
+ if err := Invoke (ctx , "/foo/bar" , & expectedRequest , & reply , cc , FailFast (false )); internal . Code (err ) == codes .DeadlineExceeded {
304
305
cancel ()
305
306
break
306
307
}
@@ -348,7 +349,7 @@ func TestOneServerDown(t *testing.T) {
348
349
var reply string
349
350
// Loop until servers[1] is up
350
351
for {
351
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [1 ].port {
352
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [1 ].port {
352
353
break
353
354
}
354
355
time .Sleep (10 * time .Millisecond )
@@ -401,7 +402,7 @@ func TestOneAddressRemoval(t *testing.T) {
401
402
var reply string
402
403
// Loop until servers[1] is up
403
404
for {
404
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [1 ].port {
405
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [1 ].port {
405
406
break
406
407
}
407
408
time .Sleep (10 * time .Millisecond )
@@ -450,7 +451,7 @@ func checkServerUp(t *testing.T, currentServer *server) {
450
451
defer cc .Close ()
451
452
var reply string
452
453
for {
453
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == port {
454
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == port {
454
455
break
455
456
}
456
457
time .Sleep (10 * time .Millisecond )
@@ -511,7 +512,7 @@ func TestPickFirstCloseWithPendingRPC(t *testing.T) {
511
512
// Loop until the above update applies.
512
513
for {
513
514
ctx , cancel := context .WithTimeout (context .Background (), 10 * time .Millisecond )
514
- if err := Invoke (ctx , "/foo/bar" , & expectedRequest , & reply , cc , FailFast (false )); Code (err ) == codes .DeadlineExceeded {
515
+ if err := Invoke (ctx , "/foo/bar" , & expectedRequest , & reply , cc , FailFast (false )); internal . Code (err ) == codes .DeadlineExceeded {
515
516
cancel ()
516
517
break
517
518
}
@@ -574,7 +575,7 @@ func TestPickFirstOrderAllServerUp(t *testing.T) {
574
575
req := "port"
575
576
var reply string
576
577
for i := 0 ; i < 20 ; i ++ {
577
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [0 ].port {
578
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [0 ].port {
578
579
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 0 , err , servers [0 ].port )
579
580
}
580
581
time .Sleep (10 * time .Millisecond )
@@ -589,13 +590,13 @@ func TestPickFirstOrderAllServerUp(t *testing.T) {
589
590
r .w .inject ([]* naming.Update {u })
590
591
// Loop until it changes to server[1]
591
592
for {
592
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [1 ].port {
593
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [1 ].port {
593
594
break
594
595
}
595
596
time .Sleep (10 * time .Millisecond )
596
597
}
597
598
for i := 0 ; i < 20 ; i ++ {
598
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [1 ].port {
599
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [1 ].port {
599
600
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 1 , err , servers [1 ].port )
600
601
}
601
602
time .Sleep (10 * time .Millisecond )
@@ -609,7 +610,7 @@ func TestPickFirstOrderAllServerUp(t *testing.T) {
609
610
}
610
611
r .w .inject ([]* naming.Update {u })
611
612
for i := 0 ; i < 20 ; i ++ {
612
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [1 ].port {
613
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [1 ].port {
613
614
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 1 , err , servers [1 ].port )
614
615
}
615
616
time .Sleep (10 * time .Millisecond )
@@ -622,13 +623,13 @@ func TestPickFirstOrderAllServerUp(t *testing.T) {
622
623
}
623
624
r .w .inject ([]* naming.Update {u })
624
625
for {
625
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [2 ].port {
626
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [2 ].port {
626
627
break
627
628
}
628
629
time .Sleep (1 * time .Second )
629
630
}
630
631
for i := 0 ; i < 20 ; i ++ {
631
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [2 ].port {
632
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [2 ].port {
632
633
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 2 , err , servers [2 ].port )
633
634
}
634
635
time .Sleep (10 * time .Millisecond )
@@ -641,13 +642,13 @@ func TestPickFirstOrderAllServerUp(t *testing.T) {
641
642
}
642
643
r .w .inject ([]* naming.Update {u })
643
644
for {
644
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [0 ].port {
645
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [0 ].port {
645
646
break
646
647
}
647
648
time .Sleep (1 * time .Second )
648
649
}
649
650
for i := 0 ; i < 20 ; i ++ {
650
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [0 ].port {
651
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [0 ].port {
651
652
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 0 , err , servers [0 ].port )
652
653
}
653
654
time .Sleep (10 * time .Millisecond )
@@ -687,7 +688,7 @@ func TestPickFirstOrderOneServerDown(t *testing.T) {
687
688
req := "port"
688
689
var reply string
689
690
for i := 0 ; i < 20 ; i ++ {
690
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [0 ].port {
691
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [0 ].port {
691
692
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 0 , err , servers [0 ].port )
692
693
}
693
694
time .Sleep (10 * time .Millisecond )
@@ -698,13 +699,13 @@ func TestPickFirstOrderOneServerDown(t *testing.T) {
698
699
servers [0 ].stop ()
699
700
// Loop until it changes to server[1]
700
701
for {
701
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [1 ].port {
702
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [1 ].port {
702
703
break
703
704
}
704
705
time .Sleep (10 * time .Millisecond )
705
706
}
706
707
for i := 0 ; i < 20 ; i ++ {
707
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [1 ].port {
708
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [1 ].port {
708
709
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 1 , err , servers [1 ].port )
709
710
}
710
711
time .Sleep (10 * time .Millisecond )
@@ -719,7 +720,7 @@ func TestPickFirstOrderOneServerDown(t *testing.T) {
719
720
checkServerUp (t , servers [0 ])
720
721
721
722
for i := 0 ; i < 20 ; i ++ {
722
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [1 ].port {
723
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [1 ].port {
723
724
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 1 , err , servers [1 ].port )
724
725
}
725
726
time .Sleep (10 * time .Millisecond )
@@ -732,13 +733,13 @@ func TestPickFirstOrderOneServerDown(t *testing.T) {
732
733
}
733
734
r .w .inject ([]* naming.Update {u })
734
735
for {
735
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && ErrorDesc (err ) == servers [0 ].port {
736
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err != nil && errorDesc (err ) == servers [0 ].port {
736
737
break
737
738
}
738
739
time .Sleep (1 * time .Second )
739
740
}
740
741
for i := 0 ; i < 20 ; i ++ {
741
- if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || ErrorDesc (err ) != servers [0 ].port {
742
+ if err := Invoke (context .Background (), "/foo/bar" , & req , & reply , cc ); err == nil || errorDesc (err ) != servers [0 ].port {
742
743
t .Fatalf ("Index %d: Invoke(_, _, _, _, _) = %v, want %s" , 0 , err , servers [0 ].port )
743
744
}
744
745
time .Sleep (10 * time .Millisecond )
0 commit comments