Codebase list golang-github-go-kit-kit / f4d5e15
Simplify retry test Peter Bourgon 8 years ago
1 changed file(s) with 6 addition(s) and 18 deletion(s). Raw diff Collapse all Expand all
1717 lb = loadbalancer.RoundRobin(p)
1818 )
1919
20 {
21 max := 999
22 e := loadbalancer.Retry(max, lb)
23 if _, err := e(context.Background(), struct{}{}); err == nil {
24 t.Errorf("expected error, got none")
25 }
20 if _, err := loadbalancer.Retry(999, lb)(context.Background(), struct{}{}); err == nil {
21 t.Errorf("expected error, got none")
2622 }
2723
2824 endpoints = []endpoint.Endpoint{
3329 p.Replace(endpoints)
3430 runtime.Gosched()
3531
36 {
37 max := len(endpoints) - 1
38 e := loadbalancer.Retry(max, lb)
39 if _, err := e(context.Background(), struct{}{}); err == nil {
40 t.Errorf("expected error, got none")
41 }
32 if _, err := loadbalancer.Retry(len(endpoints)-1, lb)(context.Background(), struct{}{}); err == nil {
33 t.Errorf("expected error, got none")
4234 }
4335
44 {
45 max := len(endpoints)
46 e := loadbalancer.Retry(max, lb)
47 if _, err := e(context.Background(), struct{}{}); err != nil {
48 t.Error(err)
49 }
36 if _, err := loadbalancer.Retry(len(endpoints), lb)(context.Background(), struct{}{}); err != nil {
37 t.Error(err)
5038 }
5139 }