remove node selector test from client
This commit is contained in:
		| @@ -1,70 +0,0 @@ | ||||
| package client | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/micro/go-micro/registry" | ||||
| 	"golang.org/x/net/context" | ||||
| ) | ||||
|  | ||||
| type mockRegistry struct{} | ||||
|  | ||||
| func (m *mockRegistry) GetService(service string) ([]*registry.Service, error) { | ||||
| 	return []*registry.Service{ | ||||
| 		{ | ||||
| 			Name:    "foo", | ||||
| 			Version: "1.0.0", | ||||
| 			Nodes: []*registry.Node{ | ||||
| 				{ | ||||
| 					Id:      "foo-123", | ||||
| 					Address: "localhost", | ||||
| 					Port:    9999, | ||||
| 				}, | ||||
| 			}, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Name:    "foo", | ||||
| 			Version: "1.0.1", | ||||
| 			Nodes: []*registry.Node{ | ||||
| 				{ | ||||
| 					Id:      "foo-321", | ||||
| 					Address: "localhost", | ||||
| 					Port:    6666, | ||||
| 				}, | ||||
| 			}, | ||||
| 		}, | ||||
| 	}, nil | ||||
| } | ||||
|  | ||||
| func (m *mockRegistry) ListServices() ([]*registry.Service, error) { | ||||
| 	return []*registry.Service{}, nil | ||||
| } | ||||
|  | ||||
| func (m *mockRegistry) Register(s *registry.Service) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *mockRegistry) Deregister(s *registry.Service) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *mockRegistry) Watch() (registry.Watcher, error) { | ||||
| 	return nil, nil | ||||
| } | ||||
|  | ||||
| func TestNodeSelector(t *testing.T) { | ||||
| 	counts := map[string]int{} | ||||
| 	n := &nodeSelector{ | ||||
| 		&mockRegistry{}, | ||||
| 	} | ||||
|  | ||||
| 	for i := 0; i < 100; i++ { | ||||
| 		n, err := n.Select(context.Background(), newRpcRequest("foo", "Foo.Bar", nil, "")) | ||||
| 		if err != nil { | ||||
| 			t.Errorf("Expected node, got err: %v", err) | ||||
| 		} | ||||
| 		counts[n.Id]++ | ||||
| 	} | ||||
|  | ||||
| 	t.Logf("Counts %v", counts) | ||||
| } | ||||
		Reference in New Issue
	
	Block a user