integration: use maintenance API adapter in tests

Signed-off-by: Gyu-Ho Lee <gyuhox@gmail.com>
This commit is contained in:
Gyu-Ho Lee 2017-03-27 14:30:47 -07:00
parent 0bf110e27f
commit c90a4b96d1

View File

@ -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}