test

client/client.go

@@ -44,3 +44,7 @@ func getDialInsecure(r *registry.Registry, logger log.Logger,
 	connMap[endpointNameKey] = conn
 	return conn, nil
 }
+
+func GetConn(endpointNameKey string) *grpc2.ClientConn {
+	return connMap[endpointNameKey]
+}