func buildMockDiscoveryServer(port uint) (*mockDiscoveryServer, *grpc.Server) { lis, err := net.Listen("tcp", fmt.Sprintf(":%d", port)) if err != nil { panic(err) } d := &mockDiscoveryServer{} s := grpc.NewServer() pb.RegisterDiscoveryServer(s, d) go s.Serve(lis) return d, s }
// RegisterRPC registers the local discovery with a gRPC server func (d *discovery) RegisterRPC(s *grpc.Server) { server := &discoveryServer{d} pb.RegisterDiscoveryServer(s, server) }