Bläddra i källkod

integration: use cluster adapter in tests

Signed-off-by: Gyu-Ho Lee <gyuhox@gmail.com>
Gyu-Ho Lee 8 år sedan
förälder
incheckning
1e3274dfa2
1 ändrade filer med 2 tillägg och 1 borttagningar
  1. 2 1
      integration/cluster_proxy.go

+ 2 - 1
integration/cluster_proxy.go

@@ -57,9 +57,10 @@ func toGRPC(c *clientv3.Client) grpcAPI {
 	wp, wpch := grpcproxy.NewWatchProxy(c)
 	wp, wpch := grpcproxy.NewWatchProxy(c)
 	lp, lpch := grpcproxy.NewLeaseProxy(c)
 	lp, lpch := grpcproxy.NewLeaseProxy(c)
 	mp := grpcproxy.NewMaintenanceProxy(c)
 	mp := grpcproxy.NewMaintenanceProxy(c)
+	clp, _ := grpcproxy.NewClusterProxy(c, "", "") // without registering proxy URLs
 
 
 	grpc := grpcAPI{
 	grpc := grpcAPI{
-		pb.NewClusterClient(c.ActiveConnection()),
+		adapter.ClusterServerToClusterClient(clp),
 		adapter.KvServerToKvClient(kvp),
 		adapter.KvServerToKvClient(kvp),
 		adapter.LeaseServerToLeaseClient(lp),
 		adapter.LeaseServerToLeaseClient(lp),
 		adapter.WatchServerToWatchClient(wp),
 		adapter.WatchServerToWatchClient(wp),