From c90a4b96d15a12af4059a5d76a3b6e36b0f0e42a Mon Sep 17 00:00:00 2001 From: Gyu-Ho Lee Date: Mon, 27 Mar 2017 14:30:47 -0700 Subject: [PATCH] integration: use maintenance API adapter in tests Signed-off-by: Gyu-Ho Lee --- integration/cluster_proxy.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/integration/cluster_proxy.go b/integration/cluster_proxy.go index 5d0b97abc..30984ddef 100644 --- a/integration/cluster_proxy.go +++ b/integration/cluster_proxy.go @@ -56,13 +56,14 @@ func toGRPC(c *clientv3.Client) grpcAPI { kvp, kvpch := grpcproxy.NewKvProxy(c) wp, wpch := grpcproxy.NewWatchProxy(c) lp, lpch := grpcproxy.NewLeaseProxy(c) + mp := grpcproxy.NewMaintenanceProxy(c) grpc := grpcAPI{ pb.NewClusterClient(c.ActiveConnection()), adapter.KvServerToKvClient(kvp), adapter.LeaseServerToLeaseClient(lp), adapter.WatchServerToWatchClient(wp), - pb.NewMaintenanceClient(c.ActiveConnection()), + adapter.MaintenanceServerToMaintenanceClient(mp), pb.NewAuthClient(c.ActiveConnection()), } proxies[c] = grpcClientProxy{grpc: grpc, wdonec: wpch, kvdonec: kvpch, lpdonec: lpch}