diff --git a/asset/apiv1/mock_test.go b/asset/apiv1/mock_test.go
deleted file mode 100644
index 61ae4b99f9c..00000000000
--- a/asset/apiv1/mock_test.go
+++ /dev/null
@@ -1,265 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package asset
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	assetpb "google.golang.org/genproto/googleapis/cloud/asset/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockAssetServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	assetpb.AssetServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockAssetServer) ExportAssets(ctx context.Context, req *assetpb.ExportAssetsRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAssetServer) BatchGetAssetsHistory(ctx context.Context, req *assetpb.BatchGetAssetsHistoryRequest) (*assetpb.BatchGetAssetsHistoryResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*assetpb.BatchGetAssetsHistoryResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockAsset mockAssetServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	assetpb.RegisterAssetServiceServer(serv, &mockAsset)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestAssetServiceExportAssets(t *testing.T) {
-	var expectedResponse *assetpb.ExportAssetsResponse = &assetpb.ExportAssetsResponse{}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAsset.resps = append(mockAsset.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var parent string = "parent-995424086"
-	var outputConfig *assetpb.OutputConfig = &assetpb.OutputConfig{}
-	var request = &assetpb.ExportAssetsRequest{
-		Parent:       parent,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportAssets(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAssetServiceExportAssetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = nil
-	mockAsset.resps = append(mockAsset.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var parent string = "parent-995424086"
-	var outputConfig *assetpb.OutputConfig = &assetpb.OutputConfig{}
-	var request = &assetpb.ExportAssetsRequest{
-		Parent:       parent,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportAssets(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAssetServiceBatchGetAssetsHistory(t *testing.T) {
-	var expectedResponse *assetpb.BatchGetAssetsHistoryResponse = &assetpb.BatchGetAssetsHistoryResponse{}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	mockAsset.resps = append(mockAsset.resps[:0], expectedResponse)
-
-	var parent string = "parent-995424086"
-	var contentType assetpb.ContentType = assetpb.ContentType_CONTENT_TYPE_UNSPECIFIED
-	var readTimeWindow *assetpb.TimeWindow = &assetpb.TimeWindow{}
-	var request = &assetpb.BatchGetAssetsHistoryRequest{
-		Parent:         parent,
-		ContentType:    contentType,
-		ReadTimeWindow: readTimeWindow,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchGetAssetsHistory(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAssetServiceBatchGetAssetsHistoryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = gstatus.Error(errCode, "test error")
-
-	var parent string = "parent-995424086"
-	var contentType assetpb.ContentType = assetpb.ContentType_CONTENT_TYPE_UNSPECIFIED
-	var readTimeWindow *assetpb.TimeWindow = &assetpb.TimeWindow{}
-	var request = &assetpb.BatchGetAssetsHistoryRequest{
-		Parent:         parent,
-		ContentType:    contentType,
-		ReadTimeWindow: readTimeWindow,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchGetAssetsHistory(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/asset/apiv1p2beta1/mock_test.go b/asset/apiv1p2beta1/mock_test.go
deleted file mode 100644
index 5908702e67e..00000000000
--- a/asset/apiv1p2beta1/mock_test.go
+++ /dev/null
@@ -1,449 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package asset
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	assetpb "google.golang.org/genproto/googleapis/cloud/asset/v1p2beta1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockAssetServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	assetpb.AssetServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockAssetServer) CreateFeed(ctx context.Context, req *assetpb.CreateFeedRequest) (*assetpb.Feed, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*assetpb.Feed), nil
-}
-
-func (s *mockAssetServer) GetFeed(ctx context.Context, req *assetpb.GetFeedRequest) (*assetpb.Feed, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*assetpb.Feed), nil
-}
-
-func (s *mockAssetServer) ListFeeds(ctx context.Context, req *assetpb.ListFeedsRequest) (*assetpb.ListFeedsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*assetpb.ListFeedsResponse), nil
-}
-
-func (s *mockAssetServer) UpdateFeed(ctx context.Context, req *assetpb.UpdateFeedRequest) (*assetpb.Feed, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*assetpb.Feed), nil
-}
-
-func (s *mockAssetServer) DeleteFeed(ctx context.Context, req *assetpb.DeleteFeedRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockAsset mockAssetServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	assetpb.RegisterAssetServiceServer(serv, &mockAsset)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestAssetServiceCreateFeed(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &assetpb.Feed{
-		Name: name,
-	}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	mockAsset.resps = append(mockAsset.resps[:0], expectedResponse)
-
-	var parent string = "parent-995424086"
-	var feedId string = "feedId-976011428"
-	var feed *assetpb.Feed = &assetpb.Feed{}
-	var request = &assetpb.CreateFeedRequest{
-		Parent: parent,
-		FeedId: feedId,
-		Feed:   feed,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateFeed(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAssetServiceCreateFeedError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = gstatus.Error(errCode, "test error")
-
-	var parent string = "parent-995424086"
-	var feedId string = "feedId-976011428"
-	var feed *assetpb.Feed = &assetpb.Feed{}
-	var request = &assetpb.CreateFeedRequest{
-		Parent: parent,
-		FeedId: feedId,
-		Feed:   feed,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateFeed(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAssetServiceGetFeed(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &assetpb.Feed{
-		Name: name2,
-	}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	mockAsset.resps = append(mockAsset.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/feeds/%s", "[PROJECT]", "[FEED]")
-	var request = &assetpb.GetFeedRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetFeed(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAssetServiceGetFeedError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/feeds/%s", "[PROJECT]", "[FEED]")
-	var request = &assetpb.GetFeedRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetFeed(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAssetServiceListFeeds(t *testing.T) {
-	var expectedResponse *assetpb.ListFeedsResponse = &assetpb.ListFeedsResponse{}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	mockAsset.resps = append(mockAsset.resps[:0], expectedResponse)
-
-	var parent string = "parent-995424086"
-	var request = &assetpb.ListFeedsRequest{
-		Parent: parent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFeeds(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAssetServiceListFeedsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = gstatus.Error(errCode, "test error")
-
-	var parent string = "parent-995424086"
-	var request = &assetpb.ListFeedsRequest{
-		Parent: parent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFeeds(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAssetServiceUpdateFeed(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &assetpb.Feed{
-		Name: name,
-	}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	mockAsset.resps = append(mockAsset.resps[:0], expectedResponse)
-
-	var feed *assetpb.Feed = &assetpb.Feed{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &assetpb.UpdateFeedRequest{
-		Feed:       feed,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateFeed(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAssetServiceUpdateFeedError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = gstatus.Error(errCode, "test error")
-
-	var feed *assetpb.Feed = &assetpb.Feed{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &assetpb.UpdateFeedRequest{
-		Feed:       feed,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateFeed(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAssetServiceDeleteFeed(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAsset.err = nil
-	mockAsset.reqs = nil
-
-	mockAsset.resps = append(mockAsset.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/feeds/%s", "[PROJECT]", "[FEED]")
-	var request = &assetpb.DeleteFeedRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteFeed(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAsset.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAssetServiceDeleteFeedError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAsset.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/feeds/%s", "[PROJECT]", "[FEED]")
-	var request = &assetpb.DeleteFeedRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteFeed(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
diff --git a/automl/apiv1/mock_test.go b/automl/apiv1/mock_test.go
deleted file mode 100644
index 59548484f4d..00000000000
--- a/automl/apiv1/mock_test.go
+++ /dev/null
@@ -1,1870 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package automl
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	automlpb "google.golang.org/genproto/googleapis/cloud/automl/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockAutoMlServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	automlpb.AutoMlServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockAutoMlServer) CreateDataset(ctx context.Context, req *automlpb.CreateDatasetRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetDataset(ctx context.Context, req *automlpb.GetDatasetRequest) (*automlpb.Dataset, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Dataset), nil
-}
-
-func (s *mockAutoMlServer) ListDatasets(ctx context.Context, req *automlpb.ListDatasetsRequest) (*automlpb.ListDatasetsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListDatasetsResponse), nil
-}
-
-func (s *mockAutoMlServer) UpdateDataset(ctx context.Context, req *automlpb.UpdateDatasetRequest) (*automlpb.Dataset, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Dataset), nil
-}
-
-func (s *mockAutoMlServer) DeleteDataset(ctx context.Context, req *automlpb.DeleteDatasetRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ImportData(ctx context.Context, req *automlpb.ImportDataRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ExportData(ctx context.Context, req *automlpb.ExportDataRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetAnnotationSpec(ctx context.Context, req *automlpb.GetAnnotationSpecRequest) (*automlpb.AnnotationSpec, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.AnnotationSpec), nil
-}
-
-func (s *mockAutoMlServer) CreateModel(ctx context.Context, req *automlpb.CreateModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetModel(ctx context.Context, req *automlpb.GetModelRequest) (*automlpb.Model, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Model), nil
-}
-
-func (s *mockAutoMlServer) ListModels(ctx context.Context, req *automlpb.ListModelsRequest) (*automlpb.ListModelsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListModelsResponse), nil
-}
-
-func (s *mockAutoMlServer) DeleteModel(ctx context.Context, req *automlpb.DeleteModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) UpdateModel(ctx context.Context, req *automlpb.UpdateModelRequest) (*automlpb.Model, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Model), nil
-}
-
-func (s *mockAutoMlServer) DeployModel(ctx context.Context, req *automlpb.DeployModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) UndeployModel(ctx context.Context, req *automlpb.UndeployModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ExportModel(ctx context.Context, req *automlpb.ExportModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetModelEvaluation(ctx context.Context, req *automlpb.GetModelEvaluationRequest) (*automlpb.ModelEvaluation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ModelEvaluation), nil
-}
-
-func (s *mockAutoMlServer) ListModelEvaluations(ctx context.Context, req *automlpb.ListModelEvaluationsRequest) (*automlpb.ListModelEvaluationsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListModelEvaluationsResponse), nil
-}
-
-type mockPredictionServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	automlpb.PredictionServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockPredictionServer) Predict(ctx context.Context, req *automlpb.PredictRequest) (*automlpb.PredictResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.PredictResponse), nil
-}
-
-func (s *mockPredictionServer) BatchPredict(ctx context.Context, req *automlpb.BatchPredictRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockAutoMl     mockAutoMlServer
-	mockPrediction mockPredictionServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	automlpb.RegisterAutoMlServer(serv, &mockAutoMl)
-	automlpb.RegisterPredictionServiceServer(serv, &mockPrediction)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestAutoMlCreateDataset(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var exampleCount int32 = 1517063674
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Dataset{
-		Name:         name,
-		DisplayName:  displayName,
-		Description:  description,
-		ExampleCount: exampleCount,
-		Etag:         etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var request = &automlpb.CreateDatasetRequest{
-		Parent:  formattedParent,
-		Dataset: dataset,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateDataset(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlCreateDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var request = &automlpb.CreateDatasetRequest{
-		Parent:  formattedParent,
-		Dataset: dataset,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateDataset(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlUpdateDataset(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var exampleCount int32 = 1517063674
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Dataset{
-		Name:         name,
-		DisplayName:  displayName,
-		Description:  description,
-		ExampleCount: exampleCount,
-		Etag:         etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &automlpb.UpdateDatasetRequest{
-		Dataset:    dataset,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDataset(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlUpdateDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &automlpb.UpdateDatasetRequest{
-		Dataset:    dataset,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDataset(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetDataset(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var exampleCount int32 = 1517063674
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Dataset{
-		Name:         name2,
-		DisplayName:  displayName,
-		Description:  description,
-		ExampleCount: exampleCount,
-		Etag:         etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.GetDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDataset(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.GetDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDataset(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListDatasets(t *testing.T) {
-	var nextPageToken string = ""
-	var datasetsElement *automlpb.Dataset = &automlpb.Dataset{}
-	var datasets = []*automlpb.Dataset{datasetsElement}
-	var expectedResponse = &automlpb.ListDatasetsResponse{
-		NextPageToken: nextPageToken,
-		Datasets:      datasets,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListDatasetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDatasets(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Datasets[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListDatasetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListDatasetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDatasets(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlDeleteDataset(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.DeleteDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteDataset(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlDeleteDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.DeleteDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteDataset(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlImportData(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var inputConfig *automlpb.InputConfig = &automlpb.InputConfig{}
-	var request = &automlpb.ImportDataRequest{
-		Name:        formattedName,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlImportDataError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var inputConfig *automlpb.InputConfig = &automlpb.InputConfig{}
-	var request = &automlpb.ImportDataRequest{
-		Name:        formattedName,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlExportData(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var outputConfig *automlpb.OutputConfig = &automlpb.OutputConfig{}
-	var request = &automlpb.ExportDataRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlExportDataError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var outputConfig *automlpb.OutputConfig = &automlpb.OutputConfig{}
-	var request = &automlpb.ExportDataRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlGetAnnotationSpec(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var exampleCount int32 = 1517063674
-	var expectedResponse = &automlpb.AnnotationSpec{
-		Name:         name2,
-		DisplayName:  displayName,
-		ExampleCount: exampleCount,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/annotationSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[ANNOTATION_SPEC]")
-	var request = &automlpb.GetAnnotationSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAnnotationSpec(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetAnnotationSpecError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/annotationSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[ANNOTATION_SPEC]")
-	var request = &automlpb.GetAnnotationSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAnnotationSpec(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlCreateModel(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var datasetId string = "datasetId-2115646910"
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Model{
-		Name:        name,
-		DisplayName: displayName,
-		DatasetId:   datasetId,
-		Etag:        etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var model *automlpb.Model = &automlpb.Model{}
-	var request = &automlpb.CreateModelRequest{
-		Parent: formattedParent,
-		Model:  model,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlCreateModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var model *automlpb.Model = &automlpb.Model{}
-	var request = &automlpb.CreateModelRequest{
-		Parent: formattedParent,
-		Model:  model,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetModel(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var datasetId string = "datasetId-2115646910"
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Model{
-		Name:        name2,
-		DisplayName: displayName,
-		DatasetId:   datasetId,
-		Etag:        etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.GetModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.GetModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlUpdateModel(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var datasetId string = "datasetId-2115646910"
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Model{
-		Name:        name,
-		DisplayName: displayName,
-		DatasetId:   datasetId,
-		Etag:        etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var model *automlpb.Model = &automlpb.Model{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &automlpb.UpdateModelRequest{
-		Model:      model,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateModel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlUpdateModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var model *automlpb.Model = &automlpb.Model{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &automlpb.UpdateModelRequest{
-		Model:      model,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateModel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListModels(t *testing.T) {
-	var nextPageToken string = ""
-	var modelElement *automlpb.Model = &automlpb.Model{}
-	var model = []*automlpb.Model{modelElement}
-	var expectedResponse = &automlpb.ListModelsResponse{
-		NextPageToken: nextPageToken,
-		Model:         model,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListModelsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModels(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Model[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListModelsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListModelsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModels(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlDeleteModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeleteModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlDeleteModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeleteModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlDeployModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlDeployModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlUndeployModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.UndeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UndeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlUndeployModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.UndeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UndeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlExportModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var outputConfig *automlpb.ModelExportOutputConfig = &automlpb.ModelExportOutputConfig{}
-	var request = &automlpb.ExportModelRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlExportModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var outputConfig *automlpb.ModelExportOutputConfig = &automlpb.ModelExportOutputConfig{}
-	var request = &automlpb.ExportModelRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlGetModelEvaluation(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var annotationSpecId string = "annotationSpecId60690191"
-	var displayName string = "displayName1615086568"
-	var evaluatedExampleCount int32 = 277565350
-	var expectedResponse = &automlpb.ModelEvaluation{
-		Name:                  name2,
-		AnnotationSpecId:      annotationSpecId,
-		DisplayName:           displayName,
-		EvaluatedExampleCount: evaluatedExampleCount,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s/modelEvaluations/%s", "[PROJECT]", "[LOCATION]", "[MODEL]", "[MODEL_EVALUATION]")
-	var request = &automlpb.GetModelEvaluationRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModelEvaluation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetModelEvaluationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s/modelEvaluations/%s", "[PROJECT]", "[LOCATION]", "[MODEL]", "[MODEL_EVALUATION]")
-	var request = &automlpb.GetModelEvaluationRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModelEvaluation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListModelEvaluations(t *testing.T) {
-	var nextPageToken string = ""
-	var modelEvaluationElement *automlpb.ModelEvaluation = &automlpb.ModelEvaluation{}
-	var modelEvaluation = []*automlpb.ModelEvaluation{modelEvaluationElement}
-	var expectedResponse = &automlpb.ListModelEvaluationsResponse{
-		NextPageToken:   nextPageToken,
-		ModelEvaluation: modelEvaluation,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var filter string = "filter-1274492040"
-	var request = &automlpb.ListModelEvaluationsRequest{
-		Parent: formattedParent,
-		Filter: filter,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModelEvaluations(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ModelEvaluation[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListModelEvaluationsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var filter string = "filter-1274492040"
-	var request = &automlpb.ListModelEvaluationsRequest{
-		Parent: formattedParent,
-		Filter: filter,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModelEvaluations(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPredictionServicePredict(t *testing.T) {
-	var expectedResponse *automlpb.PredictResponse = &automlpb.PredictResponse{}
-
-	mockPrediction.err = nil
-	mockPrediction.reqs = nil
-
-	mockPrediction.resps = append(mockPrediction.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var payload *automlpb.ExamplePayload = &automlpb.ExamplePayload{}
-	var request = &automlpb.PredictRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Predict(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPrediction.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPredictionServicePredictError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPrediction.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var payload *automlpb.ExamplePayload = &automlpb.ExamplePayload{}
-	var request = &automlpb.PredictRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Predict(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPredictionServiceBatchPredict(t *testing.T) {
-	var expectedResponse *automlpb.BatchPredictResult = &automlpb.BatchPredictResult{}
-
-	mockPrediction.err = nil
-	mockPrediction.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockPrediction.resps = append(mockPrediction.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var inputConfig *automlpb.BatchPredictInputConfig = &automlpb.BatchPredictInputConfig{}
-	var outputConfig *automlpb.BatchPredictOutputConfig = &automlpb.BatchPredictOutputConfig{}
-	var request = &automlpb.BatchPredictRequest{
-		Name:         formattedName,
-		InputConfig:  inputConfig,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchPredict(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPrediction.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPredictionServiceBatchPredictError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPrediction.err = nil
-	mockPrediction.resps = append(mockPrediction.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var inputConfig *automlpb.BatchPredictInputConfig = &automlpb.BatchPredictInputConfig{}
-	var outputConfig *automlpb.BatchPredictOutputConfig = &automlpb.BatchPredictOutputConfig{}
-	var request = &automlpb.BatchPredictRequest{
-		Name:         formattedName,
-		InputConfig:  inputConfig,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchPredict(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/automl/apiv1beta1/mock_test.go b/automl/apiv1beta1/mock_test.go
deleted file mode 100644
index 2584d8a8b64..00000000000
--- a/automl/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,2324 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package automl
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	automlpb "google.golang.org/genproto/googleapis/cloud/automl/v1beta1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockAutoMlServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	automlpb.AutoMlServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockAutoMlServer) CreateDataset(ctx context.Context, req *automlpb.CreateDatasetRequest) (*automlpb.Dataset, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Dataset), nil
-}
-
-func (s *mockAutoMlServer) GetDataset(ctx context.Context, req *automlpb.GetDatasetRequest) (*automlpb.Dataset, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Dataset), nil
-}
-
-func (s *mockAutoMlServer) ListDatasets(ctx context.Context, req *automlpb.ListDatasetsRequest) (*automlpb.ListDatasetsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListDatasetsResponse), nil
-}
-
-func (s *mockAutoMlServer) UpdateDataset(ctx context.Context, req *automlpb.UpdateDatasetRequest) (*automlpb.Dataset, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Dataset), nil
-}
-
-func (s *mockAutoMlServer) DeleteDataset(ctx context.Context, req *automlpb.DeleteDatasetRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ImportData(ctx context.Context, req *automlpb.ImportDataRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ExportData(ctx context.Context, req *automlpb.ExportDataRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetAnnotationSpec(ctx context.Context, req *automlpb.GetAnnotationSpecRequest) (*automlpb.AnnotationSpec, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.AnnotationSpec), nil
-}
-
-func (s *mockAutoMlServer) GetTableSpec(ctx context.Context, req *automlpb.GetTableSpecRequest) (*automlpb.TableSpec, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.TableSpec), nil
-}
-
-func (s *mockAutoMlServer) ListTableSpecs(ctx context.Context, req *automlpb.ListTableSpecsRequest) (*automlpb.ListTableSpecsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListTableSpecsResponse), nil
-}
-
-func (s *mockAutoMlServer) UpdateTableSpec(ctx context.Context, req *automlpb.UpdateTableSpecRequest) (*automlpb.TableSpec, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.TableSpec), nil
-}
-
-func (s *mockAutoMlServer) GetColumnSpec(ctx context.Context, req *automlpb.GetColumnSpecRequest) (*automlpb.ColumnSpec, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ColumnSpec), nil
-}
-
-func (s *mockAutoMlServer) ListColumnSpecs(ctx context.Context, req *automlpb.ListColumnSpecsRequest) (*automlpb.ListColumnSpecsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListColumnSpecsResponse), nil
-}
-
-func (s *mockAutoMlServer) UpdateColumnSpec(ctx context.Context, req *automlpb.UpdateColumnSpecRequest) (*automlpb.ColumnSpec, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ColumnSpec), nil
-}
-
-func (s *mockAutoMlServer) CreateModel(ctx context.Context, req *automlpb.CreateModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetModel(ctx context.Context, req *automlpb.GetModelRequest) (*automlpb.Model, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.Model), nil
-}
-
-func (s *mockAutoMlServer) ListModels(ctx context.Context, req *automlpb.ListModelsRequest) (*automlpb.ListModelsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListModelsResponse), nil
-}
-
-func (s *mockAutoMlServer) DeleteModel(ctx context.Context, req *automlpb.DeleteModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) DeployModel(ctx context.Context, req *automlpb.DeployModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) UndeployModel(ctx context.Context, req *automlpb.UndeployModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ExportModel(ctx context.Context, req *automlpb.ExportModelRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) ExportEvaluatedExamples(ctx context.Context, req *automlpb.ExportEvaluatedExamplesRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAutoMlServer) GetModelEvaluation(ctx context.Context, req *automlpb.GetModelEvaluationRequest) (*automlpb.ModelEvaluation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ModelEvaluation), nil
-}
-
-func (s *mockAutoMlServer) ListModelEvaluations(ctx context.Context, req *automlpb.ListModelEvaluationsRequest) (*automlpb.ListModelEvaluationsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.ListModelEvaluationsResponse), nil
-}
-
-type mockPredictionServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	automlpb.PredictionServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockPredictionServer) Predict(ctx context.Context, req *automlpb.PredictRequest) (*automlpb.PredictResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*automlpb.PredictResponse), nil
-}
-
-func (s *mockPredictionServer) BatchPredict(ctx context.Context, req *automlpb.BatchPredictRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockAutoMl     mockAutoMlServer
-	mockPrediction mockPredictionServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	automlpb.RegisterAutoMlServer(serv, &mockAutoMl)
-	automlpb.RegisterPredictionServiceServer(serv, &mockPrediction)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestAutoMlCreateDataset(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var exampleCount int32 = 1517063674
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Dataset{
-		Name:         name,
-		DisplayName:  displayName,
-		Description:  description,
-		ExampleCount: exampleCount,
-		Etag:         etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var request = &automlpb.CreateDatasetRequest{
-		Parent:  formattedParent,
-		Dataset: dataset,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDataset(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlCreateDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var request = &automlpb.CreateDatasetRequest{
-		Parent:  formattedParent,
-		Dataset: dataset,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDataset(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlUpdateDataset(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var exampleCount int32 = 1517063674
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Dataset{
-		Name:         name,
-		DisplayName:  displayName,
-		Description:  description,
-		ExampleCount: exampleCount,
-		Etag:         etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var request = &automlpb.UpdateDatasetRequest{
-		Dataset: dataset,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDataset(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlUpdateDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var dataset *automlpb.Dataset = &automlpb.Dataset{}
-	var request = &automlpb.UpdateDatasetRequest{
-		Dataset: dataset,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDataset(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetDataset(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var exampleCount int32 = 1517063674
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.Dataset{
-		Name:         name2,
-		DisplayName:  displayName,
-		Description:  description,
-		ExampleCount: exampleCount,
-		Etag:         etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.GetDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDataset(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.GetDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDataset(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListDatasets(t *testing.T) {
-	var nextPageToken string = ""
-	var datasetsElement *automlpb.Dataset = &automlpb.Dataset{}
-	var datasets = []*automlpb.Dataset{datasetsElement}
-	var expectedResponse = &automlpb.ListDatasetsResponse{
-		NextPageToken: nextPageToken,
-		Datasets:      datasets,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListDatasetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDatasets(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Datasets[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListDatasetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListDatasetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDatasets(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlDeleteDataset(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.DeleteDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteDataset(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlDeleteDatasetError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.DeleteDatasetRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteDataset(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlImportData(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var inputConfig *automlpb.InputConfig = &automlpb.InputConfig{}
-	var request = &automlpb.ImportDataRequest{
-		Name:        formattedName,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlImportDataError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var inputConfig *automlpb.InputConfig = &automlpb.InputConfig{}
-	var request = &automlpb.ImportDataRequest{
-		Name:        formattedName,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlExportData(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var outputConfig *automlpb.OutputConfig = &automlpb.OutputConfig{}
-	var request = &automlpb.ExportDataRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlExportDataError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var outputConfig *automlpb.OutputConfig = &automlpb.OutputConfig{}
-	var request = &automlpb.ExportDataRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportData(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlCreateModel(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var datasetId string = "datasetId-2115646910"
-	var expectedResponse = &automlpb.Model{
-		Name:        name,
-		DisplayName: displayName,
-		DatasetId:   datasetId,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var model *automlpb.Model = &automlpb.Model{}
-	var request = &automlpb.CreateModelRequest{
-		Parent: formattedParent,
-		Model:  model,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlCreateModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var model *automlpb.Model = &automlpb.Model{}
-	var request = &automlpb.CreateModelRequest{
-		Parent: formattedParent,
-		Model:  model,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetModel(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var datasetId string = "datasetId-2115646910"
-	var expectedResponse = &automlpb.Model{
-		Name:        name2,
-		DisplayName: displayName,
-		DatasetId:   datasetId,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.GetModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.GetModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListModels(t *testing.T) {
-	var nextPageToken string = ""
-	var modelElement *automlpb.Model = &automlpb.Model{}
-	var model = []*automlpb.Model{modelElement}
-	var expectedResponse = &automlpb.ListModelsResponse{
-		NextPageToken: nextPageToken,
-		Model:         model,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListModelsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModels(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Model[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListModelsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &automlpb.ListModelsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModels(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlDeleteModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeleteModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlDeleteModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeleteModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlDeployModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlDeployModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.DeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlUndeployModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.UndeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UndeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlUndeployModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.UndeployModelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UndeployModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlGetModelEvaluation(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var annotationSpecId string = "annotationSpecId60690191"
-	var displayName string = "displayName1615086568"
-	var evaluatedExampleCount int32 = 277565350
-	var expectedResponse = &automlpb.ModelEvaluation{
-		Name:                  name2,
-		AnnotationSpecId:      annotationSpecId,
-		DisplayName:           displayName,
-		EvaluatedExampleCount: evaluatedExampleCount,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s/modelEvaluations/%s", "[PROJECT]", "[LOCATION]", "[MODEL]", "[MODEL_EVALUATION]")
-	var request = &automlpb.GetModelEvaluationRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModelEvaluation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetModelEvaluationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s/modelEvaluations/%s", "[PROJECT]", "[LOCATION]", "[MODEL]", "[MODEL_EVALUATION]")
-	var request = &automlpb.GetModelEvaluationRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetModelEvaluation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlExportModel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var outputConfig *automlpb.ModelExportOutputConfig = &automlpb.ModelExportOutputConfig{}
-	var request = &automlpb.ExportModelRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlExportModelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var outputConfig *automlpb.ModelExportOutputConfig = &automlpb.ModelExportOutputConfig{}
-	var request = &automlpb.ExportModelRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportModel(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlExportEvaluatedExamples(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var outputConfig *automlpb.ExportEvaluatedExamplesOutputConfig = &automlpb.ExportEvaluatedExamplesOutputConfig{}
-	var request = &automlpb.ExportEvaluatedExamplesRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportEvaluatedExamples(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAutoMlExportEvaluatedExamplesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = nil
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var outputConfig *automlpb.ExportEvaluatedExamplesOutputConfig = &automlpb.ExportEvaluatedExamplesOutputConfig{}
-	var request = &automlpb.ExportEvaluatedExamplesRequest{
-		Name:         formattedName,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportEvaluatedExamples(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAutoMlListModelEvaluations(t *testing.T) {
-	var nextPageToken string = ""
-	var modelEvaluationElement *automlpb.ModelEvaluation = &automlpb.ModelEvaluation{}
-	var modelEvaluation = []*automlpb.ModelEvaluation{modelEvaluationElement}
-	var expectedResponse = &automlpb.ListModelEvaluationsResponse{
-		NextPageToken:   nextPageToken,
-		ModelEvaluation: modelEvaluation,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.ListModelEvaluationsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModelEvaluations(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ModelEvaluation[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListModelEvaluationsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var request = &automlpb.ListModelEvaluationsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListModelEvaluations(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetAnnotationSpec(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var exampleCount int32 = 1517063674
-	var expectedResponse = &automlpb.AnnotationSpec{
-		Name:         name2,
-		DisplayName:  displayName,
-		ExampleCount: exampleCount,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/annotationSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[ANNOTATION_SPEC]")
-	var request = &automlpb.GetAnnotationSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAnnotationSpec(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetAnnotationSpecError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/annotationSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[ANNOTATION_SPEC]")
-	var request = &automlpb.GetAnnotationSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAnnotationSpec(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetTableSpec(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var timeColumnSpecId string = "timeColumnSpecId1558734824"
-	var rowCount int64 = 1340416618
-	var validRowCount int64 = 406068761
-	var columnCount int64 = 122671386
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.TableSpec{
-		Name:             name2,
-		TimeColumnSpecId: timeColumnSpecId,
-		RowCount:         rowCount,
-		ValidRowCount:    validRowCount,
-		ColumnCount:      columnCount,
-		Etag:             etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/tableSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[TABLE_SPEC]")
-	var request = &automlpb.GetTableSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTableSpec(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetTableSpecError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/tableSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[TABLE_SPEC]")
-	var request = &automlpb.GetTableSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTableSpec(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListTableSpecs(t *testing.T) {
-	var nextPageToken string = ""
-	var tableSpecsElement *automlpb.TableSpec = &automlpb.TableSpec{}
-	var tableSpecs = []*automlpb.TableSpec{tableSpecsElement}
-	var expectedResponse = &automlpb.ListTableSpecsResponse{
-		NextPageToken: nextPageToken,
-		TableSpecs:    tableSpecs,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.ListTableSpecsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTableSpecs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.TableSpecs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListTableSpecsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s", "[PROJECT]", "[LOCATION]", "[DATASET]")
-	var request = &automlpb.ListTableSpecsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTableSpecs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlUpdateTableSpec(t *testing.T) {
-	var name string = "name3373707"
-	var timeColumnSpecId string = "timeColumnSpecId1558734824"
-	var rowCount int64 = 1340416618
-	var validRowCount int64 = 406068761
-	var columnCount int64 = 122671386
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.TableSpec{
-		Name:             name,
-		TimeColumnSpecId: timeColumnSpecId,
-		RowCount:         rowCount,
-		ValidRowCount:    validRowCount,
-		ColumnCount:      columnCount,
-		Etag:             etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var tableSpec *automlpb.TableSpec = &automlpb.TableSpec{}
-	var request = &automlpb.UpdateTableSpecRequest{
-		TableSpec: tableSpec,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateTableSpec(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlUpdateTableSpecError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var tableSpec *automlpb.TableSpec = &automlpb.TableSpec{}
-	var request = &automlpb.UpdateTableSpecRequest{
-		TableSpec: tableSpec,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateTableSpec(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlGetColumnSpec(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.ColumnSpec{
-		Name:        name2,
-		DisplayName: displayName,
-		Etag:        etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/tableSpecs/%s/columnSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[TABLE_SPEC]", "[COLUMN_SPEC]")
-	var request = &automlpb.GetColumnSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetColumnSpec(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlGetColumnSpecError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/tableSpecs/%s/columnSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[TABLE_SPEC]", "[COLUMN_SPEC]")
-	var request = &automlpb.GetColumnSpecRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetColumnSpec(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlListColumnSpecs(t *testing.T) {
-	var nextPageToken string = ""
-	var columnSpecsElement *automlpb.ColumnSpec = &automlpb.ColumnSpec{}
-	var columnSpecs = []*automlpb.ColumnSpec{columnSpecsElement}
-	var expectedResponse = &automlpb.ListColumnSpecsResponse{
-		NextPageToken: nextPageToken,
-		ColumnSpecs:   columnSpecs,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/tableSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[TABLE_SPEC]")
-	var request = &automlpb.ListColumnSpecsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListColumnSpecs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ColumnSpecs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlListColumnSpecsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/datasets/%s/tableSpecs/%s", "[PROJECT]", "[LOCATION]", "[DATASET]", "[TABLE_SPEC]")
-	var request = &automlpb.ListColumnSpecsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListColumnSpecs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAutoMlUpdateColumnSpec(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var etag string = "etag3123477"
-	var expectedResponse = &automlpb.ColumnSpec{
-		Name:        name,
-		DisplayName: displayName,
-		Etag:        etag,
-	}
-
-	mockAutoMl.err = nil
-	mockAutoMl.reqs = nil
-
-	mockAutoMl.resps = append(mockAutoMl.resps[:0], expectedResponse)
-
-	var columnSpec *automlpb.ColumnSpec = &automlpb.ColumnSpec{}
-	var request = &automlpb.UpdateColumnSpecRequest{
-		ColumnSpec: columnSpec,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateColumnSpec(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAutoMl.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAutoMlUpdateColumnSpecError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAutoMl.err = gstatus.Error(errCode, "test error")
-
-	var columnSpec *automlpb.ColumnSpec = &automlpb.ColumnSpec{}
-	var request = &automlpb.UpdateColumnSpecRequest{
-		ColumnSpec: columnSpec,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateColumnSpec(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPredictionServicePredict(t *testing.T) {
-	var expectedResponse *automlpb.PredictResponse = &automlpb.PredictResponse{}
-
-	mockPrediction.err = nil
-	mockPrediction.reqs = nil
-
-	mockPrediction.resps = append(mockPrediction.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var payload *automlpb.ExamplePayload = &automlpb.ExamplePayload{}
-	var request = &automlpb.PredictRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Predict(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPrediction.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPredictionServicePredictError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPrediction.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var payload *automlpb.ExamplePayload = &automlpb.ExamplePayload{}
-	var request = &automlpb.PredictRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Predict(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPredictionServiceBatchPredict(t *testing.T) {
-	var expectedResponse *automlpb.BatchPredictResult = &automlpb.BatchPredictResult{}
-
-	mockPrediction.err = nil
-	mockPrediction.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockPrediction.resps = append(mockPrediction.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var inputConfig *automlpb.BatchPredictInputConfig = &automlpb.BatchPredictInputConfig{}
-	var outputConfig *automlpb.BatchPredictOutputConfig = &automlpb.BatchPredictOutputConfig{}
-	var request = &automlpb.BatchPredictRequest{
-		Name:         formattedName,
-		InputConfig:  inputConfig,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchPredict(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPrediction.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPredictionServiceBatchPredictError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPrediction.err = nil
-	mockPrediction.resps = append(mockPrediction.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/models/%s", "[PROJECT]", "[LOCATION]", "[MODEL]")
-	var inputConfig *automlpb.BatchPredictInputConfig = &automlpb.BatchPredictInputConfig{}
-	var outputConfig *automlpb.BatchPredictOutputConfig = &automlpb.BatchPredictOutputConfig{}
-	var request = &automlpb.BatchPredictRequest{
-		Name:         formattedName,
-		InputConfig:  inputConfig,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewPredictionClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchPredict(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/bigquery/datatransfer/apiv1/mock_test.go b/bigquery/datatransfer/apiv1/mock_test.go
deleted file mode 100644
index b4d3f4aa22e..00000000000
--- a/bigquery/datatransfer/apiv1/mock_test.go
+++ /dev/null
@@ -1,1215 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package datatransfer
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	timestamppb "github.com/golang/protobuf/ptypes/timestamp"
-	"google.golang.org/api/option"
-	datatransferpb "google.golang.org/genproto/googleapis/cloud/bigquery/datatransfer/v1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockDataTransferServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	datatransferpb.DataTransferServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockDataTransferServer) GetDataSource(ctx context.Context, req *datatransferpb.GetDataSourceRequest) (*datatransferpb.DataSource, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.DataSource), nil
-}
-
-func (s *mockDataTransferServer) ListDataSources(ctx context.Context, req *datatransferpb.ListDataSourcesRequest) (*datatransferpb.ListDataSourcesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.ListDataSourcesResponse), nil
-}
-
-func (s *mockDataTransferServer) CreateTransferConfig(ctx context.Context, req *datatransferpb.CreateTransferConfigRequest) (*datatransferpb.TransferConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.TransferConfig), nil
-}
-
-func (s *mockDataTransferServer) UpdateTransferConfig(ctx context.Context, req *datatransferpb.UpdateTransferConfigRequest) (*datatransferpb.TransferConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.TransferConfig), nil
-}
-
-func (s *mockDataTransferServer) DeleteTransferConfig(ctx context.Context, req *datatransferpb.DeleteTransferConfigRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDataTransferServer) GetTransferConfig(ctx context.Context, req *datatransferpb.GetTransferConfigRequest) (*datatransferpb.TransferConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.TransferConfig), nil
-}
-
-func (s *mockDataTransferServer) ListTransferConfigs(ctx context.Context, req *datatransferpb.ListTransferConfigsRequest) (*datatransferpb.ListTransferConfigsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.ListTransferConfigsResponse), nil
-}
-
-func (s *mockDataTransferServer) ScheduleTransferRuns(ctx context.Context, req *datatransferpb.ScheduleTransferRunsRequest) (*datatransferpb.ScheduleTransferRunsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.ScheduleTransferRunsResponse), nil
-}
-
-func (s *mockDataTransferServer) StartManualTransferRuns(ctx context.Context, req *datatransferpb.StartManualTransferRunsRequest) (*datatransferpb.StartManualTransferRunsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.StartManualTransferRunsResponse), nil
-}
-
-func (s *mockDataTransferServer) GetTransferRun(ctx context.Context, req *datatransferpb.GetTransferRunRequest) (*datatransferpb.TransferRun, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.TransferRun), nil
-}
-
-func (s *mockDataTransferServer) DeleteTransferRun(ctx context.Context, req *datatransferpb.DeleteTransferRunRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDataTransferServer) ListTransferRuns(ctx context.Context, req *datatransferpb.ListTransferRunsRequest) (*datatransferpb.ListTransferRunsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.ListTransferRunsResponse), nil
-}
-
-func (s *mockDataTransferServer) ListTransferLogs(ctx context.Context, req *datatransferpb.ListTransferLogsRequest) (*datatransferpb.ListTransferLogsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.ListTransferLogsResponse), nil
-}
-
-func (s *mockDataTransferServer) CheckValidCreds(ctx context.Context, req *datatransferpb.CheckValidCredsRequest) (*datatransferpb.CheckValidCredsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*datatransferpb.CheckValidCredsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockDataTransfer mockDataTransferServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	datatransferpb.RegisterDataTransferServiceServer(serv, &mockDataTransfer)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestDataTransferServiceGetDataSource(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var dataSourceId string = "dataSourceId-1015796374"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var clientId string = "clientId-1904089585"
-	var supportsMultipleTransfers bool = true
-	var updateDeadlineSeconds int32 = 991471694
-	var defaultSchedule string = "defaultSchedule-800168235"
-	var supportsCustomSchedule bool = true
-	var helpUrl string = "helpUrl-789431439"
-	var defaultDataRefreshWindowDays int32 = 1804935157
-	var manualRunsDisabled bool = true
-	var expectedResponse = &datatransferpb.DataSource{
-		Name:                         name2,
-		DataSourceId:                 dataSourceId,
-		DisplayName:                  displayName,
-		Description:                  description,
-		ClientId:                     clientId,
-		SupportsMultipleTransfers:    supportsMultipleTransfers,
-		UpdateDeadlineSeconds:        updateDeadlineSeconds,
-		DefaultSchedule:              defaultSchedule,
-		SupportsCustomSchedule:       supportsCustomSchedule,
-		HelpUrl:                      helpUrl,
-		DefaultDataRefreshWindowDays: defaultDataRefreshWindowDays,
-		ManualRunsDisabled:           manualRunsDisabled,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/dataSources/%s", "[PROJECT]", "[DATA_SOURCE]")
-	var request = &datatransferpb.GetDataSourceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDataSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceGetDataSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/dataSources/%s", "[PROJECT]", "[DATA_SOURCE]")
-	var request = &datatransferpb.GetDataSourceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDataSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceListDataSources(t *testing.T) {
-	var nextPageToken string = ""
-	var dataSourcesElement *datatransferpb.DataSource = &datatransferpb.DataSource{}
-	var dataSources = []*datatransferpb.DataSource{dataSourcesElement}
-	var expectedResponse = &datatransferpb.ListDataSourcesResponse{
-		NextPageToken: nextPageToken,
-		DataSources:   dataSources,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &datatransferpb.ListDataSourcesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDataSources(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.DataSources[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceListDataSourcesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &datatransferpb.ListDataSourcesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDataSources(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceCreateTransferConfig(t *testing.T) {
-	var name string = "name3373707"
-	var destinationDatasetId string = "destinationDatasetId1541564179"
-	var displayName string = "displayName1615086568"
-	var dataSourceId string = "dataSourceId-1015796374"
-	var schedule string = "schedule-697920873"
-	var dataRefreshWindowDays int32 = 327632845
-	var disabled bool = true
-	var userId int64 = 147132913
-	var datasetRegion string = "datasetRegion959248539"
-	var expectedResponse = &datatransferpb.TransferConfig{
-		Name: name,
-		Destination: &datatransferpb.TransferConfig_DestinationDatasetId{
-			DestinationDatasetId: destinationDatasetId,
-		},
-		DisplayName:           displayName,
-		DataSourceId:          dataSourceId,
-		Schedule:              schedule,
-		DataRefreshWindowDays: dataRefreshWindowDays,
-		Disabled:              disabled,
-		UserId:                userId,
-		DatasetRegion:         datasetRegion,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var transferConfig *datatransferpb.TransferConfig = &datatransferpb.TransferConfig{}
-	var request = &datatransferpb.CreateTransferConfigRequest{
-		Parent:         formattedParent,
-		TransferConfig: transferConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTransferConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceCreateTransferConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var transferConfig *datatransferpb.TransferConfig = &datatransferpb.TransferConfig{}
-	var request = &datatransferpb.CreateTransferConfigRequest{
-		Parent:         formattedParent,
-		TransferConfig: transferConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTransferConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceUpdateTransferConfig(t *testing.T) {
-	var name string = "name3373707"
-	var destinationDatasetId string = "destinationDatasetId1541564179"
-	var displayName string = "displayName1615086568"
-	var dataSourceId string = "dataSourceId-1015796374"
-	var schedule string = "schedule-697920873"
-	var dataRefreshWindowDays int32 = 327632845
-	var disabled bool = true
-	var userId int64 = 147132913
-	var datasetRegion string = "datasetRegion959248539"
-	var expectedResponse = &datatransferpb.TransferConfig{
-		Name: name,
-		Destination: &datatransferpb.TransferConfig_DestinationDatasetId{
-			DestinationDatasetId: destinationDatasetId,
-		},
-		DisplayName:           displayName,
-		DataSourceId:          dataSourceId,
-		Schedule:              schedule,
-		DataRefreshWindowDays: dataRefreshWindowDays,
-		Disabled:              disabled,
-		UserId:                userId,
-		DatasetRegion:         datasetRegion,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var transferConfig *datatransferpb.TransferConfig = &datatransferpb.TransferConfig{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &datatransferpb.UpdateTransferConfigRequest{
-		TransferConfig: transferConfig,
-		UpdateMask:     updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateTransferConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceUpdateTransferConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var transferConfig *datatransferpb.TransferConfig = &datatransferpb.TransferConfig{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &datatransferpb.UpdateTransferConfigRequest{
-		TransferConfig: transferConfig,
-		UpdateMask:     updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateTransferConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceDeleteTransferConfig(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var request = &datatransferpb.DeleteTransferConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTransferConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDataTransferServiceDeleteTransferConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var request = &datatransferpb.DeleteTransferConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTransferConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDataTransferServiceGetTransferConfig(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var destinationDatasetId string = "destinationDatasetId1541564179"
-	var displayName string = "displayName1615086568"
-	var dataSourceId string = "dataSourceId-1015796374"
-	var schedule string = "schedule-697920873"
-	var dataRefreshWindowDays int32 = 327632845
-	var disabled bool = true
-	var userId int64 = 147132913
-	var datasetRegion string = "datasetRegion959248539"
-	var expectedResponse = &datatransferpb.TransferConfig{
-		Name: name2,
-		Destination: &datatransferpb.TransferConfig_DestinationDatasetId{
-			DestinationDatasetId: destinationDatasetId,
-		},
-		DisplayName:           displayName,
-		DataSourceId:          dataSourceId,
-		Schedule:              schedule,
-		DataRefreshWindowDays: dataRefreshWindowDays,
-		Disabled:              disabled,
-		UserId:                userId,
-		DatasetRegion:         datasetRegion,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var request = &datatransferpb.GetTransferConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTransferConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceGetTransferConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var request = &datatransferpb.GetTransferConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTransferConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceListTransferConfigs(t *testing.T) {
-	var nextPageToken string = ""
-	var transferConfigsElement *datatransferpb.TransferConfig = &datatransferpb.TransferConfig{}
-	var transferConfigs = []*datatransferpb.TransferConfig{transferConfigsElement}
-	var expectedResponse = &datatransferpb.ListTransferConfigsResponse{
-		NextPageToken:   nextPageToken,
-		TransferConfigs: transferConfigs,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &datatransferpb.ListTransferConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTransferConfigs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.TransferConfigs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceListTransferConfigsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &datatransferpb.ListTransferConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTransferConfigs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceScheduleTransferRuns(t *testing.T) {
-	var expectedResponse *datatransferpb.ScheduleTransferRunsResponse = &datatransferpb.ScheduleTransferRunsResponse{}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var endTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &datatransferpb.ScheduleTransferRunsRequest{
-		Parent:    formattedParent,
-		StartTime: startTime,
-		EndTime:   endTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ScheduleTransferRuns(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceScheduleTransferRunsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var endTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &datatransferpb.ScheduleTransferRunsRequest{
-		Parent:    formattedParent,
-		StartTime: startTime,
-		EndTime:   endTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ScheduleTransferRuns(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceGetTransferRun(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var destinationDatasetId string = "destinationDatasetId1541564179"
-	var dataSourceId string = "dataSourceId-1015796374"
-	var userId int64 = 147132913
-	var schedule string = "schedule-697920873"
-	var expectedResponse = &datatransferpb.TransferRun{
-		Name: name2,
-		Destination: &datatransferpb.TransferRun_DestinationDatasetId{
-			DestinationDatasetId: destinationDatasetId,
-		},
-		DataSourceId: dataSourceId,
-		UserId:       userId,
-		Schedule:     schedule,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s/runs/%s", "[PROJECT]", "[TRANSFER_CONFIG]", "[RUN]")
-	var request = &datatransferpb.GetTransferRunRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTransferRun(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceGetTransferRunError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s/runs/%s", "[PROJECT]", "[TRANSFER_CONFIG]", "[RUN]")
-	var request = &datatransferpb.GetTransferRunRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTransferRun(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceDeleteTransferRun(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s/runs/%s", "[PROJECT]", "[TRANSFER_CONFIG]", "[RUN]")
-	var request = &datatransferpb.DeleteTransferRunRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTransferRun(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDataTransferServiceDeleteTransferRunError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/transferConfigs/%s/runs/%s", "[PROJECT]", "[TRANSFER_CONFIG]", "[RUN]")
-	var request = &datatransferpb.DeleteTransferRunRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTransferRun(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDataTransferServiceListTransferRuns(t *testing.T) {
-	var nextPageToken string = ""
-	var transferRunsElement *datatransferpb.TransferRun = &datatransferpb.TransferRun{}
-	var transferRuns = []*datatransferpb.TransferRun{transferRunsElement}
-	var expectedResponse = &datatransferpb.ListTransferRunsResponse{
-		NextPageToken: nextPageToken,
-		TransferRuns:  transferRuns,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var request = &datatransferpb.ListTransferRunsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTransferRuns(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.TransferRuns[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceListTransferRunsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/transferConfigs/%s", "[PROJECT]", "[TRANSFER_CONFIG]")
-	var request = &datatransferpb.ListTransferRunsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTransferRuns(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceListTransferLogs(t *testing.T) {
-	var nextPageToken string = ""
-	var transferMessagesElement *datatransferpb.TransferMessage = &datatransferpb.TransferMessage{}
-	var transferMessages = []*datatransferpb.TransferMessage{transferMessagesElement}
-	var expectedResponse = &datatransferpb.ListTransferLogsResponse{
-		NextPageToken:    nextPageToken,
-		TransferMessages: transferMessages,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/transferConfigs/%s/runs/%s", "[PROJECT]", "[TRANSFER_CONFIG]", "[RUN]")
-	var request = &datatransferpb.ListTransferLogsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTransferLogs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.TransferMessages[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceListTransferLogsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/transferConfigs/%s/runs/%s", "[PROJECT]", "[TRANSFER_CONFIG]", "[RUN]")
-	var request = &datatransferpb.ListTransferLogsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTransferLogs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceCheckValidCreds(t *testing.T) {
-	var hasValidCreds bool = false
-	var expectedResponse = &datatransferpb.CheckValidCredsResponse{
-		HasValidCreds: hasValidCreds,
-	}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/dataSources/%s", "[PROJECT]", "[DATA_SOURCE]")
-	var request = &datatransferpb.CheckValidCredsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CheckValidCreds(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceCheckValidCredsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/dataSources/%s", "[PROJECT]", "[DATA_SOURCE]")
-	var request = &datatransferpb.CheckValidCredsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CheckValidCreds(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDataTransferServiceStartManualTransferRuns(t *testing.T) {
-	var expectedResponse *datatransferpb.StartManualTransferRunsResponse = &datatransferpb.StartManualTransferRunsResponse{}
-
-	mockDataTransfer.err = nil
-	mockDataTransfer.reqs = nil
-
-	mockDataTransfer.resps = append(mockDataTransfer.resps[:0], expectedResponse)
-
-	var request *datatransferpb.StartManualTransferRunsRequest = &datatransferpb.StartManualTransferRunsRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.StartManualTransferRuns(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDataTransfer.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDataTransferServiceStartManualTransferRunsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDataTransfer.err = gstatus.Error(errCode, "test error")
-
-	var request *datatransferpb.StartManualTransferRunsRequest = &datatransferpb.StartManualTransferRunsRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.StartManualTransferRuns(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/bigquery/storage/apiv1beta1/mock_test.go b/bigquery/storage/apiv1beta1/mock_test.go
deleted file mode 100644
index feb6fcc9c51..00000000000
--- a/bigquery/storage/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,454 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package storage
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	storagepb "google.golang.org/genproto/googleapis/cloud/bigquery/storage/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockBigQueryStorageServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	storagepb.BigQueryStorageServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockBigQueryStorageServer) CreateReadSession(ctx context.Context, req *storagepb.CreateReadSessionRequest) (*storagepb.ReadSession, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*storagepb.ReadSession), nil
-}
-
-func (s *mockBigQueryStorageServer) ReadRows(req *storagepb.ReadRowsRequest, stream storagepb.BigQueryStorage_ReadRowsServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*storagepb.ReadRowsResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockBigQueryStorageServer) BatchCreateReadSessionStreams(ctx context.Context, req *storagepb.BatchCreateReadSessionStreamsRequest) (*storagepb.BatchCreateReadSessionStreamsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*storagepb.BatchCreateReadSessionStreamsResponse), nil
-}
-
-func (s *mockBigQueryStorageServer) FinalizeStream(ctx context.Context, req *storagepb.FinalizeStreamRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockBigQueryStorageServer) SplitReadStream(ctx context.Context, req *storagepb.SplitReadStreamRequest) (*storagepb.SplitReadStreamResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*storagepb.SplitReadStreamResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockBigQueryStorage mockBigQueryStorageServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	storagepb.RegisterBigQueryStorageServer(serv, &mockBigQueryStorage)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestBigQueryStorageCreateReadSession(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &storagepb.ReadSession{
-		Name: name,
-	}
-
-	mockBigQueryStorage.err = nil
-	mockBigQueryStorage.reqs = nil
-
-	mockBigQueryStorage.resps = append(mockBigQueryStorage.resps[:0], expectedResponse)
-
-	var tableReference *storagepb.TableReference = &storagepb.TableReference{}
-	var parent string = "parent-995424086"
-	var request = &storagepb.CreateReadSessionRequest{
-		TableReference: tableReference,
-		Parent:         parent,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateReadSession(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockBigQueryStorage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestBigQueryStorageCreateReadSessionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockBigQueryStorage.err = gstatus.Error(errCode, "test error")
-
-	var tableReference *storagepb.TableReference = &storagepb.TableReference{}
-	var parent string = "parent-995424086"
-	var request = &storagepb.CreateReadSessionRequest{
-		TableReference: tableReference,
-		Parent:         parent,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateReadSession(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestBigQueryStorageReadRows(t *testing.T) {
-	var rowCount int64 = 1340416618
-	var expectedResponse = &storagepb.ReadRowsResponse{
-		RowCount: rowCount,
-	}
-
-	mockBigQueryStorage.err = nil
-	mockBigQueryStorage.reqs = nil
-
-	mockBigQueryStorage.resps = append(mockBigQueryStorage.resps[:0], expectedResponse)
-
-	var readPosition *storagepb.StreamPosition = &storagepb.StreamPosition{}
-	var request = &storagepb.ReadRowsRequest{
-		ReadPosition: readPosition,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.ReadRows(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockBigQueryStorage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestBigQueryStorageReadRowsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockBigQueryStorage.err = gstatus.Error(errCode, "test error")
-
-	var readPosition *storagepb.StreamPosition = &storagepb.StreamPosition{}
-	var request = &storagepb.ReadRowsRequest{
-		ReadPosition: readPosition,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.ReadRows(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestBigQueryStorageBatchCreateReadSessionStreams(t *testing.T) {
-	var expectedResponse *storagepb.BatchCreateReadSessionStreamsResponse = &storagepb.BatchCreateReadSessionStreamsResponse{}
-
-	mockBigQueryStorage.err = nil
-	mockBigQueryStorage.reqs = nil
-
-	mockBigQueryStorage.resps = append(mockBigQueryStorage.resps[:0], expectedResponse)
-
-	var session *storagepb.ReadSession = &storagepb.ReadSession{}
-	var requestedStreams int32 = 1017221410
-	var request = &storagepb.BatchCreateReadSessionStreamsRequest{
-		Session:          session,
-		RequestedStreams: requestedStreams,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateReadSessionStreams(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockBigQueryStorage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestBigQueryStorageBatchCreateReadSessionStreamsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockBigQueryStorage.err = gstatus.Error(errCode, "test error")
-
-	var session *storagepb.ReadSession = &storagepb.ReadSession{}
-	var requestedStreams int32 = 1017221410
-	var request = &storagepb.BatchCreateReadSessionStreamsRequest{
-		Session:          session,
-		RequestedStreams: requestedStreams,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateReadSessionStreams(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestBigQueryStorageFinalizeStream(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockBigQueryStorage.err = nil
-	mockBigQueryStorage.reqs = nil
-
-	mockBigQueryStorage.resps = append(mockBigQueryStorage.resps[:0], expectedResponse)
-
-	var stream *storagepb.Stream = &storagepb.Stream{}
-	var request = &storagepb.FinalizeStreamRequest{
-		Stream: stream,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.FinalizeStream(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockBigQueryStorage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestBigQueryStorageFinalizeStreamError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockBigQueryStorage.err = gstatus.Error(errCode, "test error")
-
-	var stream *storagepb.Stream = &storagepb.Stream{}
-	var request = &storagepb.FinalizeStreamRequest{
-		Stream: stream,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.FinalizeStream(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestBigQueryStorageSplitReadStream(t *testing.T) {
-	var expectedResponse *storagepb.SplitReadStreamResponse = &storagepb.SplitReadStreamResponse{}
-
-	mockBigQueryStorage.err = nil
-	mockBigQueryStorage.reqs = nil
-
-	mockBigQueryStorage.resps = append(mockBigQueryStorage.resps[:0], expectedResponse)
-
-	var originalStream *storagepb.Stream = &storagepb.Stream{}
-	var request = &storagepb.SplitReadStreamRequest{
-		OriginalStream: originalStream,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SplitReadStream(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockBigQueryStorage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestBigQueryStorageSplitReadStreamError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockBigQueryStorage.err = gstatus.Error(errCode, "test error")
-
-	var originalStream *storagepb.Stream = &storagepb.Stream{}
-	var request = &storagepb.SplitReadStreamRequest{
-		OriginalStream: originalStream,
-	}
-
-	c, err := NewBigQueryStorageClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SplitReadStream(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/cloudtasks/apiv2/mock_test.go b/cloudtasks/apiv2/mock_test.go
deleted file mode 100644
index 6a8c5f06b68..00000000000
--- a/cloudtasks/apiv2/mock_test.go
+++ /dev/null
@@ -1,1269 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package cloudtasks
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	taskspb "google.golang.org/genproto/googleapis/cloud/tasks/v2"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudTasksServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	taskspb.CloudTasksServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudTasksServer) ListQueues(ctx context.Context, req *taskspb.ListQueuesRequest) (*taskspb.ListQueuesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.ListQueuesResponse), nil
-}
-
-func (s *mockCloudTasksServer) GetQueue(ctx context.Context, req *taskspb.GetQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) CreateQueue(ctx context.Context, req *taskspb.CreateQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) UpdateQueue(ctx context.Context, req *taskspb.UpdateQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) DeleteQueue(ctx context.Context, req *taskspb.DeleteQueueRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) PurgeQueue(ctx context.Context, req *taskspb.PurgeQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) PauseQueue(ctx context.Context, req *taskspb.PauseQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) ResumeQueue(ctx context.Context, req *taskspb.ResumeQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockCloudTasksServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockCloudTasksServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockCloudTasksServer) ListTasks(ctx context.Context, req *taskspb.ListTasksRequest) (*taskspb.ListTasksResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.ListTasksResponse), nil
-}
-
-func (s *mockCloudTasksServer) GetTask(ctx context.Context, req *taskspb.GetTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) CreateTask(ctx context.Context, req *taskspb.CreateTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) DeleteTask(ctx context.Context, req *taskspb.DeleteTaskRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) RunTask(ctx context.Context, req *taskspb.RunTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudTasks mockCloudTasksServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	taskspb.RegisterCloudTasksServer(serv, &mockCloudTasks)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudTasksListQueues(t *testing.T) {
-	var nextPageToken string = ""
-	var queuesElement *taskspb.Queue = &taskspb.Queue{}
-	var queues = []*taskspb.Queue{queuesElement}
-	var expectedResponse = &taskspb.ListQueuesResponse{
-		NextPageToken: nextPageToken,
-		Queues:        queues,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &taskspb.ListQueuesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListQueues(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Queues[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksListQueuesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &taskspb.ListQueuesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListQueues(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.GetQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.GetQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCreateQueue(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Queue{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.CreateQueueRequest{
-		Parent: formattedParent,
-		Queue:  queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCreateQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.CreateQueueRequest{
-		Parent: formattedParent,
-		Queue:  queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksUpdateQueue(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Queue{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.UpdateQueueRequest{
-		Queue: queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksUpdateQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.UpdateQueueRequest{
-		Queue: queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksDeleteQueue(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.DeleteQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksDeleteQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.DeleteQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksPurgeQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PurgeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PurgeQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksPurgeQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PurgeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PurgeQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksPauseQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PauseQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksPauseQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PauseQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksResumeQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ResumeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksResumeQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ResumeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksListTasks(t *testing.T) {
-	var nextPageToken string = ""
-	var tasksElement *taskspb.Task = &taskspb.Task{}
-	var tasks = []*taskspb.Task{tasksElement}
-	var expectedResponse = &taskspb.ListTasksResponse{
-		NextPageToken: nextPageToken,
-		Tasks:         tasks,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ListTasksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTasks(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Tasks[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksListTasksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ListTasksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTasks(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetTask(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var dispatchCount int32 = 1217252086
-	var responseCount int32 = 424727441
-	var expectedResponse = &taskspb.Task{
-		Name:          name2,
-		DispatchCount: dispatchCount,
-		ResponseCount: responseCount,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.GetTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.GetTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCreateTask(t *testing.T) {
-	var name string = "name3373707"
-	var dispatchCount int32 = 1217252086
-	var responseCount int32 = 424727441
-	var expectedResponse = &taskspb.Task{
-		Name:          name,
-		DispatchCount: dispatchCount,
-		ResponseCount: responseCount,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var task *taskspb.Task = &taskspb.Task{}
-	var request = &taskspb.CreateTaskRequest{
-		Parent: formattedParent,
-		Task:   task,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCreateTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var task *taskspb.Task = &taskspb.Task{}
-	var request = &taskspb.CreateTaskRequest{
-		Parent: formattedParent,
-		Task:   task,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksDeleteTask(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.DeleteTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksDeleteTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.DeleteTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksRunTask(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var dispatchCount int32 = 1217252086
-	var responseCount int32 = 424727441
-	var expectedResponse = &taskspb.Task{
-		Name:          name2,
-		DispatchCount: dispatchCount,
-		ResponseCount: responseCount,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.RunTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksRunTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.RunTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/cloudtasks/apiv2beta2/mock_test.go b/cloudtasks/apiv2beta2/mock_test.go
deleted file mode 100644
index 306468fb8a1..00000000000
--- a/cloudtasks/apiv2beta2/mock_test.go
+++ /dev/null
@@ -1,1553 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package cloudtasks
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	durationpb "github.com/golang/protobuf/ptypes/duration"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	timestamppb "github.com/golang/protobuf/ptypes/timestamp"
-	"google.golang.org/api/option"
-	taskspb "google.golang.org/genproto/googleapis/cloud/tasks/v2beta2"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudTasksServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	taskspb.CloudTasksServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudTasksServer) ListQueues(ctx context.Context, req *taskspb.ListQueuesRequest) (*taskspb.ListQueuesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.ListQueuesResponse), nil
-}
-
-func (s *mockCloudTasksServer) GetQueue(ctx context.Context, req *taskspb.GetQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) CreateQueue(ctx context.Context, req *taskspb.CreateQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) UpdateQueue(ctx context.Context, req *taskspb.UpdateQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) DeleteQueue(ctx context.Context, req *taskspb.DeleteQueueRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) PurgeQueue(ctx context.Context, req *taskspb.PurgeQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) PauseQueue(ctx context.Context, req *taskspb.PauseQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) ResumeQueue(ctx context.Context, req *taskspb.ResumeQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockCloudTasksServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockCloudTasksServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockCloudTasksServer) ListTasks(ctx context.Context, req *taskspb.ListTasksRequest) (*taskspb.ListTasksResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.ListTasksResponse), nil
-}
-
-func (s *mockCloudTasksServer) GetTask(ctx context.Context, req *taskspb.GetTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) CreateTask(ctx context.Context, req *taskspb.CreateTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) DeleteTask(ctx context.Context, req *taskspb.DeleteTaskRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) LeaseTasks(ctx context.Context, req *taskspb.LeaseTasksRequest) (*taskspb.LeaseTasksResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.LeaseTasksResponse), nil
-}
-
-func (s *mockCloudTasksServer) AcknowledgeTask(ctx context.Context, req *taskspb.AcknowledgeTaskRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) RenewLease(ctx context.Context, req *taskspb.RenewLeaseRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) CancelLease(ctx context.Context, req *taskspb.CancelLeaseRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) RunTask(ctx context.Context, req *taskspb.RunTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudTasks mockCloudTasksServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	taskspb.RegisterCloudTasksServer(serv, &mockCloudTasks)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudTasksListQueues(t *testing.T) {
-	var nextPageToken string = ""
-	var queuesElement *taskspb.Queue = &taskspb.Queue{}
-	var queues = []*taskspb.Queue{queuesElement}
-	var expectedResponse = &taskspb.ListQueuesResponse{
-		NextPageToken: nextPageToken,
-		Queues:        queues,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &taskspb.ListQueuesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListQueues(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Queues[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksListQueuesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &taskspb.ListQueuesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListQueues(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.GetQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.GetQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCreateQueue(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Queue{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.CreateQueueRequest{
-		Parent: formattedParent,
-		Queue:  queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCreateQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.CreateQueueRequest{
-		Parent: formattedParent,
-		Queue:  queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksUpdateQueue(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Queue{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.UpdateQueueRequest{
-		Queue: queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksUpdateQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.UpdateQueueRequest{
-		Queue: queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksDeleteQueue(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.DeleteQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksDeleteQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.DeleteQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksPurgeQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PurgeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PurgeQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksPurgeQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PurgeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PurgeQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksPauseQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PauseQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksPauseQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PauseQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksResumeQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ResumeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksResumeQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ResumeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksListTasks(t *testing.T) {
-	var nextPageToken string = ""
-	var tasksElement *taskspb.Task = &taskspb.Task{}
-	var tasks = []*taskspb.Task{tasksElement}
-	var expectedResponse = &taskspb.ListTasksResponse{
-		NextPageToken: nextPageToken,
-		Tasks:         tasks,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ListTasksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTasks(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Tasks[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksListTasksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ListTasksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTasks(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetTask(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Task{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.GetTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.GetTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCreateTask(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Task{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var task *taskspb.Task = &taskspb.Task{}
-	var request = &taskspb.CreateTaskRequest{
-		Parent: formattedParent,
-		Task:   task,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCreateTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var task *taskspb.Task = &taskspb.Task{}
-	var request = &taskspb.CreateTaskRequest{
-		Parent: formattedParent,
-		Task:   task,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksDeleteTask(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.DeleteTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksDeleteTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.DeleteTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksLeaseTasks(t *testing.T) {
-	var expectedResponse *taskspb.LeaseTasksResponse = &taskspb.LeaseTasksResponse{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var leaseDuration *durationpb.Duration = &durationpb.Duration{}
-	var request = &taskspb.LeaseTasksRequest{
-		Parent:        formattedParent,
-		LeaseDuration: leaseDuration,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.LeaseTasks(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksLeaseTasksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var leaseDuration *durationpb.Duration = &durationpb.Duration{}
-	var request = &taskspb.LeaseTasksRequest{
-		Parent:        formattedParent,
-		LeaseDuration: leaseDuration,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.LeaseTasks(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksAcknowledgeTask(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &taskspb.AcknowledgeTaskRequest{
-		Name:         formattedName,
-		ScheduleTime: scheduleTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.AcknowledgeTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksAcknowledgeTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &taskspb.AcknowledgeTaskRequest{
-		Name:         formattedName,
-		ScheduleTime: scheduleTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.AcknowledgeTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksRenewLease(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Task{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var leaseDuration *durationpb.Duration = &durationpb.Duration{}
-	var request = &taskspb.RenewLeaseRequest{
-		Name:          formattedName,
-		ScheduleTime:  scheduleTime,
-		LeaseDuration: leaseDuration,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RenewLease(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksRenewLeaseError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var leaseDuration *durationpb.Duration = &durationpb.Duration{}
-	var request = &taskspb.RenewLeaseRequest{
-		Name:          formattedName,
-		ScheduleTime:  scheduleTime,
-		LeaseDuration: leaseDuration,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RenewLease(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCancelLease(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Task{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &taskspb.CancelLeaseRequest{
-		Name:         formattedName,
-		ScheduleTime: scheduleTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CancelLease(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCancelLeaseError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var scheduleTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &taskspb.CancelLeaseRequest{
-		Name:         formattedName,
-		ScheduleTime: scheduleTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CancelLease(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksRunTask(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Task{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.RunTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksRunTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.RunTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/cloudtasks/apiv2beta3/mock_test.go b/cloudtasks/apiv2beta3/mock_test.go
deleted file mode 100644
index 3e1eb864347..00000000000
--- a/cloudtasks/apiv2beta3/mock_test.go
+++ /dev/null
@@ -1,1269 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package cloudtasks
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	taskspb "google.golang.org/genproto/googleapis/cloud/tasks/v2beta3"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudTasksServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	taskspb.CloudTasksServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudTasksServer) ListQueues(ctx context.Context, req *taskspb.ListQueuesRequest) (*taskspb.ListQueuesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.ListQueuesResponse), nil
-}
-
-func (s *mockCloudTasksServer) GetQueue(ctx context.Context, req *taskspb.GetQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) CreateQueue(ctx context.Context, req *taskspb.CreateQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) UpdateQueue(ctx context.Context, req *taskspb.UpdateQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) DeleteQueue(ctx context.Context, req *taskspb.DeleteQueueRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) PurgeQueue(ctx context.Context, req *taskspb.PurgeQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) PauseQueue(ctx context.Context, req *taskspb.PauseQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) ResumeQueue(ctx context.Context, req *taskspb.ResumeQueueRequest) (*taskspb.Queue, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Queue), nil
-}
-
-func (s *mockCloudTasksServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockCloudTasksServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockCloudTasksServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockCloudTasksServer) ListTasks(ctx context.Context, req *taskspb.ListTasksRequest) (*taskspb.ListTasksResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.ListTasksResponse), nil
-}
-
-func (s *mockCloudTasksServer) GetTask(ctx context.Context, req *taskspb.GetTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) CreateTask(ctx context.Context, req *taskspb.CreateTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-func (s *mockCloudTasksServer) DeleteTask(ctx context.Context, req *taskspb.DeleteTaskRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudTasksServer) RunTask(ctx context.Context, req *taskspb.RunTaskRequest) (*taskspb.Task, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*taskspb.Task), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudTasks mockCloudTasksServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	taskspb.RegisterCloudTasksServer(serv, &mockCloudTasks)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudTasksListQueues(t *testing.T) {
-	var nextPageToken string = ""
-	var queuesElement *taskspb.Queue = &taskspb.Queue{}
-	var queues = []*taskspb.Queue{queuesElement}
-	var expectedResponse = &taskspb.ListQueuesResponse{
-		NextPageToken: nextPageToken,
-		Queues:        queues,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &taskspb.ListQueuesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListQueues(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Queues[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksListQueuesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &taskspb.ListQueuesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListQueues(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.GetQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.GetQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCreateQueue(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Queue{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.CreateQueueRequest{
-		Parent: formattedParent,
-		Queue:  queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCreateQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.CreateQueueRequest{
-		Parent: formattedParent,
-		Queue:  queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksUpdateQueue(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &taskspb.Queue{
-		Name: name,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.UpdateQueueRequest{
-		Queue: queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksUpdateQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var queue *taskspb.Queue = &taskspb.Queue{}
-	var request = &taskspb.UpdateQueueRequest{
-		Queue: queue,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksDeleteQueue(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.DeleteQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksDeleteQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.DeleteQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksPurgeQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PurgeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PurgeQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksPurgeQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PurgeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PurgeQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksPauseQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PauseQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksPauseQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.PauseQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksResumeQueue(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &taskspb.Queue{
-		Name: name2,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ResumeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeQueue(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksResumeQueueError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ResumeQueueRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeQueue(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksListTasks(t *testing.T) {
-	var nextPageToken string = ""
-	var tasksElement *taskspb.Task = &taskspb.Task{}
-	var tasks = []*taskspb.Task{tasksElement}
-	var expectedResponse = &taskspb.ListTasksResponse{
-		NextPageToken: nextPageToken,
-		Tasks:         tasks,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ListTasksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTasks(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Tasks[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksListTasksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var request = &taskspb.ListTasksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTasks(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksGetTask(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var dispatchCount int32 = 1217252086
-	var responseCount int32 = 424727441
-	var expectedResponse = &taskspb.Task{
-		Name:          name2,
-		DispatchCount: dispatchCount,
-		ResponseCount: responseCount,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.GetTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksGetTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.GetTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksCreateTask(t *testing.T) {
-	var name string = "name3373707"
-	var dispatchCount int32 = 1217252086
-	var responseCount int32 = 424727441
-	var expectedResponse = &taskspb.Task{
-		Name:          name,
-		DispatchCount: dispatchCount,
-		ResponseCount: responseCount,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var task *taskspb.Task = &taskspb.Task{}
-	var request = &taskspb.CreateTaskRequest{
-		Parent: formattedParent,
-		Task:   task,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksCreateTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/queues/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]")
-	var task *taskspb.Task = &taskspb.Task{}
-	var request = &taskspb.CreateTaskRequest{
-		Parent: formattedParent,
-		Task:   task,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudTasksDeleteTask(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.DeleteTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudTasksDeleteTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.DeleteTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudTasksRunTask(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var dispatchCount int32 = 1217252086
-	var responseCount int32 = 424727441
-	var expectedResponse = &taskspb.Task{
-		Name:          name2,
-		DispatchCount: dispatchCount,
-		ResponseCount: responseCount,
-	}
-
-	mockCloudTasks.err = nil
-	mockCloudTasks.reqs = nil
-
-	mockCloudTasks.resps = append(mockCloudTasks.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.RunTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunTask(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudTasks.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudTasksRunTaskError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudTasks.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/queues/%s/tasks/%s", "[PROJECT]", "[LOCATION]", "[QUEUE]", "[TASK]")
-	var request = &taskspb.RunTaskRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunTask(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/container/apiv1/mock_test.go b/container/apiv1/mock_test.go
deleted file mode 100644
index 7e55e1a07a0..00000000000
--- a/container/apiv1/mock_test.go
+++ /dev/null
@@ -1,2911 +0,0 @@
-// Copyright 2018 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// AUTO-GENERATED CODE. DO NOT EDIT.
-
-package container
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	containerpb "google.golang.org/genproto/googleapis/container/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockClusterManagerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	containerpb.ClusterManagerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockClusterManagerServer) ListClusters(ctx context.Context, req *containerpb.ListClustersRequest) (*containerpb.ListClustersResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.ListClustersResponse), nil
-}
-
-func (s *mockClusterManagerServer) GetCluster(ctx context.Context, req *containerpb.GetClusterRequest) (*containerpb.Cluster, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Cluster), nil
-}
-
-func (s *mockClusterManagerServer) CreateCluster(ctx context.Context, req *containerpb.CreateClusterRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) UpdateCluster(ctx context.Context, req *containerpb.UpdateClusterRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) UpdateNodePool(ctx context.Context, req *containerpb.UpdateNodePoolRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetNodePoolAutoscaling(ctx context.Context, req *containerpb.SetNodePoolAutoscalingRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetLoggingService(ctx context.Context, req *containerpb.SetLoggingServiceRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetMonitoringService(ctx context.Context, req *containerpb.SetMonitoringServiceRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetAddonsConfig(ctx context.Context, req *containerpb.SetAddonsConfigRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetLocations(ctx context.Context, req *containerpb.SetLocationsRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) UpdateMaster(ctx context.Context, req *containerpb.UpdateMasterRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetMasterAuth(ctx context.Context, req *containerpb.SetMasterAuthRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) DeleteCluster(ctx context.Context, req *containerpb.DeleteClusterRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) ListOperations(ctx context.Context, req *containerpb.ListOperationsRequest) (*containerpb.ListOperationsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.ListOperationsResponse), nil
-}
-
-func (s *mockClusterManagerServer) GetOperation(ctx context.Context, req *containerpb.GetOperationRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) CancelOperation(ctx context.Context, req *containerpb.CancelOperationRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockClusterManagerServer) GetServerConfig(ctx context.Context, req *containerpb.GetServerConfigRequest) (*containerpb.ServerConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.ServerConfig), nil
-}
-
-func (s *mockClusterManagerServer) ListNodePools(ctx context.Context, req *containerpb.ListNodePoolsRequest) (*containerpb.ListNodePoolsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.ListNodePoolsResponse), nil
-}
-
-func (s *mockClusterManagerServer) GetNodePool(ctx context.Context, req *containerpb.GetNodePoolRequest) (*containerpb.NodePool, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.NodePool), nil
-}
-
-func (s *mockClusterManagerServer) CreateNodePool(ctx context.Context, req *containerpb.CreateNodePoolRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) DeleteNodePool(ctx context.Context, req *containerpb.DeleteNodePoolRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) RollbackNodePoolUpgrade(ctx context.Context, req *containerpb.RollbackNodePoolUpgradeRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetNodePoolManagement(ctx context.Context, req *containerpb.SetNodePoolManagementRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetLabels(ctx context.Context, req *containerpb.SetLabelsRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetLegacyAbac(ctx context.Context, req *containerpb.SetLegacyAbacRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) StartIPRotation(ctx context.Context, req *containerpb.StartIPRotationRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) CompleteIPRotation(ctx context.Context, req *containerpb.CompleteIPRotationRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetNodePoolSize(ctx context.Context, req *containerpb.SetNodePoolSizeRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetNetworkPolicy(ctx context.Context, req *containerpb.SetNetworkPolicyRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-func (s *mockClusterManagerServer) SetMaintenancePolicy(ctx context.Context, req *containerpb.SetMaintenancePolicyRequest) (*containerpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containerpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockClusterManager mockClusterManagerServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	containerpb.RegisterClusterManagerServer(serv, &mockClusterManager)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestClusterManagerListClusters(t *testing.T) {
-	var expectedResponse *containerpb.ListClustersResponse = &containerpb.ListClustersResponse{}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var request = &containerpb.ListClustersRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListClusters(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerListClustersError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var request = &containerpb.ListClustersRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListClusters(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerGetCluster(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var initialNodeCount int32 = 1682564205
-	var loggingService string = "loggingService-1700501035"
-	var monitoringService string = "monitoringService1469270462"
-	var network string = "network1843485230"
-	var clusterIpv4Cidr string = "clusterIpv4Cidr-141875831"
-	var subnetwork string = "subnetwork-1302785042"
-	var enableKubernetesAlpha bool = false
-	var labelFingerprint string = "labelFingerprint714995737"
-	var selfLink string = "selfLink-1691268851"
-	var zone2 string = "zone2-696322977"
-	var endpoint string = "endpoint1741102485"
-	var initialClusterVersion string = "initialClusterVersion-276373352"
-	var currentMasterVersion string = "currentMasterVersion-920953983"
-	var currentNodeVersion string = "currentNodeVersion-407476063"
-	var createTime string = "createTime-493574096"
-	var statusMessage string = "statusMessage-239442758"
-	var nodeIpv4CidrSize int32 = 1181176815
-	var servicesIpv4Cidr string = "servicesIpv4Cidr1966438125"
-	var currentNodeCount int32 = 178977560
-	var expireTime string = "expireTime-96179731"
-	var expectedResponse = &containerpb.Cluster{
-		Name:                  name,
-		Description:           description,
-		InitialNodeCount:      initialNodeCount,
-		LoggingService:        loggingService,
-		MonitoringService:     monitoringService,
-		Network:               network,
-		ClusterIpv4Cidr:       clusterIpv4Cidr,
-		Subnetwork:            subnetwork,
-		EnableKubernetesAlpha: enableKubernetesAlpha,
-		LabelFingerprint:      labelFingerprint,
-		SelfLink:              selfLink,
-		Zone:                  zone2,
-		Endpoint:              endpoint,
-		InitialClusterVersion: initialClusterVersion,
-		CurrentMasterVersion:  currentMasterVersion,
-		CurrentNodeVersion:    currentNodeVersion,
-		CreateTime:            createTime,
-		StatusMessage:         statusMessage,
-		NodeIpv4CidrSize:      nodeIpv4CidrSize,
-		ServicesIpv4Cidr:      servicesIpv4Cidr,
-		CurrentNodeCount:      currentNodeCount,
-		ExpireTime:            expireTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.GetClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCluster(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerGetClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.GetClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCluster(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerCreateCluster(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var cluster *containerpb.Cluster = &containerpb.Cluster{}
-	var request = &containerpb.CreateClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		Cluster:   cluster,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateCluster(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerCreateClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var cluster *containerpb.Cluster = &containerpb.Cluster{}
-	var request = &containerpb.CreateClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		Cluster:   cluster,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateCluster(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerUpdateCluster(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var update *containerpb.ClusterUpdate = &containerpb.ClusterUpdate{}
-	var request = &containerpb.UpdateClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Update:    update,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCluster(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerUpdateClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var update *containerpb.ClusterUpdate = &containerpb.ClusterUpdate{}
-	var request = &containerpb.UpdateClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Update:    update,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCluster(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerUpdateNodePool(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var nodeVersion string = "nodeVersion1790136219"
-	var imageType string = "imageType-1442758754"
-	var request = &containerpb.UpdateNodePoolRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		ClusterId:   clusterId,
-		NodePoolId:  nodePoolId,
-		NodeVersion: nodeVersion,
-		ImageType:   imageType,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNodePool(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerUpdateNodePoolError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var nodeVersion string = "nodeVersion1790136219"
-	var imageType string = "imageType-1442758754"
-	var request = &containerpb.UpdateNodePoolRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		ClusterId:   clusterId,
-		NodePoolId:  nodePoolId,
-		NodeVersion: nodeVersion,
-		ImageType:   imageType,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNodePool(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetNodePoolAutoscaling(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var autoscaling *containerpb.NodePoolAutoscaling = &containerpb.NodePoolAutoscaling{}
-	var request = &containerpb.SetNodePoolAutoscalingRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		ClusterId:   clusterId,
-		NodePoolId:  nodePoolId,
-		Autoscaling: autoscaling,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNodePoolAutoscaling(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetNodePoolAutoscalingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var autoscaling *containerpb.NodePoolAutoscaling = &containerpb.NodePoolAutoscaling{}
-	var request = &containerpb.SetNodePoolAutoscalingRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		ClusterId:   clusterId,
-		NodePoolId:  nodePoolId,
-		Autoscaling: autoscaling,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNodePoolAutoscaling(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetLoggingService(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var loggingService string = "loggingService-1700501035"
-	var request = &containerpb.SetLoggingServiceRequest{
-		ProjectId:      projectId,
-		Zone:           zone,
-		ClusterId:      clusterId,
-		LoggingService: loggingService,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLoggingService(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetLoggingServiceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var loggingService string = "loggingService-1700501035"
-	var request = &containerpb.SetLoggingServiceRequest{
-		ProjectId:      projectId,
-		Zone:           zone,
-		ClusterId:      clusterId,
-		LoggingService: loggingService,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLoggingService(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetMonitoringService(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var monitoringService string = "monitoringService1469270462"
-	var request = &containerpb.SetMonitoringServiceRequest{
-		ProjectId:         projectId,
-		Zone:              zone,
-		ClusterId:         clusterId,
-		MonitoringService: monitoringService,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetMonitoringService(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetMonitoringServiceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var monitoringService string = "monitoringService1469270462"
-	var request = &containerpb.SetMonitoringServiceRequest{
-		ProjectId:         projectId,
-		Zone:              zone,
-		ClusterId:         clusterId,
-		MonitoringService: monitoringService,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetMonitoringService(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetAddonsConfig(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var addonsConfig *containerpb.AddonsConfig = &containerpb.AddonsConfig{}
-	var request = &containerpb.SetAddonsConfigRequest{
-		ProjectId:    projectId,
-		Zone:         zone,
-		ClusterId:    clusterId,
-		AddonsConfig: addonsConfig,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetAddonsConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetAddonsConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var addonsConfig *containerpb.AddonsConfig = &containerpb.AddonsConfig{}
-	var request = &containerpb.SetAddonsConfigRequest{
-		ProjectId:    projectId,
-		Zone:         zone,
-		ClusterId:    clusterId,
-		AddonsConfig: addonsConfig,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetAddonsConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetLocations(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var locations []string = nil
-	var request = &containerpb.SetLocationsRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Locations: locations,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLocations(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetLocationsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var locations []string = nil
-	var request = &containerpb.SetLocationsRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Locations: locations,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLocations(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerUpdateMaster(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var masterVersion string = "masterVersion-2139460613"
-	var request = &containerpb.UpdateMasterRequest{
-		ProjectId:     projectId,
-		Zone:          zone,
-		ClusterId:     clusterId,
-		MasterVersion: masterVersion,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateMaster(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerUpdateMasterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var masterVersion string = "masterVersion-2139460613"
-	var request = &containerpb.UpdateMasterRequest{
-		ProjectId:     projectId,
-		Zone:          zone,
-		ClusterId:     clusterId,
-		MasterVersion: masterVersion,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateMaster(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetMasterAuth(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var action containerpb.SetMasterAuthRequest_Action = containerpb.SetMasterAuthRequest_UNKNOWN
-	var update *containerpb.MasterAuth = &containerpb.MasterAuth{}
-	var request = &containerpb.SetMasterAuthRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Action:    action,
-		Update:    update,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetMasterAuth(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetMasterAuthError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var action containerpb.SetMasterAuthRequest_Action = containerpb.SetMasterAuthRequest_UNKNOWN
-	var update *containerpb.MasterAuth = &containerpb.MasterAuth{}
-	var request = &containerpb.SetMasterAuthRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Action:    action,
-		Update:    update,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetMasterAuth(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerDeleteCluster(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.DeleteClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteCluster(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerDeleteClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.DeleteClusterRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteCluster(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerListOperations(t *testing.T) {
-	var expectedResponse *containerpb.ListOperationsResponse = &containerpb.ListOperationsResponse{}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var request = &containerpb.ListOperationsRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListOperations(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerListOperationsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var request = &containerpb.ListOperationsRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListOperations(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerGetOperation(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var operationId string = "operationId-274116877"
-	var request = &containerpb.GetOperationRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		OperationId: operationId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOperation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerGetOperationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var operationId string = "operationId-274116877"
-	var request = &containerpb.GetOperationRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		OperationId: operationId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOperation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerCancelOperation(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var operationId string = "operationId-274116877"
-	var request = &containerpb.CancelOperationRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		OperationId: operationId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.CancelOperation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestClusterManagerCancelOperationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var operationId string = "operationId-274116877"
-	var request = &containerpb.CancelOperationRequest{
-		ProjectId:   projectId,
-		Zone:        zone,
-		OperationId: operationId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.CancelOperation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestClusterManagerGetServerConfig(t *testing.T) {
-	var defaultClusterVersion string = "defaultClusterVersion111003029"
-	var defaultImageType string = "defaultImageType-918225828"
-	var expectedResponse = &containerpb.ServerConfig{
-		DefaultClusterVersion: defaultClusterVersion,
-		DefaultImageType:      defaultImageType,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var request = &containerpb.GetServerConfigRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServerConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerGetServerConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var request = &containerpb.GetServerConfigRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServerConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerListNodePools(t *testing.T) {
-	var expectedResponse *containerpb.ListNodePoolsResponse = &containerpb.ListNodePoolsResponse{}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.ListNodePoolsRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNodePools(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerListNodePoolsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.ListNodePoolsRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNodePools(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerGetNodePool(t *testing.T) {
-	var name string = "name3373707"
-	var initialNodeCount int32 = 1682564205
-	var selfLink string = "selfLink-1691268851"
-	var version string = "version351608024"
-	var statusMessage string = "statusMessage-239442758"
-	var expectedResponse = &containerpb.NodePool{
-		Name:             name,
-		InitialNodeCount: initialNodeCount,
-		SelfLink:         selfLink,
-		Version:          version,
-		StatusMessage:    statusMessage,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var request = &containerpb.GetNodePoolRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNodePool(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerGetNodePoolError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var request = &containerpb.GetNodePoolRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNodePool(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerCreateNodePool(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePool *containerpb.NodePool = &containerpb.NodePool{}
-	var request = &containerpb.CreateNodePoolRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		NodePool:  nodePool,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNodePool(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerCreateNodePoolError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePool *containerpb.NodePool = &containerpb.NodePool{}
-	var request = &containerpb.CreateNodePoolRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		NodePool:  nodePool,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNodePool(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerDeleteNodePool(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var request = &containerpb.DeleteNodePoolRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteNodePool(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerDeleteNodePoolError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var request = &containerpb.DeleteNodePoolRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteNodePool(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerRollbackNodePoolUpgrade(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var request = &containerpb.RollbackNodePoolUpgradeRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RollbackNodePoolUpgrade(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerRollbackNodePoolUpgradeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var request = &containerpb.RollbackNodePoolUpgradeRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RollbackNodePoolUpgrade(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetNodePoolManagement(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var management *containerpb.NodeManagement = &containerpb.NodeManagement{}
-	var request = &containerpb.SetNodePoolManagementRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-		Management: management,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNodePoolManagement(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetNodePoolManagementError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var management *containerpb.NodeManagement = &containerpb.NodeManagement{}
-	var request = &containerpb.SetNodePoolManagementRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-		Management: management,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNodePoolManagement(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetLabels(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var resourceLabels map[string]string = nil
-	var labelFingerprint string = "labelFingerprint714995737"
-	var request = &containerpb.SetLabelsRequest{
-		ProjectId:        projectId,
-		Zone:             zone,
-		ClusterId:        clusterId,
-		ResourceLabels:   resourceLabels,
-		LabelFingerprint: labelFingerprint,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLabels(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetLabelsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var resourceLabels map[string]string = nil
-	var labelFingerprint string = "labelFingerprint714995737"
-	var request = &containerpb.SetLabelsRequest{
-		ProjectId:        projectId,
-		Zone:             zone,
-		ClusterId:        clusterId,
-		ResourceLabels:   resourceLabels,
-		LabelFingerprint: labelFingerprint,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLabels(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetLegacyAbac(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var enabled bool = false
-	var request = &containerpb.SetLegacyAbacRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Enabled:   enabled,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLegacyAbac(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetLegacyAbacError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var enabled bool = false
-	var request = &containerpb.SetLegacyAbacRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-		Enabled:   enabled,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetLegacyAbac(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerStartIPRotation(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.StartIPRotationRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.StartIPRotation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerStartIPRotationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.StartIPRotationRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.StartIPRotation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerCompleteIPRotation(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.CompleteIPRotationRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CompleteIPRotation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerCompleteIPRotationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var request = &containerpb.CompleteIPRotationRequest{
-		ProjectId: projectId,
-		Zone:      zone,
-		ClusterId: clusterId,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CompleteIPRotation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetNodePoolSize(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var nodeCount int32 = 1539922066
-	var request = &containerpb.SetNodePoolSizeRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-		NodeCount:  nodeCount,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNodePoolSize(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetNodePoolSizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var nodePoolId string = "nodePoolId1043384033"
-	var nodeCount int32 = 1539922066
-	var request = &containerpb.SetNodePoolSizeRequest{
-		ProjectId:  projectId,
-		Zone:       zone,
-		ClusterId:  clusterId,
-		NodePoolId: nodePoolId,
-		NodeCount:  nodeCount,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNodePoolSize(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetNetworkPolicy(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var networkPolicy *containerpb.NetworkPolicy = &containerpb.NetworkPolicy{}
-	var request = &containerpb.SetNetworkPolicyRequest{
-		ProjectId:     projectId,
-		Zone:          zone,
-		ClusterId:     clusterId,
-		NetworkPolicy: networkPolicy,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNetworkPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetNetworkPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var networkPolicy *containerpb.NetworkPolicy = &containerpb.NetworkPolicy{}
-	var request = &containerpb.SetNetworkPolicyRequest{
-		ProjectId:     projectId,
-		Zone:          zone,
-		ClusterId:     clusterId,
-		NetworkPolicy: networkPolicy,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetNetworkPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterManagerSetMaintenancePolicy(t *testing.T) {
-	var name string = "name3373707"
-	var zone2 string = "zone2-696322977"
-	var detail string = "detail-1335224239"
-	var statusMessage string = "statusMessage-239442758"
-	var selfLink string = "selfLink-1691268851"
-	var targetLink string = "targetLink-2084812312"
-	var startTime string = "startTime-1573145462"
-	var endTime string = "endTime1725551537"
-	var expectedResponse = &containerpb.Operation{
-		Name:          name,
-		Zone:          zone2,
-		Detail:        detail,
-		StatusMessage: statusMessage,
-		SelfLink:      selfLink,
-		TargetLink:    targetLink,
-		StartTime:     startTime,
-		EndTime:       endTime,
-	}
-
-	mockClusterManager.err = nil
-	mockClusterManager.reqs = nil
-
-	mockClusterManager.resps = append(mockClusterManager.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var maintenancePolicy *containerpb.MaintenancePolicy = &containerpb.MaintenancePolicy{}
-	var request = &containerpb.SetMaintenancePolicyRequest{
-		ProjectId:         projectId,
-		Zone:              zone,
-		ClusterId:         clusterId,
-		MaintenancePolicy: maintenancePolicy,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetMaintenancePolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterManagerSetMaintenancePolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterManager.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var zone string = "zone3744684"
-	var clusterId string = "clusterId240280960"
-	var maintenancePolicy *containerpb.MaintenancePolicy = &containerpb.MaintenancePolicy{}
-	var request = &containerpb.SetMaintenancePolicyRequest{
-		ProjectId:         projectId,
-		Zone:              zone,
-		ClusterId:         clusterId,
-		MaintenancePolicy: maintenancePolicy,
-	}
-
-	c, err := NewClusterManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetMaintenancePolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/containeranalysis/apiv1/mock_test.go b/containeranalysis/apiv1/mock_test.go
deleted file mode 100644
index 9fd2eba984b..00000000000
--- a/containeranalysis/apiv1/mock_test.go
+++ /dev/null
@@ -1,311 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package containeranalysis
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	containeranalysispb "google.golang.org/genproto/googleapis/devtools/containeranalysis/v1"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockContainerAnalysisServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	containeranalysispb.ContainerAnalysisServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockContainerAnalysisServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockContainerAnalysisServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockContainerAnalysisServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockContainerAnalysis mockContainerAnalysisServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	containeranalysispb.RegisterContainerAnalysisServer(serv, &mockContainerAnalysis)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestContainerAnalysisSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockContainerAnalysis.err = nil
-	mockContainerAnalysis.reqs = nil
-
-	mockContainerAnalysis.resps = append(mockContainerAnalysis.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysis.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockContainerAnalysis.err = nil
-	mockContainerAnalysis.reqs = nil
-
-	mockContainerAnalysis.resps = append(mockContainerAnalysis.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysis.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockContainerAnalysis.err = nil
-	mockContainerAnalysis.reqs = nil
-
-	mockContainerAnalysis.resps = append(mockContainerAnalysis.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysis.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/containeranalysis/apiv1beta1/mock_test.go b/containeranalysis/apiv1beta1/mock_test.go
deleted file mode 100644
index 8a71e8f2a4a..00000000000
--- a/containeranalysis/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,1707 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package containeranalysis
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	containeranalysispb "google.golang.org/genproto/googleapis/devtools/containeranalysis/v1beta1"
-	grafeaspb "google.golang.org/genproto/googleapis/devtools/containeranalysis/v1beta1/grafeas"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockContainerAnalysisV1Beta1Server struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	containeranalysispb.ContainerAnalysisV1Beta1Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockContainerAnalysisV1Beta1Server) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockContainerAnalysisV1Beta1Server) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockContainerAnalysisV1Beta1Server) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockContainerAnalysisV1Beta1Server) GetScanConfig(ctx context.Context, req *containeranalysispb.GetScanConfigRequest) (*containeranalysispb.ScanConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containeranalysispb.ScanConfig), nil
-}
-
-func (s *mockContainerAnalysisV1Beta1Server) ListScanConfigs(ctx context.Context, req *containeranalysispb.ListScanConfigsRequest) (*containeranalysispb.ListScanConfigsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containeranalysispb.ListScanConfigsResponse), nil
-}
-
-func (s *mockContainerAnalysisV1Beta1Server) UpdateScanConfig(ctx context.Context, req *containeranalysispb.UpdateScanConfigRequest) (*containeranalysispb.ScanConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*containeranalysispb.ScanConfig), nil
-}
-
-type mockGrafeasV1Beta1Server struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	grafeaspb.GrafeasV1Beta1Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockGrafeasV1Beta1Server) GetOccurrence(ctx context.Context, req *grafeaspb.GetOccurrenceRequest) (*grafeaspb.Occurrence, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Occurrence), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) ListOccurrences(ctx context.Context, req *grafeaspb.ListOccurrencesRequest) (*grafeaspb.ListOccurrencesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.ListOccurrencesResponse), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) DeleteOccurrence(ctx context.Context, req *grafeaspb.DeleteOccurrenceRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) CreateOccurrence(ctx context.Context, req *grafeaspb.CreateOccurrenceRequest) (*grafeaspb.Occurrence, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Occurrence), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) BatchCreateOccurrences(ctx context.Context, req *grafeaspb.BatchCreateOccurrencesRequest) (*grafeaspb.BatchCreateOccurrencesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.BatchCreateOccurrencesResponse), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) UpdateOccurrence(ctx context.Context, req *grafeaspb.UpdateOccurrenceRequest) (*grafeaspb.Occurrence, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Occurrence), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) GetOccurrenceNote(ctx context.Context, req *grafeaspb.GetOccurrenceNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) GetNote(ctx context.Context, req *grafeaspb.GetNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) ListNotes(ctx context.Context, req *grafeaspb.ListNotesRequest) (*grafeaspb.ListNotesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.ListNotesResponse), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) DeleteNote(ctx context.Context, req *grafeaspb.DeleteNoteRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) CreateNote(ctx context.Context, req *grafeaspb.CreateNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) BatchCreateNotes(ctx context.Context, req *grafeaspb.BatchCreateNotesRequest) (*grafeaspb.BatchCreateNotesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.BatchCreateNotesResponse), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) UpdateNote(ctx context.Context, req *grafeaspb.UpdateNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) ListNoteOccurrences(ctx context.Context, req *grafeaspb.ListNoteOccurrencesRequest) (*grafeaspb.ListNoteOccurrencesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.ListNoteOccurrencesResponse), nil
-}
-
-func (s *mockGrafeasV1Beta1Server) GetVulnerabilityOccurrencesSummary(ctx context.Context, req *grafeaspb.GetVulnerabilityOccurrencesSummaryRequest) (*grafeaspb.VulnerabilityOccurrencesSummary, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.VulnerabilityOccurrencesSummary), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockContainerAnalysisV1Beta1 mockContainerAnalysisV1Beta1Server
-	mockGrafeasV1Beta1           mockGrafeasV1Beta1Server
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	containeranalysispb.RegisterContainerAnalysisV1Beta1Server(serv, &mockContainerAnalysisV1Beta1)
-	grafeaspb.RegisterGrafeasV1Beta1Server(serv, &mockGrafeasV1Beta1)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestContainerAnalysisV1Beta1SetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockContainerAnalysisV1Beta1.err = nil
-	mockContainerAnalysisV1Beta1.reqs = nil
-
-	mockContainerAnalysisV1Beta1.resps = append(mockContainerAnalysisV1Beta1.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysisV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisV1Beta1SetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysisV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisV1Beta1GetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockContainerAnalysisV1Beta1.err = nil
-	mockContainerAnalysisV1Beta1.reqs = nil
-
-	mockContainerAnalysisV1Beta1.resps = append(mockContainerAnalysisV1Beta1.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysisV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisV1Beta1GetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysisV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisV1Beta1TestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockContainerAnalysisV1Beta1.err = nil
-	mockContainerAnalysisV1Beta1.reqs = nil
-
-	mockContainerAnalysisV1Beta1.resps = append(mockContainerAnalysisV1Beta1.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysisV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisV1Beta1TestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysisV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisV1Beta1GetScanConfig(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var enabled bool = false
-	var expectedResponse = &containeranalysispb.ScanConfig{
-		Name:        name2,
-		Description: description,
-		Enabled:     enabled,
-	}
-
-	mockContainerAnalysisV1Beta1.err = nil
-	mockContainerAnalysisV1Beta1.reqs = nil
-
-	mockContainerAnalysisV1Beta1.resps = append(mockContainerAnalysisV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/scanConfigs/%s", "[PROJECT]", "[SCAN_CONFIG]")
-	var request = &containeranalysispb.GetScanConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetScanConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysisV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisV1Beta1GetScanConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysisV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/scanConfigs/%s", "[PROJECT]", "[SCAN_CONFIG]")
-	var request = &containeranalysispb.GetScanConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetScanConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisV1Beta1ListScanConfigs(t *testing.T) {
-	var nextPageToken string = ""
-	var scanConfigsElement *containeranalysispb.ScanConfig = &containeranalysispb.ScanConfig{}
-	var scanConfigs = []*containeranalysispb.ScanConfig{scanConfigsElement}
-	var expectedResponse = &containeranalysispb.ListScanConfigsResponse{
-		NextPageToken: nextPageToken,
-		ScanConfigs:   scanConfigs,
-	}
-
-	mockContainerAnalysisV1Beta1.err = nil
-	mockContainerAnalysisV1Beta1.reqs = nil
-
-	mockContainerAnalysisV1Beta1.resps = append(mockContainerAnalysisV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &containeranalysispb.ListScanConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListScanConfigs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysisV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ScanConfigs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisV1Beta1ListScanConfigsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysisV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &containeranalysispb.ListScanConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListScanConfigs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContainerAnalysisV1Beta1UpdateScanConfig(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var enabled bool = false
-	var expectedResponse = &containeranalysispb.ScanConfig{
-		Name:        name2,
-		Description: description,
-		Enabled:     enabled,
-	}
-
-	mockContainerAnalysisV1Beta1.err = nil
-	mockContainerAnalysisV1Beta1.reqs = nil
-
-	mockContainerAnalysisV1Beta1.resps = append(mockContainerAnalysisV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/scanConfigs/%s", "[PROJECT]", "[SCAN_CONFIG]")
-	var scanConfig *containeranalysispb.ScanConfig = &containeranalysispb.ScanConfig{}
-	var request = &containeranalysispb.UpdateScanConfigRequest{
-		Name:       formattedName,
-		ScanConfig: scanConfig,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateScanConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContainerAnalysisV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContainerAnalysisV1Beta1UpdateScanConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContainerAnalysisV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/scanConfigs/%s", "[PROJECT]", "[SCAN_CONFIG]")
-	var scanConfig *containeranalysispb.ScanConfig = &containeranalysispb.ScanConfig{}
-	var request = &containeranalysispb.UpdateScanConfigRequest{
-		Name:       formattedName,
-		ScanConfig: scanConfig,
-	}
-
-	c, err := NewContainerAnalysisV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateScanConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1GetOccurrence(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var noteName string = "noteName1780787896"
-	var remediation string = "remediation779381797"
-	var expectedResponse = &grafeaspb.Occurrence{
-		Name:        name2,
-		NoteName:    noteName,
-		Remediation: remediation,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1GetOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1ListOccurrences(t *testing.T) {
-	var nextPageToken string = ""
-	var occurrencesElement *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var occurrences = []*grafeaspb.Occurrence{occurrencesElement}
-	var expectedResponse = &grafeaspb.ListOccurrencesResponse{
-		NextPageToken: nextPageToken,
-		Occurrences:   occurrences,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListOccurrencesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListOccurrences(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Occurrences[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1ListOccurrencesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListOccurrencesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListOccurrences(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1DeleteOccurrence(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.DeleteOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestGrafeasV1Beta1DeleteOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.DeleteOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestGrafeasV1Beta1CreateOccurrence(t *testing.T) {
-	var name string = "name3373707"
-	var noteName string = "noteName1780787896"
-	var remediation string = "remediation779381797"
-	var expectedResponse = &grafeaspb.Occurrence{
-		Name:        name,
-		NoteName:    noteName,
-		Remediation: remediation,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.CreateOccurrenceRequest{
-		Parent:     formattedParent,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1CreateOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.CreateOccurrenceRequest{
-		Parent:     formattedParent,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1BatchCreateOccurrences(t *testing.T) {
-	var expectedResponse *grafeaspb.BatchCreateOccurrencesResponse = &grafeaspb.BatchCreateOccurrencesResponse{}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrences []*grafeaspb.Occurrence = nil
-	var request = &grafeaspb.BatchCreateOccurrencesRequest{
-		Parent:      formattedParent,
-		Occurrences: occurrences,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateOccurrences(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1BatchCreateOccurrencesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrences []*grafeaspb.Occurrence = nil
-	var request = &grafeaspb.BatchCreateOccurrencesRequest{
-		Parent:      formattedParent,
-		Occurrences: occurrences,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateOccurrences(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1UpdateOccurrence(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var noteName string = "noteName1780787896"
-	var remediation string = "remediation779381797"
-	var expectedResponse = &grafeaspb.Occurrence{
-		Name:        name2,
-		NoteName:    noteName,
-		Remediation: remediation,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.UpdateOccurrenceRequest{
-		Name:       formattedName,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1UpdateOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.UpdateOccurrenceRequest{
-		Name:       formattedName,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1GetOccurrenceNote(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name2,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrenceNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1GetOccurrenceNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrenceNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1GetNote(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name2,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.GetNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1GetNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.GetNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1ListNotes(t *testing.T) {
-	var nextPageToken string = ""
-	var notesElement *grafeaspb.Note = &grafeaspb.Note{}
-	var notes = []*grafeaspb.Note{notesElement}
-	var expectedResponse = &grafeaspb.ListNotesResponse{
-		NextPageToken: nextPageToken,
-		Notes:         notes,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListNotesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotes(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Notes[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1ListNotesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListNotesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotes(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1DeleteNote(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.DeleteNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestGrafeasV1Beta1DeleteNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.DeleteNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestGrafeasV1Beta1CreateNote(t *testing.T) {
-	var name string = "name3373707"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var noteId string = "noteId2129224840"
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.CreateNoteRequest{
-		Parent: formattedParent,
-		NoteId: noteId,
-		Note:   note,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1CreateNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var noteId string = "noteId2129224840"
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.CreateNoteRequest{
-		Parent: formattedParent,
-		NoteId: noteId,
-		Note:   note,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1BatchCreateNotes(t *testing.T) {
-	var expectedResponse *grafeaspb.BatchCreateNotesResponse = &grafeaspb.BatchCreateNotesResponse{}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var notes map[string]*grafeaspb.Note = nil
-	var request = &grafeaspb.BatchCreateNotesRequest{
-		Parent: formattedParent,
-		Notes:  notes,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateNotes(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1BatchCreateNotesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var notes map[string]*grafeaspb.Note = nil
-	var request = &grafeaspb.BatchCreateNotesRequest{
-		Parent: formattedParent,
-		Notes:  notes,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateNotes(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1UpdateNote(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name2,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.UpdateNoteRequest{
-		Name: formattedName,
-		Note: note,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1UpdateNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.UpdateNoteRequest{
-		Name: formattedName,
-		Note: note,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1ListNoteOccurrences(t *testing.T) {
-	var nextPageToken string = ""
-	var occurrencesElement *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var occurrences = []*grafeaspb.Occurrence{occurrencesElement}
-	var expectedResponse = &grafeaspb.ListNoteOccurrencesResponse{
-		NextPageToken: nextPageToken,
-		Occurrences:   occurrences,
-	}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.ListNoteOccurrencesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNoteOccurrences(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Occurrences[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1ListNoteOccurrencesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.ListNoteOccurrencesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNoteOccurrences(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasV1Beta1GetVulnerabilityOccurrencesSummary(t *testing.T) {
-	var expectedResponse *grafeaspb.VulnerabilityOccurrencesSummary = &grafeaspb.VulnerabilityOccurrencesSummary{}
-
-	mockGrafeasV1Beta1.err = nil
-	mockGrafeasV1Beta1.reqs = nil
-
-	mockGrafeasV1Beta1.resps = append(mockGrafeasV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.GetVulnerabilityOccurrencesSummaryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetVulnerabilityOccurrencesSummary(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeasV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasV1Beta1GetVulnerabilityOccurrencesSummaryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeasV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.GetVulnerabilityOccurrencesSummaryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewGrafeasV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetVulnerabilityOccurrencesSummary(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/dataproc/apiv1/mock_test.go b/dataproc/apiv1/mock_test.go
deleted file mode 100644
index 10df9b6e9c7..00000000000
--- a/dataproc/apiv1/mock_test.go
+++ /dev/null
@@ -1,1781 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package dataproc
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	dataprocpb "google.golang.org/genproto/googleapis/cloud/dataproc/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockClusterControllerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dataprocpb.ClusterControllerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockClusterControllerServer) CreateCluster(ctx context.Context, req *dataprocpb.CreateClusterRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockClusterControllerServer) UpdateCluster(ctx context.Context, req *dataprocpb.UpdateClusterRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockClusterControllerServer) DeleteCluster(ctx context.Context, req *dataprocpb.DeleteClusterRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockClusterControllerServer) GetCluster(ctx context.Context, req *dataprocpb.GetClusterRequest) (*dataprocpb.Cluster, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.Cluster), nil
-}
-
-func (s *mockClusterControllerServer) ListClusters(ctx context.Context, req *dataprocpb.ListClustersRequest) (*dataprocpb.ListClustersResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.ListClustersResponse), nil
-}
-
-func (s *mockClusterControllerServer) DiagnoseCluster(ctx context.Context, req *dataprocpb.DiagnoseClusterRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-type mockJobControllerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dataprocpb.JobControllerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockJobControllerServer) SubmitJob(ctx context.Context, req *dataprocpb.SubmitJobRequest) (*dataprocpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.Job), nil
-}
-
-func (s *mockJobControllerServer) GetJob(ctx context.Context, req *dataprocpb.GetJobRequest) (*dataprocpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.Job), nil
-}
-
-func (s *mockJobControllerServer) ListJobs(ctx context.Context, req *dataprocpb.ListJobsRequest) (*dataprocpb.ListJobsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.ListJobsResponse), nil
-}
-
-func (s *mockJobControllerServer) UpdateJob(ctx context.Context, req *dataprocpb.UpdateJobRequest) (*dataprocpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.Job), nil
-}
-
-func (s *mockJobControllerServer) CancelJob(ctx context.Context, req *dataprocpb.CancelJobRequest) (*dataprocpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.Job), nil
-}
-
-func (s *mockJobControllerServer) DeleteJob(ctx context.Context, req *dataprocpb.DeleteJobRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-type mockWorkflowTemplateServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dataprocpb.WorkflowTemplateServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockWorkflowTemplateServer) CreateWorkflowTemplate(ctx context.Context, req *dataprocpb.CreateWorkflowTemplateRequest) (*dataprocpb.WorkflowTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.WorkflowTemplate), nil
-}
-
-func (s *mockWorkflowTemplateServer) GetWorkflowTemplate(ctx context.Context, req *dataprocpb.GetWorkflowTemplateRequest) (*dataprocpb.WorkflowTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.WorkflowTemplate), nil
-}
-
-func (s *mockWorkflowTemplateServer) InstantiateWorkflowTemplate(ctx context.Context, req *dataprocpb.InstantiateWorkflowTemplateRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockWorkflowTemplateServer) InstantiateInlineWorkflowTemplate(ctx context.Context, req *dataprocpb.InstantiateInlineWorkflowTemplateRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockWorkflowTemplateServer) UpdateWorkflowTemplate(ctx context.Context, req *dataprocpb.UpdateWorkflowTemplateRequest) (*dataprocpb.WorkflowTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.WorkflowTemplate), nil
-}
-
-func (s *mockWorkflowTemplateServer) ListWorkflowTemplates(ctx context.Context, req *dataprocpb.ListWorkflowTemplatesRequest) (*dataprocpb.ListWorkflowTemplatesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dataprocpb.ListWorkflowTemplatesResponse), nil
-}
-
-func (s *mockWorkflowTemplateServer) DeleteWorkflowTemplate(ctx context.Context, req *dataprocpb.DeleteWorkflowTemplateRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockClusterController mockClusterControllerServer
-	mockJobController     mockJobControllerServer
-	mockWorkflowTemplate  mockWorkflowTemplateServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	dataprocpb.RegisterClusterControllerServer(serv, &mockClusterController)
-	dataprocpb.RegisterJobControllerServer(serv, &mockJobController)
-	dataprocpb.RegisterWorkflowTemplateServiceServer(serv, &mockWorkflowTemplate)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestClusterControllerCreateCluster(t *testing.T) {
-	var projectId2 string = "projectId2939242356"
-	var clusterName string = "clusterName-1018081872"
-	var clusterUuid string = "clusterUuid-1017854240"
-	var expectedResponse = &dataprocpb.Cluster{
-		ProjectId:   projectId2,
-		ClusterName: clusterName,
-		ClusterUuid: clusterUuid,
-	}
-
-	mockClusterController.err = nil
-	mockClusterController.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var cluster *dataprocpb.Cluster = &dataprocpb.Cluster{}
-	var request = &dataprocpb.CreateClusterRequest{
-		ProjectId: projectId,
-		Region:    region,
-		Cluster:   cluster,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterControllerCreateClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterController.err = nil
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var cluster *dataprocpb.Cluster = &dataprocpb.Cluster{}
-	var request = &dataprocpb.CreateClusterRequest{
-		ProjectId: projectId,
-		Region:    region,
-		Cluster:   cluster,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterControllerUpdateCluster(t *testing.T) {
-	var projectId2 string = "projectId2939242356"
-	var clusterName2 string = "clusterName2875867491"
-	var clusterUuid string = "clusterUuid-1017854240"
-	var expectedResponse = &dataprocpb.Cluster{
-		ProjectId:   projectId2,
-		ClusterName: clusterName2,
-		ClusterUuid: clusterUuid,
-	}
-
-	mockClusterController.err = nil
-	mockClusterController.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var cluster *dataprocpb.Cluster = &dataprocpb.Cluster{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &dataprocpb.UpdateClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-		Cluster:     cluster,
-		UpdateMask:  updateMask,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterControllerUpdateClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterController.err = nil
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var cluster *dataprocpb.Cluster = &dataprocpb.Cluster{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &dataprocpb.UpdateClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-		Cluster:     cluster,
-		UpdateMask:  updateMask,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterControllerDeleteCluster(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockClusterController.err = nil
-	mockClusterController.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var request = &dataprocpb.DeleteClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestClusterControllerDeleteClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterController.err = nil
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var request = &dataprocpb.DeleteClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestClusterControllerGetCluster(t *testing.T) {
-	var projectId2 string = "projectId2939242356"
-	var clusterName2 string = "clusterName2875867491"
-	var clusterUuid string = "clusterUuid-1017854240"
-	var expectedResponse = &dataprocpb.Cluster{
-		ProjectId:   projectId2,
-		ClusterName: clusterName2,
-		ClusterUuid: clusterUuid,
-	}
-
-	mockClusterController.err = nil
-	mockClusterController.reqs = nil
-
-	mockClusterController.resps = append(mockClusterController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var request = &dataprocpb.GetClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCluster(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterControllerGetClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var request = &dataprocpb.GetClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCluster(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterControllerListClusters(t *testing.T) {
-	var nextPageToken string = ""
-	var clustersElement *dataprocpb.Cluster = &dataprocpb.Cluster{}
-	var clusters = []*dataprocpb.Cluster{clustersElement}
-	var expectedResponse = &dataprocpb.ListClustersResponse{
-		NextPageToken: nextPageToken,
-		Clusters:      clusters,
-	}
-
-	mockClusterController.err = nil
-	mockClusterController.reqs = nil
-
-	mockClusterController.resps = append(mockClusterController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var request = &dataprocpb.ListClustersRequest{
-		ProjectId: projectId,
-		Region:    region,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListClusters(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Clusters[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestClusterControllerListClustersError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var request = &dataprocpb.ListClustersRequest{
-		ProjectId: projectId,
-		Region:    region,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListClusters(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestClusterControllerDiagnoseCluster(t *testing.T) {
-	var expectedResponse *dataprocpb.DiagnoseClusterResults = &dataprocpb.DiagnoseClusterResults{}
-
-	mockClusterController.err = nil
-	mockClusterController.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var request = &dataprocpb.DiagnoseClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DiagnoseCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	_, err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockClusterController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestClusterControllerDiagnoseClusterError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockClusterController.err = nil
-	mockClusterController.resps = append(mockClusterController.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var clusterName string = "clusterName-1018081872"
-	var request = &dataprocpb.DiagnoseClusterRequest{
-		ProjectId:   projectId,
-		Region:      region,
-		ClusterName: clusterName,
-	}
-
-	c, err := NewClusterControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DiagnoseCluster(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	_, err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestJobControllerSubmitJob(t *testing.T) {
-	var driverOutputResourceUri string = "driverOutputResourceUri-542229086"
-	var driverControlFilesUri string = "driverControlFilesUri207057643"
-	var jobUuid string = "jobUuid-1615012099"
-	var expectedResponse = &dataprocpb.Job{
-		DriverOutputResourceUri: driverOutputResourceUri,
-		DriverControlFilesUri:   driverControlFilesUri,
-		JobUuid:                 jobUuid,
-	}
-
-	mockJobController.err = nil
-	mockJobController.reqs = nil
-
-	mockJobController.resps = append(mockJobController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var job *dataprocpb.Job = &dataprocpb.Job{}
-	var request = &dataprocpb.SubmitJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		Job:       job,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SubmitJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockJobController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestJobControllerSubmitJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockJobController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var job *dataprocpb.Job = &dataprocpb.Job{}
-	var request = &dataprocpb.SubmitJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		Job:       job,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SubmitJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestJobControllerGetJob(t *testing.T) {
-	var driverOutputResourceUri string = "driverOutputResourceUri-542229086"
-	var driverControlFilesUri string = "driverControlFilesUri207057643"
-	var jobUuid string = "jobUuid-1615012099"
-	var expectedResponse = &dataprocpb.Job{
-		DriverOutputResourceUri: driverOutputResourceUri,
-		DriverControlFilesUri:   driverControlFilesUri,
-		JobUuid:                 jobUuid,
-	}
-
-	mockJobController.err = nil
-	mockJobController.reqs = nil
-
-	mockJobController.resps = append(mockJobController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var request = &dataprocpb.GetJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		JobId:     jobId,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockJobController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestJobControllerGetJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockJobController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var request = &dataprocpb.GetJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		JobId:     jobId,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestJobControllerListJobs(t *testing.T) {
-	var nextPageToken string = ""
-	var jobsElement *dataprocpb.Job = &dataprocpb.Job{}
-	var jobs = []*dataprocpb.Job{jobsElement}
-	var expectedResponse = &dataprocpb.ListJobsResponse{
-		NextPageToken: nextPageToken,
-		Jobs:          jobs,
-	}
-
-	mockJobController.err = nil
-	mockJobController.reqs = nil
-
-	mockJobController.resps = append(mockJobController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var request = &dataprocpb.ListJobsRequest{
-		ProjectId: projectId,
-		Region:    region,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockJobController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Jobs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestJobControllerListJobsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockJobController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var request = &dataprocpb.ListJobsRequest{
-		ProjectId: projectId,
-		Region:    region,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestJobControllerUpdateJob(t *testing.T) {
-	var driverOutputResourceUri string = "driverOutputResourceUri-542229086"
-	var driverControlFilesUri string = "driverControlFilesUri207057643"
-	var jobUuid string = "jobUuid-1615012099"
-	var expectedResponse = &dataprocpb.Job{
-		DriverOutputResourceUri: driverOutputResourceUri,
-		DriverControlFilesUri:   driverControlFilesUri,
-		JobUuid:                 jobUuid,
-	}
-
-	mockJobController.err = nil
-	mockJobController.reqs = nil
-
-	mockJobController.resps = append(mockJobController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var job *dataprocpb.Job = &dataprocpb.Job{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &dataprocpb.UpdateJobRequest{
-		ProjectId:  projectId,
-		Region:     region,
-		JobId:      jobId,
-		Job:        job,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockJobController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestJobControllerUpdateJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockJobController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var job *dataprocpb.Job = &dataprocpb.Job{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &dataprocpb.UpdateJobRequest{
-		ProjectId:  projectId,
-		Region:     region,
-		JobId:      jobId,
-		Job:        job,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestJobControllerCancelJob(t *testing.T) {
-	var driverOutputResourceUri string = "driverOutputResourceUri-542229086"
-	var driverControlFilesUri string = "driverControlFilesUri207057643"
-	var jobUuid string = "jobUuid-1615012099"
-	var expectedResponse = &dataprocpb.Job{
-		DriverOutputResourceUri: driverOutputResourceUri,
-		DriverControlFilesUri:   driverControlFilesUri,
-		JobUuid:                 jobUuid,
-	}
-
-	mockJobController.err = nil
-	mockJobController.reqs = nil
-
-	mockJobController.resps = append(mockJobController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var request = &dataprocpb.CancelJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		JobId:     jobId,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CancelJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockJobController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestJobControllerCancelJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockJobController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var request = &dataprocpb.CancelJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		JobId:     jobId,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CancelJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestJobControllerDeleteJob(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockJobController.err = nil
-	mockJobController.reqs = nil
-
-	mockJobController.resps = append(mockJobController.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var request = &dataprocpb.DeleteJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		JobId:     jobId,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockJobController.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestJobControllerDeleteJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockJobController.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var region string = "region-934795532"
-	var jobId string = "jobId-1154752291"
-	var request = &dataprocpb.DeleteJobRequest{
-		ProjectId: projectId,
-		Region:    region,
-		JobId:     jobId,
-	}
-
-	c, err := NewJobControllerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestWorkflowTemplateServiceCreateWorkflowTemplate(t *testing.T) {
-	var id string = "id3355"
-	var name string = "name3373707"
-	var version int32 = 351608024
-	var expectedResponse = &dataprocpb.WorkflowTemplate{
-		Id:      id,
-		Name:    name,
-		Version: version,
-	}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/regions/%s", "[PROJECT]", "[REGION]")
-	var template *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var request = &dataprocpb.CreateWorkflowTemplateRequest{
-		Parent:   formattedParent,
-		Template: template,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateWorkflowTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWorkflowTemplateServiceCreateWorkflowTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/regions/%s", "[PROJECT]", "[REGION]")
-	var template *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var request = &dataprocpb.CreateWorkflowTemplateRequest{
-		Parent:   formattedParent,
-		Template: template,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateWorkflowTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestWorkflowTemplateServiceGetWorkflowTemplate(t *testing.T) {
-	var id string = "id3355"
-	var name2 string = "name2-1052831874"
-	var version int32 = 351608024
-	var expectedResponse = &dataprocpb.WorkflowTemplate{
-		Id:      id,
-		Name:    name2,
-		Version: version,
-	}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/regions/%s/workflowTemplates/%s", "[PROJECT]", "[REGION]", "[WORKFLOW_TEMPLATE]")
-	var request = &dataprocpb.GetWorkflowTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetWorkflowTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWorkflowTemplateServiceGetWorkflowTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/regions/%s/workflowTemplates/%s", "[PROJECT]", "[REGION]", "[WORKFLOW_TEMPLATE]")
-	var request = &dataprocpb.GetWorkflowTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetWorkflowTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestWorkflowTemplateServiceInstantiateWorkflowTemplate(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/regions/%s/workflowTemplates/%s", "[PROJECT]", "[REGION]", "[WORKFLOW_TEMPLATE]")
-	var request = &dataprocpb.InstantiateWorkflowTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.InstantiateWorkflowTemplate(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestWorkflowTemplateServiceInstantiateWorkflowTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/regions/%s/workflowTemplates/%s", "[PROJECT]", "[REGION]", "[WORKFLOW_TEMPLATE]")
-	var request = &dataprocpb.InstantiateWorkflowTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.InstantiateWorkflowTemplate(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestWorkflowTemplateServiceInstantiateInlineWorkflowTemplate(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/regions/%s", "[PROJECT]", "[REGION]")
-	var template *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var request = &dataprocpb.InstantiateInlineWorkflowTemplateRequest{
-		Parent:   formattedParent,
-		Template: template,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.InstantiateInlineWorkflowTemplate(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestWorkflowTemplateServiceInstantiateInlineWorkflowTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/regions/%s", "[PROJECT]", "[REGION]")
-	var template *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var request = &dataprocpb.InstantiateInlineWorkflowTemplateRequest{
-		Parent:   formattedParent,
-		Template: template,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.InstantiateInlineWorkflowTemplate(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestWorkflowTemplateServiceUpdateWorkflowTemplate(t *testing.T) {
-	var id string = "id3355"
-	var name string = "name3373707"
-	var version int32 = 351608024
-	var expectedResponse = &dataprocpb.WorkflowTemplate{
-		Id:      id,
-		Name:    name,
-		Version: version,
-	}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], expectedResponse)
-
-	var template *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var request = &dataprocpb.UpdateWorkflowTemplateRequest{
-		Template: template,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateWorkflowTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWorkflowTemplateServiceUpdateWorkflowTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = gstatus.Error(errCode, "test error")
-
-	var template *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var request = &dataprocpb.UpdateWorkflowTemplateRequest{
-		Template: template,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateWorkflowTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestWorkflowTemplateServiceListWorkflowTemplates(t *testing.T) {
-	var nextPageToken string = ""
-	var templatesElement *dataprocpb.WorkflowTemplate = &dataprocpb.WorkflowTemplate{}
-	var templates = []*dataprocpb.WorkflowTemplate{templatesElement}
-	var expectedResponse = &dataprocpb.ListWorkflowTemplatesResponse{
-		NextPageToken: nextPageToken,
-		Templates:     templates,
-	}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/regions/%s", "[PROJECT]", "[REGION]")
-	var request = &dataprocpb.ListWorkflowTemplatesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListWorkflowTemplates(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Templates[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWorkflowTemplateServiceListWorkflowTemplatesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/regions/%s", "[PROJECT]", "[REGION]")
-	var request = &dataprocpb.ListWorkflowTemplatesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListWorkflowTemplates(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestWorkflowTemplateServiceDeleteWorkflowTemplate(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockWorkflowTemplate.err = nil
-	mockWorkflowTemplate.reqs = nil
-
-	mockWorkflowTemplate.resps = append(mockWorkflowTemplate.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/regions/%s/workflowTemplates/%s", "[PROJECT]", "[REGION]", "[WORKFLOW_TEMPLATE]")
-	var request = &dataprocpb.DeleteWorkflowTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteWorkflowTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWorkflowTemplate.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestWorkflowTemplateServiceDeleteWorkflowTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWorkflowTemplate.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/regions/%s/workflowTemplates/%s", "[PROJECT]", "[REGION]", "[WORKFLOW_TEMPLATE]")
-	var request = &dataprocpb.DeleteWorkflowTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewWorkflowTemplateClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteWorkflowTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
diff --git a/dialogflow/apiv2/mock_test.go b/dialogflow/apiv2/mock_test.go
deleted file mode 100644
index 7242a10b24d..00000000000
--- a/dialogflow/apiv2/mock_test.go
+++ /dev/null
@@ -1,3307 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package dialogflow
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	dialogflowpb "google.golang.org/genproto/googleapis/cloud/dialogflow/v2"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockAgentsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dialogflowpb.AgentsServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockAgentsServer) GetAgent(ctx context.Context, req *dialogflowpb.GetAgentRequest) (*dialogflowpb.Agent, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Agent), nil
-}
-
-func (s *mockAgentsServer) SetAgent(ctx context.Context, req *dialogflowpb.SetAgentRequest) (*dialogflowpb.Agent, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Agent), nil
-}
-
-func (s *mockAgentsServer) DeleteAgent(ctx context.Context, req *dialogflowpb.DeleteAgentRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockAgentsServer) SearchAgents(ctx context.Context, req *dialogflowpb.SearchAgentsRequest) (*dialogflowpb.SearchAgentsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.SearchAgentsResponse), nil
-}
-
-func (s *mockAgentsServer) TrainAgent(ctx context.Context, req *dialogflowpb.TrainAgentRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAgentsServer) ExportAgent(ctx context.Context, req *dialogflowpb.ExportAgentRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAgentsServer) ImportAgent(ctx context.Context, req *dialogflowpb.ImportAgentRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockAgentsServer) RestoreAgent(ctx context.Context, req *dialogflowpb.RestoreAgentRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-type mockContextsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dialogflowpb.ContextsServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockContextsServer) ListContexts(ctx context.Context, req *dialogflowpb.ListContextsRequest) (*dialogflowpb.ListContextsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.ListContextsResponse), nil
-}
-
-func (s *mockContextsServer) GetContext(ctx context.Context, req *dialogflowpb.GetContextRequest) (*dialogflowpb.Context, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Context), nil
-}
-
-func (s *mockContextsServer) CreateContext(ctx context.Context, req *dialogflowpb.CreateContextRequest) (*dialogflowpb.Context, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Context), nil
-}
-
-func (s *mockContextsServer) UpdateContext(ctx context.Context, req *dialogflowpb.UpdateContextRequest) (*dialogflowpb.Context, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Context), nil
-}
-
-func (s *mockContextsServer) DeleteContext(ctx context.Context, req *dialogflowpb.DeleteContextRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockContextsServer) DeleteAllContexts(ctx context.Context, req *dialogflowpb.DeleteAllContextsRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-type mockEntityTypesServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dialogflowpb.EntityTypesServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockEntityTypesServer) ListEntityTypes(ctx context.Context, req *dialogflowpb.ListEntityTypesRequest) (*dialogflowpb.ListEntityTypesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.ListEntityTypesResponse), nil
-}
-
-func (s *mockEntityTypesServer) GetEntityType(ctx context.Context, req *dialogflowpb.GetEntityTypeRequest) (*dialogflowpb.EntityType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.EntityType), nil
-}
-
-func (s *mockEntityTypesServer) CreateEntityType(ctx context.Context, req *dialogflowpb.CreateEntityTypeRequest) (*dialogflowpb.EntityType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.EntityType), nil
-}
-
-func (s *mockEntityTypesServer) UpdateEntityType(ctx context.Context, req *dialogflowpb.UpdateEntityTypeRequest) (*dialogflowpb.EntityType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.EntityType), nil
-}
-
-func (s *mockEntityTypesServer) DeleteEntityType(ctx context.Context, req *dialogflowpb.DeleteEntityTypeRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockEntityTypesServer) BatchUpdateEntityTypes(ctx context.Context, req *dialogflowpb.BatchUpdateEntityTypesRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockEntityTypesServer) BatchDeleteEntityTypes(ctx context.Context, req *dialogflowpb.BatchDeleteEntityTypesRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockEntityTypesServer) BatchCreateEntities(ctx context.Context, req *dialogflowpb.BatchCreateEntitiesRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockEntityTypesServer) BatchUpdateEntities(ctx context.Context, req *dialogflowpb.BatchUpdateEntitiesRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockEntityTypesServer) BatchDeleteEntities(ctx context.Context, req *dialogflowpb.BatchDeleteEntitiesRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-type mockIntentsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dialogflowpb.IntentsServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockIntentsServer) ListIntents(ctx context.Context, req *dialogflowpb.ListIntentsRequest) (*dialogflowpb.ListIntentsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.ListIntentsResponse), nil
-}
-
-func (s *mockIntentsServer) GetIntent(ctx context.Context, req *dialogflowpb.GetIntentRequest) (*dialogflowpb.Intent, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Intent), nil
-}
-
-func (s *mockIntentsServer) CreateIntent(ctx context.Context, req *dialogflowpb.CreateIntentRequest) (*dialogflowpb.Intent, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Intent), nil
-}
-
-func (s *mockIntentsServer) UpdateIntent(ctx context.Context, req *dialogflowpb.UpdateIntentRequest) (*dialogflowpb.Intent, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.Intent), nil
-}
-
-func (s *mockIntentsServer) DeleteIntent(ctx context.Context, req *dialogflowpb.DeleteIntentRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockIntentsServer) BatchUpdateIntents(ctx context.Context, req *dialogflowpb.BatchUpdateIntentsRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockIntentsServer) BatchDeleteIntents(ctx context.Context, req *dialogflowpb.BatchDeleteIntentsRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-type mockSessionEntityTypesServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dialogflowpb.SessionEntityTypesServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSessionEntityTypesServer) ListSessionEntityTypes(ctx context.Context, req *dialogflowpb.ListSessionEntityTypesRequest) (*dialogflowpb.ListSessionEntityTypesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.ListSessionEntityTypesResponse), nil
-}
-
-func (s *mockSessionEntityTypesServer) GetSessionEntityType(ctx context.Context, req *dialogflowpb.GetSessionEntityTypeRequest) (*dialogflowpb.SessionEntityType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.SessionEntityType), nil
-}
-
-func (s *mockSessionEntityTypesServer) CreateSessionEntityType(ctx context.Context, req *dialogflowpb.CreateSessionEntityTypeRequest) (*dialogflowpb.SessionEntityType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.SessionEntityType), nil
-}
-
-func (s *mockSessionEntityTypesServer) UpdateSessionEntityType(ctx context.Context, req *dialogflowpb.UpdateSessionEntityTypeRequest) (*dialogflowpb.SessionEntityType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.SessionEntityType), nil
-}
-
-func (s *mockSessionEntityTypesServer) DeleteSessionEntityType(ctx context.Context, req *dialogflowpb.DeleteSessionEntityTypeRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-type mockSessionsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dialogflowpb.SessionsServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSessionsServer) DetectIntent(ctx context.Context, req *dialogflowpb.DetectIntentRequest) (*dialogflowpb.DetectIntentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dialogflowpb.DetectIntentResponse), nil
-}
-
-func (s *mockSessionsServer) StreamingDetectIntent(stream dialogflowpb.Sessions_StreamingDetectIntentServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*dialogflowpb.StreamingDetectIntentResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockAgents             mockAgentsServer
-	mockContexts           mockContextsServer
-	mockEntityTypes        mockEntityTypesServer
-	mockIntents            mockIntentsServer
-	mockSessionEntityTypes mockSessionEntityTypesServer
-	mockSessions           mockSessionsServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	dialogflowpb.RegisterAgentsServer(serv, &mockAgents)
-	dialogflowpb.RegisterContextsServer(serv, &mockContexts)
-	dialogflowpb.RegisterEntityTypesServer(serv, &mockEntityTypes)
-	dialogflowpb.RegisterIntentsServer(serv, &mockIntents)
-	dialogflowpb.RegisterSessionEntityTypesServer(serv, &mockSessionEntityTypes)
-	dialogflowpb.RegisterSessionsServer(serv, &mockSessions)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestAgentsSetAgent(t *testing.T) {
-	var parent string = "parent-995424086"
-	var displayName string = "displayName1615086568"
-	var defaultLanguageCode string = "defaultLanguageCode856575222"
-	var timeZone string = "timeZone36848094"
-	var description string = "description-1724546052"
-	var avatarUri string = "avatarUri-402824826"
-	var enableLogging bool = false
-	var classificationThreshold float32 = 1.11581064e8
-	var expectedResponse = &dialogflowpb.Agent{
-		Parent:                  parent,
-		DisplayName:             displayName,
-		DefaultLanguageCode:     defaultLanguageCode,
-		TimeZone:                timeZone,
-		Description:             description,
-		AvatarUri:               avatarUri,
-		EnableLogging:           enableLogging,
-		ClassificationThreshold: classificationThreshold,
-	}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	mockAgents.resps = append(mockAgents.resps[:0], expectedResponse)
-
-	var agent *dialogflowpb.Agent = &dialogflowpb.Agent{}
-	var request = &dialogflowpb.SetAgentRequest{
-		Agent: agent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetAgent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAgentsSetAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = gstatus.Error(errCode, "test error")
-
-	var agent *dialogflowpb.Agent = &dialogflowpb.Agent{}
-	var request = &dialogflowpb.SetAgentRequest{
-		Agent: agent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetAgent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAgentsDeleteAgent(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	mockAgents.resps = append(mockAgents.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.DeleteAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteAgent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAgentsDeleteAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.DeleteAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteAgent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAgentsGetAgent(t *testing.T) {
-	var parent2 string = "parent21175163357"
-	var displayName string = "displayName1615086568"
-	var defaultLanguageCode string = "defaultLanguageCode856575222"
-	var timeZone string = "timeZone36848094"
-	var description string = "description-1724546052"
-	var avatarUri string = "avatarUri-402824826"
-	var enableLogging bool = false
-	var classificationThreshold float32 = 1.11581064e8
-	var expectedResponse = &dialogflowpb.Agent{
-		Parent:                  parent2,
-		DisplayName:             displayName,
-		DefaultLanguageCode:     defaultLanguageCode,
-		TimeZone:                timeZone,
-		Description:             description,
-		AvatarUri:               avatarUri,
-		EnableLogging:           enableLogging,
-		ClassificationThreshold: classificationThreshold,
-	}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	mockAgents.resps = append(mockAgents.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.GetAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAgent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAgentsGetAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.GetAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAgent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAgentsSearchAgents(t *testing.T) {
-	var nextPageToken string = ""
-	var agentsElement *dialogflowpb.Agent = &dialogflowpb.Agent{}
-	var agents = []*dialogflowpb.Agent{agentsElement}
-	var expectedResponse = &dialogflowpb.SearchAgentsResponse{
-		NextPageToken: nextPageToken,
-		Agents:        agents,
-	}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	mockAgents.resps = append(mockAgents.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.SearchAgentsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SearchAgents(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Agents[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAgentsSearchAgentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.SearchAgentsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SearchAgents(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAgentsTrainAgent(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.TrainAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.TrainAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAgentsTrainAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = nil
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.TrainAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.TrainAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAgentsExportAgent(t *testing.T) {
-	var agentUri string = "agentUri-1700713166"
-	var expectedResponse = &dialogflowpb.ExportAgentResponse{
-		Agent: &dialogflowpb.ExportAgentResponse_AgentUri{
-			AgentUri: agentUri,
-		},
-	}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.ExportAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAgentsExportAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = nil
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.ExportAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAgentsImportAgent(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.ImportAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAgentsImportAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = nil
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.ImportAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAgentsRestoreAgent(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAgents.err = nil
-	mockAgents.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.RestoreAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.RestoreAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAgents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAgentsRestoreAgentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAgents.err = nil
-	mockAgents.resps = append(mockAgents.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dialogflowpb.RestoreAgentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewAgentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.RestoreAgent(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestContextsListContexts(t *testing.T) {
-	var nextPageToken string = ""
-	var contextsElement *dialogflowpb.Context = &dialogflowpb.Context{}
-	var contexts = []*dialogflowpb.Context{contextsElement}
-	var expectedResponse = &dialogflowpb.ListContextsResponse{
-		NextPageToken: nextPageToken,
-		Contexts:      contexts,
-	}
-
-	mockContexts.err = nil
-	mockContexts.reqs = nil
-
-	mockContexts.resps = append(mockContexts.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var request = &dialogflowpb.ListContextsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListContexts(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContexts.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Contexts[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContextsListContextsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContexts.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var request = &dialogflowpb.ListContextsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListContexts(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContextsGetContext(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var lifespanCount int32 = 1178775510
-	var expectedResponse = &dialogflowpb.Context{
-		Name:          name2,
-		LifespanCount: lifespanCount,
-	}
-
-	mockContexts.err = nil
-	mockContexts.reqs = nil
-
-	mockContexts.resps = append(mockContexts.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/contexts/%s", "[PROJECT]", "[SESSION]", "[CONTEXT]")
-	var request = &dialogflowpb.GetContextRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetContext(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContexts.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContextsGetContextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContexts.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/contexts/%s", "[PROJECT]", "[SESSION]", "[CONTEXT]")
-	var request = &dialogflowpb.GetContextRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetContext(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContextsCreateContext(t *testing.T) {
-	var name string = "name3373707"
-	var lifespanCount int32 = 1178775510
-	var expectedResponse = &dialogflowpb.Context{
-		Name:          name,
-		LifespanCount: lifespanCount,
-	}
-
-	mockContexts.err = nil
-	mockContexts.reqs = nil
-
-	mockContexts.resps = append(mockContexts.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var context_ *dialogflowpb.Context = &dialogflowpb.Context{}
-	var request = &dialogflowpb.CreateContextRequest{
-		Parent:  formattedParent,
-		Context: context_,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateContext(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContexts.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContextsCreateContextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContexts.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var context_ *dialogflowpb.Context = &dialogflowpb.Context{}
-	var request = &dialogflowpb.CreateContextRequest{
-		Parent:  formattedParent,
-		Context: context_,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateContext(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContextsUpdateContext(t *testing.T) {
-	var name string = "name3373707"
-	var lifespanCount int32 = 1178775510
-	var expectedResponse = &dialogflowpb.Context{
-		Name:          name,
-		LifespanCount: lifespanCount,
-	}
-
-	mockContexts.err = nil
-	mockContexts.reqs = nil
-
-	mockContexts.resps = append(mockContexts.resps[:0], expectedResponse)
-
-	var context_ *dialogflowpb.Context = &dialogflowpb.Context{}
-	var request = &dialogflowpb.UpdateContextRequest{
-		Context: context_,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateContext(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContexts.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestContextsUpdateContextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContexts.err = gstatus.Error(errCode, "test error")
-
-	var context_ *dialogflowpb.Context = &dialogflowpb.Context{}
-	var request = &dialogflowpb.UpdateContextRequest{
-		Context: context_,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateContext(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestContextsDeleteContext(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockContexts.err = nil
-	mockContexts.reqs = nil
-
-	mockContexts.resps = append(mockContexts.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/contexts/%s", "[PROJECT]", "[SESSION]", "[CONTEXT]")
-	var request = &dialogflowpb.DeleteContextRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteContext(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContexts.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestContextsDeleteContextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContexts.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/contexts/%s", "[PROJECT]", "[SESSION]", "[CONTEXT]")
-	var request = &dialogflowpb.DeleteContextRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteContext(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestContextsDeleteAllContexts(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockContexts.err = nil
-	mockContexts.reqs = nil
-
-	mockContexts.resps = append(mockContexts.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var request = &dialogflowpb.DeleteAllContextsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteAllContexts(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockContexts.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestContextsDeleteAllContextsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockContexts.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var request = &dialogflowpb.DeleteAllContextsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewContextsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteAllContexts(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestEntityTypesListEntityTypes(t *testing.T) {
-	var nextPageToken string = ""
-	var entityTypesElement *dialogflowpb.EntityType = &dialogflowpb.EntityType{}
-	var entityTypes = []*dialogflowpb.EntityType{entityTypesElement}
-	var expectedResponse = &dialogflowpb.ListEntityTypesResponse{
-		NextPageToken: nextPageToken,
-		EntityTypes:   entityTypes,
-	}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var request = &dialogflowpb.ListEntityTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListEntityTypes(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.EntityTypes[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestEntityTypesListEntityTypesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var request = &dialogflowpb.ListEntityTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListEntityTypes(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestEntityTypesGetEntityType(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var enableFuzzyExtraction bool = true
-	var expectedResponse = &dialogflowpb.EntityType{
-		Name:                  name2,
-		DisplayName:           displayName,
-		EnableFuzzyExtraction: enableFuzzyExtraction,
-	}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.GetEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestEntityTypesGetEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.GetEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestEntityTypesCreateEntityType(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var enableFuzzyExtraction bool = true
-	var expectedResponse = &dialogflowpb.EntityType{
-		Name:                  name,
-		DisplayName:           displayName,
-		EnableFuzzyExtraction: enableFuzzyExtraction,
-	}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var entityType *dialogflowpb.EntityType = &dialogflowpb.EntityType{}
-	var request = &dialogflowpb.CreateEntityTypeRequest{
-		Parent:     formattedParent,
-		EntityType: entityType,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestEntityTypesCreateEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var entityType *dialogflowpb.EntityType = &dialogflowpb.EntityType{}
-	var request = &dialogflowpb.CreateEntityTypeRequest{
-		Parent:     formattedParent,
-		EntityType: entityType,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestEntityTypesUpdateEntityType(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var enableFuzzyExtraction bool = true
-	var expectedResponse = &dialogflowpb.EntityType{
-		Name:                  name,
-		DisplayName:           displayName,
-		EnableFuzzyExtraction: enableFuzzyExtraction,
-	}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], expectedResponse)
-
-	var entityType *dialogflowpb.EntityType = &dialogflowpb.EntityType{}
-	var request = &dialogflowpb.UpdateEntityTypeRequest{
-		EntityType: entityType,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestEntityTypesUpdateEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var entityType *dialogflowpb.EntityType = &dialogflowpb.EntityType{}
-	var request = &dialogflowpb.UpdateEntityTypeRequest{
-		EntityType: entityType,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestEntityTypesDeleteEntityType(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.DeleteEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestEntityTypesDeleteEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.DeleteEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestEntityTypesBatchUpdateEntityTypes(t *testing.T) {
-	var expectedResponse *dialogflowpb.BatchUpdateEntityTypesResponse = &dialogflowpb.BatchUpdateEntityTypesResponse{}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var request = &dialogflowpb.BatchUpdateEntityTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchUpdateEntityTypes(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestEntityTypesBatchUpdateEntityTypesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = nil
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var request = &dialogflowpb.BatchUpdateEntityTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchUpdateEntityTypes(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestEntityTypesBatchDeleteEntityTypes(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var entityTypeNames []string = nil
-	var request = &dialogflowpb.BatchDeleteEntityTypesRequest{
-		Parent:          formattedParent,
-		EntityTypeNames: entityTypeNames,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchDeleteEntityTypes(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestEntityTypesBatchDeleteEntityTypesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = nil
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var entityTypeNames []string = nil
-	var request = &dialogflowpb.BatchDeleteEntityTypesRequest{
-		Parent:          formattedParent,
-		EntityTypeNames: entityTypeNames,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchDeleteEntityTypes(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestEntityTypesBatchCreateEntities(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var entities []*dialogflowpb.EntityType_Entity = nil
-	var request = &dialogflowpb.BatchCreateEntitiesRequest{
-		Parent:   formattedParent,
-		Entities: entities,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchCreateEntities(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestEntityTypesBatchCreateEntitiesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = nil
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var entities []*dialogflowpb.EntityType_Entity = nil
-	var request = &dialogflowpb.BatchCreateEntitiesRequest{
-		Parent:   formattedParent,
-		Entities: entities,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchCreateEntities(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestEntityTypesBatchUpdateEntities(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var entities []*dialogflowpb.EntityType_Entity = nil
-	var request = &dialogflowpb.BatchUpdateEntitiesRequest{
-		Parent:   formattedParent,
-		Entities: entities,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchUpdateEntities(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestEntityTypesBatchUpdateEntitiesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = nil
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var entities []*dialogflowpb.EntityType_Entity = nil
-	var request = &dialogflowpb.BatchUpdateEntitiesRequest{
-		Parent:   formattedParent,
-		Entities: entities,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchUpdateEntities(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestEntityTypesBatchDeleteEntities(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockEntityTypes.err = nil
-	mockEntityTypes.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var entityValues []string = nil
-	var request = &dialogflowpb.BatchDeleteEntitiesRequest{
-		Parent:       formattedParent,
-		EntityValues: entityValues,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchDeleteEntities(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestEntityTypesBatchDeleteEntitiesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockEntityTypes.err = nil
-	mockEntityTypes.resps = append(mockEntityTypes.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/entityTypes/%s", "[PROJECT]", "[ENTITY_TYPE]")
-	var entityValues []string = nil
-	var request = &dialogflowpb.BatchDeleteEntitiesRequest{
-		Parent:       formattedParent,
-		EntityValues: entityValues,
-	}
-
-	c, err := NewEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchDeleteEntities(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestIntentsListIntents(t *testing.T) {
-	var nextPageToken string = ""
-	var intentsElement *dialogflowpb.Intent = &dialogflowpb.Intent{}
-	var intents = []*dialogflowpb.Intent{intentsElement}
-	var expectedResponse = &dialogflowpb.ListIntentsResponse{
-		NextPageToken: nextPageToken,
-		Intents:       intents,
-	}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	mockIntents.resps = append(mockIntents.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var request = &dialogflowpb.ListIntentsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListIntents(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Intents[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIntentsListIntentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var request = &dialogflowpb.ListIntentsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListIntents(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIntentsGetIntent(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var priority int32 = 1165461084
-	var isFallback bool = false
-	var mlDisabled bool = true
-	var action string = "action-1422950858"
-	var resetContexts bool = true
-	var rootFollowupIntentName string = "rootFollowupIntentName402253784"
-	var parentFollowupIntentName string = "parentFollowupIntentName-1131901680"
-	var expectedResponse = &dialogflowpb.Intent{
-		Name:                     name2,
-		DisplayName:              displayName,
-		Priority:                 priority,
-		IsFallback:               isFallback,
-		MlDisabled:               mlDisabled,
-		Action:                   action,
-		ResetContexts:            resetContexts,
-		RootFollowupIntentName:   rootFollowupIntentName,
-		ParentFollowupIntentName: parentFollowupIntentName,
-	}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	mockIntents.resps = append(mockIntents.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/intents/%s", "[PROJECT]", "[INTENT]")
-	var request = &dialogflowpb.GetIntentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIntent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIntentsGetIntentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/intents/%s", "[PROJECT]", "[INTENT]")
-	var request = &dialogflowpb.GetIntentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIntent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIntentsCreateIntent(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var priority int32 = 1165461084
-	var isFallback bool = false
-	var mlDisabled bool = true
-	var action string = "action-1422950858"
-	var resetContexts bool = true
-	var rootFollowupIntentName string = "rootFollowupIntentName402253784"
-	var parentFollowupIntentName string = "parentFollowupIntentName-1131901680"
-	var expectedResponse = &dialogflowpb.Intent{
-		Name:                     name,
-		DisplayName:              displayName,
-		Priority:                 priority,
-		IsFallback:               isFallback,
-		MlDisabled:               mlDisabled,
-		Action:                   action,
-		ResetContexts:            resetContexts,
-		RootFollowupIntentName:   rootFollowupIntentName,
-		ParentFollowupIntentName: parentFollowupIntentName,
-	}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	mockIntents.resps = append(mockIntents.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var intent *dialogflowpb.Intent = &dialogflowpb.Intent{}
-	var request = &dialogflowpb.CreateIntentRequest{
-		Parent: formattedParent,
-		Intent: intent,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateIntent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIntentsCreateIntentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var intent *dialogflowpb.Intent = &dialogflowpb.Intent{}
-	var request = &dialogflowpb.CreateIntentRequest{
-		Parent: formattedParent,
-		Intent: intent,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateIntent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIntentsUpdateIntent(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var priority int32 = 1165461084
-	var isFallback bool = false
-	var mlDisabled bool = true
-	var action string = "action-1422950858"
-	var resetContexts bool = true
-	var rootFollowupIntentName string = "rootFollowupIntentName402253784"
-	var parentFollowupIntentName string = "parentFollowupIntentName-1131901680"
-	var expectedResponse = &dialogflowpb.Intent{
-		Name:                     name,
-		DisplayName:              displayName,
-		Priority:                 priority,
-		IsFallback:               isFallback,
-		MlDisabled:               mlDisabled,
-		Action:                   action,
-		ResetContexts:            resetContexts,
-		RootFollowupIntentName:   rootFollowupIntentName,
-		ParentFollowupIntentName: parentFollowupIntentName,
-	}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	mockIntents.resps = append(mockIntents.resps[:0], expectedResponse)
-
-	var intent *dialogflowpb.Intent = &dialogflowpb.Intent{}
-	var languageCode string = "languageCode-412800396"
-	var request = &dialogflowpb.UpdateIntentRequest{
-		Intent:       intent,
-		LanguageCode: languageCode,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateIntent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIntentsUpdateIntentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = gstatus.Error(errCode, "test error")
-
-	var intent *dialogflowpb.Intent = &dialogflowpb.Intent{}
-	var languageCode string = "languageCode-412800396"
-	var request = &dialogflowpb.UpdateIntentRequest{
-		Intent:       intent,
-		LanguageCode: languageCode,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateIntent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIntentsDeleteIntent(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	mockIntents.resps = append(mockIntents.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/intents/%s", "[PROJECT]", "[INTENT]")
-	var request = &dialogflowpb.DeleteIntentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteIntent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestIntentsDeleteIntentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/intents/%s", "[PROJECT]", "[INTENT]")
-	var request = &dialogflowpb.DeleteIntentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteIntent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestIntentsBatchUpdateIntents(t *testing.T) {
-	var expectedResponse *dialogflowpb.BatchUpdateIntentsResponse = &dialogflowpb.BatchUpdateIntentsResponse{}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockIntents.resps = append(mockIntents.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var languageCode string = "languageCode-412800396"
-	var request = &dialogflowpb.BatchUpdateIntentsRequest{
-		Parent:       formattedParent,
-		LanguageCode: languageCode,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchUpdateIntents(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIntentsBatchUpdateIntentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = nil
-	mockIntents.resps = append(mockIntents.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var languageCode string = "languageCode-412800396"
-	var request = &dialogflowpb.BatchUpdateIntentsRequest{
-		Parent:       formattedParent,
-		LanguageCode: languageCode,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchUpdateIntents(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIntentsBatchDeleteIntents(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockIntents.err = nil
-	mockIntents.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockIntents.resps = append(mockIntents.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var intents []*dialogflowpb.Intent = nil
-	var request = &dialogflowpb.BatchDeleteIntentsRequest{
-		Parent:  formattedParent,
-		Intents: intents,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchDeleteIntents(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIntents.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestIntentsBatchDeleteIntentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIntents.err = nil
-	mockIntents.resps = append(mockIntents.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent", "[PROJECT]")
-	var intents []*dialogflowpb.Intent = nil
-	var request = &dialogflowpb.BatchDeleteIntentsRequest{
-		Parent:  formattedParent,
-		Intents: intents,
-	}
-
-	c, err := NewIntentsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchDeleteIntents(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSessionEntityTypesListSessionEntityTypes(t *testing.T) {
-	var nextPageToken string = ""
-	var sessionEntityTypesElement *dialogflowpb.SessionEntityType = &dialogflowpb.SessionEntityType{}
-	var sessionEntityTypes = []*dialogflowpb.SessionEntityType{sessionEntityTypesElement}
-	var expectedResponse = &dialogflowpb.ListSessionEntityTypesResponse{
-		NextPageToken:      nextPageToken,
-		SessionEntityTypes: sessionEntityTypes,
-	}
-
-	mockSessionEntityTypes.err = nil
-	mockSessionEntityTypes.reqs = nil
-
-	mockSessionEntityTypes.resps = append(mockSessionEntityTypes.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var request = &dialogflowpb.ListSessionEntityTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSessionEntityTypes(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessionEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.SessionEntityTypes[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSessionEntityTypesListSessionEntityTypesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessionEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var request = &dialogflowpb.ListSessionEntityTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSessionEntityTypes(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSessionEntityTypesGetSessionEntityType(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &dialogflowpb.SessionEntityType{
-		Name: name2,
-	}
-
-	mockSessionEntityTypes.err = nil
-	mockSessionEntityTypes.reqs = nil
-
-	mockSessionEntityTypes.resps = append(mockSessionEntityTypes.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/entityTypes/%s", "[PROJECT]", "[SESSION]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.GetSessionEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSessionEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessionEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSessionEntityTypesGetSessionEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessionEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/entityTypes/%s", "[PROJECT]", "[SESSION]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.GetSessionEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSessionEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSessionEntityTypesCreateSessionEntityType(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &dialogflowpb.SessionEntityType{
-		Name: name,
-	}
-
-	mockSessionEntityTypes.err = nil
-	mockSessionEntityTypes.reqs = nil
-
-	mockSessionEntityTypes.resps = append(mockSessionEntityTypes.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var sessionEntityType *dialogflowpb.SessionEntityType = &dialogflowpb.SessionEntityType{}
-	var request = &dialogflowpb.CreateSessionEntityTypeRequest{
-		Parent:            formattedParent,
-		SessionEntityType: sessionEntityType,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSessionEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessionEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSessionEntityTypesCreateSessionEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessionEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var sessionEntityType *dialogflowpb.SessionEntityType = &dialogflowpb.SessionEntityType{}
-	var request = &dialogflowpb.CreateSessionEntityTypeRequest{
-		Parent:            formattedParent,
-		SessionEntityType: sessionEntityType,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSessionEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSessionEntityTypesUpdateSessionEntityType(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &dialogflowpb.SessionEntityType{
-		Name: name,
-	}
-
-	mockSessionEntityTypes.err = nil
-	mockSessionEntityTypes.reqs = nil
-
-	mockSessionEntityTypes.resps = append(mockSessionEntityTypes.resps[:0], expectedResponse)
-
-	var sessionEntityType *dialogflowpb.SessionEntityType = &dialogflowpb.SessionEntityType{}
-	var request = &dialogflowpb.UpdateSessionEntityTypeRequest{
-		SessionEntityType: sessionEntityType,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSessionEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessionEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSessionEntityTypesUpdateSessionEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessionEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var sessionEntityType *dialogflowpb.SessionEntityType = &dialogflowpb.SessionEntityType{}
-	var request = &dialogflowpb.UpdateSessionEntityTypeRequest{
-		SessionEntityType: sessionEntityType,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSessionEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSessionEntityTypesDeleteSessionEntityType(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSessionEntityTypes.err = nil
-	mockSessionEntityTypes.reqs = nil
-
-	mockSessionEntityTypes.resps = append(mockSessionEntityTypes.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/entityTypes/%s", "[PROJECT]", "[SESSION]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.DeleteSessionEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSessionEntityType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessionEntityTypes.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSessionEntityTypesDeleteSessionEntityTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessionEntityTypes.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/agent/sessions/%s/entityTypes/%s", "[PROJECT]", "[SESSION]", "[ENTITY_TYPE]")
-	var request = &dialogflowpb.DeleteSessionEntityTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewSessionEntityTypesClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSessionEntityType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSessionsDetectIntent(t *testing.T) {
-	var responseId string = "responseId1847552473"
-	var outputAudio []byte = []byte("24")
-	var expectedResponse = &dialogflowpb.DetectIntentResponse{
-		ResponseId:  responseId,
-		OutputAudio: outputAudio,
-	}
-
-	mockSessions.err = nil
-	mockSessions.reqs = nil
-
-	mockSessions.resps = append(mockSessions.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var queryInput *dialogflowpb.QueryInput = &dialogflowpb.QueryInput{}
-	var request = &dialogflowpb.DetectIntentRequest{
-		Session:    formattedSession,
-		QueryInput: queryInput,
-	}
-
-	c, err := NewSessionsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DetectIntent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessions.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSessionsDetectIntentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessions.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/agent/sessions/%s", "[PROJECT]", "[SESSION]")
-	var queryInput *dialogflowpb.QueryInput = &dialogflowpb.QueryInput{}
-	var request = &dialogflowpb.DetectIntentRequest{
-		Session:    formattedSession,
-		QueryInput: queryInput,
-	}
-
-	c, err := NewSessionsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DetectIntent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSessionsStreamingDetectIntent(t *testing.T) {
-	var responseId string = "responseId1847552473"
-	var outputAudio []byte = []byte("24")
-	var expectedResponse = &dialogflowpb.StreamingDetectIntentResponse{
-		ResponseId:  responseId,
-		OutputAudio: outputAudio,
-	}
-
-	mockSessions.err = nil
-	mockSessions.reqs = nil
-
-	mockSessions.resps = append(mockSessions.resps[:0], expectedResponse)
-
-	var session string = "session1984987798"
-	var queryInput *dialogflowpb.QueryInput = &dialogflowpb.QueryInput{}
-	var request = &dialogflowpb.StreamingDetectIntentRequest{
-		Session:    session,
-		QueryInput: queryInput,
-	}
-
-	c, err := NewSessionsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingDetectIntent(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSessions.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSessionsStreamingDetectIntentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSessions.err = gstatus.Error(errCode, "test error")
-
-	var session string = "session1984987798"
-	var queryInput *dialogflowpb.QueryInput = &dialogflowpb.QueryInput{}
-	var request = &dialogflowpb.StreamingDetectIntentRequest{
-		Session:    session,
-		QueryInput: queryInput,
-	}
-
-	c, err := NewSessionsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingDetectIntent(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/dlp/apiv2/mock_test.go b/dlp/apiv2/mock_test.go
deleted file mode 100644
index 8cc8e0446a8..00000000000
--- a/dlp/apiv2/mock_test.go
+++ /dev/null
@@ -1,2262 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package dlp
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	dlppb "google.golang.org/genproto/googleapis/privacy/dlp/v2"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockDlpServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	dlppb.DlpServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockDlpServer) InspectContent(ctx context.Context, req *dlppb.InspectContentRequest) (*dlppb.InspectContentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.InspectContentResponse), nil
-}
-
-func (s *mockDlpServer) RedactImage(ctx context.Context, req *dlppb.RedactImageRequest) (*dlppb.RedactImageResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.RedactImageResponse), nil
-}
-
-func (s *mockDlpServer) DeidentifyContent(ctx context.Context, req *dlppb.DeidentifyContentRequest) (*dlppb.DeidentifyContentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.DeidentifyContentResponse), nil
-}
-
-func (s *mockDlpServer) ReidentifyContent(ctx context.Context, req *dlppb.ReidentifyContentRequest) (*dlppb.ReidentifyContentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ReidentifyContentResponse), nil
-}
-
-func (s *mockDlpServer) ListInfoTypes(ctx context.Context, req *dlppb.ListInfoTypesRequest) (*dlppb.ListInfoTypesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ListInfoTypesResponse), nil
-}
-
-func (s *mockDlpServer) CreateInspectTemplate(ctx context.Context, req *dlppb.CreateInspectTemplateRequest) (*dlppb.InspectTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.InspectTemplate), nil
-}
-
-func (s *mockDlpServer) UpdateInspectTemplate(ctx context.Context, req *dlppb.UpdateInspectTemplateRequest) (*dlppb.InspectTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.InspectTemplate), nil
-}
-
-func (s *mockDlpServer) GetInspectTemplate(ctx context.Context, req *dlppb.GetInspectTemplateRequest) (*dlppb.InspectTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.InspectTemplate), nil
-}
-
-func (s *mockDlpServer) ListInspectTemplates(ctx context.Context, req *dlppb.ListInspectTemplatesRequest) (*dlppb.ListInspectTemplatesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ListInspectTemplatesResponse), nil
-}
-
-func (s *mockDlpServer) DeleteInspectTemplate(ctx context.Context, req *dlppb.DeleteInspectTemplateRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDlpServer) CreateDeidentifyTemplate(ctx context.Context, req *dlppb.CreateDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.DeidentifyTemplate), nil
-}
-
-func (s *mockDlpServer) UpdateDeidentifyTemplate(ctx context.Context, req *dlppb.UpdateDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.DeidentifyTemplate), nil
-}
-
-func (s *mockDlpServer) GetDeidentifyTemplate(ctx context.Context, req *dlppb.GetDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.DeidentifyTemplate), nil
-}
-
-func (s *mockDlpServer) ListDeidentifyTemplates(ctx context.Context, req *dlppb.ListDeidentifyTemplatesRequest) (*dlppb.ListDeidentifyTemplatesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ListDeidentifyTemplatesResponse), nil
-}
-
-func (s *mockDlpServer) DeleteDeidentifyTemplate(ctx context.Context, req *dlppb.DeleteDeidentifyTemplateRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDlpServer) CreateJobTrigger(ctx context.Context, req *dlppb.CreateJobTriggerRequest) (*dlppb.JobTrigger, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.JobTrigger), nil
-}
-
-func (s *mockDlpServer) UpdateJobTrigger(ctx context.Context, req *dlppb.UpdateJobTriggerRequest) (*dlppb.JobTrigger, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.JobTrigger), nil
-}
-
-func (s *mockDlpServer) GetJobTrigger(ctx context.Context, req *dlppb.GetJobTriggerRequest) (*dlppb.JobTrigger, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.JobTrigger), nil
-}
-
-func (s *mockDlpServer) ListJobTriggers(ctx context.Context, req *dlppb.ListJobTriggersRequest) (*dlppb.ListJobTriggersResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ListJobTriggersResponse), nil
-}
-
-func (s *mockDlpServer) DeleteJobTrigger(ctx context.Context, req *dlppb.DeleteJobTriggerRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDlpServer) CreateDlpJob(ctx context.Context, req *dlppb.CreateDlpJobRequest) (*dlppb.DlpJob, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.DlpJob), nil
-}
-
-func (s *mockDlpServer) ListDlpJobs(ctx context.Context, req *dlppb.ListDlpJobsRequest) (*dlppb.ListDlpJobsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ListDlpJobsResponse), nil
-}
-
-func (s *mockDlpServer) GetDlpJob(ctx context.Context, req *dlppb.GetDlpJobRequest) (*dlppb.DlpJob, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.DlpJob), nil
-}
-
-func (s *mockDlpServer) DeleteDlpJob(ctx context.Context, req *dlppb.DeleteDlpJobRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDlpServer) CancelDlpJob(ctx context.Context, req *dlppb.CancelDlpJobRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDlpServer) CreateStoredInfoType(ctx context.Context, req *dlppb.CreateStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.StoredInfoType), nil
-}
-
-func (s *mockDlpServer) UpdateStoredInfoType(ctx context.Context, req *dlppb.UpdateStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.StoredInfoType), nil
-}
-
-func (s *mockDlpServer) GetStoredInfoType(ctx context.Context, req *dlppb.GetStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.StoredInfoType), nil
-}
-
-func (s *mockDlpServer) ListStoredInfoTypes(ctx context.Context, req *dlppb.ListStoredInfoTypesRequest) (*dlppb.ListStoredInfoTypesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*dlppb.ListStoredInfoTypesResponse), nil
-}
-
-func (s *mockDlpServer) DeleteStoredInfoType(ctx context.Context, req *dlppb.DeleteStoredInfoTypeRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockDlp mockDlpServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	dlppb.RegisterDlpServiceServer(serv, &mockDlp)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestDlpServiceInspectContent(t *testing.T) {
-	var expectedResponse *dlppb.InspectContentResponse = &dlppb.InspectContentResponse{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.InspectContentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.InspectContent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceInspectContentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.InspectContentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.InspectContent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceRedactImage(t *testing.T) {
-	var redactedImage []byte = []byte("28")
-	var extractedText string = "extractedText998260012"
-	var expectedResponse = &dlppb.RedactImageResponse{
-		RedactedImage: redactedImage,
-		ExtractedText: extractedText,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.RedactImageRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RedactImage(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceRedactImageError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.RedactImageRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RedactImage(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceDeidentifyContent(t *testing.T) {
-	var expectedResponse *dlppb.DeidentifyContentResponse = &dlppb.DeidentifyContentResponse{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.DeidentifyContentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeidentifyContent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceDeidentifyContentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.DeidentifyContentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeidentifyContent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceReidentifyContent(t *testing.T) {
-	var expectedResponse *dlppb.ReidentifyContentResponse = &dlppb.ReidentifyContentResponse{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.ReidentifyContentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ReidentifyContent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceReidentifyContentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.ReidentifyContentRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ReidentifyContent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceListInfoTypes(t *testing.T) {
-	var expectedResponse *dlppb.ListInfoTypesResponse = &dlppb.ListInfoTypesResponse{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInfoTypes(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceListInfoTypesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInfoTypes(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceCreateInspectTemplate(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.InspectTemplate{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.CreateInspectTemplateRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateInspectTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceCreateInspectTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.CreateInspectTemplateRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateInspectTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceUpdateInspectTemplate(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.InspectTemplate{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
-	var request = &dlppb.UpdateInspectTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateInspectTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceUpdateInspectTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
-	var request = &dlppb.UpdateInspectTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateInspectTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceGetInspectTemplate(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.InspectTemplate{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInspectTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceGetInspectTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInspectTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceListInspectTemplates(t *testing.T) {
-	var nextPageToken string = ""
-	var inspectTemplatesElement *dlppb.InspectTemplate = &dlppb.InspectTemplate{}
-	var inspectTemplates = []*dlppb.InspectTemplate{inspectTemplatesElement}
-	var expectedResponse = &dlppb.ListInspectTemplatesResponse{
-		NextPageToken:    nextPageToken,
-		InspectTemplates: inspectTemplates,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.ListInspectTemplatesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInspectTemplates(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.InspectTemplates[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceListInspectTemplatesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.ListInspectTemplatesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInspectTemplates(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceDeleteInspectTemplate(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
-	var request = &dlppb.DeleteInspectTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteInspectTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDlpServiceDeleteInspectTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
-	var request = &dlppb.DeleteInspectTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteInspectTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDlpServiceCreateDeidentifyTemplate(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.DeidentifyTemplate{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.CreateDeidentifyTemplateRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDeidentifyTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceCreateDeidentifyTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.CreateDeidentifyTemplateRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDeidentifyTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceUpdateDeidentifyTemplate(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.DeidentifyTemplate{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
-	var request = &dlppb.UpdateDeidentifyTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDeidentifyTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceUpdateDeidentifyTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
-	var request = &dlppb.UpdateDeidentifyTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDeidentifyTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceGetDeidentifyTemplate(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.DeidentifyTemplate{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
-	var request = &dlppb.GetDeidentifyTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDeidentifyTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceGetDeidentifyTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
-	var request = &dlppb.GetDeidentifyTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDeidentifyTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceListDeidentifyTemplates(t *testing.T) {
-	var nextPageToken string = ""
-	var deidentifyTemplatesElement *dlppb.DeidentifyTemplate = &dlppb.DeidentifyTemplate{}
-	var deidentifyTemplates = []*dlppb.DeidentifyTemplate{deidentifyTemplatesElement}
-	var expectedResponse = &dlppb.ListDeidentifyTemplatesResponse{
-		NextPageToken:       nextPageToken,
-		DeidentifyTemplates: deidentifyTemplates,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.ListDeidentifyTemplatesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.DeidentifyTemplates[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceListDeidentifyTemplatesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.ListDeidentifyTemplatesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceDeleteDeidentifyTemplate(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
-	var request = &dlppb.DeleteDeidentifyTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDeidentifyTemplate(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDlpServiceDeleteDeidentifyTemplateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
-	var request = &dlppb.DeleteDeidentifyTemplateRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDeidentifyTemplate(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDlpServiceCreateDlpJob(t *testing.T) {
-	var name string = "name3373707"
-	var jobTriggerName string = "jobTriggerName1819490804"
-	var expectedResponse = &dlppb.DlpJob{
-		Name:           name,
-		JobTriggerName: jobTriggerName,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.CreateDlpJobRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDlpJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceCreateDlpJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.CreateDlpJobRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDlpJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceListDlpJobs(t *testing.T) {
-	var nextPageToken string = ""
-	var jobsElement *dlppb.DlpJob = &dlppb.DlpJob{}
-	var jobs = []*dlppb.DlpJob{jobsElement}
-	var expectedResponse = &dlppb.ListDlpJobsResponse{
-		NextPageToken: nextPageToken,
-		Jobs:          jobs,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.ListDlpJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDlpJobs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Jobs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceListDlpJobsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.ListDlpJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDlpJobs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceGetDlpJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var jobTriggerName string = "jobTriggerName1819490804"
-	var expectedResponse = &dlppb.DlpJob{
-		Name:           name2,
-		JobTriggerName: jobTriggerName,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
-	var request = &dlppb.GetDlpJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDlpJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceGetDlpJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
-	var request = &dlppb.GetDlpJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDlpJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceDeleteDlpJob(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
-	var request = &dlppb.DeleteDlpJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDlpJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDlpServiceDeleteDlpJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
-	var request = &dlppb.DeleteDlpJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDlpJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDlpServiceCancelDlpJob(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
-	var request = &dlppb.CancelDlpJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.CancelDlpJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDlpServiceCancelDlpJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
-	var request = &dlppb.CancelDlpJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.CancelDlpJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDlpServiceListJobTriggers(t *testing.T) {
-	var nextPageToken string = ""
-	var jobTriggersElement *dlppb.JobTrigger = &dlppb.JobTrigger{}
-	var jobTriggers = []*dlppb.JobTrigger{jobTriggersElement}
-	var expectedResponse = &dlppb.ListJobTriggersResponse{
-		NextPageToken: nextPageToken,
-		JobTriggers:   jobTriggers,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.ListJobTriggersRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobTriggers(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.JobTriggers[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceListJobTriggersError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.ListJobTriggersRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobTriggers(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceGetJobTrigger(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.JobTrigger{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
-	var request = &dlppb.GetJobTriggerRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJobTrigger(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceGetJobTriggerError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
-	var request = &dlppb.GetJobTriggerRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJobTrigger(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceDeleteJobTrigger(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var name string = "name3373707"
-	var request = &dlppb.DeleteJobTriggerRequest{
-		Name: name,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJobTrigger(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDlpServiceDeleteJobTriggerError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var name string = "name3373707"
-	var request = &dlppb.DeleteJobTriggerRequest{
-		Name: name,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJobTrigger(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDlpServiceUpdateJobTrigger(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.JobTrigger{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
-	var request = &dlppb.UpdateJobTriggerRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJobTrigger(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceUpdateJobTriggerError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
-	var request = &dlppb.UpdateJobTriggerRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJobTrigger(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceCreateJobTrigger(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &dlppb.JobTrigger{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.CreateJobTriggerRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateJobTrigger(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceCreateJobTriggerError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &dlppb.CreateJobTriggerRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateJobTrigger(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceCreateStoredInfoType(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &dlppb.StoredInfoType{
-		Name: name,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.CreateStoredInfoTypeRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateStoredInfoType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceCreateStoredInfoTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.CreateStoredInfoTypeRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateStoredInfoType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceUpdateStoredInfoType(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &dlppb.StoredInfoType{
-		Name: name2,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
-	var request = &dlppb.UpdateStoredInfoTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateStoredInfoType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceUpdateStoredInfoTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
-	var request = &dlppb.UpdateStoredInfoTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateStoredInfoType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceGetStoredInfoType(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &dlppb.StoredInfoType{
-		Name: name2,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
-	var request = &dlppb.GetStoredInfoTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetStoredInfoType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceGetStoredInfoTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
-	var request = &dlppb.GetStoredInfoTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetStoredInfoType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceListStoredInfoTypes(t *testing.T) {
-	var nextPageToken string = ""
-	var storedInfoTypesElement *dlppb.StoredInfoType = &dlppb.StoredInfoType{}
-	var storedInfoTypes = []*dlppb.StoredInfoType{storedInfoTypesElement}
-	var expectedResponse = &dlppb.ListStoredInfoTypesResponse{
-		NextPageToken:   nextPageToken,
-		StoredInfoTypes: storedInfoTypes,
-	}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.ListStoredInfoTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListStoredInfoTypes(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.StoredInfoTypes[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDlpServiceListStoredInfoTypesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &dlppb.ListStoredInfoTypesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListStoredInfoTypes(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDlpServiceDeleteStoredInfoType(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDlp.err = nil
-	mockDlp.reqs = nil
-
-	mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
-	var request = &dlppb.DeleteStoredInfoTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteStoredInfoType(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDlpServiceDeleteStoredInfoTypeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDlp.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
-	var request = &dlppb.DeleteStoredInfoTypeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteStoredInfoType(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
diff --git a/errorreporting/apiv1beta1/mock_test.go b/errorreporting/apiv1beta1/mock_test.go
deleted file mode 100644
index 1f1ba4506fd..00000000000
--- a/errorreporting/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,582 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package errorreporting
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	clouderrorreportingpb "google.golang.org/genproto/googleapis/devtools/clouderrorreporting/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockReportErrorsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	clouderrorreportingpb.ReportErrorsServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockReportErrorsServer) ReportErrorEvent(ctx context.Context, req *clouderrorreportingpb.ReportErrorEventRequest) (*clouderrorreportingpb.ReportErrorEventResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*clouderrorreportingpb.ReportErrorEventResponse), nil
-}
-
-type mockErrorStatsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	clouderrorreportingpb.ErrorStatsServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockErrorStatsServer) ListGroupStats(ctx context.Context, req *clouderrorreportingpb.ListGroupStatsRequest) (*clouderrorreportingpb.ListGroupStatsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*clouderrorreportingpb.ListGroupStatsResponse), nil
-}
-
-func (s *mockErrorStatsServer) ListEvents(ctx context.Context, req *clouderrorreportingpb.ListEventsRequest) (*clouderrorreportingpb.ListEventsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*clouderrorreportingpb.ListEventsResponse), nil
-}
-
-func (s *mockErrorStatsServer) DeleteEvents(ctx context.Context, req *clouderrorreportingpb.DeleteEventsRequest) (*clouderrorreportingpb.DeleteEventsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*clouderrorreportingpb.DeleteEventsResponse), nil
-}
-
-type mockErrorGroupServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	clouderrorreportingpb.ErrorGroupServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockErrorGroupServer) GetGroup(ctx context.Context, req *clouderrorreportingpb.GetGroupRequest) (*clouderrorreportingpb.ErrorGroup, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*clouderrorreportingpb.ErrorGroup), nil
-}
-
-func (s *mockErrorGroupServer) UpdateGroup(ctx context.Context, req *clouderrorreportingpb.UpdateGroupRequest) (*clouderrorreportingpb.ErrorGroup, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*clouderrorreportingpb.ErrorGroup), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockReportErrors mockReportErrorsServer
-	mockErrorStats   mockErrorStatsServer
-	mockErrorGroup   mockErrorGroupServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	clouderrorreportingpb.RegisterReportErrorsServiceServer(serv, &mockReportErrors)
-	clouderrorreportingpb.RegisterErrorStatsServiceServer(serv, &mockErrorStats)
-	clouderrorreportingpb.RegisterErrorGroupServiceServer(serv, &mockErrorGroup)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestReportErrorsServiceReportErrorEvent(t *testing.T) {
-	var expectedResponse *clouderrorreportingpb.ReportErrorEventResponse = &clouderrorreportingpb.ReportErrorEventResponse{}
-
-	mockReportErrors.err = nil
-	mockReportErrors.reqs = nil
-
-	mockReportErrors.resps = append(mockReportErrors.resps[:0], expectedResponse)
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var event *clouderrorreportingpb.ReportedErrorEvent = &clouderrorreportingpb.ReportedErrorEvent{}
-	var request = &clouderrorreportingpb.ReportErrorEventRequest{
-		ProjectName: formattedProjectName,
-		Event:       event,
-	}
-
-	c, err := NewReportErrorsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ReportErrorEvent(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockReportErrors.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestReportErrorsServiceReportErrorEventError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockReportErrors.err = gstatus.Error(errCode, "test error")
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var event *clouderrorreportingpb.ReportedErrorEvent = &clouderrorreportingpb.ReportedErrorEvent{}
-	var request = &clouderrorreportingpb.ReportErrorEventRequest{
-		ProjectName: formattedProjectName,
-		Event:       event,
-	}
-
-	c, err := NewReportErrorsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ReportErrorEvent(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestErrorStatsServiceListGroupStats(t *testing.T) {
-	var nextPageToken string = ""
-	var errorGroupStatsElement *clouderrorreportingpb.ErrorGroupStats = &clouderrorreportingpb.ErrorGroupStats{}
-	var errorGroupStats = []*clouderrorreportingpb.ErrorGroupStats{errorGroupStatsElement}
-	var expectedResponse = &clouderrorreportingpb.ListGroupStatsResponse{
-		NextPageToken:   nextPageToken,
-		ErrorGroupStats: errorGroupStats,
-	}
-
-	mockErrorStats.err = nil
-	mockErrorStats.reqs = nil
-
-	mockErrorStats.resps = append(mockErrorStats.resps[:0], expectedResponse)
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &clouderrorreportingpb.ListGroupStatsRequest{
-		ProjectName: formattedProjectName,
-	}
-
-	c, err := NewErrorStatsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGroupStats(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockErrorStats.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ErrorGroupStats[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestErrorStatsServiceListGroupStatsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockErrorStats.err = gstatus.Error(errCode, "test error")
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &clouderrorreportingpb.ListGroupStatsRequest{
-		ProjectName: formattedProjectName,
-	}
-
-	c, err := NewErrorStatsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGroupStats(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestErrorStatsServiceListEvents(t *testing.T) {
-	var nextPageToken string = ""
-	var errorEventsElement *clouderrorreportingpb.ErrorEvent = &clouderrorreportingpb.ErrorEvent{}
-	var errorEvents = []*clouderrorreportingpb.ErrorEvent{errorEventsElement}
-	var expectedResponse = &clouderrorreportingpb.ListEventsResponse{
-		NextPageToken: nextPageToken,
-		ErrorEvents:   errorEvents,
-	}
-
-	mockErrorStats.err = nil
-	mockErrorStats.reqs = nil
-
-	mockErrorStats.resps = append(mockErrorStats.resps[:0], expectedResponse)
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var groupId string = "groupId506361563"
-	var request = &clouderrorreportingpb.ListEventsRequest{
-		ProjectName: formattedProjectName,
-		GroupId:     groupId,
-	}
-
-	c, err := NewErrorStatsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListEvents(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockErrorStats.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ErrorEvents[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestErrorStatsServiceListEventsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockErrorStats.err = gstatus.Error(errCode, "test error")
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var groupId string = "groupId506361563"
-	var request = &clouderrorreportingpb.ListEventsRequest{
-		ProjectName: formattedProjectName,
-		GroupId:     groupId,
-	}
-
-	c, err := NewErrorStatsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListEvents(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestErrorStatsServiceDeleteEvents(t *testing.T) {
-	var expectedResponse *clouderrorreportingpb.DeleteEventsResponse = &clouderrorreportingpb.DeleteEventsResponse{}
-
-	mockErrorStats.err = nil
-	mockErrorStats.reqs = nil
-
-	mockErrorStats.resps = append(mockErrorStats.resps[:0], expectedResponse)
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &clouderrorreportingpb.DeleteEventsRequest{
-		ProjectName: formattedProjectName,
-	}
-
-	c, err := NewErrorStatsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteEvents(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockErrorStats.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestErrorStatsServiceDeleteEventsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockErrorStats.err = gstatus.Error(errCode, "test error")
-
-	var formattedProjectName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &clouderrorreportingpb.DeleteEventsRequest{
-		ProjectName: formattedProjectName,
-	}
-
-	c, err := NewErrorStatsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteEvents(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestErrorGroupServiceGetGroup(t *testing.T) {
-	var name string = "name3373707"
-	var groupId string = "groupId506361563"
-	var expectedResponse = &clouderrorreportingpb.ErrorGroup{
-		Name:    name,
-		GroupId: groupId,
-	}
-
-	mockErrorGroup.err = nil
-	mockErrorGroup.reqs = nil
-
-	mockErrorGroup.resps = append(mockErrorGroup.resps[:0], expectedResponse)
-
-	var formattedGroupName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &clouderrorreportingpb.GetGroupRequest{
-		GroupName: formattedGroupName,
-	}
-
-	c, err := NewErrorGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetGroup(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockErrorGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestErrorGroupServiceGetGroupError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockErrorGroup.err = gstatus.Error(errCode, "test error")
-
-	var formattedGroupName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &clouderrorreportingpb.GetGroupRequest{
-		GroupName: formattedGroupName,
-	}
-
-	c, err := NewErrorGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetGroup(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestErrorGroupServiceUpdateGroup(t *testing.T) {
-	var name string = "name3373707"
-	var groupId string = "groupId506361563"
-	var expectedResponse = &clouderrorreportingpb.ErrorGroup{
-		Name:    name,
-		GroupId: groupId,
-	}
-
-	mockErrorGroup.err = nil
-	mockErrorGroup.reqs = nil
-
-	mockErrorGroup.resps = append(mockErrorGroup.resps[:0], expectedResponse)
-
-	var group *clouderrorreportingpb.ErrorGroup = &clouderrorreportingpb.ErrorGroup{}
-	var request = &clouderrorreportingpb.UpdateGroupRequest{
-		Group: group,
-	}
-
-	c, err := NewErrorGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateGroup(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockErrorGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestErrorGroupServiceUpdateGroupError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockErrorGroup.err = gstatus.Error(errCode, "test error")
-
-	var group *clouderrorreportingpb.ErrorGroup = &clouderrorreportingpb.ErrorGroup{}
-	var request = &clouderrorreportingpb.UpdateGroupRequest{
-		Group: group,
-	}
-
-	c, err := NewErrorGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateGroup(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/firestore/apiv1/admin/mock_test.go b/firestore/apiv1/admin/mock_test.go
deleted file mode 100644
index 80e5b0291d6..00000000000
--- a/firestore/apiv1/admin/mock_test.go
+++ /dev/null
@@ -1,782 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package apiv1
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	adminpb "google.golang.org/genproto/googleapis/firestore/admin/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockFirestoreAdminServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	adminpb.FirestoreAdminServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockFirestoreAdminServer) CreateIndex(ctx context.Context, req *adminpb.CreateIndexRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockFirestoreAdminServer) ListIndexes(ctx context.Context, req *adminpb.ListIndexesRequest) (*adminpb.ListIndexesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ListIndexesResponse), nil
-}
-
-func (s *mockFirestoreAdminServer) GetIndex(ctx context.Context, req *adminpb.GetIndexRequest) (*adminpb.Index, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Index), nil
-}
-
-func (s *mockFirestoreAdminServer) DeleteIndex(ctx context.Context, req *adminpb.DeleteIndexRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockFirestoreAdminServer) GetField(ctx context.Context, req *adminpb.GetFieldRequest) (*adminpb.Field, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Field), nil
-}
-
-func (s *mockFirestoreAdminServer) UpdateField(ctx context.Context, req *adminpb.UpdateFieldRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockFirestoreAdminServer) ListFields(ctx context.Context, req *adminpb.ListFieldsRequest) (*adminpb.ListFieldsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ListFieldsResponse), nil
-}
-
-func (s *mockFirestoreAdminServer) ExportDocuments(ctx context.Context, req *adminpb.ExportDocumentsRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockFirestoreAdminServer) ImportDocuments(ctx context.Context, req *adminpb.ImportDocumentsRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockFirestoreAdmin mockFirestoreAdminServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	adminpb.RegisterFirestoreAdminServer(serv, &mockFirestoreAdmin)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestFirestoreAdminCreateIndex(t *testing.T) {
-	var expectedResponse = &adminpb.Index{}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]")
-	var index *adminpb.Index = &adminpb.Index{}
-	var request = &adminpb.CreateIndexRequest{
-		Parent: formattedParent,
-		Index:  index,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateIndex(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminCreateIndexError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]")
-	var index *adminpb.Index = &adminpb.Index{}
-	var request = &adminpb.CreateIndexRequest{
-		Parent: formattedParent,
-		Index:  index,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateIndex(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminListIndexes(t *testing.T) {
-	var nextPageToken string = ""
-	var indexesElement *adminpb.Index = &adminpb.Index{}
-	var indexes = []*adminpb.Index{indexesElement}
-	var expectedResponse = &adminpb.ListIndexesResponse{
-		NextPageToken: nextPageToken,
-		Indexes:       indexes,
-	}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]")
-	var request = &adminpb.ListIndexesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListIndexes(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Indexes[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminListIndexesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]")
-	var request = &adminpb.ListIndexesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListIndexes(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminGetIndex(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &adminpb.Index{
-		Name: name2,
-	}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s/indexes/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]", "[INDEX_ID]")
-	var request = &adminpb.GetIndexRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIndex(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminGetIndexError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s/indexes/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]", "[INDEX_ID]")
-	var request = &adminpb.GetIndexRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIndex(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminDeleteIndex(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s/indexes/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]", "[INDEX_ID]")
-	var request = &adminpb.DeleteIndexRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteIndex(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestFirestoreAdminDeleteIndexError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s/indexes/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]", "[INDEX_ID]")
-	var request = &adminpb.DeleteIndexRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteIndex(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestFirestoreAdminImportDocuments(t *testing.T) {
-	var expectedResponse = &emptypb.Empty{}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &adminpb.ImportDocumentsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportDocuments(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-}
-
-func TestFirestoreAdminImportDocumentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &adminpb.ImportDocumentsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportDocuments(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminExportDocuments(t *testing.T) {
-	var expectedResponse = &adminpb.ExportDocumentsResponse{}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &adminpb.ExportDocumentsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportDocuments(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminExportDocumentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &adminpb.ExportDocumentsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ExportDocuments(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminGetField(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &adminpb.Field{
-		Name: name2,
-	}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s/fields/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]", "[FIELD_ID]")
-	var request = &adminpb.GetFieldRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetField(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminGetFieldError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s/fields/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]", "[FIELD_ID]")
-	var request = &adminpb.GetFieldRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetField(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminListFields(t *testing.T) {
-	var nextPageToken string = ""
-	var fieldsElement *adminpb.Field = &adminpb.Field{}
-	var fields = []*adminpb.Field{fieldsElement}
-	var expectedResponse = &adminpb.ListFieldsResponse{
-		NextPageToken: nextPageToken,
-		Fields:        fields,
-	}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]")
-	var request = &adminpb.ListFieldsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFields(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Fields[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminListFieldsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/collectionGroups/%s", "[PROJECT]", "[DATABASE]", "[COLLECTION_ID]")
-	var request = &adminpb.ListFieldsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFields(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreAdminUpdateField(t *testing.T) {
-	var expectedResponse = &adminpb.Field{}
-
-	mockFirestoreAdmin.err = nil
-	mockFirestoreAdmin.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockFirestoreAdmin.resps = append(mockFirestoreAdmin.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var field *adminpb.Field = &adminpb.Field{}
-	var request = &adminpb.UpdateFieldRequest{
-		Field: field,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateField(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestoreAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreAdminUpdateFieldError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestoreAdmin.err = gstatus.Error(errCode, "test error")
-
-	var field *adminpb.Field = &adminpb.Field{}
-	var request = &adminpb.UpdateFieldRequest{
-		Field: field,
-	}
-
-	c, err := NewFirestoreAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateField(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/firestore/apiv1/mock_test.go b/firestore/apiv1/mock_test.go
deleted file mode 100644
index f7bdf76b031..00000000000
--- a/firestore/apiv1/mock_test.go
+++ /dev/null
@@ -1,1152 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package firestore
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	firestorepb "google.golang.org/genproto/googleapis/firestore/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockFirestoreServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	firestorepb.FirestoreServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockFirestoreServer) GetDocument(ctx context.Context, req *firestorepb.GetDocumentRequest) (*firestorepb.Document, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.Document), nil
-}
-
-func (s *mockFirestoreServer) ListDocuments(ctx context.Context, req *firestorepb.ListDocumentsRequest) (*firestorepb.ListDocumentsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.ListDocumentsResponse), nil
-}
-
-func (s *mockFirestoreServer) CreateDocument(ctx context.Context, req *firestorepb.CreateDocumentRequest) (*firestorepb.Document, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.Document), nil
-}
-
-func (s *mockFirestoreServer) UpdateDocument(ctx context.Context, req *firestorepb.UpdateDocumentRequest) (*firestorepb.Document, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.Document), nil
-}
-
-func (s *mockFirestoreServer) DeleteDocument(ctx context.Context, req *firestorepb.DeleteDocumentRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockFirestoreServer) BatchGetDocuments(req *firestorepb.BatchGetDocumentsRequest, stream firestorepb.Firestore_BatchGetDocumentsServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.BatchGetDocumentsResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) BeginTransaction(ctx context.Context, req *firestorepb.BeginTransactionRequest) (*firestorepb.BeginTransactionResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.BeginTransactionResponse), nil
-}
-
-func (s *mockFirestoreServer) Commit(ctx context.Context, req *firestorepb.CommitRequest) (*firestorepb.CommitResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.CommitResponse), nil
-}
-
-func (s *mockFirestoreServer) Rollback(ctx context.Context, req *firestorepb.RollbackRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockFirestoreServer) RunQuery(req *firestorepb.RunQueryRequest, stream firestorepb.Firestore_RunQueryServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.RunQueryResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) Write(stream firestorepb.Firestore_WriteServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.WriteResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) Listen(stream firestorepb.Firestore_ListenServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.ListenResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) ListCollectionIds(ctx context.Context, req *firestorepb.ListCollectionIdsRequest) (*firestorepb.ListCollectionIdsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.ListCollectionIdsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockFirestore mockFirestoreServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	firestorepb.RegisterFirestoreServer(serv, &mockFirestore)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestFirestoreGetDocument(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &firestorepb.Document{
-		Name: name2,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.GetDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreGetDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.GetDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreListDocuments(t *testing.T) {
-	var nextPageToken string = ""
-	var documentsElement *firestorepb.Document = &firestorepb.Document{}
-	var documents = []*firestorepb.Document{documentsElement}
-	var expectedResponse = &firestorepb.ListDocumentsResponse{
-		NextPageToken: nextPageToken,
-		Documents:     documents,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var request = &firestorepb.ListDocumentsRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDocuments(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Documents[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreListDocumentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var request = &firestorepb.ListDocumentsRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDocuments(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreCreateDocument(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &firestorepb.Document{
-		Name: name,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var documentId string = "documentId506676927"
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var request = &firestorepb.CreateDocumentRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-		DocumentId:   documentId,
-		Document:     document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreCreateDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var documentId string = "documentId506676927"
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var request = &firestorepb.CreateDocumentRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-		DocumentId:   documentId,
-		Document:     document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreUpdateDocument(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &firestorepb.Document{
-		Name: name,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var updateMask *firestorepb.DocumentMask = &firestorepb.DocumentMask{}
-	var request = &firestorepb.UpdateDocumentRequest{
-		Document:   document,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreUpdateDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var updateMask *firestorepb.DocumentMask = &firestorepb.DocumentMask{}
-	var request = &firestorepb.UpdateDocumentRequest{
-		Document:   document,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreDeleteDocument(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.DeleteDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestFirestoreDeleteDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.DeleteDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestFirestoreBatchGetDocuments(t *testing.T) {
-	var missing string = "missing1069449574"
-	var transaction []byte = []byte("-34")
-	var expectedResponse = &firestorepb.BatchGetDocumentsResponse{
-		Result: &firestorepb.BatchGetDocumentsResponse_Missing{
-			Missing: missing,
-		},
-		Transaction: transaction,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var documents []string = nil
-	var request = &firestorepb.BatchGetDocumentsRequest{
-		Database:  formattedDatabase,
-		Documents: documents,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.BatchGetDocuments(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreBatchGetDocumentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var documents []string = nil
-	var request = &firestorepb.BatchGetDocumentsRequest{
-		Database:  formattedDatabase,
-		Documents: documents,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.BatchGetDocuments(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreBeginTransaction(t *testing.T) {
-	var transaction []byte = []byte("-34")
-	var expectedResponse = &firestorepb.BeginTransactionResponse{
-		Transaction: transaction,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.BeginTransactionRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BeginTransaction(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreBeginTransactionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.BeginTransactionRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BeginTransaction(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreCommit(t *testing.T) {
-	var expectedResponse *firestorepb.CommitResponse = &firestorepb.CommitResponse{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var writes []*firestorepb.Write = nil
-	var request = &firestorepb.CommitRequest{
-		Database: formattedDatabase,
-		Writes:   writes,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Commit(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreCommitError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var writes []*firestorepb.Write = nil
-	var request = &firestorepb.CommitRequest{
-		Database: formattedDatabase,
-		Writes:   writes,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Commit(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreRollback(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var transaction []byte = []byte("-34")
-	var request = &firestorepb.RollbackRequest{
-		Database:    formattedDatabase,
-		Transaction: transaction,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Rollback(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestFirestoreRollbackError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var transaction []byte = []byte("-34")
-	var request = &firestorepb.RollbackRequest{
-		Database:    formattedDatabase,
-		Transaction: transaction,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Rollback(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestFirestoreRunQuery(t *testing.T) {
-	var transaction []byte = []byte("-34")
-	var skippedResults int32 = 880286183
-	var expectedResponse = &firestorepb.RunQueryResponse{
-		Transaction:    transaction,
-		SkippedResults: skippedResults,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.RunQueryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.RunQuery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreRunQueryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.RunQueryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.RunQuery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreWrite(t *testing.T) {
-	var streamId string = "streamId-315624902"
-	var streamToken []byte = []byte("122")
-	var expectedResponse = &firestorepb.WriteResponse{
-		StreamId:    streamId,
-		StreamToken: streamToken,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.WriteRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Write(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreWriteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.WriteRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Write(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreListen(t *testing.T) {
-	var expectedResponse *firestorepb.ListenResponse = &firestorepb.ListenResponse{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.ListenRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Listen(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreListenError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.ListenRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Listen(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreListCollectionIds(t *testing.T) {
-	var nextPageToken string = ""
-	var collectionIdsElement string = "collectionIdsElement1368994900"
-	var collectionIds = []string{collectionIdsElement}
-	var expectedResponse = &firestorepb.ListCollectionIdsResponse{
-		NextPageToken: nextPageToken,
-		CollectionIds: collectionIds,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.ListCollectionIdsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCollectionIds(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.CollectionIds[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreListCollectionIdsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.ListCollectionIdsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCollectionIds(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/firestore/apiv1beta1/mock_test.go b/firestore/apiv1beta1/mock_test.go
deleted file mode 100644
index 75ef4fe15d6..00000000000
--- a/firestore/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,1152 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package firestore
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	firestorepb "google.golang.org/genproto/googleapis/firestore/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockFirestoreServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	firestorepb.FirestoreServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockFirestoreServer) GetDocument(ctx context.Context, req *firestorepb.GetDocumentRequest) (*firestorepb.Document, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.Document), nil
-}
-
-func (s *mockFirestoreServer) ListDocuments(ctx context.Context, req *firestorepb.ListDocumentsRequest) (*firestorepb.ListDocumentsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.ListDocumentsResponse), nil
-}
-
-func (s *mockFirestoreServer) CreateDocument(ctx context.Context, req *firestorepb.CreateDocumentRequest) (*firestorepb.Document, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.Document), nil
-}
-
-func (s *mockFirestoreServer) UpdateDocument(ctx context.Context, req *firestorepb.UpdateDocumentRequest) (*firestorepb.Document, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.Document), nil
-}
-
-func (s *mockFirestoreServer) DeleteDocument(ctx context.Context, req *firestorepb.DeleteDocumentRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockFirestoreServer) BatchGetDocuments(req *firestorepb.BatchGetDocumentsRequest, stream firestorepb.Firestore_BatchGetDocumentsServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.BatchGetDocumentsResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) BeginTransaction(ctx context.Context, req *firestorepb.BeginTransactionRequest) (*firestorepb.BeginTransactionResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.BeginTransactionResponse), nil
-}
-
-func (s *mockFirestoreServer) Commit(ctx context.Context, req *firestorepb.CommitRequest) (*firestorepb.CommitResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.CommitResponse), nil
-}
-
-func (s *mockFirestoreServer) Rollback(ctx context.Context, req *firestorepb.RollbackRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockFirestoreServer) RunQuery(req *firestorepb.RunQueryRequest, stream firestorepb.Firestore_RunQueryServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.RunQueryResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) Write(stream firestorepb.Firestore_WriteServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.WriteResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) Listen(stream firestorepb.Firestore_ListenServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*firestorepb.ListenResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockFirestoreServer) ListCollectionIds(ctx context.Context, req *firestorepb.ListCollectionIdsRequest) (*firestorepb.ListCollectionIdsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*firestorepb.ListCollectionIdsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockFirestore mockFirestoreServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	firestorepb.RegisterFirestoreServer(serv, &mockFirestore)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestFirestoreGetDocument(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &firestorepb.Document{
-		Name: name2,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.GetDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreGetDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.GetDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreListDocuments(t *testing.T) {
-	var nextPageToken string = ""
-	var documentsElement *firestorepb.Document = &firestorepb.Document{}
-	var documents = []*firestorepb.Document{documentsElement}
-	var expectedResponse = &firestorepb.ListDocumentsResponse{
-		NextPageToken: nextPageToken,
-		Documents:     documents,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var request = &firestorepb.ListDocumentsRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDocuments(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Documents[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreListDocumentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var request = &firestorepb.ListDocumentsRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDocuments(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreCreateDocument(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &firestorepb.Document{
-		Name: name,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var documentId string = "documentId506676927"
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var request = &firestorepb.CreateDocumentRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-		DocumentId:   documentId,
-		Document:     document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreCreateDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var collectionId string = "collectionId-821242276"
-	var documentId string = "documentId506676927"
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var request = &firestorepb.CreateDocumentRequest{
-		Parent:       formattedParent,
-		CollectionId: collectionId,
-		DocumentId:   documentId,
-		Document:     document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreUpdateDocument(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &firestorepb.Document{
-		Name: name,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var updateMask *firestorepb.DocumentMask = &firestorepb.DocumentMask{}
-	var request = &firestorepb.UpdateDocumentRequest{
-		Document:   document,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreUpdateDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var document *firestorepb.Document = &firestorepb.Document{}
-	var updateMask *firestorepb.DocumentMask = &firestorepb.DocumentMask{}
-	var request = &firestorepb.UpdateDocumentRequest{
-		Document:   document,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreDeleteDocument(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.DeleteDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDocument(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestFirestoreDeleteDocumentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.DeleteDocumentRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDocument(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestFirestoreBatchGetDocuments(t *testing.T) {
-	var missing string = "missing1069449574"
-	var transaction []byte = []byte("-34")
-	var expectedResponse = &firestorepb.BatchGetDocumentsResponse{
-		Result: &firestorepb.BatchGetDocumentsResponse_Missing{
-			Missing: missing,
-		},
-		Transaction: transaction,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var documents []string = nil
-	var request = &firestorepb.BatchGetDocumentsRequest{
-		Database:  formattedDatabase,
-		Documents: documents,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.BatchGetDocuments(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreBatchGetDocumentsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var documents []string = nil
-	var request = &firestorepb.BatchGetDocumentsRequest{
-		Database:  formattedDatabase,
-		Documents: documents,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.BatchGetDocuments(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreBeginTransaction(t *testing.T) {
-	var transaction []byte = []byte("-34")
-	var expectedResponse = &firestorepb.BeginTransactionResponse{
-		Transaction: transaction,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.BeginTransactionRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BeginTransaction(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreBeginTransactionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.BeginTransactionRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BeginTransaction(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreCommit(t *testing.T) {
-	var expectedResponse *firestorepb.CommitResponse = &firestorepb.CommitResponse{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var writes []*firestorepb.Write = nil
-	var request = &firestorepb.CommitRequest{
-		Database: formattedDatabase,
-		Writes:   writes,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Commit(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreCommitError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var writes []*firestorepb.Write = nil
-	var request = &firestorepb.CommitRequest{
-		Database: formattedDatabase,
-		Writes:   writes,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Commit(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreRollback(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var transaction []byte = []byte("-34")
-	var request = &firestorepb.RollbackRequest{
-		Database:    formattedDatabase,
-		Transaction: transaction,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Rollback(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestFirestoreRollbackError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var transaction []byte = []byte("-34")
-	var request = &firestorepb.RollbackRequest{
-		Database:    formattedDatabase,
-		Transaction: transaction,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Rollback(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestFirestoreRunQuery(t *testing.T) {
-	var transaction []byte = []byte("-34")
-	var skippedResults int32 = 880286183
-	var expectedResponse = &firestorepb.RunQueryResponse{
-		Transaction:    transaction,
-		SkippedResults: skippedResults,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.RunQueryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.RunQuery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreRunQueryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.RunQueryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.RunQuery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreWrite(t *testing.T) {
-	var streamId string = "streamId-315624902"
-	var streamToken []byte = []byte("122")
-	var expectedResponse = &firestorepb.WriteResponse{
-		StreamId:    streamId,
-		StreamToken: streamToken,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.WriteRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Write(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreWriteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.WriteRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Write(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreListen(t *testing.T) {
-	var expectedResponse *firestorepb.ListenResponse = &firestorepb.ListenResponse{}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.ListenRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Listen(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreListenError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/databases/%s", "[PROJECT]", "[DATABASE]")
-	var request = &firestorepb.ListenRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.Listen(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestFirestoreListCollectionIds(t *testing.T) {
-	var nextPageToken string = ""
-	var collectionIdsElement string = "collectionIdsElement1368994900"
-	var collectionIds = []string{collectionIdsElement}
-	var expectedResponse = &firestorepb.ListCollectionIdsResponse{
-		NextPageToken: nextPageToken,
-		CollectionIds: collectionIds,
-	}
-
-	mockFirestore.err = nil
-	mockFirestore.reqs = nil
-
-	mockFirestore.resps = append(mockFirestore.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.ListCollectionIdsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCollectionIds(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockFirestore.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.CollectionIds[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestFirestoreListCollectionIdsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockFirestore.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/databases/%s/documents/%s/%s", "[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]")
-	var request = &firestorepb.ListCollectionIdsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCollectionIds(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/grafeas/apiv1/mock_test.go b/grafeas/apiv1/mock_test.go
deleted file mode 100644
index 4f10b02b038..00000000000
--- a/grafeas/apiv1/mock_test.go
+++ /dev/null
@@ -1,1166 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package grafeas
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	grafeaspb "google.golang.org/genproto/googleapis/grafeas/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockGrafeasServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	grafeaspb.GrafeasServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockGrafeasServer) GetOccurrence(ctx context.Context, req *grafeaspb.GetOccurrenceRequest) (*grafeaspb.Occurrence, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Occurrence), nil
-}
-
-func (s *mockGrafeasServer) ListOccurrences(ctx context.Context, req *grafeaspb.ListOccurrencesRequest) (*grafeaspb.ListOccurrencesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.ListOccurrencesResponse), nil
-}
-
-func (s *mockGrafeasServer) DeleteOccurrence(ctx context.Context, req *grafeaspb.DeleteOccurrenceRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockGrafeasServer) CreateOccurrence(ctx context.Context, req *grafeaspb.CreateOccurrenceRequest) (*grafeaspb.Occurrence, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Occurrence), nil
-}
-
-func (s *mockGrafeasServer) BatchCreateOccurrences(ctx context.Context, req *grafeaspb.BatchCreateOccurrencesRequest) (*grafeaspb.BatchCreateOccurrencesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.BatchCreateOccurrencesResponse), nil
-}
-
-func (s *mockGrafeasServer) UpdateOccurrence(ctx context.Context, req *grafeaspb.UpdateOccurrenceRequest) (*grafeaspb.Occurrence, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Occurrence), nil
-}
-
-func (s *mockGrafeasServer) GetOccurrenceNote(ctx context.Context, req *grafeaspb.GetOccurrenceNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasServer) GetNote(ctx context.Context, req *grafeaspb.GetNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasServer) ListNotes(ctx context.Context, req *grafeaspb.ListNotesRequest) (*grafeaspb.ListNotesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.ListNotesResponse), nil
-}
-
-func (s *mockGrafeasServer) DeleteNote(ctx context.Context, req *grafeaspb.DeleteNoteRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockGrafeasServer) CreateNote(ctx context.Context, req *grafeaspb.CreateNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasServer) BatchCreateNotes(ctx context.Context, req *grafeaspb.BatchCreateNotesRequest) (*grafeaspb.BatchCreateNotesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.BatchCreateNotesResponse), nil
-}
-
-func (s *mockGrafeasServer) UpdateNote(ctx context.Context, req *grafeaspb.UpdateNoteRequest) (*grafeaspb.Note, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.Note), nil
-}
-
-func (s *mockGrafeasServer) ListNoteOccurrences(ctx context.Context, req *grafeaspb.ListNoteOccurrencesRequest) (*grafeaspb.ListNoteOccurrencesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*grafeaspb.ListNoteOccurrencesResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockGrafeas mockGrafeasServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	grafeaspb.RegisterGrafeasServer(serv, &mockGrafeas)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestGrafeasGetOccurrence(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var resourceUri string = "resourceUri-384040517"
-	var noteName string = "noteName1780787896"
-	var remediation string = "remediation779381797"
-	var expectedResponse = &grafeaspb.Occurrence{
-		Name:        name2,
-		ResourceUri: resourceUri,
-		NoteName:    noteName,
-		Remediation: remediation,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasGetOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasListOccurrences(t *testing.T) {
-	var nextPageToken string = ""
-	var occurrencesElement *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var occurrences = []*grafeaspb.Occurrence{occurrencesElement}
-	var expectedResponse = &grafeaspb.ListOccurrencesResponse{
-		NextPageToken: nextPageToken,
-		Occurrences:   occurrences,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListOccurrencesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListOccurrences(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Occurrences[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasListOccurrencesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListOccurrencesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListOccurrences(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasDeleteOccurrence(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.DeleteOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestGrafeasDeleteOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.DeleteOccurrenceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestGrafeasCreateOccurrence(t *testing.T) {
-	var name string = "name3373707"
-	var resourceUri string = "resourceUri-384040517"
-	var noteName string = "noteName1780787896"
-	var remediation string = "remediation779381797"
-	var expectedResponse = &grafeaspb.Occurrence{
-		Name:        name,
-		ResourceUri: resourceUri,
-		NoteName:    noteName,
-		Remediation: remediation,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.CreateOccurrenceRequest{
-		Parent:     formattedParent,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasCreateOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.CreateOccurrenceRequest{
-		Parent:     formattedParent,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasBatchCreateOccurrences(t *testing.T) {
-	var expectedResponse *grafeaspb.BatchCreateOccurrencesResponse = &grafeaspb.BatchCreateOccurrencesResponse{}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrences []*grafeaspb.Occurrence = nil
-	var request = &grafeaspb.BatchCreateOccurrencesRequest{
-		Parent:      formattedParent,
-		Occurrences: occurrences,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateOccurrences(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasBatchCreateOccurrencesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var occurrences []*grafeaspb.Occurrence = nil
-	var request = &grafeaspb.BatchCreateOccurrencesRequest{
-		Parent:      formattedParent,
-		Occurrences: occurrences,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateOccurrences(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasUpdateOccurrence(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var resourceUri string = "resourceUri-384040517"
-	var noteName string = "noteName1780787896"
-	var remediation string = "remediation779381797"
-	var expectedResponse = &grafeaspb.Occurrence{
-		Name:        name2,
-		ResourceUri: resourceUri,
-		NoteName:    noteName,
-		Remediation: remediation,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.UpdateOccurrenceRequest{
-		Name:       formattedName,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOccurrence(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasUpdateOccurrenceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var occurrence *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var request = &grafeaspb.UpdateOccurrenceRequest{
-		Name:       formattedName,
-		Occurrence: occurrence,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOccurrence(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasGetOccurrenceNote(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name2,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrenceNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasGetOccurrenceNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/occurrences/%s", "[PROJECT]", "[OCCURRENCE]")
-	var request = &grafeaspb.GetOccurrenceNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOccurrenceNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasGetNote(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name2,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.GetNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasGetNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.GetNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasListNotes(t *testing.T) {
-	var nextPageToken string = ""
-	var notesElement *grafeaspb.Note = &grafeaspb.Note{}
-	var notes = []*grafeaspb.Note{notesElement}
-	var expectedResponse = &grafeaspb.ListNotesResponse{
-		NextPageToken: nextPageToken,
-		Notes:         notes,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListNotesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotes(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Notes[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasListNotesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &grafeaspb.ListNotesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotes(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasDeleteNote(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.DeleteNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestGrafeasDeleteNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.DeleteNoteRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestGrafeasCreateNote(t *testing.T) {
-	var name string = "name3373707"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var noteId string = "noteId2129224840"
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.CreateNoteRequest{
-		Parent: formattedParent,
-		NoteId: noteId,
-		Note:   note,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasCreateNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var noteId string = "noteId2129224840"
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.CreateNoteRequest{
-		Parent: formattedParent,
-		NoteId: noteId,
-		Note:   note,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasBatchCreateNotes(t *testing.T) {
-	var expectedResponse *grafeaspb.BatchCreateNotesResponse = &grafeaspb.BatchCreateNotesResponse{}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var notes map[string]*grafeaspb.Note = nil
-	var request = &grafeaspb.BatchCreateNotesRequest{
-		Parent: formattedParent,
-		Notes:  notes,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateNotes(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasBatchCreateNotesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var notes map[string]*grafeaspb.Note = nil
-	var request = &grafeaspb.BatchCreateNotesRequest{
-		Parent: formattedParent,
-		Notes:  notes,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateNotes(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasUpdateNote(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var shortDescription string = "shortDescription-235369287"
-	var longDescription string = "longDescription-1747792199"
-	var expectedResponse = &grafeaspb.Note{
-		Name:             name2,
-		ShortDescription: shortDescription,
-		LongDescription:  longDescription,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.UpdateNoteRequest{
-		Name: formattedName,
-		Note: note,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNote(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasUpdateNoteError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var note *grafeaspb.Note = &grafeaspb.Note{}
-	var request = &grafeaspb.UpdateNoteRequest{
-		Name: formattedName,
-		Note: note,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNote(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGrafeasListNoteOccurrences(t *testing.T) {
-	var nextPageToken string = ""
-	var occurrencesElement *grafeaspb.Occurrence = &grafeaspb.Occurrence{}
-	var occurrences = []*grafeaspb.Occurrence{occurrencesElement}
-	var expectedResponse = &grafeaspb.ListNoteOccurrencesResponse{
-		NextPageToken: nextPageToken,
-		Occurrences:   occurrences,
-	}
-
-	mockGrafeas.err = nil
-	mockGrafeas.reqs = nil
-
-	mockGrafeas.resps = append(mockGrafeas.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.ListNoteOccurrencesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNoteOccurrences(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGrafeas.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Occurrences[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGrafeasListNoteOccurrencesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGrafeas.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notes/%s", "[PROJECT]", "[NOTE]")
-	var request = &grafeaspb.ListNoteOccurrencesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNoteOccurrences(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/iam/admin/apiv1/mock_test.go b/iam/admin/apiv1/mock_test.go
deleted file mode 100644
index b5d8c884fdd..00000000000
--- a/iam/admin/apiv1/mock_test.go
+++ /dev/null
@@ -1,1717 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package admin
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	adminpb "google.golang.org/genproto/googleapis/iam/admin/v1"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockIamServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	adminpb.IAMServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockIamServer) ListServiceAccounts(ctx context.Context, req *adminpb.ListServiceAccountsRequest) (*adminpb.ListServiceAccountsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ListServiceAccountsResponse), nil
-}
-
-func (s *mockIamServer) GetServiceAccount(ctx context.Context, req *adminpb.GetServiceAccountRequest) (*adminpb.ServiceAccount, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ServiceAccount), nil
-}
-
-func (s *mockIamServer) CreateServiceAccount(ctx context.Context, req *adminpb.CreateServiceAccountRequest) (*adminpb.ServiceAccount, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ServiceAccount), nil
-}
-
-func (s *mockIamServer) UpdateServiceAccount(ctx context.Context, req *adminpb.ServiceAccount) (*adminpb.ServiceAccount, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ServiceAccount), nil
-}
-
-func (s *mockIamServer) DeleteServiceAccount(ctx context.Context, req *adminpb.DeleteServiceAccountRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockIamServer) ListServiceAccountKeys(ctx context.Context, req *adminpb.ListServiceAccountKeysRequest) (*adminpb.ListServiceAccountKeysResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ListServiceAccountKeysResponse), nil
-}
-
-func (s *mockIamServer) GetServiceAccountKey(ctx context.Context, req *adminpb.GetServiceAccountKeyRequest) (*adminpb.ServiceAccountKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ServiceAccountKey), nil
-}
-
-func (s *mockIamServer) CreateServiceAccountKey(ctx context.Context, req *adminpb.CreateServiceAccountKeyRequest) (*adminpb.ServiceAccountKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ServiceAccountKey), nil
-}
-
-func (s *mockIamServer) DeleteServiceAccountKey(ctx context.Context, req *adminpb.DeleteServiceAccountKeyRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockIamServer) SignBlob(ctx context.Context, req *adminpb.SignBlobRequest) (*adminpb.SignBlobResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.SignBlobResponse), nil
-}
-
-func (s *mockIamServer) SignJwt(ctx context.Context, req *adminpb.SignJwtRequest) (*adminpb.SignJwtResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.SignJwtResponse), nil
-}
-
-func (s *mockIamServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockIamServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockIamServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockIamServer) QueryGrantableRoles(ctx context.Context, req *adminpb.QueryGrantableRolesRequest) (*adminpb.QueryGrantableRolesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.QueryGrantableRolesResponse), nil
-}
-
-func (s *mockIamServer) ListRoles(ctx context.Context, req *adminpb.ListRolesRequest) (*adminpb.ListRolesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.ListRolesResponse), nil
-}
-
-func (s *mockIamServer) GetRole(ctx context.Context, req *adminpb.GetRoleRequest) (*adminpb.Role, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Role), nil
-}
-
-func (s *mockIamServer) CreateRole(ctx context.Context, req *adminpb.CreateRoleRequest) (*adminpb.Role, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Role), nil
-}
-
-func (s *mockIamServer) UpdateRole(ctx context.Context, req *adminpb.UpdateRoleRequest) (*adminpb.Role, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Role), nil
-}
-
-func (s *mockIamServer) DeleteRole(ctx context.Context, req *adminpb.DeleteRoleRequest) (*adminpb.Role, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Role), nil
-}
-
-func (s *mockIamServer) UndeleteRole(ctx context.Context, req *adminpb.UndeleteRoleRequest) (*adminpb.Role, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.Role), nil
-}
-
-func (s *mockIamServer) QueryTestablePermissions(ctx context.Context, req *adminpb.QueryTestablePermissionsRequest) (*adminpb.QueryTestablePermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*adminpb.QueryTestablePermissionsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockIam mockIamServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	adminpb.RegisterIAMServer(serv, &mockIam)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestIamListServiceAccounts(t *testing.T) {
-	var nextPageToken string = ""
-	var accountsElement *adminpb.ServiceAccount = &adminpb.ServiceAccount{}
-	var accounts = []*adminpb.ServiceAccount{accountsElement}
-	var expectedResponse = &adminpb.ListServiceAccountsResponse{
-		NextPageToken: nextPageToken,
-		Accounts:      accounts,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &adminpb.ListServiceAccountsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServiceAccounts(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Accounts[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamListServiceAccountsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &adminpb.ListServiceAccountsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServiceAccounts(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamGetServiceAccount(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var projectId string = "projectId-1969970175"
-	var uniqueId string = "uniqueId-538310583"
-	var email string = "email96619420"
-	var displayName string = "displayName1615086568"
-	var etag []byte = []byte("21")
-	var oauth2ClientId string = "oauth2ClientId-1833466037"
-	var expectedResponse = &adminpb.ServiceAccount{
-		Name:           name2,
-		ProjectId:      projectId,
-		UniqueId:       uniqueId,
-		Email:          email,
-		DisplayName:    displayName,
-		Etag:           etag,
-		Oauth2ClientId: oauth2ClientId,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.GetServiceAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServiceAccount(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamGetServiceAccountError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.GetServiceAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServiceAccount(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamCreateServiceAccount(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var projectId string = "projectId-1969970175"
-	var uniqueId string = "uniqueId-538310583"
-	var email string = "email96619420"
-	var displayName string = "displayName1615086568"
-	var etag []byte = []byte("21")
-	var oauth2ClientId string = "oauth2ClientId-1833466037"
-	var expectedResponse = &adminpb.ServiceAccount{
-		Name:           name2,
-		ProjectId:      projectId,
-		UniqueId:       uniqueId,
-		Email:          email,
-		DisplayName:    displayName,
-		Etag:           etag,
-		Oauth2ClientId: oauth2ClientId,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var accountId string = "accountId-803333011"
-	var request = &adminpb.CreateServiceAccountRequest{
-		Name:      formattedName,
-		AccountId: accountId,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateServiceAccount(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCreateServiceAccountError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var accountId string = "accountId-803333011"
-	var request = &adminpb.CreateServiceAccountRequest{
-		Name:      formattedName,
-		AccountId: accountId,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateServiceAccount(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamUpdateServiceAccount(t *testing.T) {
-	var name string = "name3373707"
-	var projectId string = "projectId-1969970175"
-	var uniqueId string = "uniqueId-538310583"
-	var email string = "email96619420"
-	var displayName string = "displayName1615086568"
-	var etag2 []byte = []byte("-120")
-	var oauth2ClientId string = "oauth2ClientId-1833466037"
-	var expectedResponse = &adminpb.ServiceAccount{
-		Name:           name,
-		ProjectId:      projectId,
-		UniqueId:       uniqueId,
-		Email:          email,
-		DisplayName:    displayName,
-		Etag:           etag2,
-		Oauth2ClientId: oauth2ClientId,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var etag []byte = []byte("21")
-	var request = &adminpb.ServiceAccount{
-		Etag: etag,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateServiceAccount(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamUpdateServiceAccountError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var etag []byte = []byte("21")
-	var request = &adminpb.ServiceAccount{
-		Etag: etag,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateServiceAccount(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamDeleteServiceAccount(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.DeleteServiceAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteServiceAccount(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestIamDeleteServiceAccountError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.DeleteServiceAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteServiceAccount(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestIamListServiceAccountKeys(t *testing.T) {
-	var expectedResponse *adminpb.ListServiceAccountKeysResponse = &adminpb.ListServiceAccountKeysResponse{}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.ListServiceAccountKeysRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServiceAccountKeys(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamListServiceAccountKeysError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.ListServiceAccountKeysRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServiceAccountKeys(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamGetServiceAccountKey(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var privateKeyData []byte = []byte("-58")
-	var publicKeyData []byte = []byte("-96")
-	var expectedResponse = &adminpb.ServiceAccountKey{
-		Name:           name2,
-		PrivateKeyData: privateKeyData,
-		PublicKeyData:  publicKeyData,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s/keys/%s", "[PROJECT]", "[SERVICE_ACCOUNT]", "[KEY]")
-	var request = &adminpb.GetServiceAccountKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServiceAccountKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamGetServiceAccountKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s/keys/%s", "[PROJECT]", "[SERVICE_ACCOUNT]", "[KEY]")
-	var request = &adminpb.GetServiceAccountKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServiceAccountKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamCreateServiceAccountKey(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var privateKeyData []byte = []byte("-58")
-	var publicKeyData []byte = []byte("-96")
-	var expectedResponse = &adminpb.ServiceAccountKey{
-		Name:           name2,
-		PrivateKeyData: privateKeyData,
-		PublicKeyData:  publicKeyData,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.CreateServiceAccountKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateServiceAccountKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCreateServiceAccountKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &adminpb.CreateServiceAccountKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateServiceAccountKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamDeleteServiceAccountKey(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s/keys/%s", "[PROJECT]", "[SERVICE_ACCOUNT]", "[KEY]")
-	var request = &adminpb.DeleteServiceAccountKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteServiceAccountKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestIamDeleteServiceAccountKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s/keys/%s", "[PROJECT]", "[SERVICE_ACCOUNT]", "[KEY]")
-	var request = &adminpb.DeleteServiceAccountKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteServiceAccountKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestIamSignBlob(t *testing.T) {
-	var keyId string = "keyId-1134673157"
-	var signature []byte = []byte("-72")
-	var expectedResponse = &adminpb.SignBlobResponse{
-		KeyId:     keyId,
-		Signature: signature,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var bytesToSign []byte = []byte("45")
-	var request = &adminpb.SignBlobRequest{
-		Name:        formattedName,
-		BytesToSign: bytesToSign,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignBlob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamSignBlobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var bytesToSign []byte = []byte("45")
-	var request = &adminpb.SignBlobRequest{
-		Name:        formattedName,
-		BytesToSign: bytesToSign,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignBlob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.getIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.getIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.setIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.setIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamQueryGrantableRoles(t *testing.T) {
-	var nextPageToken string = "nextPageToken-1530815211"
-	var expectedResponse = &adminpb.QueryGrantableRolesResponse{
-		NextPageToken: nextPageToken,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var fullResourceName string = "fullResourceName1300993644"
-	var request = &adminpb.QueryGrantableRolesRequest{
-		FullResourceName: fullResourceName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.QueryGrantableRoles(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamQueryGrantableRolesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var fullResourceName string = "fullResourceName1300993644"
-	var request = &adminpb.QueryGrantableRolesRequest{
-		FullResourceName: fullResourceName,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.QueryGrantableRoles(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamSignJwt(t *testing.T) {
-	var keyId string = "keyId-1134673157"
-	var signedJwt string = "signedJwt-979546844"
-	var expectedResponse = &adminpb.SignJwtResponse{
-		KeyId:     keyId,
-		SignedJwt: signedJwt,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var name string = "name3373707"
-	var payload string = "payload-786701938"
-	var request = &adminpb.SignJwtRequest{
-		Name:    name,
-		Payload: payload,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignJwt(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamSignJwtError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var name string = "name3373707"
-	var payload string = "payload-786701938"
-	var request = &adminpb.SignJwtRequest{
-		Name:    name,
-		Payload: payload,
-	}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignJwt(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamListRoles(t *testing.T) {
-	var nextPageToken string = "nextPageToken-1530815211"
-	var expectedResponse = &adminpb.ListRolesResponse{
-		NextPageToken: nextPageToken,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.ListRolesRequest = &adminpb.ListRolesRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListRoles(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamListRolesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.ListRolesRequest = &adminpb.ListRolesRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListRoles(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamGetRole(t *testing.T) {
-	var name string = "name3373707"
-	var title string = "title110371416"
-	var description string = "description-1724546052"
-	var etag []byte = []byte("21")
-	var deleted bool = false
-	var expectedResponse = &adminpb.Role{
-		Name:        name,
-		Title:       title,
-		Description: description,
-		Etag:        etag,
-		Deleted:     deleted,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.GetRoleRequest = &adminpb.GetRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetRole(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamGetRoleError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.GetRoleRequest = &adminpb.GetRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetRole(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamCreateRole(t *testing.T) {
-	var name string = "name3373707"
-	var title string = "title110371416"
-	var description string = "description-1724546052"
-	var etag []byte = []byte("21")
-	var deleted bool = false
-	var expectedResponse = &adminpb.Role{
-		Name:        name,
-		Title:       title,
-		Description: description,
-		Etag:        etag,
-		Deleted:     deleted,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.CreateRoleRequest = &adminpb.CreateRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateRole(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCreateRoleError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.CreateRoleRequest = &adminpb.CreateRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateRole(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamUpdateRole(t *testing.T) {
-	var name string = "name3373707"
-	var title string = "title110371416"
-	var description string = "description-1724546052"
-	var etag []byte = []byte("21")
-	var deleted bool = false
-	var expectedResponse = &adminpb.Role{
-		Name:        name,
-		Title:       title,
-		Description: description,
-		Etag:        etag,
-		Deleted:     deleted,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.UpdateRoleRequest = &adminpb.UpdateRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateRole(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamUpdateRoleError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.UpdateRoleRequest = &adminpb.UpdateRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateRole(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamDeleteRole(t *testing.T) {
-	var name string = "name3373707"
-	var title string = "title110371416"
-	var description string = "description-1724546052"
-	var etag []byte = []byte("21")
-	var deleted bool = false
-	var expectedResponse = &adminpb.Role{
-		Name:        name,
-		Title:       title,
-		Description: description,
-		Etag:        etag,
-		Deleted:     deleted,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.DeleteRoleRequest = &adminpb.DeleteRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteRole(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamDeleteRoleError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.DeleteRoleRequest = &adminpb.DeleteRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DeleteRole(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamUndeleteRole(t *testing.T) {
-	var name string = "name3373707"
-	var title string = "title110371416"
-	var description string = "description-1724546052"
-	var etag []byte = []byte("21")
-	var deleted bool = false
-	var expectedResponse = &adminpb.Role{
-		Name:        name,
-		Title:       title,
-		Description: description,
-		Etag:        etag,
-		Deleted:     deleted,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.UndeleteRoleRequest = &adminpb.UndeleteRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UndeleteRole(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamUndeleteRoleError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.UndeleteRoleRequest = &adminpb.UndeleteRoleRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UndeleteRole(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamQueryTestablePermissions(t *testing.T) {
-	var nextPageToken string = "nextPageToken-1530815211"
-	var expectedResponse = &adminpb.QueryTestablePermissionsResponse{
-		NextPageToken: nextPageToken,
-	}
-
-	mockIam.err = nil
-	mockIam.reqs = nil
-
-	mockIam.resps = append(mockIam.resps[:0], expectedResponse)
-
-	var request *adminpb.QueryTestablePermissionsRequest = &adminpb.QueryTestablePermissionsRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.QueryTestablePermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIam.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamQueryTestablePermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIam.err = gstatus.Error(errCode, "test error")
-
-	var request *adminpb.QueryTestablePermissionsRequest = &adminpb.QueryTestablePermissionsRequest{}
-
-	c, err := NewIamClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.QueryTestablePermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/iam/credentials/apiv1/mock_test.go b/iam/credentials/apiv1/mock_test.go
deleted file mode 100644
index d24f492c53c..00000000000
--- a/iam/credentials/apiv1/mock_test.go
+++ /dev/null
@@ -1,394 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package credentials
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	credentialspb "google.golang.org/genproto/googleapis/iam/credentials/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockIamCredentialsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	credentialspb.IAMCredentialsServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockIamCredentialsServer) GenerateAccessToken(ctx context.Context, req *credentialspb.GenerateAccessTokenRequest) (*credentialspb.GenerateAccessTokenResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*credentialspb.GenerateAccessTokenResponse), nil
-}
-
-func (s *mockIamCredentialsServer) GenerateIdToken(ctx context.Context, req *credentialspb.GenerateIdTokenRequest) (*credentialspb.GenerateIdTokenResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*credentialspb.GenerateIdTokenResponse), nil
-}
-
-func (s *mockIamCredentialsServer) SignBlob(ctx context.Context, req *credentialspb.SignBlobRequest) (*credentialspb.SignBlobResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*credentialspb.SignBlobResponse), nil
-}
-
-func (s *mockIamCredentialsServer) SignJwt(ctx context.Context, req *credentialspb.SignJwtRequest) (*credentialspb.SignJwtResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*credentialspb.SignJwtResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockIamCredentials mockIamCredentialsServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	credentialspb.RegisterIAMCredentialsServer(serv, &mockIamCredentials)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestIamCredentialsGenerateAccessToken(t *testing.T) {
-	var accessToken string = "accessToken-1938933922"
-	var expectedResponse = &credentialspb.GenerateAccessTokenResponse{
-		AccessToken: accessToken,
-	}
-
-	mockIamCredentials.err = nil
-	mockIamCredentials.reqs = nil
-
-	mockIamCredentials.resps = append(mockIamCredentials.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var scope []string = nil
-	var request = &credentialspb.GenerateAccessTokenRequest{
-		Name:  formattedName,
-		Scope: scope,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GenerateAccessToken(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIamCredentials.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCredentialsGenerateAccessTokenError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIamCredentials.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var scope []string = nil
-	var request = &credentialspb.GenerateAccessTokenRequest{
-		Name:  formattedName,
-		Scope: scope,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GenerateAccessToken(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamCredentialsGenerateIdToken(t *testing.T) {
-	var token string = "token110541305"
-	var expectedResponse = &credentialspb.GenerateIdTokenResponse{
-		Token: token,
-	}
-
-	mockIamCredentials.err = nil
-	mockIamCredentials.reqs = nil
-
-	mockIamCredentials.resps = append(mockIamCredentials.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var audience string = "audience975628804"
-	var request = &credentialspb.GenerateIdTokenRequest{
-		Name:     formattedName,
-		Audience: audience,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GenerateIdToken(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIamCredentials.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCredentialsGenerateIdTokenError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIamCredentials.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var audience string = "audience975628804"
-	var request = &credentialspb.GenerateIdTokenRequest{
-		Name:     formattedName,
-		Audience: audience,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GenerateIdToken(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamCredentialsSignBlob(t *testing.T) {
-	var keyId string = "keyId-1134673157"
-	var signedBlob []byte = []byte("-32")
-	var expectedResponse = &credentialspb.SignBlobResponse{
-		KeyId:      keyId,
-		SignedBlob: signedBlob,
-	}
-
-	mockIamCredentials.err = nil
-	mockIamCredentials.reqs = nil
-
-	mockIamCredentials.resps = append(mockIamCredentials.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var payload []byte = []byte("-114")
-	var request = &credentialspb.SignBlobRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignBlob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIamCredentials.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCredentialsSignBlobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIamCredentials.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var payload []byte = []byte("-114")
-	var request = &credentialspb.SignBlobRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignBlob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestIamCredentialsSignJwt(t *testing.T) {
-	var keyId string = "keyId-1134673157"
-	var signedJwt string = "signedJwt-979546844"
-	var expectedResponse = &credentialspb.SignJwtResponse{
-		KeyId:     keyId,
-		SignedJwt: signedJwt,
-	}
-
-	mockIamCredentials.err = nil
-	mockIamCredentials.reqs = nil
-
-	mockIamCredentials.resps = append(mockIamCredentials.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var payload string = "-114"
-	var request = &credentialspb.SignJwtRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignJwt(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockIamCredentials.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestIamCredentialsSignJwtError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockIamCredentials.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/serviceAccounts/%s", "[PROJECT]", "[SERVICE_ACCOUNT]")
-	var payload string = "-114"
-	var request = &credentialspb.SignJwtRequest{
-		Name:    formattedName,
-		Payload: payload,
-	}
-
-	c, err := NewIamCredentialsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SignJwt(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/internal/examples/mock/mock_test.go b/internal/examples/mock/mock_test.go
deleted file mode 100644
index 59b20a88513..00000000000
--- a/internal/examples/mock/mock_test.go
+++ /dev/null
@@ -1,47 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-package mock
-
-import (
-	"context"
-	"testing"
-
-	"github.com/googleapis/gax-go/v2"
-	translatepb "google.golang.org/genproto/googleapis/cloud/translate/v3"
-)
-
-// mockClient fullfills the TranslationClient interface and is used as a mock
-// standin for a `translate.Client` that is only used to TranslateText.
-type mockClient struct{}
-
-func (*mockClient) TranslateText(_ context.Context, req *translatepb.TranslateTextRequest, opts ...gax.CallOption) (*translatepb.TranslateTextResponse, error) {
-	resp := &translatepb.TranslateTextResponse{
-		Translations: []*translatepb.Translation{
-			{TranslatedText: "Hello World"},
-		},
-	}
-	return resp, nil
-}
-
-func TestTranslateTextWithInterfaceClient(t *testing.T) {
-	client := &mockClient{}
-	text, err := TranslateTextWithInterfaceClient(client, "Hola Mundo", "en-US")
-	if err != nil {
-		t.Fatal(err)
-	}
-	if text != "Hello World" {
-		t.Fatalf("got %q, want Hello World", text)
-	}
-}
diff --git a/iot/apiv1/mock_test.go b/iot/apiv1/mock_test.go
deleted file mode 100644
index 89c2c4e5529..00000000000
--- a/iot/apiv1/mock_test.go
+++ /dev/null
@@ -1,1514 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package iot
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	iotpb "google.golang.org/genproto/googleapis/cloud/iot/v1"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockDeviceManagerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	iotpb.DeviceManagerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockDeviceManagerServer) CreateDeviceRegistry(ctx context.Context, req *iotpb.CreateDeviceRegistryRequest) (*iotpb.DeviceRegistry, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.DeviceRegistry), nil
-}
-
-func (s *mockDeviceManagerServer) GetDeviceRegistry(ctx context.Context, req *iotpb.GetDeviceRegistryRequest) (*iotpb.DeviceRegistry, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.DeviceRegistry), nil
-}
-
-func (s *mockDeviceManagerServer) UpdateDeviceRegistry(ctx context.Context, req *iotpb.UpdateDeviceRegistryRequest) (*iotpb.DeviceRegistry, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.DeviceRegistry), nil
-}
-
-func (s *mockDeviceManagerServer) DeleteDeviceRegistry(ctx context.Context, req *iotpb.DeleteDeviceRegistryRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDeviceManagerServer) ListDeviceRegistries(ctx context.Context, req *iotpb.ListDeviceRegistriesRequest) (*iotpb.ListDeviceRegistriesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.ListDeviceRegistriesResponse), nil
-}
-
-func (s *mockDeviceManagerServer) CreateDevice(ctx context.Context, req *iotpb.CreateDeviceRequest) (*iotpb.Device, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.Device), nil
-}
-
-func (s *mockDeviceManagerServer) GetDevice(ctx context.Context, req *iotpb.GetDeviceRequest) (*iotpb.Device, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.Device), nil
-}
-
-func (s *mockDeviceManagerServer) UpdateDevice(ctx context.Context, req *iotpb.UpdateDeviceRequest) (*iotpb.Device, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.Device), nil
-}
-
-func (s *mockDeviceManagerServer) DeleteDevice(ctx context.Context, req *iotpb.DeleteDeviceRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockDeviceManagerServer) ListDevices(ctx context.Context, req *iotpb.ListDevicesRequest) (*iotpb.ListDevicesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.ListDevicesResponse), nil
-}
-
-func (s *mockDeviceManagerServer) ModifyCloudToDeviceConfig(ctx context.Context, req *iotpb.ModifyCloudToDeviceConfigRequest) (*iotpb.DeviceConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.DeviceConfig), nil
-}
-
-func (s *mockDeviceManagerServer) ListDeviceConfigVersions(ctx context.Context, req *iotpb.ListDeviceConfigVersionsRequest) (*iotpb.ListDeviceConfigVersionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.ListDeviceConfigVersionsResponse), nil
-}
-
-func (s *mockDeviceManagerServer) ListDeviceStates(ctx context.Context, req *iotpb.ListDeviceStatesRequest) (*iotpb.ListDeviceStatesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.ListDeviceStatesResponse), nil
-}
-
-func (s *mockDeviceManagerServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockDeviceManagerServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockDeviceManagerServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockDeviceManagerServer) SendCommandToDevice(ctx context.Context, req *iotpb.SendCommandToDeviceRequest) (*iotpb.SendCommandToDeviceResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.SendCommandToDeviceResponse), nil
-}
-
-func (s *mockDeviceManagerServer) BindDeviceToGateway(ctx context.Context, req *iotpb.BindDeviceToGatewayRequest) (*iotpb.BindDeviceToGatewayResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.BindDeviceToGatewayResponse), nil
-}
-
-func (s *mockDeviceManagerServer) UnbindDeviceFromGateway(ctx context.Context, req *iotpb.UnbindDeviceFromGatewayRequest) (*iotpb.UnbindDeviceFromGatewayResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iotpb.UnbindDeviceFromGatewayResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockDeviceManager mockDeviceManagerServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	iotpb.RegisterDeviceManagerServer(serv, &mockDeviceManager)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestDeviceManagerCreateDeviceRegistry(t *testing.T) {
-	var id string = "id3355"
-	var name string = "name3373707"
-	var expectedResponse = &iotpb.DeviceRegistry{
-		Id:   id,
-		Name: name,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var deviceRegistry *iotpb.DeviceRegistry = &iotpb.DeviceRegistry{}
-	var request = &iotpb.CreateDeviceRegistryRequest{
-		Parent:         formattedParent,
-		DeviceRegistry: deviceRegistry,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDeviceRegistry(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerCreateDeviceRegistryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var deviceRegistry *iotpb.DeviceRegistry = &iotpb.DeviceRegistry{}
-	var request = &iotpb.CreateDeviceRegistryRequest{
-		Parent:         formattedParent,
-		DeviceRegistry: deviceRegistry,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDeviceRegistry(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerGetDeviceRegistry(t *testing.T) {
-	var id string = "id3355"
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &iotpb.DeviceRegistry{
-		Id:   id,
-		Name: name2,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var request = &iotpb.GetDeviceRegistryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDeviceRegistry(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerGetDeviceRegistryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var request = &iotpb.GetDeviceRegistryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDeviceRegistry(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerUpdateDeviceRegistry(t *testing.T) {
-	var id string = "id3355"
-	var name string = "name3373707"
-	var expectedResponse = &iotpb.DeviceRegistry{
-		Id:   id,
-		Name: name,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var deviceRegistry *iotpb.DeviceRegistry = &iotpb.DeviceRegistry{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &iotpb.UpdateDeviceRegistryRequest{
-		DeviceRegistry: deviceRegistry,
-		UpdateMask:     updateMask,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDeviceRegistry(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerUpdateDeviceRegistryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var deviceRegistry *iotpb.DeviceRegistry = &iotpb.DeviceRegistry{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &iotpb.UpdateDeviceRegistryRequest{
-		DeviceRegistry: deviceRegistry,
-		UpdateMask:     updateMask,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDeviceRegistry(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerDeleteDeviceRegistry(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var request = &iotpb.DeleteDeviceRegistryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDeviceRegistry(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDeviceManagerDeleteDeviceRegistryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var request = &iotpb.DeleteDeviceRegistryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDeviceRegistry(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDeviceManagerListDeviceRegistries(t *testing.T) {
-	var nextPageToken string = ""
-	var deviceRegistriesElement *iotpb.DeviceRegistry = &iotpb.DeviceRegistry{}
-	var deviceRegistries = []*iotpb.DeviceRegistry{deviceRegistriesElement}
-	var expectedResponse = &iotpb.ListDeviceRegistriesResponse{
-		NextPageToken:    nextPageToken,
-		DeviceRegistries: deviceRegistries,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &iotpb.ListDeviceRegistriesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeviceRegistries(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.DeviceRegistries[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerListDeviceRegistriesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &iotpb.ListDeviceRegistriesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeviceRegistries(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerCreateDevice(t *testing.T) {
-	var id string = "id3355"
-	var name string = "name3373707"
-	var numId uint64 = 1034366860
-	var blocked bool = true
-	var expectedResponse = &iotpb.Device{
-		Id:      id,
-		Name:    name,
-		NumId:   numId,
-		Blocked: blocked,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var device *iotpb.Device = &iotpb.Device{}
-	var request = &iotpb.CreateDeviceRequest{
-		Parent: formattedParent,
-		Device: device,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDevice(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerCreateDeviceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var device *iotpb.Device = &iotpb.Device{}
-	var request = &iotpb.CreateDeviceRequest{
-		Parent: formattedParent,
-		Device: device,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateDevice(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerGetDevice(t *testing.T) {
-	var id string = "id3355"
-	var name2 string = "name2-1052831874"
-	var numId uint64 = 1034366860
-	var blocked bool = true
-	var expectedResponse = &iotpb.Device{
-		Id:      id,
-		Name:    name2,
-		NumId:   numId,
-		Blocked: blocked,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.GetDeviceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDevice(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerGetDeviceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.GetDeviceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetDevice(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerUpdateDevice(t *testing.T) {
-	var id string = "id3355"
-	var name string = "name3373707"
-	var numId uint64 = 1034366860
-	var blocked bool = true
-	var expectedResponse = &iotpb.Device{
-		Id:      id,
-		Name:    name,
-		NumId:   numId,
-		Blocked: blocked,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var device *iotpb.Device = &iotpb.Device{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &iotpb.UpdateDeviceRequest{
-		Device:     device,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDevice(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerUpdateDeviceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var device *iotpb.Device = &iotpb.Device{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &iotpb.UpdateDeviceRequest{
-		Device:     device,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateDevice(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerDeleteDevice(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.DeleteDeviceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDevice(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestDeviceManagerDeleteDeviceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.DeleteDeviceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteDevice(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestDeviceManagerListDevices(t *testing.T) {
-	var nextPageToken string = ""
-	var devicesElement *iotpb.Device = &iotpb.Device{}
-	var devices = []*iotpb.Device{devicesElement}
-	var expectedResponse = &iotpb.ListDevicesResponse{
-		NextPageToken: nextPageToken,
-		Devices:       devices,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var request = &iotpb.ListDevicesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDevices(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Devices[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerListDevicesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var request = &iotpb.ListDevicesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDevices(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerModifyCloudToDeviceConfig(t *testing.T) {
-	var version int64 = 351608024
-	var binaryData2 []byte = []byte("-37")
-	var expectedResponse = &iotpb.DeviceConfig{
-		Version:    version,
-		BinaryData: binaryData2,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var binaryData []byte = []byte("40")
-	var request = &iotpb.ModifyCloudToDeviceConfigRequest{
-		Name:       formattedName,
-		BinaryData: binaryData,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ModifyCloudToDeviceConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerModifyCloudToDeviceConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var binaryData []byte = []byte("40")
-	var request = &iotpb.ModifyCloudToDeviceConfigRequest{
-		Name:       formattedName,
-		BinaryData: binaryData,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ModifyCloudToDeviceConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerListDeviceConfigVersions(t *testing.T) {
-	var expectedResponse *iotpb.ListDeviceConfigVersionsResponse = &iotpb.ListDeviceConfigVersionsResponse{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.ListDeviceConfigVersionsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeviceConfigVersions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerListDeviceConfigVersionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.ListDeviceConfigVersionsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeviceConfigVersions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerListDeviceStates(t *testing.T) {
-	var expectedResponse *iotpb.ListDeviceStatesResponse = &iotpb.ListDeviceStatesResponse{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.ListDeviceStatesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeviceStates(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerListDeviceStatesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var request = &iotpb.ListDeviceStatesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListDeviceStates(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var resource string = "resource-341064690"
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: resource,
-		Policy:   policy,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var resource string = "resource-341064690"
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: resource,
-		Policy:   policy,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var resource string = "resource-341064690"
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: resource,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var resource string = "resource-341064690"
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: resource,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var resource string = "resource-341064690"
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    resource,
-		Permissions: permissions,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var resource string = "resource-341064690"
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    resource,
-		Permissions: permissions,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerSendCommandToDevice(t *testing.T) {
-	var expectedResponse *iotpb.SendCommandToDeviceResponse = &iotpb.SendCommandToDeviceResponse{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var binaryData []byte = []byte("40")
-	var request = &iotpb.SendCommandToDeviceRequest{
-		Name:       formattedName,
-		BinaryData: binaryData,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SendCommandToDevice(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerSendCommandToDeviceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/registries/%s/devices/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]", "[DEVICE]")
-	var binaryData []byte = []byte("40")
-	var request = &iotpb.SendCommandToDeviceRequest{
-		Name:       formattedName,
-		BinaryData: binaryData,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SendCommandToDevice(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerBindDeviceToGateway(t *testing.T) {
-	var expectedResponse *iotpb.BindDeviceToGatewayResponse = &iotpb.BindDeviceToGatewayResponse{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var gatewayId string = "gatewayId955798774"
-	var deviceId string = "deviceId25209764"
-	var request = &iotpb.BindDeviceToGatewayRequest{
-		Parent:    formattedParent,
-		GatewayId: gatewayId,
-		DeviceId:  deviceId,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BindDeviceToGateway(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerBindDeviceToGatewayError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var gatewayId string = "gatewayId955798774"
-	var deviceId string = "deviceId25209764"
-	var request = &iotpb.BindDeviceToGatewayRequest{
-		Parent:    formattedParent,
-		GatewayId: gatewayId,
-		DeviceId:  deviceId,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BindDeviceToGateway(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestDeviceManagerUnbindDeviceFromGateway(t *testing.T) {
-	var expectedResponse *iotpb.UnbindDeviceFromGatewayResponse = &iotpb.UnbindDeviceFromGatewayResponse{}
-
-	mockDeviceManager.err = nil
-	mockDeviceManager.reqs = nil
-
-	mockDeviceManager.resps = append(mockDeviceManager.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var gatewayId string = "gatewayId955798774"
-	var deviceId string = "deviceId25209764"
-	var request = &iotpb.UnbindDeviceFromGatewayRequest{
-		Parent:    formattedParent,
-		GatewayId: gatewayId,
-		DeviceId:  deviceId,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UnbindDeviceFromGateway(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockDeviceManager.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestDeviceManagerUnbindDeviceFromGatewayError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockDeviceManager.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/registries/%s", "[PROJECT]", "[LOCATION]", "[REGISTRY]")
-	var gatewayId string = "gatewayId955798774"
-	var deviceId string = "deviceId25209764"
-	var request = &iotpb.UnbindDeviceFromGatewayRequest{
-		Parent:    formattedParent,
-		GatewayId: gatewayId,
-		DeviceId:  deviceId,
-	}
-
-	c, err := NewDeviceManagerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UnbindDeviceFromGateway(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/kms/apiv1/mock_test.go b/kms/apiv1/mock_test.go
deleted file mode 100644
index 5664811e3be..00000000000
--- a/kms/apiv1/mock_test.go
+++ /dev/null
@@ -1,1916 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package kms
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	durationpb "github.com/golang/protobuf/ptypes/duration"
-	timestamppb "github.com/golang/protobuf/ptypes/timestamp"
-	"google.golang.org/api/option"
-	kmspb "google.golang.org/genproto/googleapis/cloud/kms/v1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockKeyManagementServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	kmspb.KeyManagementServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockKeyManagementServer) ListKeyRings(ctx context.Context, req *kmspb.ListKeyRingsRequest) (*kmspb.ListKeyRingsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.ListKeyRingsResponse), nil
-}
-
-func (s *mockKeyManagementServer) ListCryptoKeys(ctx context.Context, req *kmspb.ListCryptoKeysRequest) (*kmspb.ListCryptoKeysResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.ListCryptoKeysResponse), nil
-}
-
-func (s *mockKeyManagementServer) ListCryptoKeyVersions(ctx context.Context, req *kmspb.ListCryptoKeyVersionsRequest) (*kmspb.ListCryptoKeyVersionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.ListCryptoKeyVersionsResponse), nil
-}
-
-func (s *mockKeyManagementServer) ListImportJobs(ctx context.Context, req *kmspb.ListImportJobsRequest) (*kmspb.ListImportJobsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.ListImportJobsResponse), nil
-}
-
-func (s *mockKeyManagementServer) GetKeyRing(ctx context.Context, req *kmspb.GetKeyRingRequest) (*kmspb.KeyRing, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.KeyRing), nil
-}
-
-func (s *mockKeyManagementServer) GetCryptoKey(ctx context.Context, req *kmspb.GetCryptoKeyRequest) (*kmspb.CryptoKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKey), nil
-}
-
-func (s *mockKeyManagementServer) GetCryptoKeyVersion(ctx context.Context, req *kmspb.GetCryptoKeyVersionRequest) (*kmspb.CryptoKeyVersion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKeyVersion), nil
-}
-
-func (s *mockKeyManagementServer) GetPublicKey(ctx context.Context, req *kmspb.GetPublicKeyRequest) (*kmspb.PublicKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.PublicKey), nil
-}
-
-func (s *mockKeyManagementServer) GetImportJob(ctx context.Context, req *kmspb.GetImportJobRequest) (*kmspb.ImportJob, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.ImportJob), nil
-}
-
-func (s *mockKeyManagementServer) CreateKeyRing(ctx context.Context, req *kmspb.CreateKeyRingRequest) (*kmspb.KeyRing, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.KeyRing), nil
-}
-
-func (s *mockKeyManagementServer) CreateCryptoKey(ctx context.Context, req *kmspb.CreateCryptoKeyRequest) (*kmspb.CryptoKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKey), nil
-}
-
-func (s *mockKeyManagementServer) CreateCryptoKeyVersion(ctx context.Context, req *kmspb.CreateCryptoKeyVersionRequest) (*kmspb.CryptoKeyVersion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKeyVersion), nil
-}
-
-func (s *mockKeyManagementServer) ImportCryptoKeyVersion(ctx context.Context, req *kmspb.ImportCryptoKeyVersionRequest) (*kmspb.CryptoKeyVersion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKeyVersion), nil
-}
-
-func (s *mockKeyManagementServer) CreateImportJob(ctx context.Context, req *kmspb.CreateImportJobRequest) (*kmspb.ImportJob, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.ImportJob), nil
-}
-
-func (s *mockKeyManagementServer) UpdateCryptoKey(ctx context.Context, req *kmspb.UpdateCryptoKeyRequest) (*kmspb.CryptoKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKey), nil
-}
-
-func (s *mockKeyManagementServer) UpdateCryptoKeyVersion(ctx context.Context, req *kmspb.UpdateCryptoKeyVersionRequest) (*kmspb.CryptoKeyVersion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKeyVersion), nil
-}
-
-func (s *mockKeyManagementServer) Encrypt(ctx context.Context, req *kmspb.EncryptRequest) (*kmspb.EncryptResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.EncryptResponse), nil
-}
-
-func (s *mockKeyManagementServer) Decrypt(ctx context.Context, req *kmspb.DecryptRequest) (*kmspb.DecryptResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.DecryptResponse), nil
-}
-
-func (s *mockKeyManagementServer) AsymmetricSign(ctx context.Context, req *kmspb.AsymmetricSignRequest) (*kmspb.AsymmetricSignResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.AsymmetricSignResponse), nil
-}
-
-func (s *mockKeyManagementServer) AsymmetricDecrypt(ctx context.Context, req *kmspb.AsymmetricDecryptRequest) (*kmspb.AsymmetricDecryptResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.AsymmetricDecryptResponse), nil
-}
-
-func (s *mockKeyManagementServer) UpdateCryptoKeyPrimaryVersion(ctx context.Context, req *kmspb.UpdateCryptoKeyPrimaryVersionRequest) (*kmspb.CryptoKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKey), nil
-}
-
-func (s *mockKeyManagementServer) DestroyCryptoKeyVersion(ctx context.Context, req *kmspb.DestroyCryptoKeyVersionRequest) (*kmspb.CryptoKeyVersion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKeyVersion), nil
-}
-
-func (s *mockKeyManagementServer) RestoreCryptoKeyVersion(ctx context.Context, req *kmspb.RestoreCryptoKeyVersionRequest) (*kmspb.CryptoKeyVersion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*kmspb.CryptoKeyVersion), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockKeyManagement mockKeyManagementServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	kmspb.RegisterKeyManagementServiceServer(serv, &mockKeyManagement)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestKeyManagementServiceListKeyRings(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var keyRingsElement *kmspb.KeyRing = &kmspb.KeyRing{}
-	var keyRings = []*kmspb.KeyRing{keyRingsElement}
-	var expectedResponse = &kmspb.ListKeyRingsResponse{
-		NextPageToken: nextPageToken,
-		TotalSize:     totalSize,
-		KeyRings:      keyRings,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &kmspb.ListKeyRingsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListKeyRings(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.KeyRings[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceListKeyRingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &kmspb.ListKeyRingsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListKeyRings(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceListImportJobs(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var importJobsElement *kmspb.ImportJob = &kmspb.ImportJob{}
-	var importJobs = []*kmspb.ImportJob{importJobsElement}
-	var expectedResponse = &kmspb.ListImportJobsResponse{
-		NextPageToken: nextPageToken,
-		TotalSize:     totalSize,
-		ImportJobs:    importJobs,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var request = &kmspb.ListImportJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListImportJobs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ImportJobs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceListImportJobsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var request = &kmspb.ListImportJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListImportJobs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceListCryptoKeys(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var cryptoKeysElement *kmspb.CryptoKey = &kmspb.CryptoKey{}
-	var cryptoKeys = []*kmspb.CryptoKey{cryptoKeysElement}
-	var expectedResponse = &kmspb.ListCryptoKeysResponse{
-		NextPageToken: nextPageToken,
-		TotalSize:     totalSize,
-		CryptoKeys:    cryptoKeys,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var request = &kmspb.ListCryptoKeysRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCryptoKeys(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.CryptoKeys[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceListCryptoKeysError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var request = &kmspb.ListCryptoKeysRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCryptoKeys(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceListCryptoKeyVersions(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var cryptoKeyVersionsElement *kmspb.CryptoKeyVersion = &kmspb.CryptoKeyVersion{}
-	var cryptoKeyVersions = []*kmspb.CryptoKeyVersion{cryptoKeyVersionsElement}
-	var expectedResponse = &kmspb.ListCryptoKeyVersionsResponse{
-		NextPageToken:     nextPageToken,
-		TotalSize:         totalSize,
-		CryptoKeyVersions: cryptoKeyVersions,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var request = &kmspb.ListCryptoKeyVersionsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCryptoKeyVersions(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.CryptoKeyVersions[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceListCryptoKeyVersionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var request = &kmspb.ListCryptoKeyVersionsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListCryptoKeyVersions(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceGetKeyRing(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &kmspb.KeyRing{
-		Name: name2,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var request = &kmspb.GetKeyRingRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetKeyRing(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceGetKeyRingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var request = &kmspb.GetKeyRingRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetKeyRing(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceGetImportJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &kmspb.ImportJob{
-		Name: name2,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/importJobs/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[IMPORT_JOB]")
-	var request = &kmspb.GetImportJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetImportJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceGetImportJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/importJobs/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[IMPORT_JOB]")
-	var request = &kmspb.GetImportJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetImportJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceGetCryptoKey(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &kmspb.CryptoKey{
-		Name: name2,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var request = &kmspb.GetCryptoKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCryptoKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceGetCryptoKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var request = &kmspb.GetCryptoKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCryptoKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceGetCryptoKeyVersion(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var importJob string = "importJob2125587491"
-	var importFailureReason string = "importFailureReason-494073229"
-	var expectedResponse = &kmspb.CryptoKeyVersion{
-		Name:                name2,
-		ImportJob:           importJob,
-		ImportFailureReason: importFailureReason,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.GetCryptoKeyVersionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCryptoKeyVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceGetCryptoKeyVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.GetCryptoKeyVersionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCryptoKeyVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceCreateKeyRing(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &kmspb.KeyRing{
-		Name: name,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var keyRingId string = "keyRingId-2056646742"
-	var keyRing *kmspb.KeyRing = &kmspb.KeyRing{}
-	var request = &kmspb.CreateKeyRingRequest{
-		Parent:    formattedParent,
-		KeyRingId: keyRingId,
-		KeyRing:   keyRing,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateKeyRing(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceCreateKeyRingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var keyRingId string = "keyRingId-2056646742"
-	var keyRing *kmspb.KeyRing = &kmspb.KeyRing{}
-	var request = &kmspb.CreateKeyRingRequest{
-		Parent:    formattedParent,
-		KeyRingId: keyRingId,
-		KeyRing:   keyRing,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateKeyRing(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceCreateImportJob(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &kmspb.ImportJob{
-		Name: name,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var importJobId string = "my-import-job"
-	var importMethod kmspb.ImportJob_ImportMethod = kmspb.ImportJob_RSA_OAEP_3072_SHA1_AES_256
-	var protectionLevel kmspb.ProtectionLevel = kmspb.ProtectionLevel_HSM
-	var importJob = &kmspb.ImportJob{
-		ImportMethod:    importMethod,
-		ProtectionLevel: protectionLevel,
-	}
-	var request = &kmspb.CreateImportJobRequest{
-		Parent:      formattedParent,
-		ImportJobId: importJobId,
-		ImportJob:   importJob,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateImportJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceCreateImportJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var importJobId string = "my-import-job"
-	var importMethod kmspb.ImportJob_ImportMethod = kmspb.ImportJob_RSA_OAEP_3072_SHA1_AES_256
-	var protectionLevel kmspb.ProtectionLevel = kmspb.ProtectionLevel_HSM
-	var importJob = &kmspb.ImportJob{
-		ImportMethod:    importMethod,
-		ProtectionLevel: protectionLevel,
-	}
-	var request = &kmspb.CreateImportJobRequest{
-		Parent:      formattedParent,
-		ImportJobId: importJobId,
-		ImportJob:   importJob,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateImportJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceCreateCryptoKey(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &kmspb.CryptoKey{
-		Name: name,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var cryptoKeyId string = "my-app-key"
-	var purpose kmspb.CryptoKey_CryptoKeyPurpose = kmspb.CryptoKey_ENCRYPT_DECRYPT
-	var seconds int64 = 2147483647
-	var nextRotationTime = &timestamppb.Timestamp{
-		Seconds: seconds,
-	}
-	var seconds2 int64 = 604800
-	var rotationPeriod = &durationpb.Duration{
-		Seconds: seconds2,
-	}
-	var cryptoKey = &kmspb.CryptoKey{
-		Purpose:          purpose,
-		NextRotationTime: nextRotationTime,
-		RotationSchedule: &kmspb.CryptoKey_RotationPeriod{
-			RotationPeriod: rotationPeriod,
-		},
-	}
-	var request = &kmspb.CreateCryptoKeyRequest{
-		Parent:      formattedParent,
-		CryptoKeyId: cryptoKeyId,
-		CryptoKey:   cryptoKey,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateCryptoKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceCreateCryptoKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]")
-	var cryptoKeyId string = "my-app-key"
-	var purpose kmspb.CryptoKey_CryptoKeyPurpose = kmspb.CryptoKey_ENCRYPT_DECRYPT
-	var seconds int64 = 2147483647
-	var nextRotationTime = &timestamppb.Timestamp{
-		Seconds: seconds,
-	}
-	var seconds2 int64 = 604800
-	var rotationPeriod = &durationpb.Duration{
-		Seconds: seconds2,
-	}
-	var cryptoKey = &kmspb.CryptoKey{
-		Purpose:          purpose,
-		NextRotationTime: nextRotationTime,
-		RotationSchedule: &kmspb.CryptoKey_RotationPeriod{
-			RotationPeriod: rotationPeriod,
-		},
-	}
-	var request = &kmspb.CreateCryptoKeyRequest{
-		Parent:      formattedParent,
-		CryptoKeyId: cryptoKeyId,
-		CryptoKey:   cryptoKey,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateCryptoKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceCreateCryptoKeyVersion(t *testing.T) {
-	var name string = "name3373707"
-	var importJob string = "importJob2125587491"
-	var importFailureReason string = "importFailureReason-494073229"
-	var expectedResponse = &kmspb.CryptoKeyVersion{
-		Name:                name,
-		ImportJob:           importJob,
-		ImportFailureReason: importFailureReason,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var cryptoKeyVersion *kmspb.CryptoKeyVersion = &kmspb.CryptoKeyVersion{}
-	var request = &kmspb.CreateCryptoKeyVersionRequest{
-		Parent:           formattedParent,
-		CryptoKeyVersion: cryptoKeyVersion,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateCryptoKeyVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceCreateCryptoKeyVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var cryptoKeyVersion *kmspb.CryptoKeyVersion = &kmspb.CryptoKeyVersion{}
-	var request = &kmspb.CreateCryptoKeyVersionRequest{
-		Parent:           formattedParent,
-		CryptoKeyVersion: cryptoKeyVersion,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateCryptoKeyVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceImportCryptoKeyVersion(t *testing.T) {
-	var name string = "name3373707"
-	var importJob2 string = "importJob2-1714851050"
-	var importFailureReason string = "importFailureReason-494073229"
-	var expectedResponse = &kmspb.CryptoKeyVersion{
-		Name:                name,
-		ImportJob:           importJob2,
-		ImportFailureReason: importFailureReason,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var algorithm kmspb.CryptoKeyVersion_CryptoKeyVersionAlgorithm = kmspb.CryptoKeyVersion_CRYPTO_KEY_VERSION_ALGORITHM_UNSPECIFIED
-	var importJob string = "importJob2125587491"
-	var request = &kmspb.ImportCryptoKeyVersionRequest{
-		Parent:    formattedParent,
-		Algorithm: algorithm,
-		ImportJob: importJob,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportCryptoKeyVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceImportCryptoKeyVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var algorithm kmspb.CryptoKeyVersion_CryptoKeyVersionAlgorithm = kmspb.CryptoKeyVersion_CRYPTO_KEY_VERSION_ALGORITHM_UNSPECIFIED
-	var importJob string = "importJob2125587491"
-	var request = &kmspb.ImportCryptoKeyVersionRequest{
-		Parent:    formattedParent,
-		Algorithm: algorithm,
-		ImportJob: importJob,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportCryptoKeyVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceUpdateCryptoKey(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &kmspb.CryptoKey{
-		Name: name,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var cryptoKey *kmspb.CryptoKey = &kmspb.CryptoKey{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &kmspb.UpdateCryptoKeyRequest{
-		CryptoKey:  cryptoKey,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCryptoKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceUpdateCryptoKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var cryptoKey *kmspb.CryptoKey = &kmspb.CryptoKey{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &kmspb.UpdateCryptoKeyRequest{
-		CryptoKey:  cryptoKey,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCryptoKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceUpdateCryptoKeyVersion(t *testing.T) {
-	var name string = "name3373707"
-	var importJob string = "importJob2125587491"
-	var importFailureReason string = "importFailureReason-494073229"
-	var expectedResponse = &kmspb.CryptoKeyVersion{
-		Name:                name,
-		ImportJob:           importJob,
-		ImportFailureReason: importFailureReason,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var cryptoKeyVersion *kmspb.CryptoKeyVersion = &kmspb.CryptoKeyVersion{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &kmspb.UpdateCryptoKeyVersionRequest{
-		CryptoKeyVersion: cryptoKeyVersion,
-		UpdateMask:       updateMask,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCryptoKeyVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceUpdateCryptoKeyVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var cryptoKeyVersion *kmspb.CryptoKeyVersion = &kmspb.CryptoKeyVersion{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &kmspb.UpdateCryptoKeyVersionRequest{
-		CryptoKeyVersion: cryptoKeyVersion,
-		UpdateMask:       updateMask,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCryptoKeyVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceEncrypt(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var ciphertext []byte = []byte("-72")
-	var expectedResponse = &kmspb.EncryptResponse{
-		Name:       name2,
-		Ciphertext: ciphertext,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY_PATH]")
-	var plaintext []byte = []byte("-9")
-	var request = &kmspb.EncryptRequest{
-		Name:      formattedName,
-		Plaintext: plaintext,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Encrypt(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceEncryptError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY_PATH]")
-	var plaintext []byte = []byte("-9")
-	var request = &kmspb.EncryptRequest{
-		Name:      formattedName,
-		Plaintext: plaintext,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Encrypt(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceDecrypt(t *testing.T) {
-	var plaintext []byte = []byte("-9")
-	var expectedResponse = &kmspb.DecryptResponse{
-		Plaintext: plaintext,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var ciphertext []byte = []byte("-72")
-	var request = &kmspb.DecryptRequest{
-		Name:       formattedName,
-		Ciphertext: ciphertext,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Decrypt(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceDecryptError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var ciphertext []byte = []byte("-72")
-	var request = &kmspb.DecryptRequest{
-		Name:       formattedName,
-		Ciphertext: ciphertext,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Decrypt(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceUpdateCryptoKeyPrimaryVersion(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &kmspb.CryptoKey{
-		Name: name2,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var cryptoKeyVersionId string = "cryptoKeyVersionId729489152"
-	var request = &kmspb.UpdateCryptoKeyPrimaryVersionRequest{
-		Name:               formattedName,
-		CryptoKeyVersionId: cryptoKeyVersionId,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCryptoKeyPrimaryVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceUpdateCryptoKeyPrimaryVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]")
-	var cryptoKeyVersionId string = "cryptoKeyVersionId729489152"
-	var request = &kmspb.UpdateCryptoKeyPrimaryVersionRequest{
-		Name:               formattedName,
-		CryptoKeyVersionId: cryptoKeyVersionId,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCryptoKeyPrimaryVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceDestroyCryptoKeyVersion(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var importJob string = "importJob2125587491"
-	var importFailureReason string = "importFailureReason-494073229"
-	var expectedResponse = &kmspb.CryptoKeyVersion{
-		Name:                name2,
-		ImportJob:           importJob,
-		ImportFailureReason: importFailureReason,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.DestroyCryptoKeyVersionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DestroyCryptoKeyVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceDestroyCryptoKeyVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.DestroyCryptoKeyVersionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DestroyCryptoKeyVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceRestoreCryptoKeyVersion(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var importJob string = "importJob2125587491"
-	var importFailureReason string = "importFailureReason-494073229"
-	var expectedResponse = &kmspb.CryptoKeyVersion{
-		Name:                name2,
-		ImportJob:           importJob,
-		ImportFailureReason: importFailureReason,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.RestoreCryptoKeyVersionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RestoreCryptoKeyVersion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceRestoreCryptoKeyVersionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.RestoreCryptoKeyVersionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RestoreCryptoKeyVersion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceGetPublicKey(t *testing.T) {
-	var pem string = "pem110872"
-	var expectedResponse = &kmspb.PublicKey{
-		Pem: pem,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.GetPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceGetPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var request = &kmspb.GetPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceAsymmetricDecrypt(t *testing.T) {
-	var plaintext []byte = []byte("-9")
-	var expectedResponse = &kmspb.AsymmetricDecryptResponse{
-		Plaintext: plaintext,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var ciphertext []byte = []byte("-72")
-	var request = &kmspb.AsymmetricDecryptRequest{
-		Name:       formattedName,
-		Ciphertext: ciphertext,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AsymmetricDecrypt(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceAsymmetricDecryptError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var ciphertext []byte = []byte("-72")
-	var request = &kmspb.AsymmetricDecryptRequest{
-		Name:       formattedName,
-		Ciphertext: ciphertext,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AsymmetricDecrypt(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestKeyManagementServiceAsymmetricSign(t *testing.T) {
-	var signature []byte = []byte("106")
-	var expectedResponse = &kmspb.AsymmetricSignResponse{
-		Signature: signature,
-	}
-
-	mockKeyManagement.err = nil
-	mockKeyManagement.reqs = nil
-
-	mockKeyManagement.resps = append(mockKeyManagement.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var digest *kmspb.Digest = &kmspb.Digest{}
-	var request = &kmspb.AsymmetricSignRequest{
-		Name:   formattedName,
-		Digest: digest,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AsymmetricSign(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockKeyManagement.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestKeyManagementServiceAsymmetricSignError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockKeyManagement.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/keyRings/%s/cryptoKeys/%s/cryptoKeyVersions/%s", "[PROJECT]", "[LOCATION]", "[KEY_RING]", "[CRYPTO_KEY]", "[CRYPTO_KEY_VERSION]")
-	var digest *kmspb.Digest = &kmspb.Digest{}
-	var request = &kmspb.AsymmetricSignRequest{
-		Name:   formattedName,
-		Digest: digest,
-	}
-
-	c, err := NewKeyManagementClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AsymmetricSign(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/language/apiv1/mock_test.go b/language/apiv1/mock_test.go
deleted file mode 100644
index 0eb6b9b46cb..00000000000
--- a/language/apiv1/mock_test.go
+++ /dev/null
@@ -1,517 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package language
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	languagepb "google.golang.org/genproto/googleapis/cloud/language/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockLanguageServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	languagepb.LanguageServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockLanguageServer) AnalyzeSentiment(ctx context.Context, req *languagepb.AnalyzeSentimentRequest) (*languagepb.AnalyzeSentimentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeSentimentResponse), nil
-}
-
-func (s *mockLanguageServer) AnalyzeEntities(ctx context.Context, req *languagepb.AnalyzeEntitiesRequest) (*languagepb.AnalyzeEntitiesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeEntitiesResponse), nil
-}
-
-func (s *mockLanguageServer) AnalyzeEntitySentiment(ctx context.Context, req *languagepb.AnalyzeEntitySentimentRequest) (*languagepb.AnalyzeEntitySentimentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeEntitySentimentResponse), nil
-}
-
-func (s *mockLanguageServer) AnalyzeSyntax(ctx context.Context, req *languagepb.AnalyzeSyntaxRequest) (*languagepb.AnalyzeSyntaxResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeSyntaxResponse), nil
-}
-
-func (s *mockLanguageServer) ClassifyText(ctx context.Context, req *languagepb.ClassifyTextRequest) (*languagepb.ClassifyTextResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.ClassifyTextResponse), nil
-}
-
-func (s *mockLanguageServer) AnnotateText(ctx context.Context, req *languagepb.AnnotateTextRequest) (*languagepb.AnnotateTextResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnnotateTextResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockLanguage mockLanguageServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	languagepb.RegisterLanguageServiceServer(serv, &mockLanguage)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestLanguageServiceAnalyzeSentiment(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeSentimentResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSentiment(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeSentimentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSentiment(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnalyzeEntities(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeEntitiesResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitiesRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntities(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeEntitiesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitiesRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntities(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnalyzeEntitySentiment(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeEntitySentimentResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitySentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntitySentiment(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeEntitySentimentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitySentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntitySentiment(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnalyzeSyntax(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeSyntaxResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSyntaxRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSyntax(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeSyntaxError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSyntaxRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSyntax(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceClassifyText(t *testing.T) {
-	var expectedResponse *languagepb.ClassifyTextResponse = &languagepb.ClassifyTextResponse{}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.ClassifyTextRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ClassifyText(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceClassifyTextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.ClassifyTextRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ClassifyText(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnnotateText(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnnotateTextResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var features *languagepb.AnnotateTextRequest_Features = &languagepb.AnnotateTextRequest_Features{}
-	var request = &languagepb.AnnotateTextRequest{
-		Document: document,
-		Features: features,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnnotateText(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnnotateTextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var features *languagepb.AnnotateTextRequest_Features = &languagepb.AnnotateTextRequest_Features{}
-	var request = &languagepb.AnnotateTextRequest{
-		Document: document,
-		Features: features,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnnotateText(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/language/apiv1beta2/mock_test.go b/language/apiv1beta2/mock_test.go
deleted file mode 100644
index c40a592a08a..00000000000
--- a/language/apiv1beta2/mock_test.go
+++ /dev/null
@@ -1,517 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package language
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	languagepb "google.golang.org/genproto/googleapis/cloud/language/v1beta2"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockLanguageServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	languagepb.LanguageServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockLanguageServer) AnalyzeSentiment(ctx context.Context, req *languagepb.AnalyzeSentimentRequest) (*languagepb.AnalyzeSentimentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeSentimentResponse), nil
-}
-
-func (s *mockLanguageServer) AnalyzeEntities(ctx context.Context, req *languagepb.AnalyzeEntitiesRequest) (*languagepb.AnalyzeEntitiesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeEntitiesResponse), nil
-}
-
-func (s *mockLanguageServer) AnalyzeEntitySentiment(ctx context.Context, req *languagepb.AnalyzeEntitySentimentRequest) (*languagepb.AnalyzeEntitySentimentResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeEntitySentimentResponse), nil
-}
-
-func (s *mockLanguageServer) AnalyzeSyntax(ctx context.Context, req *languagepb.AnalyzeSyntaxRequest) (*languagepb.AnalyzeSyntaxResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnalyzeSyntaxResponse), nil
-}
-
-func (s *mockLanguageServer) ClassifyText(ctx context.Context, req *languagepb.ClassifyTextRequest) (*languagepb.ClassifyTextResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.ClassifyTextResponse), nil
-}
-
-func (s *mockLanguageServer) AnnotateText(ctx context.Context, req *languagepb.AnnotateTextRequest) (*languagepb.AnnotateTextResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*languagepb.AnnotateTextResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockLanguage mockLanguageServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	languagepb.RegisterLanguageServiceServer(serv, &mockLanguage)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestLanguageServiceAnalyzeSentiment(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeSentimentResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSentiment(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeSentimentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSentiment(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnalyzeEntities(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeEntitiesResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitiesRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntities(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeEntitiesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitiesRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntities(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnalyzeEntitySentiment(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeEntitySentimentResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitySentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntitySentiment(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeEntitySentimentError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeEntitySentimentRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeEntitySentiment(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnalyzeSyntax(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnalyzeSyntaxResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSyntaxRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSyntax(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnalyzeSyntaxError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.AnalyzeSyntaxRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnalyzeSyntax(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceClassifyText(t *testing.T) {
-	var expectedResponse *languagepb.ClassifyTextResponse = &languagepb.ClassifyTextResponse{}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.ClassifyTextRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ClassifyText(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceClassifyTextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var request = &languagepb.ClassifyTextRequest{
-		Document: document,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ClassifyText(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLanguageServiceAnnotateText(t *testing.T) {
-	var language string = "language-1613589672"
-	var expectedResponse = &languagepb.AnnotateTextResponse{
-		Language: language,
-	}
-
-	mockLanguage.err = nil
-	mockLanguage.reqs = nil
-
-	mockLanguage.resps = append(mockLanguage.resps[:0], expectedResponse)
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var features *languagepb.AnnotateTextRequest_Features = &languagepb.AnnotateTextRequest_Features{}
-	var request = &languagepb.AnnotateTextRequest{
-		Document: document,
-		Features: features,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnnotateText(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLanguage.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLanguageServiceAnnotateTextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLanguage.err = gstatus.Error(errCode, "test error")
-
-	var document *languagepb.Document = &languagepb.Document{}
-	var features *languagepb.AnnotateTextRequest_Features = &languagepb.AnnotateTextRequest_Features{}
-	var request = &languagepb.AnnotateTextRequest{
-		Document: document,
-		Features: features,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.AnnotateText(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/logging/apiv2/mock_test.go b/logging/apiv2/mock_test.go
deleted file mode 100644
index f6de3390b57..00000000000
--- a/logging/apiv2/mock_test.go
+++ /dev/null
@@ -1,1826 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package logging
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	monitoredrespb "google.golang.org/genproto/googleapis/api/monitoredres"
-	loggingpb "google.golang.org/genproto/googleapis/logging/v2"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockConfigServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	loggingpb.ConfigServiceV2Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockConfigServer) ListSinks(ctx context.Context, req *loggingpb.ListSinksRequest) (*loggingpb.ListSinksResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.ListSinksResponse), nil
-}
-
-func (s *mockConfigServer) GetSink(ctx context.Context, req *loggingpb.GetSinkRequest) (*loggingpb.LogSink, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogSink), nil
-}
-
-func (s *mockConfigServer) CreateSink(ctx context.Context, req *loggingpb.CreateSinkRequest) (*loggingpb.LogSink, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogSink), nil
-}
-
-func (s *mockConfigServer) UpdateSink(ctx context.Context, req *loggingpb.UpdateSinkRequest) (*loggingpb.LogSink, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogSink), nil
-}
-
-func (s *mockConfigServer) DeleteSink(ctx context.Context, req *loggingpb.DeleteSinkRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockConfigServer) ListExclusions(ctx context.Context, req *loggingpb.ListExclusionsRequest) (*loggingpb.ListExclusionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.ListExclusionsResponse), nil
-}
-
-func (s *mockConfigServer) GetExclusion(ctx context.Context, req *loggingpb.GetExclusionRequest) (*loggingpb.LogExclusion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogExclusion), nil
-}
-
-func (s *mockConfigServer) CreateExclusion(ctx context.Context, req *loggingpb.CreateExclusionRequest) (*loggingpb.LogExclusion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogExclusion), nil
-}
-
-func (s *mockConfigServer) UpdateExclusion(ctx context.Context, req *loggingpb.UpdateExclusionRequest) (*loggingpb.LogExclusion, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogExclusion), nil
-}
-
-func (s *mockConfigServer) DeleteExclusion(ctx context.Context, req *loggingpb.DeleteExclusionRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockConfigServer) GetCmekSettings(ctx context.Context, req *loggingpb.GetCmekSettingsRequest) (*loggingpb.CmekSettings, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.CmekSettings), nil
-}
-
-func (s *mockConfigServer) UpdateCmekSettings(ctx context.Context, req *loggingpb.UpdateCmekSettingsRequest) (*loggingpb.CmekSettings, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.CmekSettings), nil
-}
-
-type mockLoggingServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	loggingpb.LoggingServiceV2Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockLoggingServer) DeleteLog(ctx context.Context, req *loggingpb.DeleteLogRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockLoggingServer) WriteLogEntries(ctx context.Context, req *loggingpb.WriteLogEntriesRequest) (*loggingpb.WriteLogEntriesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.WriteLogEntriesResponse), nil
-}
-
-func (s *mockLoggingServer) ListLogEntries(ctx context.Context, req *loggingpb.ListLogEntriesRequest) (*loggingpb.ListLogEntriesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.ListLogEntriesResponse), nil
-}
-
-func (s *mockLoggingServer) ListMonitoredResourceDescriptors(ctx context.Context, req *loggingpb.ListMonitoredResourceDescriptorsRequest) (*loggingpb.ListMonitoredResourceDescriptorsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.ListMonitoredResourceDescriptorsResponse), nil
-}
-
-func (s *mockLoggingServer) ListLogs(ctx context.Context, req *loggingpb.ListLogsRequest) (*loggingpb.ListLogsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.ListLogsResponse), nil
-}
-
-type mockMetricsServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	loggingpb.MetricsServiceV2Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockMetricsServer) ListLogMetrics(ctx context.Context, req *loggingpb.ListLogMetricsRequest) (*loggingpb.ListLogMetricsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.ListLogMetricsResponse), nil
-}
-
-func (s *mockMetricsServer) GetLogMetric(ctx context.Context, req *loggingpb.GetLogMetricRequest) (*loggingpb.LogMetric, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogMetric), nil
-}
-
-func (s *mockMetricsServer) CreateLogMetric(ctx context.Context, req *loggingpb.CreateLogMetricRequest) (*loggingpb.LogMetric, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogMetric), nil
-}
-
-func (s *mockMetricsServer) UpdateLogMetric(ctx context.Context, req *loggingpb.UpdateLogMetricRequest) (*loggingpb.LogMetric, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*loggingpb.LogMetric), nil
-}
-
-func (s *mockMetricsServer) DeleteLogMetric(ctx context.Context, req *loggingpb.DeleteLogMetricRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockConfig  mockConfigServer
-	mockLogging mockLoggingServer
-	mockMetrics mockMetricsServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	loggingpb.RegisterConfigServiceV2Server(serv, &mockConfig)
-	loggingpb.RegisterLoggingServiceV2Server(serv, &mockLogging)
-	loggingpb.RegisterMetricsServiceV2Server(serv, &mockMetrics)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestConfigServiceV2ListSinks(t *testing.T) {
-	var nextPageToken string = ""
-	var sinksElement *loggingpb.LogSink = &loggingpb.LogSink{}
-	var sinks = []*loggingpb.LogSink{sinksElement}
-	var expectedResponse = &loggingpb.ListSinksResponse{
-		NextPageToken: nextPageToken,
-		Sinks:         sinks,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListSinksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSinks(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Sinks[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2ListSinksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListSinksRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSinks(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2GetSink(t *testing.T) {
-	var name string = "name3373707"
-	var destination string = "destination-1429847026"
-	var filter string = "filter-1274492040"
-	var description string = "description-1724546052"
-	var disabled bool = true
-	var writerIdentity string = "writerIdentity775638794"
-	var includeChildren bool = true
-	var expectedResponse = &loggingpb.LogSink{
-		Name:            name,
-		Destination:     destination,
-		Filter:          filter,
-		Description:     description,
-		Disabled:        disabled,
-		WriterIdentity:  writerIdentity,
-		IncludeChildren: includeChildren,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedSinkName string = fmt.Sprintf("projects/%s/sinks/%s", "[PROJECT]", "[SINK]")
-	var request = &loggingpb.GetSinkRequest{
-		SinkName: formattedSinkName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSink(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2GetSinkError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedSinkName string = fmt.Sprintf("projects/%s/sinks/%s", "[PROJECT]", "[SINK]")
-	var request = &loggingpb.GetSinkRequest{
-		SinkName: formattedSinkName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSink(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2CreateSink(t *testing.T) {
-	var name string = "name3373707"
-	var destination string = "destination-1429847026"
-	var filter string = "filter-1274492040"
-	var description string = "description-1724546052"
-	var disabled bool = true
-	var writerIdentity string = "writerIdentity775638794"
-	var includeChildren bool = true
-	var expectedResponse = &loggingpb.LogSink{
-		Name:            name,
-		Destination:     destination,
-		Filter:          filter,
-		Description:     description,
-		Disabled:        disabled,
-		WriterIdentity:  writerIdentity,
-		IncludeChildren: includeChildren,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var sink *loggingpb.LogSink = &loggingpb.LogSink{}
-	var request = &loggingpb.CreateSinkRequest{
-		Parent: formattedParent,
-		Sink:   sink,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSink(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2CreateSinkError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var sink *loggingpb.LogSink = &loggingpb.LogSink{}
-	var request = &loggingpb.CreateSinkRequest{
-		Parent: formattedParent,
-		Sink:   sink,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSink(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2UpdateSink(t *testing.T) {
-	var name string = "name3373707"
-	var destination string = "destination-1429847026"
-	var filter string = "filter-1274492040"
-	var description string = "description-1724546052"
-	var disabled bool = true
-	var writerIdentity string = "writerIdentity775638794"
-	var includeChildren bool = true
-	var expectedResponse = &loggingpb.LogSink{
-		Name:            name,
-		Destination:     destination,
-		Filter:          filter,
-		Description:     description,
-		Disabled:        disabled,
-		WriterIdentity:  writerIdentity,
-		IncludeChildren: includeChildren,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedSinkName string = fmt.Sprintf("projects/%s/sinks/%s", "[PROJECT]", "[SINK]")
-	var sink *loggingpb.LogSink = &loggingpb.LogSink{}
-	var request = &loggingpb.UpdateSinkRequest{
-		SinkName: formattedSinkName,
-		Sink:     sink,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSink(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2UpdateSinkError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedSinkName string = fmt.Sprintf("projects/%s/sinks/%s", "[PROJECT]", "[SINK]")
-	var sink *loggingpb.LogSink = &loggingpb.LogSink{}
-	var request = &loggingpb.UpdateSinkRequest{
-		SinkName: formattedSinkName,
-		Sink:     sink,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSink(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2DeleteSink(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedSinkName string = fmt.Sprintf("projects/%s/sinks/%s", "[PROJECT]", "[SINK]")
-	var request = &loggingpb.DeleteSinkRequest{
-		SinkName: formattedSinkName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSink(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestConfigServiceV2DeleteSinkError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedSinkName string = fmt.Sprintf("projects/%s/sinks/%s", "[PROJECT]", "[SINK]")
-	var request = &loggingpb.DeleteSinkRequest{
-		SinkName: formattedSinkName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSink(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestConfigServiceV2ListExclusions(t *testing.T) {
-	var nextPageToken string = ""
-	var exclusionsElement *loggingpb.LogExclusion = &loggingpb.LogExclusion{}
-	var exclusions = []*loggingpb.LogExclusion{exclusionsElement}
-	var expectedResponse = &loggingpb.ListExclusionsResponse{
-		NextPageToken: nextPageToken,
-		Exclusions:    exclusions,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListExclusionsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListExclusions(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Exclusions[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2ListExclusionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListExclusionsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListExclusions(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2GetExclusion(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var filter string = "filter-1274492040"
-	var disabled bool = true
-	var expectedResponse = &loggingpb.LogExclusion{
-		Name:        name2,
-		Description: description,
-		Filter:      filter,
-		Disabled:    disabled,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/exclusions/%s", "[PROJECT]", "[EXCLUSION]")
-	var request = &loggingpb.GetExclusionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetExclusion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2GetExclusionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/exclusions/%s", "[PROJECT]", "[EXCLUSION]")
-	var request = &loggingpb.GetExclusionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetExclusion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2CreateExclusion(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var filter string = "filter-1274492040"
-	var disabled bool = true
-	var expectedResponse = &loggingpb.LogExclusion{
-		Name:        name,
-		Description: description,
-		Filter:      filter,
-		Disabled:    disabled,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var exclusion *loggingpb.LogExclusion = &loggingpb.LogExclusion{}
-	var request = &loggingpb.CreateExclusionRequest{
-		Parent:    formattedParent,
-		Exclusion: exclusion,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateExclusion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2CreateExclusionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var exclusion *loggingpb.LogExclusion = &loggingpb.LogExclusion{}
-	var request = &loggingpb.CreateExclusionRequest{
-		Parent:    formattedParent,
-		Exclusion: exclusion,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateExclusion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2UpdateExclusion(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var filter string = "filter-1274492040"
-	var disabled bool = true
-	var expectedResponse = &loggingpb.LogExclusion{
-		Name:        name2,
-		Description: description,
-		Filter:      filter,
-		Disabled:    disabled,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/exclusions/%s", "[PROJECT]", "[EXCLUSION]")
-	var exclusion *loggingpb.LogExclusion = &loggingpb.LogExclusion{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &loggingpb.UpdateExclusionRequest{
-		Name:       formattedName,
-		Exclusion:  exclusion,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateExclusion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2UpdateExclusionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/exclusions/%s", "[PROJECT]", "[EXCLUSION]")
-	var exclusion *loggingpb.LogExclusion = &loggingpb.LogExclusion{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &loggingpb.UpdateExclusionRequest{
-		Name:       formattedName,
-		Exclusion:  exclusion,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateExclusion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2DeleteExclusion(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/exclusions/%s", "[PROJECT]", "[EXCLUSION]")
-	var request = &loggingpb.DeleteExclusionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteExclusion(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestConfigServiceV2DeleteExclusionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/exclusions/%s", "[PROJECT]", "[EXCLUSION]")
-	var request = &loggingpb.DeleteExclusionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteExclusion(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestConfigServiceV2GetCmekSettings(t *testing.T) {
-	var name string = "name3373707"
-	var kmsKeyName string = "kmsKeyName2094986649"
-	var serviceAccountId string = "serviceAccountId-111486921"
-	var expectedResponse = &loggingpb.CmekSettings{
-		Name:             name,
-		KmsKeyName:       kmsKeyName,
-		ServiceAccountId: serviceAccountId,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var request *loggingpb.GetCmekSettingsRequest = &loggingpb.GetCmekSettingsRequest{}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCmekSettings(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2GetCmekSettingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var request *loggingpb.GetCmekSettingsRequest = &loggingpb.GetCmekSettingsRequest{}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetCmekSettings(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestConfigServiceV2UpdateCmekSettings(t *testing.T) {
-	var name string = "name3373707"
-	var kmsKeyName string = "kmsKeyName2094986649"
-	var serviceAccountId string = "serviceAccountId-111486921"
-	var expectedResponse = &loggingpb.CmekSettings{
-		Name:             name,
-		KmsKeyName:       kmsKeyName,
-		ServiceAccountId: serviceAccountId,
-	}
-
-	mockConfig.err = nil
-	mockConfig.reqs = nil
-
-	mockConfig.resps = append(mockConfig.resps[:0], expectedResponse)
-
-	var request *loggingpb.UpdateCmekSettingsRequest = &loggingpb.UpdateCmekSettingsRequest{}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCmekSettings(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockConfig.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestConfigServiceV2UpdateCmekSettingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockConfig.err = gstatus.Error(errCode, "test error")
-
-	var request *loggingpb.UpdateCmekSettingsRequest = &loggingpb.UpdateCmekSettingsRequest{}
-
-	c, err := NewConfigClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateCmekSettings(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLoggingServiceV2DeleteLog(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockLogging.err = nil
-	mockLogging.reqs = nil
-
-	mockLogging.resps = append(mockLogging.resps[:0], expectedResponse)
-
-	var formattedLogName string = fmt.Sprintf("projects/%s/logs/%s", "[PROJECT]", "[LOG]")
-	var request = &loggingpb.DeleteLogRequest{
-		LogName: formattedLogName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteLog(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLogging.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestLoggingServiceV2DeleteLogError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLogging.err = gstatus.Error(errCode, "test error")
-
-	var formattedLogName string = fmt.Sprintf("projects/%s/logs/%s", "[PROJECT]", "[LOG]")
-	var request = &loggingpb.DeleteLogRequest{
-		LogName: formattedLogName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteLog(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestLoggingServiceV2WriteLogEntries(t *testing.T) {
-	var expectedResponse *loggingpb.WriteLogEntriesResponse = &loggingpb.WriteLogEntriesResponse{}
-
-	mockLogging.err = nil
-	mockLogging.reqs = nil
-
-	mockLogging.resps = append(mockLogging.resps[:0], expectedResponse)
-
-	var entries []*loggingpb.LogEntry = nil
-	var request = &loggingpb.WriteLogEntriesRequest{
-		Entries: entries,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.WriteLogEntries(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLogging.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLoggingServiceV2WriteLogEntriesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLogging.err = gstatus.Error(errCode, "test error")
-
-	var entries []*loggingpb.LogEntry = nil
-	var request = &loggingpb.WriteLogEntriesRequest{
-		Entries: entries,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.WriteLogEntries(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLoggingServiceV2ListLogEntries(t *testing.T) {
-	var nextPageToken string = ""
-	var entriesElement *loggingpb.LogEntry = &loggingpb.LogEntry{}
-	var entries = []*loggingpb.LogEntry{entriesElement}
-	var expectedResponse = &loggingpb.ListLogEntriesResponse{
-		NextPageToken: nextPageToken,
-		Entries:       entries,
-	}
-
-	mockLogging.err = nil
-	mockLogging.reqs = nil
-
-	mockLogging.resps = append(mockLogging.resps[:0], expectedResponse)
-
-	var formattedResourceNames []string = nil
-	var request = &loggingpb.ListLogEntriesRequest{
-		ResourceNames: formattedResourceNames,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListLogEntries(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLogging.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Entries[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLoggingServiceV2ListLogEntriesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLogging.err = gstatus.Error(errCode, "test error")
-
-	var formattedResourceNames []string = nil
-	var request = &loggingpb.ListLogEntriesRequest{
-		ResourceNames: formattedResourceNames,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListLogEntries(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLoggingServiceV2ListMonitoredResourceDescriptors(t *testing.T) {
-	var nextPageToken string = ""
-	var resourceDescriptorsElement *monitoredrespb.MonitoredResourceDescriptor = &monitoredrespb.MonitoredResourceDescriptor{}
-	var resourceDescriptors = []*monitoredrespb.MonitoredResourceDescriptor{resourceDescriptorsElement}
-	var expectedResponse = &loggingpb.ListMonitoredResourceDescriptorsResponse{
-		NextPageToken:       nextPageToken,
-		ResourceDescriptors: resourceDescriptors,
-	}
-
-	mockLogging.err = nil
-	mockLogging.reqs = nil
-
-	mockLogging.resps = append(mockLogging.resps[:0], expectedResponse)
-
-	var request *loggingpb.ListMonitoredResourceDescriptorsRequest = &loggingpb.ListMonitoredResourceDescriptorsRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListMonitoredResourceDescriptors(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLogging.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ResourceDescriptors[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLoggingServiceV2ListMonitoredResourceDescriptorsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLogging.err = gstatus.Error(errCode, "test error")
-
-	var request *loggingpb.ListMonitoredResourceDescriptorsRequest = &loggingpb.ListMonitoredResourceDescriptorsRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListMonitoredResourceDescriptors(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestLoggingServiceV2ListLogs(t *testing.T) {
-	var nextPageToken string = ""
-	var logNamesElement string = "logNamesElement-1079688374"
-	var logNames = []string{logNamesElement}
-	var expectedResponse = &loggingpb.ListLogsResponse{
-		NextPageToken: nextPageToken,
-		LogNames:      logNames,
-	}
-
-	mockLogging.err = nil
-	mockLogging.reqs = nil
-
-	mockLogging.resps = append(mockLogging.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListLogsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListLogs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockLogging.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.LogNames[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestLoggingServiceV2ListLogsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockLogging.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListLogsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListLogs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricsServiceV2ListLogMetrics(t *testing.T) {
-	var nextPageToken string = ""
-	var metricsElement *loggingpb.LogMetric = &loggingpb.LogMetric{}
-	var metrics = []*loggingpb.LogMetric{metricsElement}
-	var expectedResponse = &loggingpb.ListLogMetricsResponse{
-		NextPageToken: nextPageToken,
-		Metrics:       metrics,
-	}
-
-	mockMetrics.err = nil
-	mockMetrics.reqs = nil
-
-	mockMetrics.resps = append(mockMetrics.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListLogMetricsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListLogMetrics(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetrics.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Metrics[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricsServiceV2ListLogMetricsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetrics.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &loggingpb.ListLogMetricsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListLogMetrics(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricsServiceV2GetLogMetric(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var filter string = "filter-1274492040"
-	var valueExtractor string = "valueExtractor2047672534"
-	var expectedResponse = &loggingpb.LogMetric{
-		Name:           name,
-		Description:    description,
-		Filter:         filter,
-		ValueExtractor: valueExtractor,
-	}
-
-	mockMetrics.err = nil
-	mockMetrics.reqs = nil
-
-	mockMetrics.resps = append(mockMetrics.resps[:0], expectedResponse)
-
-	var formattedMetricName string = fmt.Sprintf("projects/%s/metrics/%s", "[PROJECT]", "[METRIC]")
-	var request = &loggingpb.GetLogMetricRequest{
-		MetricName: formattedMetricName,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetLogMetric(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetrics.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricsServiceV2GetLogMetricError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetrics.err = gstatus.Error(errCode, "test error")
-
-	var formattedMetricName string = fmt.Sprintf("projects/%s/metrics/%s", "[PROJECT]", "[METRIC]")
-	var request = &loggingpb.GetLogMetricRequest{
-		MetricName: formattedMetricName,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetLogMetric(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricsServiceV2CreateLogMetric(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var filter string = "filter-1274492040"
-	var valueExtractor string = "valueExtractor2047672534"
-	var expectedResponse = &loggingpb.LogMetric{
-		Name:           name,
-		Description:    description,
-		Filter:         filter,
-		ValueExtractor: valueExtractor,
-	}
-
-	mockMetrics.err = nil
-	mockMetrics.reqs = nil
-
-	mockMetrics.resps = append(mockMetrics.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var metric *loggingpb.LogMetric = &loggingpb.LogMetric{}
-	var request = &loggingpb.CreateLogMetricRequest{
-		Parent: formattedParent,
-		Metric: metric,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateLogMetric(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetrics.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricsServiceV2CreateLogMetricError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetrics.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var metric *loggingpb.LogMetric = &loggingpb.LogMetric{}
-	var request = &loggingpb.CreateLogMetricRequest{
-		Parent: formattedParent,
-		Metric: metric,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateLogMetric(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricsServiceV2UpdateLogMetric(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var filter string = "filter-1274492040"
-	var valueExtractor string = "valueExtractor2047672534"
-	var expectedResponse = &loggingpb.LogMetric{
-		Name:           name,
-		Description:    description,
-		Filter:         filter,
-		ValueExtractor: valueExtractor,
-	}
-
-	mockMetrics.err = nil
-	mockMetrics.reqs = nil
-
-	mockMetrics.resps = append(mockMetrics.resps[:0], expectedResponse)
-
-	var formattedMetricName string = fmt.Sprintf("projects/%s/metrics/%s", "[PROJECT]", "[METRIC]")
-	var metric *loggingpb.LogMetric = &loggingpb.LogMetric{}
-	var request = &loggingpb.UpdateLogMetricRequest{
-		MetricName: formattedMetricName,
-		Metric:     metric,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateLogMetric(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetrics.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricsServiceV2UpdateLogMetricError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetrics.err = gstatus.Error(errCode, "test error")
-
-	var formattedMetricName string = fmt.Sprintf("projects/%s/metrics/%s", "[PROJECT]", "[METRIC]")
-	var metric *loggingpb.LogMetric = &loggingpb.LogMetric{}
-	var request = &loggingpb.UpdateLogMetricRequest{
-		MetricName: formattedMetricName,
-		Metric:     metric,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateLogMetric(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricsServiceV2DeleteLogMetric(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockMetrics.err = nil
-	mockMetrics.reqs = nil
-
-	mockMetrics.resps = append(mockMetrics.resps[:0], expectedResponse)
-
-	var formattedMetricName string = fmt.Sprintf("projects/%s/metrics/%s", "[PROJECT]", "[METRIC]")
-	var request = &loggingpb.DeleteLogMetricRequest{
-		MetricName: formattedMetricName,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteLogMetric(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetrics.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestMetricsServiceV2DeleteLogMetricError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetrics.err = gstatus.Error(errCode, "test error")
-
-	var formattedMetricName string = fmt.Sprintf("projects/%s/metrics/%s", "[PROJECT]", "[METRIC]")
-	var request = &loggingpb.DeleteLogMetricRequest{
-		MetricName: formattedMetricName,
-	}
-
-	c, err := NewMetricsClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteLogMetric(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
diff --git a/monitoring/apiv3/mock_test.go b/monitoring/apiv3/mock_test.go
deleted file mode 100644
index 08a3b2e535c..00000000000
--- a/monitoring/apiv3/mock_test.go
+++ /dev/null
@@ -1,3624 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package monitoring
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	metricpb "google.golang.org/genproto/googleapis/api/metric"
-	monitoredrespb "google.golang.org/genproto/googleapis/api/monitoredres"
-	monitoringpb "google.golang.org/genproto/googleapis/monitoring/v3"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockAlertPolicyServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	monitoringpb.AlertPolicyServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockAlertPolicyServer) ListAlertPolicies(ctx context.Context, req *monitoringpb.ListAlertPoliciesRequest) (*monitoringpb.ListAlertPoliciesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListAlertPoliciesResponse), nil
-}
-
-func (s *mockAlertPolicyServer) GetAlertPolicy(ctx context.Context, req *monitoringpb.GetAlertPolicyRequest) (*monitoringpb.AlertPolicy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.AlertPolicy), nil
-}
-
-func (s *mockAlertPolicyServer) CreateAlertPolicy(ctx context.Context, req *monitoringpb.CreateAlertPolicyRequest) (*monitoringpb.AlertPolicy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.AlertPolicy), nil
-}
-
-func (s *mockAlertPolicyServer) DeleteAlertPolicy(ctx context.Context, req *monitoringpb.DeleteAlertPolicyRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockAlertPolicyServer) UpdateAlertPolicy(ctx context.Context, req *monitoringpb.UpdateAlertPolicyRequest) (*monitoringpb.AlertPolicy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.AlertPolicy), nil
-}
-
-type mockGroupServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	monitoringpb.GroupServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockGroupServer) ListGroups(ctx context.Context, req *monitoringpb.ListGroupsRequest) (*monitoringpb.ListGroupsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListGroupsResponse), nil
-}
-
-func (s *mockGroupServer) GetGroup(ctx context.Context, req *monitoringpb.GetGroupRequest) (*monitoringpb.Group, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.Group), nil
-}
-
-func (s *mockGroupServer) CreateGroup(ctx context.Context, req *monitoringpb.CreateGroupRequest) (*monitoringpb.Group, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.Group), nil
-}
-
-func (s *mockGroupServer) UpdateGroup(ctx context.Context, req *monitoringpb.UpdateGroupRequest) (*monitoringpb.Group, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.Group), nil
-}
-
-func (s *mockGroupServer) DeleteGroup(ctx context.Context, req *monitoringpb.DeleteGroupRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockGroupServer) ListGroupMembers(ctx context.Context, req *monitoringpb.ListGroupMembersRequest) (*monitoringpb.ListGroupMembersResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListGroupMembersResponse), nil
-}
-
-type mockMetricServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	monitoringpb.MetricServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockMetricServer) ListMonitoredResourceDescriptors(ctx context.Context, req *monitoringpb.ListMonitoredResourceDescriptorsRequest) (*monitoringpb.ListMonitoredResourceDescriptorsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListMonitoredResourceDescriptorsResponse), nil
-}
-
-func (s *mockMetricServer) GetMonitoredResourceDescriptor(ctx context.Context, req *monitoringpb.GetMonitoredResourceDescriptorRequest) (*monitoredrespb.MonitoredResourceDescriptor, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoredrespb.MonitoredResourceDescriptor), nil
-}
-
-func (s *mockMetricServer) ListMetricDescriptors(ctx context.Context, req *monitoringpb.ListMetricDescriptorsRequest) (*monitoringpb.ListMetricDescriptorsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListMetricDescriptorsResponse), nil
-}
-
-func (s *mockMetricServer) GetMetricDescriptor(ctx context.Context, req *monitoringpb.GetMetricDescriptorRequest) (*metricpb.MetricDescriptor, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*metricpb.MetricDescriptor), nil
-}
-
-func (s *mockMetricServer) CreateMetricDescriptor(ctx context.Context, req *monitoringpb.CreateMetricDescriptorRequest) (*metricpb.MetricDescriptor, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*metricpb.MetricDescriptor), nil
-}
-
-func (s *mockMetricServer) DeleteMetricDescriptor(ctx context.Context, req *monitoringpb.DeleteMetricDescriptorRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockMetricServer) ListTimeSeries(ctx context.Context, req *monitoringpb.ListTimeSeriesRequest) (*monitoringpb.ListTimeSeriesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListTimeSeriesResponse), nil
-}
-
-func (s *mockMetricServer) CreateTimeSeries(ctx context.Context, req *monitoringpb.CreateTimeSeriesRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-type mockNotificationChannelServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	monitoringpb.NotificationChannelServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockNotificationChannelServer) ListNotificationChannelDescriptors(ctx context.Context, req *monitoringpb.ListNotificationChannelDescriptorsRequest) (*monitoringpb.ListNotificationChannelDescriptorsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListNotificationChannelDescriptorsResponse), nil
-}
-
-func (s *mockNotificationChannelServer) GetNotificationChannelDescriptor(ctx context.Context, req *monitoringpb.GetNotificationChannelDescriptorRequest) (*monitoringpb.NotificationChannelDescriptor, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.NotificationChannelDescriptor), nil
-}
-
-func (s *mockNotificationChannelServer) ListNotificationChannels(ctx context.Context, req *monitoringpb.ListNotificationChannelsRequest) (*monitoringpb.ListNotificationChannelsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListNotificationChannelsResponse), nil
-}
-
-func (s *mockNotificationChannelServer) GetNotificationChannel(ctx context.Context, req *monitoringpb.GetNotificationChannelRequest) (*monitoringpb.NotificationChannel, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.NotificationChannel), nil
-}
-
-func (s *mockNotificationChannelServer) CreateNotificationChannel(ctx context.Context, req *monitoringpb.CreateNotificationChannelRequest) (*monitoringpb.NotificationChannel, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.NotificationChannel), nil
-}
-
-func (s *mockNotificationChannelServer) UpdateNotificationChannel(ctx context.Context, req *monitoringpb.UpdateNotificationChannelRequest) (*monitoringpb.NotificationChannel, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.NotificationChannel), nil
-}
-
-func (s *mockNotificationChannelServer) DeleteNotificationChannel(ctx context.Context, req *monitoringpb.DeleteNotificationChannelRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockNotificationChannelServer) SendNotificationChannelVerificationCode(ctx context.Context, req *monitoringpb.SendNotificationChannelVerificationCodeRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockNotificationChannelServer) GetNotificationChannelVerificationCode(ctx context.Context, req *monitoringpb.GetNotificationChannelVerificationCodeRequest) (*monitoringpb.GetNotificationChannelVerificationCodeResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.GetNotificationChannelVerificationCodeResponse), nil
-}
-
-func (s *mockNotificationChannelServer) VerifyNotificationChannel(ctx context.Context, req *monitoringpb.VerifyNotificationChannelRequest) (*monitoringpb.NotificationChannel, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.NotificationChannel), nil
-}
-
-type mockServiceMonitoringServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	monitoringpb.ServiceMonitoringServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockServiceMonitoringServer) CreateService(ctx context.Context, req *monitoringpb.CreateServiceRequest) (*monitoringpb.Service, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.Service), nil
-}
-
-func (s *mockServiceMonitoringServer) GetService(ctx context.Context, req *monitoringpb.GetServiceRequest) (*monitoringpb.Service, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.Service), nil
-}
-
-func (s *mockServiceMonitoringServer) ListServices(ctx context.Context, req *monitoringpb.ListServicesRequest) (*monitoringpb.ListServicesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListServicesResponse), nil
-}
-
-func (s *mockServiceMonitoringServer) UpdateService(ctx context.Context, req *monitoringpb.UpdateServiceRequest) (*monitoringpb.Service, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.Service), nil
-}
-
-func (s *mockServiceMonitoringServer) DeleteService(ctx context.Context, req *monitoringpb.DeleteServiceRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockServiceMonitoringServer) CreateServiceLevelObjective(ctx context.Context, req *monitoringpb.CreateServiceLevelObjectiveRequest) (*monitoringpb.ServiceLevelObjective, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ServiceLevelObjective), nil
-}
-
-func (s *mockServiceMonitoringServer) GetServiceLevelObjective(ctx context.Context, req *monitoringpb.GetServiceLevelObjectiveRequest) (*monitoringpb.ServiceLevelObjective, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ServiceLevelObjective), nil
-}
-
-func (s *mockServiceMonitoringServer) ListServiceLevelObjectives(ctx context.Context, req *monitoringpb.ListServiceLevelObjectivesRequest) (*monitoringpb.ListServiceLevelObjectivesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListServiceLevelObjectivesResponse), nil
-}
-
-func (s *mockServiceMonitoringServer) UpdateServiceLevelObjective(ctx context.Context, req *monitoringpb.UpdateServiceLevelObjectiveRequest) (*monitoringpb.ServiceLevelObjective, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ServiceLevelObjective), nil
-}
-
-func (s *mockServiceMonitoringServer) DeleteServiceLevelObjective(ctx context.Context, req *monitoringpb.DeleteServiceLevelObjectiveRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-type mockUptimeCheckServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	monitoringpb.UptimeCheckServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockUptimeCheckServer) ListUptimeCheckConfigs(ctx context.Context, req *monitoringpb.ListUptimeCheckConfigsRequest) (*monitoringpb.ListUptimeCheckConfigsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListUptimeCheckConfigsResponse), nil
-}
-
-func (s *mockUptimeCheckServer) GetUptimeCheckConfig(ctx context.Context, req *monitoringpb.GetUptimeCheckConfigRequest) (*monitoringpb.UptimeCheckConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.UptimeCheckConfig), nil
-}
-
-func (s *mockUptimeCheckServer) CreateUptimeCheckConfig(ctx context.Context, req *monitoringpb.CreateUptimeCheckConfigRequest) (*monitoringpb.UptimeCheckConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.UptimeCheckConfig), nil
-}
-
-func (s *mockUptimeCheckServer) UpdateUptimeCheckConfig(ctx context.Context, req *monitoringpb.UpdateUptimeCheckConfigRequest) (*monitoringpb.UptimeCheckConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.UptimeCheckConfig), nil
-}
-
-func (s *mockUptimeCheckServer) DeleteUptimeCheckConfig(ctx context.Context, req *monitoringpb.DeleteUptimeCheckConfigRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockUptimeCheckServer) ListUptimeCheckIps(ctx context.Context, req *monitoringpb.ListUptimeCheckIpsRequest) (*monitoringpb.ListUptimeCheckIpsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*monitoringpb.ListUptimeCheckIpsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockAlertPolicy         mockAlertPolicyServer
-	mockGroup               mockGroupServer
-	mockMetric              mockMetricServer
-	mockNotificationChannel mockNotificationChannelServer
-	mockServiceMonitoring   mockServiceMonitoringServer
-	mockUptimeCheck         mockUptimeCheckServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	monitoringpb.RegisterAlertPolicyServiceServer(serv, &mockAlertPolicy)
-	monitoringpb.RegisterGroupServiceServer(serv, &mockGroup)
-	monitoringpb.RegisterMetricServiceServer(serv, &mockMetric)
-	monitoringpb.RegisterNotificationChannelServiceServer(serv, &mockNotificationChannel)
-	monitoringpb.RegisterServiceMonitoringServiceServer(serv, &mockServiceMonitoring)
-	monitoringpb.RegisterUptimeCheckServiceServer(serv, &mockUptimeCheck)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestAlertPolicyServiceListAlertPolicies(t *testing.T) {
-	var nextPageToken string = ""
-	var alertPoliciesElement *monitoringpb.AlertPolicy = &monitoringpb.AlertPolicy{}
-	var alertPolicies = []*monitoringpb.AlertPolicy{alertPoliciesElement}
-	var expectedResponse = &monitoringpb.ListAlertPoliciesResponse{
-		NextPageToken: nextPageToken,
-		AlertPolicies: alertPolicies,
-	}
-
-	mockAlertPolicy.err = nil
-	mockAlertPolicy.reqs = nil
-
-	mockAlertPolicy.resps = append(mockAlertPolicy.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListAlertPoliciesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListAlertPolicies(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAlertPolicy.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.AlertPolicies[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAlertPolicyServiceListAlertPoliciesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAlertPolicy.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListAlertPoliciesRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListAlertPolicies(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAlertPolicyServiceGetAlertPolicy(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &monitoringpb.AlertPolicy{
-		Name:        name2,
-		DisplayName: displayName,
-	}
-
-	mockAlertPolicy.err = nil
-	mockAlertPolicy.reqs = nil
-
-	mockAlertPolicy.resps = append(mockAlertPolicy.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/alertPolicies/%s", "[PROJECT]", "[ALERT_POLICY]")
-	var request = &monitoringpb.GetAlertPolicyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAlertPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAlertPolicy.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAlertPolicyServiceGetAlertPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAlertPolicy.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/alertPolicies/%s", "[PROJECT]", "[ALERT_POLICY]")
-	var request = &monitoringpb.GetAlertPolicyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetAlertPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAlertPolicyServiceCreateAlertPolicy(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &monitoringpb.AlertPolicy{
-		Name:        name2,
-		DisplayName: displayName,
-	}
-
-	mockAlertPolicy.err = nil
-	mockAlertPolicy.reqs = nil
-
-	mockAlertPolicy.resps = append(mockAlertPolicy.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var alertPolicy *monitoringpb.AlertPolicy = &monitoringpb.AlertPolicy{}
-	var request = &monitoringpb.CreateAlertPolicyRequest{
-		Name:        formattedName,
-		AlertPolicy: alertPolicy,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateAlertPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAlertPolicy.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAlertPolicyServiceCreateAlertPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAlertPolicy.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var alertPolicy *monitoringpb.AlertPolicy = &monitoringpb.AlertPolicy{}
-	var request = &monitoringpb.CreateAlertPolicyRequest{
-		Name:        formattedName,
-		AlertPolicy: alertPolicy,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateAlertPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestAlertPolicyServiceDeleteAlertPolicy(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockAlertPolicy.err = nil
-	mockAlertPolicy.reqs = nil
-
-	mockAlertPolicy.resps = append(mockAlertPolicy.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/alertPolicies/%s", "[PROJECT]", "[ALERT_POLICY]")
-	var request = &monitoringpb.DeleteAlertPolicyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteAlertPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAlertPolicy.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestAlertPolicyServiceDeleteAlertPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAlertPolicy.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/alertPolicies/%s", "[PROJECT]", "[ALERT_POLICY]")
-	var request = &monitoringpb.DeleteAlertPolicyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteAlertPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestAlertPolicyServiceUpdateAlertPolicy(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &monitoringpb.AlertPolicy{
-		Name:        name,
-		DisplayName: displayName,
-	}
-
-	mockAlertPolicy.err = nil
-	mockAlertPolicy.reqs = nil
-
-	mockAlertPolicy.resps = append(mockAlertPolicy.resps[:0], expectedResponse)
-
-	var alertPolicy *monitoringpb.AlertPolicy = &monitoringpb.AlertPolicy{}
-	var request = &monitoringpb.UpdateAlertPolicyRequest{
-		AlertPolicy: alertPolicy,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateAlertPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockAlertPolicy.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestAlertPolicyServiceUpdateAlertPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockAlertPolicy.err = gstatus.Error(errCode, "test error")
-
-	var alertPolicy *monitoringpb.AlertPolicy = &monitoringpb.AlertPolicy{}
-	var request = &monitoringpb.UpdateAlertPolicyRequest{
-		AlertPolicy: alertPolicy,
-	}
-
-	c, err := NewAlertPolicyClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateAlertPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGroupServiceListGroups(t *testing.T) {
-	var nextPageToken string = ""
-	var groupElement *monitoringpb.Group = &monitoringpb.Group{}
-	var group = []*monitoringpb.Group{groupElement}
-	var expectedResponse = &monitoringpb.ListGroupsResponse{
-		NextPageToken: nextPageToken,
-		Group:         group,
-	}
-
-	mockGroup.err = nil
-	mockGroup.reqs = nil
-
-	mockGroup.resps = append(mockGroup.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListGroupsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGroups(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Group[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGroupServiceListGroupsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGroup.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListGroupsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGroups(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGroupServiceGetGroup(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var parentName string = "parentName1015022848"
-	var filter string = "filter-1274492040"
-	var isCluster bool = false
-	var expectedResponse = &monitoringpb.Group{
-		Name:        name2,
-		DisplayName: displayName,
-		ParentName:  parentName,
-		Filter:      filter,
-		IsCluster:   isCluster,
-	}
-
-	mockGroup.err = nil
-	mockGroup.reqs = nil
-
-	mockGroup.resps = append(mockGroup.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &monitoringpb.GetGroupRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetGroup(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGroupServiceGetGroupError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGroup.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &monitoringpb.GetGroupRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetGroup(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGroupServiceCreateGroup(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var parentName string = "parentName1015022848"
-	var filter string = "filter-1274492040"
-	var isCluster bool = false
-	var expectedResponse = &monitoringpb.Group{
-		Name:        name2,
-		DisplayName: displayName,
-		ParentName:  parentName,
-		Filter:      filter,
-		IsCluster:   isCluster,
-	}
-
-	mockGroup.err = nil
-	mockGroup.reqs = nil
-
-	mockGroup.resps = append(mockGroup.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var group *monitoringpb.Group = &monitoringpb.Group{}
-	var request = &monitoringpb.CreateGroupRequest{
-		Name:  formattedName,
-		Group: group,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateGroup(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGroupServiceCreateGroupError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGroup.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var group *monitoringpb.Group = &monitoringpb.Group{}
-	var request = &monitoringpb.CreateGroupRequest{
-		Name:  formattedName,
-		Group: group,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateGroup(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGroupServiceUpdateGroup(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var parentName string = "parentName1015022848"
-	var filter string = "filter-1274492040"
-	var isCluster bool = false
-	var expectedResponse = &monitoringpb.Group{
-		Name:        name,
-		DisplayName: displayName,
-		ParentName:  parentName,
-		Filter:      filter,
-		IsCluster:   isCluster,
-	}
-
-	mockGroup.err = nil
-	mockGroup.reqs = nil
-
-	mockGroup.resps = append(mockGroup.resps[:0], expectedResponse)
-
-	var group *monitoringpb.Group = &monitoringpb.Group{}
-	var request = &monitoringpb.UpdateGroupRequest{
-		Group: group,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateGroup(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGroupServiceUpdateGroupError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGroup.err = gstatus.Error(errCode, "test error")
-
-	var group *monitoringpb.Group = &monitoringpb.Group{}
-	var request = &monitoringpb.UpdateGroupRequest{
-		Group: group,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateGroup(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestGroupServiceDeleteGroup(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockGroup.err = nil
-	mockGroup.reqs = nil
-
-	mockGroup.resps = append(mockGroup.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &monitoringpb.DeleteGroupRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteGroup(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestGroupServiceDeleteGroupError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGroup.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &monitoringpb.DeleteGroupRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteGroup(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestGroupServiceListGroupMembers(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var membersElement *monitoredrespb.MonitoredResource = &monitoredrespb.MonitoredResource{}
-	var members = []*monitoredrespb.MonitoredResource{membersElement}
-	var expectedResponse = &monitoringpb.ListGroupMembersResponse{
-		NextPageToken: nextPageToken,
-		TotalSize:     totalSize,
-		Members:       members,
-	}
-
-	mockGroup.err = nil
-	mockGroup.reqs = nil
-
-	mockGroup.resps = append(mockGroup.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &monitoringpb.ListGroupMembersRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGroupMembers(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockGroup.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Members[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestGroupServiceListGroupMembersError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockGroup.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/groups/%s", "[PROJECT]", "[GROUP]")
-	var request = &monitoringpb.ListGroupMembersRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewGroupClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGroupMembers(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceListMonitoredResourceDescriptors(t *testing.T) {
-	var nextPageToken string = ""
-	var resourceDescriptorsElement *monitoredrespb.MonitoredResourceDescriptor = &monitoredrespb.MonitoredResourceDescriptor{}
-	var resourceDescriptors = []*monitoredrespb.MonitoredResourceDescriptor{resourceDescriptorsElement}
-	var expectedResponse = &monitoringpb.ListMonitoredResourceDescriptorsResponse{
-		NextPageToken:       nextPageToken,
-		ResourceDescriptors: resourceDescriptors,
-	}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListMonitoredResourceDescriptorsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListMonitoredResourceDescriptors(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ResourceDescriptors[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricServiceListMonitoredResourceDescriptorsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListMonitoredResourceDescriptorsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListMonitoredResourceDescriptors(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceGetMonitoredResourceDescriptor(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var type_ string = "type3575610"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &monitoredrespb.MonitoredResourceDescriptor{
-		Name:        name2,
-		Type:        type_,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/monitoredResourceDescriptors/%s", "[PROJECT]", "[MONITORED_RESOURCE_DESCRIPTOR]")
-	var request = &monitoringpb.GetMonitoredResourceDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetMonitoredResourceDescriptor(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricServiceGetMonitoredResourceDescriptorError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/monitoredResourceDescriptors/%s", "[PROJECT]", "[MONITORED_RESOURCE_DESCRIPTOR]")
-	var request = &monitoringpb.GetMonitoredResourceDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetMonitoredResourceDescriptor(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceListMetricDescriptors(t *testing.T) {
-	var nextPageToken string = ""
-	var metricDescriptorsElement *metricpb.MetricDescriptor = &metricpb.MetricDescriptor{}
-	var metricDescriptors = []*metricpb.MetricDescriptor{metricDescriptorsElement}
-	var expectedResponse = &monitoringpb.ListMetricDescriptorsResponse{
-		NextPageToken:     nextPageToken,
-		MetricDescriptors: metricDescriptors,
-	}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListMetricDescriptorsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListMetricDescriptors(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.MetricDescriptors[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricServiceListMetricDescriptorsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListMetricDescriptorsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListMetricDescriptors(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceGetMetricDescriptor(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var type_ string = "type3575610"
-	var unit string = "unit3594628"
-	var description string = "description-1724546052"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &metricpb.MetricDescriptor{
-		Name:        name2,
-		Type:        type_,
-		Unit:        unit,
-		Description: description,
-		DisplayName: displayName,
-	}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/metricDescriptors/%s", "[PROJECT]", "[METRIC_DESCRIPTOR]")
-	var request = &monitoringpb.GetMetricDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetMetricDescriptor(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricServiceGetMetricDescriptorError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/metricDescriptors/%s", "[PROJECT]", "[METRIC_DESCRIPTOR]")
-	var request = &monitoringpb.GetMetricDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetMetricDescriptor(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceCreateMetricDescriptor(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var type_ string = "type3575610"
-	var unit string = "unit3594628"
-	var description string = "description-1724546052"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &metricpb.MetricDescriptor{
-		Name:        name2,
-		Type:        type_,
-		Unit:        unit,
-		Description: description,
-		DisplayName: displayName,
-	}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var metricDescriptor *metricpb.MetricDescriptor = &metricpb.MetricDescriptor{}
-	var request = &monitoringpb.CreateMetricDescriptorRequest{
-		Name:             formattedName,
-		MetricDescriptor: metricDescriptor,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateMetricDescriptor(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricServiceCreateMetricDescriptorError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var metricDescriptor *metricpb.MetricDescriptor = &metricpb.MetricDescriptor{}
-	var request = &monitoringpb.CreateMetricDescriptorRequest{
-		Name:             formattedName,
-		MetricDescriptor: metricDescriptor,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateMetricDescriptor(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceDeleteMetricDescriptor(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/metricDescriptors/%s", "[PROJECT]", "[METRIC_DESCRIPTOR]")
-	var request = &monitoringpb.DeleteMetricDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteMetricDescriptor(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestMetricServiceDeleteMetricDescriptorError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/metricDescriptors/%s", "[PROJECT]", "[METRIC_DESCRIPTOR]")
-	var request = &monitoringpb.DeleteMetricDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteMetricDescriptor(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestMetricServiceListTimeSeries(t *testing.T) {
-	var nextPageToken string = ""
-	var timeSeriesElement *monitoringpb.TimeSeries = &monitoringpb.TimeSeries{}
-	var timeSeries = []*monitoringpb.TimeSeries{timeSeriesElement}
-	var expectedResponse = &monitoringpb.ListTimeSeriesResponse{
-		NextPageToken: nextPageToken,
-		TimeSeries:    timeSeries,
-	}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var filter string = "filter-1274492040"
-	var interval *monitoringpb.TimeInterval = &monitoringpb.TimeInterval{}
-	var view monitoringpb.ListTimeSeriesRequest_TimeSeriesView = monitoringpb.ListTimeSeriesRequest_FULL
-	var request = &monitoringpb.ListTimeSeriesRequest{
-		Name:     formattedName,
-		Filter:   filter,
-		Interval: interval,
-		View:     view,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTimeSeries(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.TimeSeries[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestMetricServiceListTimeSeriesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var filter string = "filter-1274492040"
-	var interval *monitoringpb.TimeInterval = &monitoringpb.TimeInterval{}
-	var view monitoringpb.ListTimeSeriesRequest_TimeSeriesView = monitoringpb.ListTimeSeriesRequest_FULL
-	var request = &monitoringpb.ListTimeSeriesRequest{
-		Name:     formattedName,
-		Filter:   filter,
-		Interval: interval,
-		View:     view,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTimeSeries(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestMetricServiceCreateTimeSeries(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockMetric.err = nil
-	mockMetric.reqs = nil
-
-	mockMetric.resps = append(mockMetric.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var timeSeries []*monitoringpb.TimeSeries = nil
-	var request = &monitoringpb.CreateTimeSeriesRequest{
-		Name:       formattedName,
-		TimeSeries: timeSeries,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.CreateTimeSeries(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockMetric.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestMetricServiceCreateTimeSeriesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockMetric.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var timeSeries []*monitoringpb.TimeSeries = nil
-	var request = &monitoringpb.CreateTimeSeriesRequest{
-		Name:       formattedName,
-		TimeSeries: timeSeries,
-	}
-
-	c, err := NewMetricClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.CreateTimeSeries(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestNotificationChannelServiceListNotificationChannelDescriptors(t *testing.T) {
-	var nextPageToken string = ""
-	var channelDescriptorsElement *monitoringpb.NotificationChannelDescriptor = &monitoringpb.NotificationChannelDescriptor{}
-	var channelDescriptors = []*monitoringpb.NotificationChannelDescriptor{channelDescriptorsElement}
-	var expectedResponse = &monitoringpb.ListNotificationChannelDescriptorsResponse{
-		NextPageToken:      nextPageToken,
-		ChannelDescriptors: channelDescriptors,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListNotificationChannelDescriptorsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotificationChannelDescriptors(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ChannelDescriptors[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceListNotificationChannelDescriptorsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListNotificationChannelDescriptorsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotificationChannelDescriptors(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceGetNotificationChannelDescriptor(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var type_ string = "type3575610"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &monitoringpb.NotificationChannelDescriptor{
-		Name:        name2,
-		Type:        type_,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannelDescriptors/%s", "[PROJECT]", "[CHANNEL_DESCRIPTOR]")
-	var request = &monitoringpb.GetNotificationChannelDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNotificationChannelDescriptor(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceGetNotificationChannelDescriptorError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannelDescriptors/%s", "[PROJECT]", "[CHANNEL_DESCRIPTOR]")
-	var request = &monitoringpb.GetNotificationChannelDescriptorRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNotificationChannelDescriptor(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceListNotificationChannels(t *testing.T) {
-	var nextPageToken string = ""
-	var notificationChannelsElement *monitoringpb.NotificationChannel = &monitoringpb.NotificationChannel{}
-	var notificationChannels = []*monitoringpb.NotificationChannel{notificationChannelsElement}
-	var expectedResponse = &monitoringpb.ListNotificationChannelsResponse{
-		NextPageToken:        nextPageToken,
-		NotificationChannels: notificationChannels,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListNotificationChannelsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotificationChannels(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.NotificationChannels[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceListNotificationChannelsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListNotificationChannelsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListNotificationChannels(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceGetNotificationChannel(t *testing.T) {
-	var type_ string = "type3575610"
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &monitoringpb.NotificationChannel{
-		Type:        type_,
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.GetNotificationChannelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNotificationChannel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceGetNotificationChannelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.GetNotificationChannelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNotificationChannel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceCreateNotificationChannel(t *testing.T) {
-	var type_ string = "type3575610"
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &monitoringpb.NotificationChannel{
-		Type:        type_,
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var notificationChannel *monitoringpb.NotificationChannel = &monitoringpb.NotificationChannel{}
-	var request = &monitoringpb.CreateNotificationChannelRequest{
-		Name:                formattedName,
-		NotificationChannel: notificationChannel,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNotificationChannel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceCreateNotificationChannelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var notificationChannel *monitoringpb.NotificationChannel = &monitoringpb.NotificationChannel{}
-	var request = &monitoringpb.CreateNotificationChannelRequest{
-		Name:                formattedName,
-		NotificationChannel: notificationChannel,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateNotificationChannel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceUpdateNotificationChannel(t *testing.T) {
-	var type_ string = "type3575610"
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &monitoringpb.NotificationChannel{
-		Type:        type_,
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var notificationChannel *monitoringpb.NotificationChannel = &monitoringpb.NotificationChannel{}
-	var request = &monitoringpb.UpdateNotificationChannelRequest{
-		NotificationChannel: notificationChannel,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNotificationChannel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceUpdateNotificationChannelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var notificationChannel *monitoringpb.NotificationChannel = &monitoringpb.NotificationChannel{}
-	var request = &monitoringpb.UpdateNotificationChannelRequest{
-		NotificationChannel: notificationChannel,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateNotificationChannel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceDeleteNotificationChannel(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.DeleteNotificationChannelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteNotificationChannel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestNotificationChannelServiceDeleteNotificationChannelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.DeleteNotificationChannelRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteNotificationChannel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestNotificationChannelServiceSendNotificationChannelVerificationCode(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.SendNotificationChannelVerificationCodeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.SendNotificationChannelVerificationCode(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestNotificationChannelServiceSendNotificationChannelVerificationCodeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.SendNotificationChannelVerificationCodeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.SendNotificationChannelVerificationCode(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestNotificationChannelServiceGetNotificationChannelVerificationCode(t *testing.T) {
-	var code string = "code3059181"
-	var expectedResponse = &monitoringpb.GetNotificationChannelVerificationCodeResponse{
-		Code: code,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.GetNotificationChannelVerificationCodeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNotificationChannelVerificationCode(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceGetNotificationChannelVerificationCodeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var request = &monitoringpb.GetNotificationChannelVerificationCodeRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetNotificationChannelVerificationCode(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestNotificationChannelServiceVerifyNotificationChannel(t *testing.T) {
-	var type_ string = "type3575610"
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &monitoringpb.NotificationChannel{
-		Type:        type_,
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockNotificationChannel.err = nil
-	mockNotificationChannel.reqs = nil
-
-	mockNotificationChannel.resps = append(mockNotificationChannel.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var code string = "code3059181"
-	var request = &monitoringpb.VerifyNotificationChannelRequest{
-		Name: formattedName,
-		Code: code,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.VerifyNotificationChannel(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockNotificationChannel.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestNotificationChannelServiceVerifyNotificationChannelError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockNotificationChannel.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/notificationChannels/%s", "[PROJECT]", "[NOTIFICATION_CHANNEL]")
-	var code string = "code3059181"
-	var request = &monitoringpb.VerifyNotificationChannelRequest{
-		Name: formattedName,
-		Code: code,
-	}
-
-	c, err := NewNotificationChannelClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.VerifyNotificationChannel(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceCreateService(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &monitoringpb.Service{
-		Name:        name,
-		DisplayName: displayName,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var service *monitoringpb.Service = &monitoringpb.Service{}
-	var request = &monitoringpb.CreateServiceRequest{
-		Parent:  formattedParent,
-		Service: service,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateService(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceCreateServiceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var service *monitoringpb.Service = &monitoringpb.Service{}
-	var request = &monitoringpb.CreateServiceRequest{
-		Parent:  formattedParent,
-		Service: service,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateService(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceGetService(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &monitoringpb.Service{
-		Name:        name2,
-		DisplayName: displayName,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var request = &monitoringpb.GetServiceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetService(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceGetServiceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var request = &monitoringpb.GetServiceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetService(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceListServices(t *testing.T) {
-	var nextPageToken string = ""
-	var servicesElement *monitoringpb.Service = &monitoringpb.Service{}
-	var services = []*monitoringpb.Service{servicesElement}
-	var expectedResponse = &monitoringpb.ListServicesResponse{
-		NextPageToken: nextPageToken,
-		Services:      services,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListServicesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServices(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Services[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceListServicesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListServicesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServices(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceUpdateService(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &monitoringpb.Service{
-		Name:        name,
-		DisplayName: displayName,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var service *monitoringpb.Service = &monitoringpb.Service{}
-	var request = &monitoringpb.UpdateServiceRequest{
-		Service: service,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateService(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceUpdateServiceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var service *monitoringpb.Service = &monitoringpb.Service{}
-	var request = &monitoringpb.UpdateServiceRequest{
-		Service: service,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateService(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceDeleteService(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var request = &monitoringpb.DeleteServiceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteService(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestServiceMonitoringServiceDeleteServiceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var request = &monitoringpb.DeleteServiceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteService(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestServiceMonitoringServiceCreateServiceLevelObjective(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var goal float64 = 317825.0
-	var expectedResponse = &monitoringpb.ServiceLevelObjective{
-		Name:        name,
-		DisplayName: displayName,
-		Goal:        goal,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var serviceLevelObjective *monitoringpb.ServiceLevelObjective = &monitoringpb.ServiceLevelObjective{}
-	var request = &monitoringpb.CreateServiceLevelObjectiveRequest{
-		Parent:                formattedParent,
-		ServiceLevelObjective: serviceLevelObjective,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateServiceLevelObjective(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceCreateServiceLevelObjectiveError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var serviceLevelObjective *monitoringpb.ServiceLevelObjective = &monitoringpb.ServiceLevelObjective{}
-	var request = &monitoringpb.CreateServiceLevelObjectiveRequest{
-		Parent:                formattedParent,
-		ServiceLevelObjective: serviceLevelObjective,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateServiceLevelObjective(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceGetServiceLevelObjective(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var goal float64 = 317825.0
-	var expectedResponse = &monitoringpb.ServiceLevelObjective{
-		Name:        name2,
-		DisplayName: displayName,
-		Goal:        goal,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s/serviceLevelObjectives/%s", "[PROJECT]", "[SERVICE]", "[SERVICE_LEVEL_OBJECTIVE]")
-	var request = &monitoringpb.GetServiceLevelObjectiveRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServiceLevelObjective(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceGetServiceLevelObjectiveError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s/serviceLevelObjectives/%s", "[PROJECT]", "[SERVICE]", "[SERVICE_LEVEL_OBJECTIVE]")
-	var request = &monitoringpb.GetServiceLevelObjectiveRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetServiceLevelObjective(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceListServiceLevelObjectives(t *testing.T) {
-	var nextPageToken string = ""
-	var serviceLevelObjectivesElement *monitoringpb.ServiceLevelObjective = &monitoringpb.ServiceLevelObjective{}
-	var serviceLevelObjectives = []*monitoringpb.ServiceLevelObjective{serviceLevelObjectivesElement}
-	var expectedResponse = &monitoringpb.ListServiceLevelObjectivesResponse{
-		NextPageToken:          nextPageToken,
-		ServiceLevelObjectives: serviceLevelObjectives,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var request = &monitoringpb.ListServiceLevelObjectivesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServiceLevelObjectives(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ServiceLevelObjectives[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceListServiceLevelObjectivesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/services/%s", "[PROJECT]", "[SERVICE]")
-	var request = &monitoringpb.ListServiceLevelObjectivesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListServiceLevelObjectives(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceUpdateServiceLevelObjective(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var goal float64 = 317825.0
-	var expectedResponse = &monitoringpb.ServiceLevelObjective{
-		Name:        name,
-		DisplayName: displayName,
-		Goal:        goal,
-	}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var serviceLevelObjective *monitoringpb.ServiceLevelObjective = &monitoringpb.ServiceLevelObjective{}
-	var request = &monitoringpb.UpdateServiceLevelObjectiveRequest{
-		ServiceLevelObjective: serviceLevelObjective,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateServiceLevelObjective(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestServiceMonitoringServiceUpdateServiceLevelObjectiveError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var serviceLevelObjective *monitoringpb.ServiceLevelObjective = &monitoringpb.ServiceLevelObjective{}
-	var request = &monitoringpb.UpdateServiceLevelObjectiveRequest{
-		ServiceLevelObjective: serviceLevelObjective,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateServiceLevelObjective(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestServiceMonitoringServiceDeleteServiceLevelObjective(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockServiceMonitoring.err = nil
-	mockServiceMonitoring.reqs = nil
-
-	mockServiceMonitoring.resps = append(mockServiceMonitoring.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s/serviceLevelObjectives/%s", "[PROJECT]", "[SERVICE]", "[SERVICE_LEVEL_OBJECTIVE]")
-	var request = &monitoringpb.DeleteServiceLevelObjectiveRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteServiceLevelObjective(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockServiceMonitoring.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestServiceMonitoringServiceDeleteServiceLevelObjectiveError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockServiceMonitoring.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/services/%s/serviceLevelObjectives/%s", "[PROJECT]", "[SERVICE]", "[SERVICE_LEVEL_OBJECTIVE]")
-	var request = &monitoringpb.DeleteServiceLevelObjectiveRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewServiceMonitoringClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteServiceLevelObjective(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestUptimeCheckServiceListUptimeCheckConfigs(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var uptimeCheckConfigsElement *monitoringpb.UptimeCheckConfig = &monitoringpb.UptimeCheckConfig{}
-	var uptimeCheckConfigs = []*monitoringpb.UptimeCheckConfig{uptimeCheckConfigsElement}
-	var expectedResponse = &monitoringpb.ListUptimeCheckConfigsResponse{
-		NextPageToken:      nextPageToken,
-		TotalSize:          totalSize,
-		UptimeCheckConfigs: uptimeCheckConfigs,
-	}
-
-	mockUptimeCheck.err = nil
-	mockUptimeCheck.reqs = nil
-
-	mockUptimeCheck.resps = append(mockUptimeCheck.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListUptimeCheckConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListUptimeCheckConfigs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockUptimeCheck.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.UptimeCheckConfigs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestUptimeCheckServiceListUptimeCheckConfigsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockUptimeCheck.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &monitoringpb.ListUptimeCheckConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListUptimeCheckConfigs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestUptimeCheckServiceGetUptimeCheckConfig(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var isInternal bool = true
-	var expectedResponse = &monitoringpb.UptimeCheckConfig{
-		Name:        name2,
-		DisplayName: displayName,
-		IsInternal:  isInternal,
-	}
-
-	mockUptimeCheck.err = nil
-	mockUptimeCheck.reqs = nil
-
-	mockUptimeCheck.resps = append(mockUptimeCheck.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/uptimeCheckConfigs/%s", "[PROJECT]", "[UPTIME_CHECK_CONFIG]")
-	var request = &monitoringpb.GetUptimeCheckConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetUptimeCheckConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockUptimeCheck.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestUptimeCheckServiceGetUptimeCheckConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockUptimeCheck.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/uptimeCheckConfigs/%s", "[PROJECT]", "[UPTIME_CHECK_CONFIG]")
-	var request = &monitoringpb.GetUptimeCheckConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetUptimeCheckConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestUptimeCheckServiceCreateUptimeCheckConfig(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var isInternal bool = true
-	var expectedResponse = &monitoringpb.UptimeCheckConfig{
-		Name:        name,
-		DisplayName: displayName,
-		IsInternal:  isInternal,
-	}
-
-	mockUptimeCheck.err = nil
-	mockUptimeCheck.reqs = nil
-
-	mockUptimeCheck.resps = append(mockUptimeCheck.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var uptimeCheckConfig *monitoringpb.UptimeCheckConfig = &monitoringpb.UptimeCheckConfig{}
-	var request = &monitoringpb.CreateUptimeCheckConfigRequest{
-		Parent:            formattedParent,
-		UptimeCheckConfig: uptimeCheckConfig,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateUptimeCheckConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockUptimeCheck.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestUptimeCheckServiceCreateUptimeCheckConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockUptimeCheck.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var uptimeCheckConfig *monitoringpb.UptimeCheckConfig = &monitoringpb.UptimeCheckConfig{}
-	var request = &monitoringpb.CreateUptimeCheckConfigRequest{
-		Parent:            formattedParent,
-		UptimeCheckConfig: uptimeCheckConfig,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateUptimeCheckConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestUptimeCheckServiceUpdateUptimeCheckConfig(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var isInternal bool = true
-	var expectedResponse = &monitoringpb.UptimeCheckConfig{
-		Name:        name,
-		DisplayName: displayName,
-		IsInternal:  isInternal,
-	}
-
-	mockUptimeCheck.err = nil
-	mockUptimeCheck.reqs = nil
-
-	mockUptimeCheck.resps = append(mockUptimeCheck.resps[:0], expectedResponse)
-
-	var uptimeCheckConfig *monitoringpb.UptimeCheckConfig = &monitoringpb.UptimeCheckConfig{}
-	var request = &monitoringpb.UpdateUptimeCheckConfigRequest{
-		UptimeCheckConfig: uptimeCheckConfig,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateUptimeCheckConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockUptimeCheck.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestUptimeCheckServiceUpdateUptimeCheckConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockUptimeCheck.err = gstatus.Error(errCode, "test error")
-
-	var uptimeCheckConfig *monitoringpb.UptimeCheckConfig = &monitoringpb.UptimeCheckConfig{}
-	var request = &monitoringpb.UpdateUptimeCheckConfigRequest{
-		UptimeCheckConfig: uptimeCheckConfig,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateUptimeCheckConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestUptimeCheckServiceDeleteUptimeCheckConfig(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockUptimeCheck.err = nil
-	mockUptimeCheck.reqs = nil
-
-	mockUptimeCheck.resps = append(mockUptimeCheck.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/uptimeCheckConfigs/%s", "[PROJECT]", "[UPTIME_CHECK_CONFIG]")
-	var request = &monitoringpb.DeleteUptimeCheckConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteUptimeCheckConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockUptimeCheck.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestUptimeCheckServiceDeleteUptimeCheckConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockUptimeCheck.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/uptimeCheckConfigs/%s", "[PROJECT]", "[UPTIME_CHECK_CONFIG]")
-	var request = &monitoringpb.DeleteUptimeCheckConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteUptimeCheckConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestUptimeCheckServiceListUptimeCheckIps(t *testing.T) {
-	var nextPageToken string = ""
-	var uptimeCheckIpsElement *monitoringpb.UptimeCheckIp = &monitoringpb.UptimeCheckIp{}
-	var uptimeCheckIps = []*monitoringpb.UptimeCheckIp{uptimeCheckIpsElement}
-	var expectedResponse = &monitoringpb.ListUptimeCheckIpsResponse{
-		NextPageToken:  nextPageToken,
-		UptimeCheckIps: uptimeCheckIps,
-	}
-
-	mockUptimeCheck.err = nil
-	mockUptimeCheck.reqs = nil
-
-	mockUptimeCheck.resps = append(mockUptimeCheck.resps[:0], expectedResponse)
-
-	var request *monitoringpb.ListUptimeCheckIpsRequest = &monitoringpb.ListUptimeCheckIpsRequest{}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListUptimeCheckIps(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockUptimeCheck.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.UptimeCheckIps[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestUptimeCheckServiceListUptimeCheckIpsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockUptimeCheck.err = gstatus.Error(errCode, "test error")
-
-	var request *monitoringpb.ListUptimeCheckIpsRequest = &monitoringpb.ListUptimeCheckIpsRequest{}
-
-	c, err := NewUptimeCheckClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListUptimeCheckIps(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/oslogin/apiv1/mock_test.go b/oslogin/apiv1/mock_test.go
deleted file mode 100644
index f3c2879fb0a..00000000000
--- a/oslogin/apiv1/mock_test.go
+++ /dev/null
@@ -1,517 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package oslogin
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	commonpb "google.golang.org/genproto/googleapis/cloud/oslogin/common"
-	osloginpb "google.golang.org/genproto/googleapis/cloud/oslogin/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockOsLoginServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	osloginpb.OsLoginServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockOsLoginServer) DeletePosixAccount(ctx context.Context, req *osloginpb.DeletePosixAccountRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockOsLoginServer) DeleteSshPublicKey(ctx context.Context, req *osloginpb.DeleteSshPublicKeyRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockOsLoginServer) GetLoginProfile(ctx context.Context, req *osloginpb.GetLoginProfileRequest) (*osloginpb.LoginProfile, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*osloginpb.LoginProfile), nil
-}
-
-func (s *mockOsLoginServer) GetSshPublicKey(ctx context.Context, req *osloginpb.GetSshPublicKeyRequest) (*commonpb.SshPublicKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*commonpb.SshPublicKey), nil
-}
-
-func (s *mockOsLoginServer) ImportSshPublicKey(ctx context.Context, req *osloginpb.ImportSshPublicKeyRequest) (*osloginpb.ImportSshPublicKeyResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*osloginpb.ImportSshPublicKeyResponse), nil
-}
-
-func (s *mockOsLoginServer) UpdateSshPublicKey(ctx context.Context, req *osloginpb.UpdateSshPublicKeyRequest) (*commonpb.SshPublicKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*commonpb.SshPublicKey), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockOsLogin mockOsLoginServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	osloginpb.RegisterOsLoginServiceServer(serv, &mockOsLogin)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestOsLoginServiceDeletePosixAccount(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/projects/%s", "[USER]", "[PROJECT]")
-	var request = &osloginpb.DeletePosixAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeletePosixAccount(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestOsLoginServiceDeletePosixAccountError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/projects/%s", "[USER]", "[PROJECT]")
-	var request = &osloginpb.DeletePosixAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeletePosixAccount(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestOsLoginServiceDeleteSshPublicKey(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.DeleteSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestOsLoginServiceDeleteSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.DeleteSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestOsLoginServiceGetLoginProfile(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &osloginpb.LoginProfile{
-		Name: name2,
-	}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s", "[USER]")
-	var request = &osloginpb.GetLoginProfileRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetLoginProfile(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceGetLoginProfileError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s", "[USER]")
-	var request = &osloginpb.GetLoginProfileRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetLoginProfile(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestOsLoginServiceGetSshPublicKey(t *testing.T) {
-	var key string = "key106079"
-	var expirationTimeUsec int64 = 2058878882
-	var fingerprint string = "fingerprint-1375934236"
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &commonpb.SshPublicKey{
-		Key:                key,
-		ExpirationTimeUsec: expirationTimeUsec,
-		Fingerprint:        fingerprint,
-		Name:               name2,
-	}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.GetSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceGetSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.GetSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestOsLoginServiceImportSshPublicKey(t *testing.T) {
-	var expectedResponse *osloginpb.ImportSshPublicKeyResponse = &osloginpb.ImportSshPublicKeyResponse{}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("users/%s", "[USER]")
-	var request = &osloginpb.ImportSshPublicKeyRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceImportSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("users/%s", "[USER]")
-	var request = &osloginpb.ImportSshPublicKeyRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestOsLoginServiceUpdateSshPublicKey(t *testing.T) {
-	var key string = "key106079"
-	var expirationTimeUsec int64 = 2058878882
-	var fingerprint string = "fingerprint-1375934236"
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &commonpb.SshPublicKey{
-		Key:                key,
-		ExpirationTimeUsec: expirationTimeUsec,
-		Fingerprint:        fingerprint,
-		Name:               name2,
-	}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var sshPublicKey *commonpb.SshPublicKey = &commonpb.SshPublicKey{}
-	var request = &osloginpb.UpdateSshPublicKeyRequest{
-		Name:         formattedName,
-		SshPublicKey: sshPublicKey,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceUpdateSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var sshPublicKey *commonpb.SshPublicKey = &commonpb.SshPublicKey{}
-	var request = &osloginpb.UpdateSshPublicKeyRequest{
-		Name:         formattedName,
-		SshPublicKey: sshPublicKey,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/oslogin/apiv1beta/mock_test.go b/oslogin/apiv1beta/mock_test.go
deleted file mode 100644
index 20a5f2d2a7b..00000000000
--- a/oslogin/apiv1beta/mock_test.go
+++ /dev/null
@@ -1,521 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package oslogin
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	commonpb "google.golang.org/genproto/googleapis/cloud/oslogin/common"
-	osloginpb "google.golang.org/genproto/googleapis/cloud/oslogin/v1beta"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockOsLoginServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	osloginpb.OsLoginServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockOsLoginServer) DeletePosixAccount(ctx context.Context, req *osloginpb.DeletePosixAccountRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockOsLoginServer) DeleteSshPublicKey(ctx context.Context, req *osloginpb.DeleteSshPublicKeyRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockOsLoginServer) GetLoginProfile(ctx context.Context, req *osloginpb.GetLoginProfileRequest) (*osloginpb.LoginProfile, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*osloginpb.LoginProfile), nil
-}
-
-func (s *mockOsLoginServer) GetSshPublicKey(ctx context.Context, req *osloginpb.GetSshPublicKeyRequest) (*commonpb.SshPublicKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*commonpb.SshPublicKey), nil
-}
-
-func (s *mockOsLoginServer) ImportSshPublicKey(ctx context.Context, req *osloginpb.ImportSshPublicKeyRequest) (*osloginpb.ImportSshPublicKeyResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*osloginpb.ImportSshPublicKeyResponse), nil
-}
-
-func (s *mockOsLoginServer) UpdateSshPublicKey(ctx context.Context, req *osloginpb.UpdateSshPublicKeyRequest) (*commonpb.SshPublicKey, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*commonpb.SshPublicKey), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockOsLogin mockOsLoginServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	osloginpb.RegisterOsLoginServiceServer(serv, &mockOsLogin)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestOsLoginServiceDeletePosixAccount(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/projects/%s", "[USER]", "[PROJECT]")
-	var request = &osloginpb.DeletePosixAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeletePosixAccount(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestOsLoginServiceDeletePosixAccountError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/projects/%s", "[USER]", "[PROJECT]")
-	var request = &osloginpb.DeletePosixAccountRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeletePosixAccount(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestOsLoginServiceDeleteSshPublicKey(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.DeleteSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestOsLoginServiceDeleteSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.DeleteSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestOsLoginServiceGetLoginProfile(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &osloginpb.LoginProfile{
-		Name: name2,
-	}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s", "[USER]")
-	var request = &osloginpb.GetLoginProfileRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetLoginProfile(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceGetLoginProfileError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s", "[USER]")
-	var request = &osloginpb.GetLoginProfileRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetLoginProfile(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestOsLoginServiceGetSshPublicKey(t *testing.T) {
-	var key string = "key106079"
-	var expirationTimeUsec int64 = 2058878882
-	var fingerprint string = "fingerprint-1375934236"
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &commonpb.SshPublicKey{
-		Key:                key,
-		ExpirationTimeUsec: expirationTimeUsec,
-		Fingerprint:        fingerprint,
-		Name:               name2,
-	}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.GetSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceGetSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var request = &osloginpb.GetSshPublicKeyRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestOsLoginServiceImportSshPublicKey(t *testing.T) {
-	var expectedResponse *osloginpb.ImportSshPublicKeyResponse = &osloginpb.ImportSshPublicKeyResponse{}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("users/%s", "[USER]")
-	var sshPublicKey *commonpb.SshPublicKey = &commonpb.SshPublicKey{}
-	var request = &osloginpb.ImportSshPublicKeyRequest{
-		Parent:       formattedParent,
-		SshPublicKey: sshPublicKey,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceImportSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("users/%s", "[USER]")
-	var sshPublicKey *commonpb.SshPublicKey = &commonpb.SshPublicKey{}
-	var request = &osloginpb.ImportSshPublicKeyRequest{
-		Parent:       formattedParent,
-		SshPublicKey: sshPublicKey,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ImportSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestOsLoginServiceUpdateSshPublicKey(t *testing.T) {
-	var key string = "key106079"
-	var expirationTimeUsec int64 = 2058878882
-	var fingerprint string = "fingerprint-1375934236"
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &commonpb.SshPublicKey{
-		Key:                key,
-		ExpirationTimeUsec: expirationTimeUsec,
-		Fingerprint:        fingerprint,
-		Name:               name2,
-	}
-
-	mockOsLogin.err = nil
-	mockOsLogin.reqs = nil
-
-	mockOsLogin.resps = append(mockOsLogin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var sshPublicKey *commonpb.SshPublicKey = &commonpb.SshPublicKey{}
-	var request = &osloginpb.UpdateSshPublicKeyRequest{
-		Name:         formattedName,
-		SshPublicKey: sshPublicKey,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSshPublicKey(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockOsLogin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestOsLoginServiceUpdateSshPublicKeyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockOsLogin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("users/%s/sshPublicKeys/%s", "[USER]", "[FINGERPRINT]")
-	var sshPublicKey *commonpb.SshPublicKey = &commonpb.SshPublicKey{}
-	var request = &osloginpb.UpdateSshPublicKeyRequest{
-		Name:         formattedName,
-		SshPublicKey: sshPublicKey,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSshPublicKey(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/phishingprotection/apiv1beta1/mock_test.go b/phishingprotection/apiv1beta1/mock_test.go
deleted file mode 100644
index 0940b39d13b..00000000000
--- a/phishingprotection/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,162 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package phishingprotection
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	phishingprotectionpb "google.golang.org/genproto/googleapis/cloud/phishingprotection/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockPhishingProtectionServiceV1Beta1Server struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	phishingprotectionpb.PhishingProtectionServiceV1Beta1Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockPhishingProtectionServiceV1Beta1Server) ReportPhishing(ctx context.Context, req *phishingprotectionpb.ReportPhishingRequest) (*phishingprotectionpb.ReportPhishingResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*phishingprotectionpb.ReportPhishingResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockPhishingProtectionServiceV1Beta1 mockPhishingProtectionServiceV1Beta1Server
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	phishingprotectionpb.RegisterPhishingProtectionServiceV1Beta1Server(serv, &mockPhishingProtectionServiceV1Beta1)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestPhishingProtectionServiceV1Beta1ReportPhishing(t *testing.T) {
-	var expectedResponse *phishingprotectionpb.ReportPhishingResponse = &phishingprotectionpb.ReportPhishingResponse{}
-
-	mockPhishingProtectionServiceV1Beta1.err = nil
-	mockPhishingProtectionServiceV1Beta1.reqs = nil
-
-	mockPhishingProtectionServiceV1Beta1.resps = append(mockPhishingProtectionServiceV1Beta1.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var uri string = "uri116076"
-	var request = &phishingprotectionpb.ReportPhishingRequest{
-		Parent: formattedParent,
-		Uri:    uri,
-	}
-
-	c, err := NewPhishingProtectionServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ReportPhishing(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPhishingProtectionServiceV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPhishingProtectionServiceV1Beta1ReportPhishingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPhishingProtectionServiceV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var uri string = "uri116076"
-	var request = &phishingprotectionpb.ReportPhishingRequest{
-		Parent: formattedParent,
-		Uri:    uri,
-	}
-
-	c, err := NewPhishingProtectionServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ReportPhishing(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/pubsub/apiv1/mock_test.go b/pubsub/apiv1/mock_test.go
deleted file mode 100644
index dc44269a36f..00000000000
--- a/pubsub/apiv1/mock_test.go
+++ /dev/null
@@ -1,1833 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package pubsub
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"google.golang.org/api/option"
-	pubsubpb "google.golang.org/genproto/googleapis/pubsub/v1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-	"google.golang.org/protobuf/proto"
-	"google.golang.org/protobuf/types/known/emptypb"
-	"google.golang.org/protobuf/types/known/timestamppb"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ status.Status
-
-type mockPublisherServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	pubsubpb.PublisherServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockPublisherServer) CreateTopic(ctx context.Context, req *pubsubpb.Topic) (*pubsubpb.Topic, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Topic), nil
-}
-
-func (s *mockPublisherServer) UpdateTopic(ctx context.Context, req *pubsubpb.UpdateTopicRequest) (*pubsubpb.Topic, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Topic), nil
-}
-
-func (s *mockPublisherServer) Publish(ctx context.Context, req *pubsubpb.PublishRequest) (*pubsubpb.PublishResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.PublishResponse), nil
-}
-
-func (s *mockPublisherServer) GetTopic(ctx context.Context, req *pubsubpb.GetTopicRequest) (*pubsubpb.Topic, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Topic), nil
-}
-
-func (s *mockPublisherServer) ListTopics(ctx context.Context, req *pubsubpb.ListTopicsRequest) (*pubsubpb.ListTopicsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.ListTopicsResponse), nil
-}
-
-func (s *mockPublisherServer) ListTopicSubscriptions(ctx context.Context, req *pubsubpb.ListTopicSubscriptionsRequest) (*pubsubpb.ListTopicSubscriptionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.ListTopicSubscriptionsResponse), nil
-}
-
-func (s *mockPublisherServer) DeleteTopic(ctx context.Context, req *pubsubpb.DeleteTopicRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-type mockSubscriberServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	pubsubpb.SubscriberServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSubscriberServer) CreateSubscription(ctx context.Context, req *pubsubpb.Subscription) (*pubsubpb.Subscription, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Subscription), nil
-}
-
-func (s *mockSubscriberServer) GetSubscription(ctx context.Context, req *pubsubpb.GetSubscriptionRequest) (*pubsubpb.Subscription, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Subscription), nil
-}
-
-func (s *mockSubscriberServer) UpdateSubscription(ctx context.Context, req *pubsubpb.UpdateSubscriptionRequest) (*pubsubpb.Subscription, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Subscription), nil
-}
-
-func (s *mockSubscriberServer) ListSubscriptions(ctx context.Context, req *pubsubpb.ListSubscriptionsRequest) (*pubsubpb.ListSubscriptionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.ListSubscriptionsResponse), nil
-}
-
-func (s *mockSubscriberServer) DeleteSubscription(ctx context.Context, req *pubsubpb.DeleteSubscriptionRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSubscriberServer) ModifyAckDeadline(ctx context.Context, req *pubsubpb.ModifyAckDeadlineRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSubscriberServer) Acknowledge(ctx context.Context, req *pubsubpb.AcknowledgeRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSubscriberServer) Pull(ctx context.Context, req *pubsubpb.PullRequest) (*pubsubpb.PullResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.PullResponse), nil
-}
-
-func (s *mockSubscriberServer) StreamingPull(stream pubsubpb.Subscriber_StreamingPullServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*pubsubpb.StreamingPullResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockSubscriberServer) ModifyPushConfig(ctx context.Context, req *pubsubpb.ModifyPushConfigRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSubscriberServer) ListSnapshots(ctx context.Context, req *pubsubpb.ListSnapshotsRequest) (*pubsubpb.ListSnapshotsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.ListSnapshotsResponse), nil
-}
-
-func (s *mockSubscriberServer) CreateSnapshot(ctx context.Context, req *pubsubpb.CreateSnapshotRequest) (*pubsubpb.Snapshot, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Snapshot), nil
-}
-
-func (s *mockSubscriberServer) UpdateSnapshot(ctx context.Context, req *pubsubpb.UpdateSnapshotRequest) (*pubsubpb.Snapshot, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.Snapshot), nil
-}
-
-func (s *mockSubscriberServer) DeleteSnapshot(ctx context.Context, req *pubsubpb.DeleteSnapshotRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSubscriberServer) Seek(ctx context.Context, req *pubsubpb.SeekRequest) (*pubsubpb.SeekResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*pubsubpb.SeekResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockPublisher  mockPublisherServer
-	mockSubscriber mockSubscriberServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	pubsubpb.RegisterPublisherServer(serv, &mockPublisher)
-	pubsubpb.RegisterSubscriberServer(serv, &mockSubscriber)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestPublisherCreateTopic(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var kmsKeyName string = "kmsKeyName2094986649"
-	var expectedResponse = &pubsubpb.Topic{
-		Name:       name2,
-		KmsKeyName: kmsKeyName,
-	}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.Topic{
-		Name: formattedName,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTopic(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPublisherCreateTopicError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.Topic{
-		Name: formattedName,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateTopic(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPublisherUpdateTopic(t *testing.T) {
-	var name string = "name3373707"
-	var kmsKeyName string = "kmsKeyName2094986649"
-	var expectedResponse = &pubsubpb.Topic{
-		Name:       name,
-		KmsKeyName: kmsKeyName,
-	}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var topic *pubsubpb.Topic = &pubsubpb.Topic{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &pubsubpb.UpdateTopicRequest{
-		Topic:      topic,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateTopic(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPublisherUpdateTopicError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var topic *pubsubpb.Topic = &pubsubpb.Topic{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &pubsubpb.UpdateTopicRequest{
-		Topic:      topic,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateTopic(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPublisherPublish(t *testing.T) {
-	var messageIdsElement string = "messageIdsElement-744837059"
-	var messageIds = []string{messageIdsElement}
-	var expectedResponse = &pubsubpb.PublishResponse{
-		MessageIds: messageIds,
-	}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var data []byte = []byte("-86")
-	var messagesElement = &pubsubpb.PubsubMessage{
-		Data: data,
-	}
-	var messages = []*pubsubpb.PubsubMessage{messagesElement}
-	var request = &pubsubpb.PublishRequest{
-		Topic:    formattedTopic,
-		Messages: messages,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Publish(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPublisherPublishError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var data []byte = []byte("-86")
-	var messagesElement = &pubsubpb.PubsubMessage{
-		Data: data,
-	}
-	var messages = []*pubsubpb.PubsubMessage{messagesElement}
-	var request = &pubsubpb.PublishRequest{
-		Topic:    formattedTopic,
-		Messages: messages,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Publish(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPublisherGetTopic(t *testing.T) {
-	var name string = "name3373707"
-	var kmsKeyName string = "kmsKeyName2094986649"
-	var expectedResponse = &pubsubpb.Topic{
-		Name:       name,
-		KmsKeyName: kmsKeyName,
-	}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.GetTopicRequest{
-		Topic: formattedTopic,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTopic(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPublisherGetTopicError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.GetTopicRequest{
-		Topic: formattedTopic,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTopic(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPublisherListTopics(t *testing.T) {
-	var nextPageToken string = ""
-	var topicsElement *pubsubpb.Topic = &pubsubpb.Topic{}
-	var topics = []*pubsubpb.Topic{topicsElement}
-	var expectedResponse = &pubsubpb.ListTopicsResponse{
-		NextPageToken: nextPageToken,
-		Topics:        topics,
-	}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var formattedProject string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &pubsubpb.ListTopicsRequest{
-		Project: formattedProject,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTopics(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Topics[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPublisherListTopicsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var formattedProject string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &pubsubpb.ListTopicsRequest{
-		Project: formattedProject,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTopics(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPublisherListTopicSubscriptions(t *testing.T) {
-	var nextPageToken string = ""
-	var subscriptionsElement string = "subscriptionsElement1698708147"
-	var subscriptions = []string{subscriptionsElement}
-	var expectedResponse = &pubsubpb.ListTopicSubscriptionsResponse{
-		NextPageToken: nextPageToken,
-		Subscriptions: subscriptions,
-	}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.ListTopicSubscriptionsRequest{
-		Topic: formattedTopic,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTopicSubscriptions(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Subscriptions[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestPublisherListTopicSubscriptionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.ListTopicSubscriptionsRequest{
-		Topic: formattedTopic,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTopicSubscriptions(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestPublisherDeleteTopic(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockPublisher.err = nil
-	mockPublisher.reqs = nil
-
-	mockPublisher.resps = append(mockPublisher.resps[:0], expectedResponse)
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.DeleteTopicRequest{
-		Topic: formattedTopic,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTopic(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockPublisher.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestPublisherDeleteTopicError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockPublisher.err = gstatus.Error(errCode, "test error")
-
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.DeleteTopicRequest{
-		Topic: formattedTopic,
-	}
-
-	c, err := NewPublisherClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteTopic(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSubscriberCreateSubscription(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var topic2 string = "topic2-1139259102"
-	var ackDeadlineSeconds int32 = 2135351438
-	var retainAckedMessages bool = false
-	var enableMessageOrdering bool = true
-	var expectedResponse = &pubsubpb.Subscription{
-		Name:                  name2,
-		Topic:                 topic2,
-		AckDeadlineSeconds:    ackDeadlineSeconds,
-		RetainAckedMessages:   retainAckedMessages,
-		EnableMessageOrdering: enableMessageOrdering,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.Subscription{
-		Name:  formattedName,
-		Topic: formattedTopic,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSubscription(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberCreateSubscriptionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var formattedTopic string = fmt.Sprintf("projects/%s/topics/%s", "[PROJECT]", "[TOPIC]")
-	var request = &pubsubpb.Subscription{
-		Name:  formattedName,
-		Topic: formattedTopic,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSubscription(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberGetSubscription(t *testing.T) {
-	var name string = "name3373707"
-	var topic string = "topic110546223"
-	var ackDeadlineSeconds int32 = 2135351438
-	var retainAckedMessages bool = false
-	var enableMessageOrdering bool = true
-	var expectedResponse = &pubsubpb.Subscription{
-		Name:                  name,
-		Topic:                 topic,
-		AckDeadlineSeconds:    ackDeadlineSeconds,
-		RetainAckedMessages:   retainAckedMessages,
-		EnableMessageOrdering: enableMessageOrdering,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.GetSubscriptionRequest{
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSubscription(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberGetSubscriptionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.GetSubscriptionRequest{
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSubscription(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberUpdateSubscription(t *testing.T) {
-	var name string = "name3373707"
-	var topic string = "topic110546223"
-	var ackDeadlineSeconds2 int32 = 921632575
-	var retainAckedMessages bool = false
-	var enableMessageOrdering bool = true
-	var expectedResponse = &pubsubpb.Subscription{
-		Name:                  name,
-		Topic:                 topic,
-		AckDeadlineSeconds:    ackDeadlineSeconds2,
-		RetainAckedMessages:   retainAckedMessages,
-		EnableMessageOrdering: enableMessageOrdering,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var ackDeadlineSeconds int32 = 42
-	var subscription = &pubsubpb.Subscription{
-		AckDeadlineSeconds: ackDeadlineSeconds,
-	}
-	var pathsElement string = "ack_deadline_seconds"
-	var paths = []string{pathsElement}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var request = &pubsubpb.UpdateSubscriptionRequest{
-		Subscription: subscription,
-		UpdateMask:   updateMask,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSubscription(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberUpdateSubscriptionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var ackDeadlineSeconds int32 = 42
-	var subscription = &pubsubpb.Subscription{
-		AckDeadlineSeconds: ackDeadlineSeconds,
-	}
-	var pathsElement string = "ack_deadline_seconds"
-	var paths = []string{pathsElement}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var request = &pubsubpb.UpdateSubscriptionRequest{
-		Subscription: subscription,
-		UpdateMask:   updateMask,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSubscription(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberListSubscriptions(t *testing.T) {
-	var nextPageToken string = ""
-	var subscriptionsElement *pubsubpb.Subscription = &pubsubpb.Subscription{}
-	var subscriptions = []*pubsubpb.Subscription{subscriptionsElement}
-	var expectedResponse = &pubsubpb.ListSubscriptionsResponse{
-		NextPageToken: nextPageToken,
-		Subscriptions: subscriptions,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedProject string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &pubsubpb.ListSubscriptionsRequest{
-		Project: formattedProject,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSubscriptions(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Subscriptions[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberListSubscriptionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedProject string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &pubsubpb.ListSubscriptionsRequest{
-		Project: formattedProject,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSubscriptions(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberDeleteSubscription(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.DeleteSubscriptionRequest{
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSubscription(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSubscriberDeleteSubscriptionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.DeleteSubscriptionRequest{
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSubscription(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSubscriberModifyAckDeadline(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var ackIds []string = nil
-	var ackDeadlineSeconds int32 = 2135351438
-	var request = &pubsubpb.ModifyAckDeadlineRequest{
-		Subscription:       formattedSubscription,
-		AckIds:             ackIds,
-		AckDeadlineSeconds: ackDeadlineSeconds,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.ModifyAckDeadline(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSubscriberModifyAckDeadlineError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var ackIds []string = nil
-	var ackDeadlineSeconds int32 = 2135351438
-	var request = &pubsubpb.ModifyAckDeadlineRequest{
-		Subscription:       formattedSubscription,
-		AckIds:             ackIds,
-		AckDeadlineSeconds: ackDeadlineSeconds,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.ModifyAckDeadline(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSubscriberAcknowledge(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var ackIds []string = nil
-	var request = &pubsubpb.AcknowledgeRequest{
-		Subscription: formattedSubscription,
-		AckIds:       ackIds,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Acknowledge(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSubscriberAcknowledgeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var ackIds []string = nil
-	var request = &pubsubpb.AcknowledgeRequest{
-		Subscription: formattedSubscription,
-		AckIds:       ackIds,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Acknowledge(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSubscriberPull(t *testing.T) {
-	var expectedResponse *pubsubpb.PullResponse = &pubsubpb.PullResponse{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var maxMessages int32 = 496131527
-	var request = &pubsubpb.PullRequest{
-		Subscription: formattedSubscription,
-		MaxMessages:  maxMessages,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Pull(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberPullError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var maxMessages int32 = 496131527
-	var request = &pubsubpb.PullRequest{
-		Subscription: formattedSubscription,
-		MaxMessages:  maxMessages,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Pull(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberStreamingPull(t *testing.T) {
-	var receivedMessagesElement *pubsubpb.ReceivedMessage = &pubsubpb.ReceivedMessage{}
-	var receivedMessages = []*pubsubpb.ReceivedMessage{receivedMessagesElement}
-	var expectedResponse = &pubsubpb.StreamingPullResponse{
-		ReceivedMessages: receivedMessages,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var streamAckDeadlineSeconds int32 = 1875467245
-	var request = &pubsubpb.StreamingPullRequest{
-		Subscription:             formattedSubscription,
-		StreamAckDeadlineSeconds: streamAckDeadlineSeconds,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingPull(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberStreamingPullError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var streamAckDeadlineSeconds int32 = 1875467245
-	var request = &pubsubpb.StreamingPullRequest{
-		Subscription:             formattedSubscription,
-		StreamAckDeadlineSeconds: streamAckDeadlineSeconds,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingPull(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberModifyPushConfig(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var pushConfig *pubsubpb.PushConfig = &pubsubpb.PushConfig{}
-	var request = &pubsubpb.ModifyPushConfigRequest{
-		Subscription: formattedSubscription,
-		PushConfig:   pushConfig,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.ModifyPushConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSubscriberModifyPushConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var pushConfig *pubsubpb.PushConfig = &pubsubpb.PushConfig{}
-	var request = &pubsubpb.ModifyPushConfigRequest{
-		Subscription: formattedSubscription,
-		PushConfig:   pushConfig,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.ModifyPushConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSubscriberListSnapshots(t *testing.T) {
-	var nextPageToken string = ""
-	var snapshotsElement *pubsubpb.Snapshot = &pubsubpb.Snapshot{}
-	var snapshots = []*pubsubpb.Snapshot{snapshotsElement}
-	var expectedResponse = &pubsubpb.ListSnapshotsResponse{
-		NextPageToken: nextPageToken,
-		Snapshots:     snapshots,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedProject string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &pubsubpb.ListSnapshotsRequest{
-		Project: formattedProject,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSnapshots(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Snapshots[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberListSnapshotsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedProject string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &pubsubpb.ListSnapshotsRequest{
-		Project: formattedProject,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSnapshots(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberCreateSnapshot(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var topic string = "topic110546223"
-	var expectedResponse = &pubsubpb.Snapshot{
-		Name:  name2,
-		Topic: topic,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/snapshots/%s", "[PROJECT]", "[SNAPSHOT]")
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.CreateSnapshotRequest{
-		Name:         formattedName,
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSnapshot(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberCreateSnapshotError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/snapshots/%s", "[PROJECT]", "[SNAPSHOT]")
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.CreateSnapshotRequest{
-		Name:         formattedName,
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSnapshot(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberUpdateSnapshot(t *testing.T) {
-	var name string = "name3373707"
-	var topic string = "topic110546223"
-	var expectedResponse = &pubsubpb.Snapshot{
-		Name:  name,
-		Topic: topic,
-	}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var seconds int64 = 123456
-	var expireTime = &timestamppb.Timestamp{
-		Seconds: seconds,
-	}
-	var snapshot = &pubsubpb.Snapshot{
-		ExpireTime: expireTime,
-	}
-	var pathsElement string = "expire_time"
-	var paths = []string{pathsElement}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var request = &pubsubpb.UpdateSnapshotRequest{
-		Snapshot:   snapshot,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSnapshot(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberUpdateSnapshotError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var seconds int64 = 123456
-	var expireTime = &timestamppb.Timestamp{
-		Seconds: seconds,
-	}
-	var snapshot = &pubsubpb.Snapshot{
-		ExpireTime: expireTime,
-	}
-	var pathsElement string = "expire_time"
-	var paths = []string{pathsElement}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var request = &pubsubpb.UpdateSnapshotRequest{
-		Snapshot:   snapshot,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSnapshot(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSubscriberDeleteSnapshot(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSnapshot string = fmt.Sprintf("projects/%s/snapshots/%s", "[PROJECT]", "[SNAPSHOT]")
-	var request = &pubsubpb.DeleteSnapshotRequest{
-		Snapshot: formattedSnapshot,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSnapshot(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSubscriberDeleteSnapshotError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSnapshot string = fmt.Sprintf("projects/%s/snapshots/%s", "[PROJECT]", "[SNAPSHOT]")
-	var request = &pubsubpb.DeleteSnapshotRequest{
-		Snapshot: formattedSnapshot,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSnapshot(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSubscriberSeek(t *testing.T) {
-	var expectedResponse *pubsubpb.SeekResponse = &pubsubpb.SeekResponse{}
-
-	mockSubscriber.err = nil
-	mockSubscriber.reqs = nil
-
-	mockSubscriber.resps = append(mockSubscriber.resps[:0], expectedResponse)
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.SeekRequest{
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Seek(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSubscriber.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSubscriberSeekError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSubscriber.err = gstatus.Error(errCode, "test error")
-
-	var formattedSubscription string = fmt.Sprintf("projects/%s/subscriptions/%s", "[PROJECT]", "[SUBSCRIPTION]")
-	var request = &pubsubpb.SeekRequest{
-		Subscription: formattedSubscription,
-	}
-
-	c, err := NewSubscriberClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Seek(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/recommender/apiv1beta1/mock_test.go b/recommender/apiv1beta1/mock_test.go
deleted file mode 100644
index 5cecc33b195..00000000000
--- a/recommender/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,494 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package recommender
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	recommenderpb "google.golang.org/genproto/googleapis/cloud/recommender/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockRecommenderServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	recommenderpb.RecommenderServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockRecommenderServer) ListRecommendations(ctx context.Context, req *recommenderpb.ListRecommendationsRequest) (*recommenderpb.ListRecommendationsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*recommenderpb.ListRecommendationsResponse), nil
-}
-
-func (s *mockRecommenderServer) GetRecommendation(ctx context.Context, req *recommenderpb.GetRecommendationRequest) (*recommenderpb.Recommendation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*recommenderpb.Recommendation), nil
-}
-
-func (s *mockRecommenderServer) MarkRecommendationClaimed(ctx context.Context, req *recommenderpb.MarkRecommendationClaimedRequest) (*recommenderpb.Recommendation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*recommenderpb.Recommendation), nil
-}
-
-func (s *mockRecommenderServer) MarkRecommendationSucceeded(ctx context.Context, req *recommenderpb.MarkRecommendationSucceededRequest) (*recommenderpb.Recommendation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*recommenderpb.Recommendation), nil
-}
-
-func (s *mockRecommenderServer) MarkRecommendationFailed(ctx context.Context, req *recommenderpb.MarkRecommendationFailedRequest) (*recommenderpb.Recommendation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*recommenderpb.Recommendation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockRecommender mockRecommenderServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	recommenderpb.RegisterRecommenderServer(serv, &mockRecommender)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestRecommenderListRecommendations(t *testing.T) {
-	var nextPageToken string = ""
-	var recommendationsElement *recommenderpb.Recommendation = &recommenderpb.Recommendation{}
-	var recommendations = []*recommenderpb.Recommendation{recommendationsElement}
-	var expectedResponse = &recommenderpb.ListRecommendationsResponse{
-		NextPageToken:   nextPageToken,
-		Recommendations: recommendations,
-	}
-
-	mockRecommender.err = nil
-	mockRecommender.reqs = nil
-
-	mockRecommender.resps = append(mockRecommender.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]")
-	var request = &recommenderpb.ListRecommendationsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListRecommendations(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockRecommender.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Recommendations[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestRecommenderListRecommendationsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockRecommender.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]")
-	var request = &recommenderpb.ListRecommendationsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListRecommendations(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestRecommenderGetRecommendation(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var recommenderSubtype string = "recommenderSubtype-1488504412"
-	var etag string = "etag3123477"
-	var expectedResponse = &recommenderpb.Recommendation{
-		Name:               name2,
-		Description:        description,
-		RecommenderSubtype: recommenderSubtype,
-		Etag:               etag,
-	}
-
-	mockRecommender.err = nil
-	mockRecommender.reqs = nil
-
-	mockRecommender.resps = append(mockRecommender.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var request = &recommenderpb.GetRecommendationRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetRecommendation(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockRecommender.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestRecommenderGetRecommendationError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockRecommender.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var request = &recommenderpb.GetRecommendationRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetRecommendation(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestRecommenderMarkRecommendationClaimed(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var recommenderSubtype string = "recommenderSubtype-1488504412"
-	var etag2 string = "etag2-1293302904"
-	var expectedResponse = &recommenderpb.Recommendation{
-		Name:               name2,
-		Description:        description,
-		RecommenderSubtype: recommenderSubtype,
-		Etag:               etag2,
-	}
-
-	mockRecommender.err = nil
-	mockRecommender.reqs = nil
-
-	mockRecommender.resps = append(mockRecommender.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var etag string = "etag3123477"
-	var request = &recommenderpb.MarkRecommendationClaimedRequest{
-		Name: formattedName,
-		Etag: etag,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.MarkRecommendationClaimed(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockRecommender.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestRecommenderMarkRecommendationClaimedError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockRecommender.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var etag string = "etag3123477"
-	var request = &recommenderpb.MarkRecommendationClaimedRequest{
-		Name: formattedName,
-		Etag: etag,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.MarkRecommendationClaimed(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestRecommenderMarkRecommendationSucceeded(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var recommenderSubtype string = "recommenderSubtype-1488504412"
-	var etag2 string = "etag2-1293302904"
-	var expectedResponse = &recommenderpb.Recommendation{
-		Name:               name2,
-		Description:        description,
-		RecommenderSubtype: recommenderSubtype,
-		Etag:               etag2,
-	}
-
-	mockRecommender.err = nil
-	mockRecommender.reqs = nil
-
-	mockRecommender.resps = append(mockRecommender.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var etag string = "etag3123477"
-	var request = &recommenderpb.MarkRecommendationSucceededRequest{
-		Name: formattedName,
-		Etag: etag,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.MarkRecommendationSucceeded(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockRecommender.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestRecommenderMarkRecommendationSucceededError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockRecommender.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var etag string = "etag3123477"
-	var request = &recommenderpb.MarkRecommendationSucceededRequest{
-		Name: formattedName,
-		Etag: etag,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.MarkRecommendationSucceeded(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestRecommenderMarkRecommendationFailed(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var recommenderSubtype string = "recommenderSubtype-1488504412"
-	var etag2 string = "etag2-1293302904"
-	var expectedResponse = &recommenderpb.Recommendation{
-		Name:               name2,
-		Description:        description,
-		RecommenderSubtype: recommenderSubtype,
-		Etag:               etag2,
-	}
-
-	mockRecommender.err = nil
-	mockRecommender.reqs = nil
-
-	mockRecommender.resps = append(mockRecommender.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var etag string = "etag3123477"
-	var request = &recommenderpb.MarkRecommendationFailedRequest{
-		Name: formattedName,
-		Etag: etag,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.MarkRecommendationFailed(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockRecommender.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestRecommenderMarkRecommendationFailedError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockRecommender.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/recommenders/%s/recommendations/%s", "[PROJECT]", "[LOCATION]", "[RECOMMENDER]", "[RECOMMENDATION]")
-	var etag string = "etag3123477"
-	var request = &recommenderpb.MarkRecommendationFailedRequest{
-		Name: formattedName,
-		Etag: etag,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.MarkRecommendationFailed(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/redis/apiv1/mock_test.go b/redis/apiv1/mock_test.go
deleted file mode 100644
index 22e9088ae64..00000000000
--- a/redis/apiv1/mock_test.go
+++ /dev/null
@@ -1,1013 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package redis
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	redispb "google.golang.org/genproto/googleapis/cloud/redis/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudRedisServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	redispb.CloudRedisServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudRedisServer) ListInstances(ctx context.Context, req *redispb.ListInstancesRequest) (*redispb.ListInstancesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*redispb.ListInstancesResponse), nil
-}
-
-func (s *mockCloudRedisServer) GetInstance(ctx context.Context, req *redispb.GetInstanceRequest) (*redispb.Instance, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*redispb.Instance), nil
-}
-
-func (s *mockCloudRedisServer) CreateInstance(ctx context.Context, req *redispb.CreateInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) UpdateInstance(ctx context.Context, req *redispb.UpdateInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) ImportInstance(ctx context.Context, req *redispb.ImportInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) ExportInstance(ctx context.Context, req *redispb.ExportInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) FailoverInstance(ctx context.Context, req *redispb.FailoverInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) DeleteInstance(ctx context.Context, req *redispb.DeleteInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudRedis mockCloudRedisServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	redispb.RegisterCloudRedisServer(serv, &mockCloudRedis)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudRedisListInstances(t *testing.T) {
-	var nextPageToken string = ""
-	var instancesElement *redispb.Instance = &redispb.Instance{}
-	var instances = []*redispb.Instance{instancesElement}
-	var expectedResponse = &redispb.ListInstancesResponse{
-		NextPageToken: nextPageToken,
-		Instances:     instances,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &redispb.ListInstancesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstances(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Instances[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisListInstancesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &redispb.ListInstancesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstances(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisGetInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.GetInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstance(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisGetInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.GetInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstance(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisCreateInstance(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb2 int32 = 1493816946
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb2,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var instanceId string = "test_instance"
-	var tier redispb.Instance_Tier = redispb.Instance_BASIC
-	var memorySizeGb int32 = 1
-	var instance = &redispb.Instance{
-		Tier:         tier,
-		MemorySizeGb: memorySizeGb,
-	}
-	var request = &redispb.CreateInstanceRequest{
-		Parent:     formattedParent,
-		InstanceId: instanceId,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisCreateInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var instanceId string = "test_instance"
-	var tier redispb.Instance_Tier = redispb.Instance_BASIC
-	var memorySizeGb int32 = 1
-	var instance = &redispb.Instance{
-		Tier:         tier,
-		MemorySizeGb: memorySizeGb,
-	}
-	var request = &redispb.CreateInstanceRequest{
-		Parent:     formattedParent,
-		InstanceId: instanceId,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisUpdateInstance(t *testing.T) {
-	var name string = "name3373707"
-	var displayName2 string = "displayName21615000987"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name,
-		DisplayName:            displayName2,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var pathsElement string = "display_name"
-	var pathsElement2 string = "memory_size_gb"
-	var paths = []string{pathsElement, pathsElement2}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var displayName string = " instance.memory_size_gb=4"
-	var instance = &redispb.Instance{
-		DisplayName: displayName,
-	}
-	var request = &redispb.UpdateInstanceRequest{
-		UpdateMask: updateMask,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisUpdateInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var pathsElement string = "display_name"
-	var pathsElement2 string = "memory_size_gb"
-	var paths = []string{pathsElement, pathsElement2}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var displayName string = " instance.memory_size_gb=4"
-	var instance = &redispb.Instance{
-		DisplayName: displayName,
-	}
-	var request = &redispb.UpdateInstanceRequest{
-		UpdateMask: updateMask,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisImportInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var name string = "name3373707"
-	var inputConfig *redispb.InputConfig = &redispb.InputConfig{}
-	var request = &redispb.ImportInstanceRequest{
-		Name:        name,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisImportInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var name string = "name3373707"
-	var inputConfig *redispb.InputConfig = &redispb.InputConfig{}
-	var request = &redispb.ImportInstanceRequest{
-		Name:        name,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisExportInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var name string = "name3373707"
-	var outputConfig *redispb.OutputConfig = &redispb.OutputConfig{}
-	var request = &redispb.ExportInstanceRequest{
-		Name:         name,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisExportInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var name string = "name3373707"
-	var outputConfig *redispb.OutputConfig = &redispb.OutputConfig{}
-	var request = &redispb.ExportInstanceRequest{
-		Name:         name,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisFailoverInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.FailoverInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.FailoverInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisFailoverInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.FailoverInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.FailoverInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisDeleteInstance(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.DeleteInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudRedisDeleteInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.DeleteInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
diff --git a/redis/apiv1beta1/mock_test.go b/redis/apiv1beta1/mock_test.go
deleted file mode 100644
index 60a6b7df6ae..00000000000
--- a/redis/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,1017 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package redis
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	redispb "google.golang.org/genproto/googleapis/cloud/redis/v1beta1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudRedisServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	redispb.CloudRedisServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudRedisServer) ListInstances(ctx context.Context, req *redispb.ListInstancesRequest) (*redispb.ListInstancesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*redispb.ListInstancesResponse), nil
-}
-
-func (s *mockCloudRedisServer) GetInstance(ctx context.Context, req *redispb.GetInstanceRequest) (*redispb.Instance, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*redispb.Instance), nil
-}
-
-func (s *mockCloudRedisServer) CreateInstance(ctx context.Context, req *redispb.CreateInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) UpdateInstance(ctx context.Context, req *redispb.UpdateInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) ImportInstance(ctx context.Context, req *redispb.ImportInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) ExportInstance(ctx context.Context, req *redispb.ExportInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) FailoverInstance(ctx context.Context, req *redispb.FailoverInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockCloudRedisServer) DeleteInstance(ctx context.Context, req *redispb.DeleteInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudRedis mockCloudRedisServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	redispb.RegisterCloudRedisServer(serv, &mockCloudRedis)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudRedisListInstances(t *testing.T) {
-	var nextPageToken string = ""
-	var instancesElement *redispb.Instance = &redispb.Instance{}
-	var instances = []*redispb.Instance{instancesElement}
-	var expectedResponse = &redispb.ListInstancesResponse{
-		NextPageToken: nextPageToken,
-		Instances:     instances,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &redispb.ListInstancesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstances(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Instances[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisListInstancesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &redispb.ListInstancesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstances(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisGetInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.GetInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstance(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisGetInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.GetInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstance(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisCreateInstance(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb2 int32 = 1493816946
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb2,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var instanceId string = "test_instance"
-	var tier redispb.Instance_Tier = redispb.Instance_BASIC
-	var memorySizeGb int32 = 1
-	var instance = &redispb.Instance{
-		Tier:         tier,
-		MemorySizeGb: memorySizeGb,
-	}
-	var request = &redispb.CreateInstanceRequest{
-		Parent:     formattedParent,
-		InstanceId: instanceId,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisCreateInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var instanceId string = "test_instance"
-	var tier redispb.Instance_Tier = redispb.Instance_BASIC
-	var memorySizeGb int32 = 1
-	var instance = &redispb.Instance{
-		Tier:         tier,
-		MemorySizeGb: memorySizeGb,
-	}
-	var request = &redispb.CreateInstanceRequest{
-		Parent:     formattedParent,
-		InstanceId: instanceId,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisUpdateInstance(t *testing.T) {
-	var name string = "name3373707"
-	var displayName2 string = "displayName21615000987"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb2 int32 = 1493816946
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name,
-		DisplayName:            displayName2,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb2,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var pathsElement string = "display_name"
-	var pathsElement2 string = "memory_size_gb"
-	var paths = []string{pathsElement, pathsElement2}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var displayName string = "UpdatedDisplayName"
-	var memorySizeGb int32 = 4
-	var instance = &redispb.Instance{
-		DisplayName:  displayName,
-		MemorySizeGb: memorySizeGb,
-	}
-	var request = &redispb.UpdateInstanceRequest{
-		UpdateMask: updateMask,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisUpdateInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var pathsElement string = "display_name"
-	var pathsElement2 string = "memory_size_gb"
-	var paths = []string{pathsElement, pathsElement2}
-	var updateMask = &field_maskpb.FieldMask{
-		Paths: paths,
-	}
-	var displayName string = "UpdatedDisplayName"
-	var memorySizeGb int32 = 4
-	var instance = &redispb.Instance{
-		DisplayName:  displayName,
-		MemorySizeGb: memorySizeGb,
-	}
-	var request = &redispb.UpdateInstanceRequest{
-		UpdateMask: updateMask,
-		Instance:   instance,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisImportInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var name string = "name3373707"
-	var inputConfig *redispb.InputConfig = &redispb.InputConfig{}
-	var request = &redispb.ImportInstanceRequest{
-		Name:        name,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisImportInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var name string = "name3373707"
-	var inputConfig *redispb.InputConfig = &redispb.InputConfig{}
-	var request = &redispb.ImportInstanceRequest{
-		Name:        name,
-		InputConfig: inputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ImportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisExportInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var name string = "name3373707"
-	var outputConfig *redispb.OutputConfig = &redispb.OutputConfig{}
-	var request = &redispb.ExportInstanceRequest{
-		Name:         name,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisExportInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var name string = "name3373707"
-	var outputConfig *redispb.OutputConfig = &redispb.OutputConfig{}
-	var request = &redispb.ExportInstanceRequest{
-		Name:         name,
-		OutputConfig: outputConfig,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.ExportInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisFailoverInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var locationId string = "locationId552319461"
-	var alternativeLocationId string = "alternativeLocationId-718920621"
-	var redisVersion string = "redisVersion-685310444"
-	var reservedIpRange string = "reservedIpRange-1082940580"
-	var host string = "host3208616"
-	var port int32 = 3446913
-	var currentLocationId string = "currentLocationId1312712735"
-	var statusMessage string = "statusMessage-239442758"
-	var memorySizeGb int32 = 34199707
-	var authorizedNetwork string = "authorizedNetwork-1733809270"
-	var persistenceIamIdentity string = "persistenceIamIdentity1061944584"
-	var expectedResponse = &redispb.Instance{
-		Name:                   name2,
-		DisplayName:            displayName,
-		LocationId:             locationId,
-		AlternativeLocationId:  alternativeLocationId,
-		RedisVersion:           redisVersion,
-		ReservedIpRange:        reservedIpRange,
-		Host:                   host,
-		Port:                   port,
-		CurrentLocationId:      currentLocationId,
-		StatusMessage:          statusMessage,
-		MemorySizeGb:           memorySizeGb,
-		AuthorizedNetwork:      authorizedNetwork,
-		PersistenceIamIdentity: persistenceIamIdentity,
-	}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.FailoverInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.FailoverInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudRedisFailoverInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.FailoverInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.FailoverInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudRedisDeleteInstance(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudRedis.err = nil
-	mockCloudRedis.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.DeleteInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudRedis.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudRedisDeleteInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudRedis.err = nil
-	mockCloudRedis.resps = append(mockCloudRedis.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/instances/%s", "[PROJECT]", "[LOCATION]", "[INSTANCE]")
-	var request = &redispb.DeleteInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudRedisClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
diff --git a/scheduler/apiv1/mock_test.go b/scheduler/apiv1/mock_test.go
deleted file mode 100644
index f451b65a8dd..00000000000
--- a/scheduler/apiv1/mock_test.go
+++ /dev/null
@@ -1,710 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package scheduler
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	schedulerpb "google.golang.org/genproto/googleapis/cloud/scheduler/v1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudSchedulerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	schedulerpb.CloudSchedulerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudSchedulerServer) ListJobs(ctx context.Context, req *schedulerpb.ListJobsRequest) (*schedulerpb.ListJobsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.ListJobsResponse), nil
-}
-
-func (s *mockCloudSchedulerServer) GetJob(ctx context.Context, req *schedulerpb.GetJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) CreateJob(ctx context.Context, req *schedulerpb.CreateJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) UpdateJob(ctx context.Context, req *schedulerpb.UpdateJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) DeleteJob(ctx context.Context, req *schedulerpb.DeleteJobRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudSchedulerServer) PauseJob(ctx context.Context, req *schedulerpb.PauseJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) ResumeJob(ctx context.Context, req *schedulerpb.ResumeJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) RunJob(ctx context.Context, req *schedulerpb.RunJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudScheduler mockCloudSchedulerServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	schedulerpb.RegisterCloudSchedulerServer(serv, &mockCloudScheduler)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudSchedulerListJobs(t *testing.T) {
-	var nextPageToken string = ""
-	var jobsElement *schedulerpb.Job = &schedulerpb.Job{}
-	var jobs = []*schedulerpb.Job{jobsElement}
-	var expectedResponse = &schedulerpb.ListJobsResponse{
-		NextPageToken: nextPageToken,
-		Jobs:          jobs,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &schedulerpb.ListJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Jobs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerListJobsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &schedulerpb.ListJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerGetJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.GetJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerGetJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.GetJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerCreateJob(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var request = &schedulerpb.CreateJobRequest{
-		Parent: formattedParent,
-		Job:    job,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerCreateJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var request = &schedulerpb.CreateJobRequest{
-		Parent: formattedParent,
-		Job:    job,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerUpdateJob(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &schedulerpb.UpdateJobRequest{
-		Job:        job,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerUpdateJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var updateMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &schedulerpb.UpdateJobRequest{
-		Job:        job,
-		UpdateMask: updateMask,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerDeleteJob(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.DeleteJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudSchedulerDeleteJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.DeleteJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudSchedulerPauseJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.PauseJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerPauseJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.PauseJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerResumeJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.ResumeJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerResumeJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.ResumeJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerRunJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.RunJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerRunJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.RunJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/scheduler/apiv1beta1/mock_test.go b/scheduler/apiv1beta1/mock_test.go
deleted file mode 100644
index 6cdd5852834..00000000000
--- a/scheduler/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,705 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package scheduler
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	schedulerpb "google.golang.org/genproto/googleapis/cloud/scheduler/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockCloudSchedulerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	schedulerpb.CloudSchedulerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockCloudSchedulerServer) ListJobs(ctx context.Context, req *schedulerpb.ListJobsRequest) (*schedulerpb.ListJobsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.ListJobsResponse), nil
-}
-
-func (s *mockCloudSchedulerServer) GetJob(ctx context.Context, req *schedulerpb.GetJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) CreateJob(ctx context.Context, req *schedulerpb.CreateJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) UpdateJob(ctx context.Context, req *schedulerpb.UpdateJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) DeleteJob(ctx context.Context, req *schedulerpb.DeleteJobRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockCloudSchedulerServer) PauseJob(ctx context.Context, req *schedulerpb.PauseJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) ResumeJob(ctx context.Context, req *schedulerpb.ResumeJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-func (s *mockCloudSchedulerServer) RunJob(ctx context.Context, req *schedulerpb.RunJobRequest) (*schedulerpb.Job, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*schedulerpb.Job), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockCloudScheduler mockCloudSchedulerServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	schedulerpb.RegisterCloudSchedulerServer(serv, &mockCloudScheduler)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestCloudSchedulerListJobs(t *testing.T) {
-	var nextPageToken string = ""
-	var jobsElement *schedulerpb.Job = &schedulerpb.Job{}
-	var jobs = []*schedulerpb.Job{jobsElement}
-	var expectedResponse = &schedulerpb.ListJobsResponse{
-		NextPageToken: nextPageToken,
-		Jobs:          jobs,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &schedulerpb.ListJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Jobs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerListJobsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &schedulerpb.ListJobsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListJobs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerGetJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.GetJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerGetJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.GetJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerCreateJob(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var request = &schedulerpb.CreateJobRequest{
-		Parent: formattedParent,
-		Job:    job,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerCreateJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var request = &schedulerpb.CreateJobRequest{
-		Parent: formattedParent,
-		Job:    job,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerUpdateJob(t *testing.T) {
-	var name string = "name3373707"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var request = &schedulerpb.UpdateJobRequest{
-		Job: job,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerUpdateJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var job *schedulerpb.Job = &schedulerpb.Job{}
-	var request = &schedulerpb.UpdateJobRequest{
-		Job: job,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerDeleteJob(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.DeleteJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestCloudSchedulerDeleteJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.DeleteJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestCloudSchedulerPauseJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.PauseJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerPauseJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.PauseJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PauseJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerResumeJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.ResumeJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerResumeJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.ResumeJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ResumeJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestCloudSchedulerRunJob(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var description string = "description-1724546052"
-	var schedule string = "schedule-697920873"
-	var timeZone string = "timeZone36848094"
-	var expectedResponse = &schedulerpb.Job{
-		Name:        name2,
-		Description: description,
-		Schedule:    schedule,
-		TimeZone:    timeZone,
-	}
-
-	mockCloudScheduler.err = nil
-	mockCloudScheduler.reqs = nil
-
-	mockCloudScheduler.resps = append(mockCloudScheduler.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.RunJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunJob(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockCloudScheduler.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestCloudSchedulerRunJobError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockCloudScheduler.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/jobs/%s", "[PROJECT]", "[LOCATION]", "[JOB]")
-	var request = &schedulerpb.RunJobRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewCloudSchedulerClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.RunJob(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/securitycenter/apiv1/mock_test.go b/securitycenter/apiv1/mock_test.go
deleted file mode 100644
index 454b1837336..00000000000
--- a/securitycenter/apiv1/mock_test.go
+++ /dev/null
@@ -1,1544 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package securitycenter
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	timestamppb "github.com/golang/protobuf/ptypes/timestamp"
-	"google.golang.org/api/option"
-	securitycenterpb "google.golang.org/genproto/googleapis/cloud/securitycenter/v1"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockSecurityCenterServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	securitycenterpb.SecurityCenterServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSecurityCenterServer) CreateSource(ctx context.Context, req *securitycenterpb.CreateSourceRequest) (*securitycenterpb.Source, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Source), nil
-}
-
-func (s *mockSecurityCenterServer) CreateFinding(ctx context.Context, req *securitycenterpb.CreateFindingRequest) (*securitycenterpb.Finding, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Finding), nil
-}
-
-func (s *mockSecurityCenterServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockSecurityCenterServer) GetOrganizationSettings(ctx context.Context, req *securitycenterpb.GetOrganizationSettingsRequest) (*securitycenterpb.OrganizationSettings, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.OrganizationSettings), nil
-}
-
-func (s *mockSecurityCenterServer) GetSource(ctx context.Context, req *securitycenterpb.GetSourceRequest) (*securitycenterpb.Source, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Source), nil
-}
-
-func (s *mockSecurityCenterServer) GroupAssets(ctx context.Context, req *securitycenterpb.GroupAssetsRequest) (*securitycenterpb.GroupAssetsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.GroupAssetsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) GroupFindings(ctx context.Context, req *securitycenterpb.GroupFindingsRequest) (*securitycenterpb.GroupFindingsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.GroupFindingsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) ListAssets(ctx context.Context, req *securitycenterpb.ListAssetsRequest) (*securitycenterpb.ListAssetsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.ListAssetsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) ListFindings(ctx context.Context, req *securitycenterpb.ListFindingsRequest) (*securitycenterpb.ListFindingsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.ListFindingsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) ListSources(ctx context.Context, req *securitycenterpb.ListSourcesRequest) (*securitycenterpb.ListSourcesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.ListSourcesResponse), nil
-}
-
-func (s *mockSecurityCenterServer) RunAssetDiscovery(ctx context.Context, req *securitycenterpb.RunAssetDiscoveryRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockSecurityCenterServer) SetFindingState(ctx context.Context, req *securitycenterpb.SetFindingStateRequest) (*securitycenterpb.Finding, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Finding), nil
-}
-
-func (s *mockSecurityCenterServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockSecurityCenterServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateFinding(ctx context.Context, req *securitycenterpb.UpdateFindingRequest) (*securitycenterpb.Finding, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Finding), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateOrganizationSettings(ctx context.Context, req *securitycenterpb.UpdateOrganizationSettingsRequest) (*securitycenterpb.OrganizationSettings, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.OrganizationSettings), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateSource(ctx context.Context, req *securitycenterpb.UpdateSourceRequest) (*securitycenterpb.Source, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Source), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateSecurityMarks(ctx context.Context, req *securitycenterpb.UpdateSecurityMarksRequest) (*securitycenterpb.SecurityMarks, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.SecurityMarks), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockSecurityCenter mockSecurityCenterServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	securitycenterpb.RegisterSecurityCenterServer(serv, &mockSecurityCenter)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestSecurityCenterCreateSource(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &securitycenterpb.Source{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.CreateSourceRequest{
-		Parent: formattedParent,
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterCreateSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.CreateSourceRequest{
-		Parent: formattedParent,
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterCreateFinding(t *testing.T) {
-	var name string = "name3373707"
-	var parent2 string = "parent21175163357"
-	var resourceName string = "resourceName979421212"
-	var category string = "category50511102"
-	var externalUri string = "externalUri-1385596168"
-	var expectedResponse = &securitycenterpb.Finding{
-		Name:         name,
-		Parent:       parent2,
-		ResourceName: resourceName,
-		Category:     category,
-		ExternalUri:  externalUri,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var findingId string = "findingId728776081"
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.CreateFindingRequest{
-		Parent:    formattedParent,
-		FindingId: findingId,
-		Finding:   finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateFinding(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterCreateFindingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var findingId string = "findingId728776081"
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.CreateFindingRequest{
-		Parent:    formattedParent,
-		FindingId: findingId,
-		Finding:   finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateFinding(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGetOrganizationSettings(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var enableAssetDiscovery bool = false
-	var expectedResponse = &securitycenterpb.OrganizationSettings{
-		Name:                 name2,
-		EnableAssetDiscovery: enableAssetDiscovery,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/organizationSettings", "[ORGANIZATION]")
-	var request = &securitycenterpb.GetOrganizationSettingsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOrganizationSettings(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGetOrganizationSettingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/organizationSettings", "[ORGANIZATION]")
-	var request = &securitycenterpb.GetOrganizationSettingsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOrganizationSettings(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGetSource(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &securitycenterpb.Source{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.GetSourceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGetSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.GetSourceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGroupAssets(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var groupByResultsElement *securitycenterpb.GroupResult = &securitycenterpb.GroupResult{}
-	var groupByResults = []*securitycenterpb.GroupResult{groupByResultsElement}
-	var expectedResponse = &securitycenterpb.GroupAssetsResponse{
-		NextPageToken:  nextPageToken,
-		TotalSize:      totalSize,
-		GroupByResults: groupByResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupAssetsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupAssets(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.GroupByResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGroupAssetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupAssetsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupAssets(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGroupFindings(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var groupByResultsElement *securitycenterpb.GroupResult = &securitycenterpb.GroupResult{}
-	var groupByResults = []*securitycenterpb.GroupResult{groupByResultsElement}
-	var expectedResponse = &securitycenterpb.GroupFindingsResponse{
-		NextPageToken:  nextPageToken,
-		TotalSize:      totalSize,
-		GroupByResults: groupByResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupFindingsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupFindings(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.GroupByResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGroupFindingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupFindingsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupFindings(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterListAssets(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var listAssetsResultsElement *securitycenterpb.ListAssetsResponse_ListAssetsResult = &securitycenterpb.ListAssetsResponse_ListAssetsResult{}
-	var listAssetsResults = []*securitycenterpb.ListAssetsResponse_ListAssetsResult{listAssetsResultsElement}
-	var expectedResponse = &securitycenterpb.ListAssetsResponse{
-		NextPageToken:     nextPageToken,
-		TotalSize:         totalSize,
-		ListAssetsResults: listAssetsResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListAssetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListAssets(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ListAssetsResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterListAssetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListAssetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListAssets(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterListFindings(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var listFindingsResultsElement *securitycenterpb.ListFindingsResponse_ListFindingsResult = &securitycenterpb.ListFindingsResponse_ListFindingsResult{}
-	var listFindingsResults = []*securitycenterpb.ListFindingsResponse_ListFindingsResult{listFindingsResultsElement}
-	var expectedResponse = &securitycenterpb.ListFindingsResponse{
-		NextPageToken:       nextPageToken,
-		TotalSize:           totalSize,
-		ListFindingsResults: listFindingsResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.ListFindingsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFindings(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ListFindingsResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterListFindingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.ListFindingsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFindings(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterListSources(t *testing.T) {
-	var nextPageToken string = ""
-	var sourcesElement *securitycenterpb.Source = &securitycenterpb.Source{}
-	var sources = []*securitycenterpb.Source{sourcesElement}
-	var expectedResponse = &securitycenterpb.ListSourcesResponse{
-		NextPageToken: nextPageToken,
-		Sources:       sources,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListSourcesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSources(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Sources[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterListSourcesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListSourcesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSources(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterRunAssetDiscovery(t *testing.T) {
-	var expectedResponse *securitycenterpb.RunAssetDiscoveryResponse = &securitycenterpb.RunAssetDiscoveryResponse{}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.RunAssetDiscoveryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.RunAssetDiscovery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterRunAssetDiscoveryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.RunAssetDiscoveryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.RunAssetDiscovery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterSetFindingState(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var parent string = "parent-995424086"
-	var resourceName string = "resourceName979421212"
-	var category string = "category50511102"
-	var externalUri string = "externalUri-1385596168"
-	var expectedResponse = &securitycenterpb.Finding{
-		Name:         name2,
-		Parent:       parent,
-		ResourceName: resourceName,
-		Category:     category,
-		ExternalUri:  externalUri,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s/findings/%s", "[ORGANIZATION]", "[SOURCE]", "[FINDING]")
-	var state securitycenterpb.Finding_State = securitycenterpb.Finding_STATE_UNSPECIFIED
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &securitycenterpb.SetFindingStateRequest{
-		Name:      formattedName,
-		State:     state,
-		StartTime: startTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetFindingState(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterSetFindingStateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s/findings/%s", "[ORGANIZATION]", "[SOURCE]", "[FINDING]")
-	var state securitycenterpb.Finding_State = securitycenterpb.Finding_STATE_UNSPECIFIED
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &securitycenterpb.SetFindingStateRequest{
-		Name:      formattedName,
-		State:     state,
-		StartTime: startTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetFindingState(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateFinding(t *testing.T) {
-	var name string = "name3373707"
-	var parent string = "parent-995424086"
-	var resourceName string = "resourceName979421212"
-	var category string = "category50511102"
-	var externalUri string = "externalUri-1385596168"
-	var expectedResponse = &securitycenterpb.Finding{
-		Name:         name,
-		Parent:       parent,
-		ResourceName: resourceName,
-		Category:     category,
-		ExternalUri:  externalUri,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.UpdateFindingRequest{
-		Finding: finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateFinding(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateFindingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.UpdateFindingRequest{
-		Finding: finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateFinding(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateOrganizationSettings(t *testing.T) {
-	var name string = "name3373707"
-	var enableAssetDiscovery bool = false
-	var expectedResponse = &securitycenterpb.OrganizationSettings{
-		Name:                 name,
-		EnableAssetDiscovery: enableAssetDiscovery,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var organizationSettings *securitycenterpb.OrganizationSettings = &securitycenterpb.OrganizationSettings{}
-	var request = &securitycenterpb.UpdateOrganizationSettingsRequest{
-		OrganizationSettings: organizationSettings,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOrganizationSettings(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateOrganizationSettingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var organizationSettings *securitycenterpb.OrganizationSettings = &securitycenterpb.OrganizationSettings{}
-	var request = &securitycenterpb.UpdateOrganizationSettingsRequest{
-		OrganizationSettings: organizationSettings,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOrganizationSettings(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateSource(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &securitycenterpb.Source{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.UpdateSourceRequest{
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.UpdateSourceRequest{
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateSecurityMarks(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &securitycenterpb.SecurityMarks{
-		Name: name,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var securityMarks *securitycenterpb.SecurityMarks = &securitycenterpb.SecurityMarks{}
-	var request = &securitycenterpb.UpdateSecurityMarksRequest{
-		SecurityMarks: securityMarks,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSecurityMarks(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateSecurityMarksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var securityMarks *securitycenterpb.SecurityMarks = &securitycenterpb.SecurityMarks{}
-	var request = &securitycenterpb.UpdateSecurityMarksRequest{
-		SecurityMarks: securityMarks,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSecurityMarks(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/securitycenter/apiv1beta1/mock_test.go b/securitycenter/apiv1beta1/mock_test.go
deleted file mode 100644
index ffc63b37862..00000000000
--- a/securitycenter/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,1537 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package securitycenter
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	timestamppb "github.com/golang/protobuf/ptypes/timestamp"
-	"google.golang.org/api/option"
-	securitycenterpb "google.golang.org/genproto/googleapis/cloud/securitycenter/v1beta1"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockSecurityCenterServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	securitycenterpb.SecurityCenterServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSecurityCenterServer) CreateSource(ctx context.Context, req *securitycenterpb.CreateSourceRequest) (*securitycenterpb.Source, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Source), nil
-}
-
-func (s *mockSecurityCenterServer) CreateFinding(ctx context.Context, req *securitycenterpb.CreateFindingRequest) (*securitycenterpb.Finding, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Finding), nil
-}
-
-func (s *mockSecurityCenterServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockSecurityCenterServer) GetOrganizationSettings(ctx context.Context, req *securitycenterpb.GetOrganizationSettingsRequest) (*securitycenterpb.OrganizationSettings, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.OrganizationSettings), nil
-}
-
-func (s *mockSecurityCenterServer) GetSource(ctx context.Context, req *securitycenterpb.GetSourceRequest) (*securitycenterpb.Source, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Source), nil
-}
-
-func (s *mockSecurityCenterServer) GroupAssets(ctx context.Context, req *securitycenterpb.GroupAssetsRequest) (*securitycenterpb.GroupAssetsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.GroupAssetsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) GroupFindings(ctx context.Context, req *securitycenterpb.GroupFindingsRequest) (*securitycenterpb.GroupFindingsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.GroupFindingsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) ListAssets(ctx context.Context, req *securitycenterpb.ListAssetsRequest) (*securitycenterpb.ListAssetsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.ListAssetsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) ListFindings(ctx context.Context, req *securitycenterpb.ListFindingsRequest) (*securitycenterpb.ListFindingsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.ListFindingsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) ListSources(ctx context.Context, req *securitycenterpb.ListSourcesRequest) (*securitycenterpb.ListSourcesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.ListSourcesResponse), nil
-}
-
-func (s *mockSecurityCenterServer) RunAssetDiscovery(ctx context.Context, req *securitycenterpb.RunAssetDiscoveryRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockSecurityCenterServer) SetFindingState(ctx context.Context, req *securitycenterpb.SetFindingStateRequest) (*securitycenterpb.Finding, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Finding), nil
-}
-
-func (s *mockSecurityCenterServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockSecurityCenterServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateFinding(ctx context.Context, req *securitycenterpb.UpdateFindingRequest) (*securitycenterpb.Finding, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Finding), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateOrganizationSettings(ctx context.Context, req *securitycenterpb.UpdateOrganizationSettingsRequest) (*securitycenterpb.OrganizationSettings, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.OrganizationSettings), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateSource(ctx context.Context, req *securitycenterpb.UpdateSourceRequest) (*securitycenterpb.Source, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.Source), nil
-}
-
-func (s *mockSecurityCenterServer) UpdateSecurityMarks(ctx context.Context, req *securitycenterpb.UpdateSecurityMarksRequest) (*securitycenterpb.SecurityMarks, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*securitycenterpb.SecurityMarks), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockSecurityCenter mockSecurityCenterServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	securitycenterpb.RegisterSecurityCenterServer(serv, &mockSecurityCenter)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestSecurityCenterCreateSource(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &securitycenterpb.Source{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.CreateSourceRequest{
-		Parent: formattedParent,
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterCreateSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.CreateSourceRequest{
-		Parent: formattedParent,
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterCreateFinding(t *testing.T) {
-	var name string = "name3373707"
-	var parent2 string = "parent21175163357"
-	var resourceName string = "resourceName979421212"
-	var category string = "category50511102"
-	var externalUri string = "externalUri-1385596168"
-	var expectedResponse = &securitycenterpb.Finding{
-		Name:         name,
-		Parent:       parent2,
-		ResourceName: resourceName,
-		Category:     category,
-		ExternalUri:  externalUri,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var findingId string = "findingId728776081"
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.CreateFindingRequest{
-		Parent:    formattedParent,
-		FindingId: findingId,
-		Finding:   finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateFinding(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterCreateFindingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var findingId string = "findingId728776081"
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.CreateFindingRequest{
-		Parent:    formattedParent,
-		FindingId: findingId,
-		Finding:   finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateFinding(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: formattedResource,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGetOrganizationSettings(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var enableAssetDiscovery bool = false
-	var expectedResponse = &securitycenterpb.OrganizationSettings{
-		Name:                 name2,
-		EnableAssetDiscovery: enableAssetDiscovery,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/organizationSettings", "[ORGANIZATION]")
-	var request = &securitycenterpb.GetOrganizationSettingsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOrganizationSettings(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGetOrganizationSettingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/organizationSettings", "[ORGANIZATION]")
-	var request = &securitycenterpb.GetOrganizationSettingsRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetOrganizationSettings(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGetSource(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &securitycenterpb.Source{
-		Name:        name2,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.GetSourceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGetSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.GetSourceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGroupAssets(t *testing.T) {
-	var nextPageToken string = ""
-	var groupByResultsElement *securitycenterpb.GroupResult = &securitycenterpb.GroupResult{}
-	var groupByResults = []*securitycenterpb.GroupResult{groupByResultsElement}
-	var expectedResponse = &securitycenterpb.GroupAssetsResponse{
-		NextPageToken:  nextPageToken,
-		GroupByResults: groupByResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupAssetsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupAssets(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.GroupByResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGroupAssetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupAssetsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupAssets(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterGroupFindings(t *testing.T) {
-	var nextPageToken string = ""
-	var groupByResultsElement *securitycenterpb.GroupResult = &securitycenterpb.GroupResult{}
-	var groupByResults = []*securitycenterpb.GroupResult{groupByResultsElement}
-	var expectedResponse = &securitycenterpb.GroupFindingsResponse{
-		NextPageToken:  nextPageToken,
-		GroupByResults: groupByResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupFindingsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupFindings(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.GroupByResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterGroupFindingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var groupBy string = "groupBy506361367"
-	var request = &securitycenterpb.GroupFindingsRequest{
-		Parent:  formattedParent,
-		GroupBy: groupBy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GroupFindings(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterListAssets(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var listAssetsResultsElement *securitycenterpb.ListAssetsResponse_ListAssetsResult = &securitycenterpb.ListAssetsResponse_ListAssetsResult{}
-	var listAssetsResults = []*securitycenterpb.ListAssetsResponse_ListAssetsResult{listAssetsResultsElement}
-	var expectedResponse = &securitycenterpb.ListAssetsResponse{
-		NextPageToken:     nextPageToken,
-		TotalSize:         totalSize,
-		ListAssetsResults: listAssetsResults,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListAssetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListAssets(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.ListAssetsResults[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterListAssetsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListAssetsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListAssets(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterListFindings(t *testing.T) {
-	var nextPageToken string = ""
-	var totalSize int32 = 705419236
-	var findingsElement *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var findings = []*securitycenterpb.Finding{findingsElement}
-	var expectedResponse = &securitycenterpb.ListFindingsResponse{
-		NextPageToken: nextPageToken,
-		TotalSize:     totalSize,
-		Findings:      findings,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.ListFindingsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFindings(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Findings[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterListFindingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var request = &securitycenterpb.ListFindingsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListFindings(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterListSources(t *testing.T) {
-	var nextPageToken string = ""
-	var sourcesElement *securitycenterpb.Source = &securitycenterpb.Source{}
-	var sources = []*securitycenterpb.Source{sourcesElement}
-	var expectedResponse = &securitycenterpb.ListSourcesResponse{
-		NextPageToken: nextPageToken,
-		Sources:       sources,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListSourcesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSources(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Sources[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterListSourcesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.ListSourcesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSources(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterRunAssetDiscovery(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.RunAssetDiscoveryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.RunAssetDiscovery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSecurityCenterRunAssetDiscoveryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
-	var request = &securitycenterpb.RunAssetDiscoveryRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.RunAssetDiscovery(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	err = respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSecurityCenterSetFindingState(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var parent string = "parent-995424086"
-	var resourceName string = "resourceName979421212"
-	var category string = "category50511102"
-	var externalUri string = "externalUri-1385596168"
-	var expectedResponse = &securitycenterpb.Finding{
-		Name:         name2,
-		Parent:       parent,
-		ResourceName: resourceName,
-		Category:     category,
-		ExternalUri:  externalUri,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s/findings/%s", "[ORGANIZATION]", "[SOURCE]", "[FINDING]")
-	var state securitycenterpb.Finding_State = securitycenterpb.Finding_STATE_UNSPECIFIED
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &securitycenterpb.SetFindingStateRequest{
-		Name:      formattedName,
-		State:     state,
-		StartTime: startTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetFindingState(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterSetFindingStateError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("organizations/%s/sources/%s/findings/%s", "[ORGANIZATION]", "[SOURCE]", "[FINDING]")
-	var state securitycenterpb.Finding_State = securitycenterpb.Finding_STATE_UNSPECIFIED
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &securitycenterpb.SetFindingStateRequest{
-		Name:      formattedName,
-		State:     state,
-		StartTime: startTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetFindingState(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: formattedResource,
-		Policy:   policy,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var formattedResource string = fmt.Sprintf("organizations/%s/sources/%s", "[ORGANIZATION]", "[SOURCE]")
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    formattedResource,
-		Permissions: permissions,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateFinding(t *testing.T) {
-	var name string = "name3373707"
-	var parent string = "parent-995424086"
-	var resourceName string = "resourceName979421212"
-	var category string = "category50511102"
-	var externalUri string = "externalUri-1385596168"
-	var expectedResponse = &securitycenterpb.Finding{
-		Name:         name,
-		Parent:       parent,
-		ResourceName: resourceName,
-		Category:     category,
-		ExternalUri:  externalUri,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.UpdateFindingRequest{
-		Finding: finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateFinding(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateFindingError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var finding *securitycenterpb.Finding = &securitycenterpb.Finding{}
-	var request = &securitycenterpb.UpdateFindingRequest{
-		Finding: finding,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateFinding(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateOrganizationSettings(t *testing.T) {
-	var name string = "name3373707"
-	var enableAssetDiscovery bool = false
-	var expectedResponse = &securitycenterpb.OrganizationSettings{
-		Name:                 name,
-		EnableAssetDiscovery: enableAssetDiscovery,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var organizationSettings *securitycenterpb.OrganizationSettings = &securitycenterpb.OrganizationSettings{}
-	var request = &securitycenterpb.UpdateOrganizationSettingsRequest{
-		OrganizationSettings: organizationSettings,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOrganizationSettings(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateOrganizationSettingsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var organizationSettings *securitycenterpb.OrganizationSettings = &securitycenterpb.OrganizationSettings{}
-	var request = &securitycenterpb.UpdateOrganizationSettingsRequest{
-		OrganizationSettings: organizationSettings,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateOrganizationSettings(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateSource(t *testing.T) {
-	var name string = "name3373707"
-	var displayName string = "displayName1615086568"
-	var description string = "description-1724546052"
-	var expectedResponse = &securitycenterpb.Source{
-		Name:        name,
-		DisplayName: displayName,
-		Description: description,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.UpdateSourceRequest{
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSource(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateSourceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var source *securitycenterpb.Source = &securitycenterpb.Source{}
-	var request = &securitycenterpb.UpdateSourceRequest{
-		Source: source,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSource(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSecurityCenterUpdateSecurityMarks(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &securitycenterpb.SecurityMarks{
-		Name: name,
-	}
-
-	mockSecurityCenter.err = nil
-	mockSecurityCenter.reqs = nil
-
-	mockSecurityCenter.resps = append(mockSecurityCenter.resps[:0], expectedResponse)
-
-	var securityMarks *securitycenterpb.SecurityMarks = &securitycenterpb.SecurityMarks{}
-	var request = &securitycenterpb.UpdateSecurityMarksRequest{
-		SecurityMarks: securityMarks,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSecurityMarks(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSecurityCenter.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSecurityCenterUpdateSecurityMarksError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSecurityCenter.err = gstatus.Error(errCode, "test error")
-
-	var securityMarks *securitycenterpb.SecurityMarks = &securitycenterpb.SecurityMarks{}
-	var request = &securitycenterpb.UpdateSecurityMarksRequest{
-		SecurityMarks: securityMarks,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.UpdateSecurityMarks(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/spanner/admin/instance/apiv1/mock_test.go b/spanner/admin/instance/apiv1/mock_test.go
deleted file mode 100644
index f894774a9a8..00000000000
--- a/spanner/admin/instance/apiv1/mock_test.go
+++ /dev/null
@@ -1,916 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package instance
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	iampb "google.golang.org/genproto/googleapis/iam/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-	instancepb "google.golang.org/genproto/googleapis/spanner/admin/instance/v1"
-	field_maskpb "google.golang.org/genproto/protobuf/field_mask"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockInstanceAdminServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	instancepb.InstanceAdminServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockInstanceAdminServer) ListInstanceConfigs(ctx context.Context, req *instancepb.ListInstanceConfigsRequest) (*instancepb.ListInstanceConfigsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*instancepb.ListInstanceConfigsResponse), nil
-}
-
-func (s *mockInstanceAdminServer) GetInstanceConfig(ctx context.Context, req *instancepb.GetInstanceConfigRequest) (*instancepb.InstanceConfig, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*instancepb.InstanceConfig), nil
-}
-
-func (s *mockInstanceAdminServer) ListInstances(ctx context.Context, req *instancepb.ListInstancesRequest) (*instancepb.ListInstancesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*instancepb.ListInstancesResponse), nil
-}
-
-func (s *mockInstanceAdminServer) GetInstance(ctx context.Context, req *instancepb.GetInstanceRequest) (*instancepb.Instance, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*instancepb.Instance), nil
-}
-
-func (s *mockInstanceAdminServer) CreateInstance(ctx context.Context, req *instancepb.CreateInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockInstanceAdminServer) UpdateInstance(ctx context.Context, req *instancepb.UpdateInstanceRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockInstanceAdminServer) DeleteInstance(ctx context.Context, req *instancepb.DeleteInstanceRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockInstanceAdminServer) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockInstanceAdminServer) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest) (*iampb.Policy, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.Policy), nil
-}
-
-func (s *mockInstanceAdminServer) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest) (*iampb.TestIamPermissionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*iampb.TestIamPermissionsResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockInstanceAdmin mockInstanceAdminServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	instancepb.RegisterInstanceAdminServer(serv, &mockInstanceAdmin)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestInstanceAdminListInstanceConfigs(t *testing.T) {
-	var nextPageToken string = ""
-	var instanceConfigsElement *instancepb.InstanceConfig = &instancepb.InstanceConfig{}
-	var instanceConfigs = []*instancepb.InstanceConfig{instanceConfigsElement}
-	var expectedResponse = &instancepb.ListInstanceConfigsResponse{
-		NextPageToken:   nextPageToken,
-		InstanceConfigs: instanceConfigs,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &instancepb.ListInstanceConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstanceConfigs(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.InstanceConfigs[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminListInstanceConfigsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &instancepb.ListInstanceConfigsRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstanceConfigs(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminGetInstanceConfig(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var displayName string = "displayName1615086568"
-	var expectedResponse = &instancepb.InstanceConfig{
-		Name:        name2,
-		DisplayName: displayName,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/instanceConfigs/%s", "[PROJECT]", "[INSTANCE_CONFIG]")
-	var request = &instancepb.GetInstanceConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstanceConfig(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminGetInstanceConfigError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/instanceConfigs/%s", "[PROJECT]", "[INSTANCE_CONFIG]")
-	var request = &instancepb.GetInstanceConfigRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstanceConfig(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminListInstances(t *testing.T) {
-	var nextPageToken string = ""
-	var instancesElement *instancepb.Instance = &instancepb.Instance{}
-	var instances = []*instancepb.Instance{instancesElement}
-	var expectedResponse = &instancepb.ListInstancesResponse{
-		NextPageToken: nextPageToken,
-		Instances:     instances,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &instancepb.ListInstancesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstances(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Instances[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminListInstancesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var request = &instancepb.ListInstancesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListInstances(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminGetInstance(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var config string = "config-1354792126"
-	var displayName string = "displayName1615086568"
-	var nodeCount int32 = 1539922066
-	var expectedResponse = &instancepb.Instance{
-		Name:        name2,
-		Config:      config,
-		DisplayName: displayName,
-		NodeCount:   nodeCount,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s", "[PROJECT]", "[INSTANCE]")
-	var request = &instancepb.GetInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstance(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminGetInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s", "[PROJECT]", "[INSTANCE]")
-	var request = &instancepb.GetInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetInstance(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminCreateInstance(t *testing.T) {
-	var name string = "name3373707"
-	var config string = "config-1354792126"
-	var displayName string = "displayName1615086568"
-	var nodeCount int32 = 1539922066
-	var expectedResponse = &instancepb.Instance{
-		Name:        name,
-		Config:      config,
-		DisplayName: displayName,
-		NodeCount:   nodeCount,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var instanceId string = "instanceId-2101995259"
-	var instance *instancepb.Instance = &instancepb.Instance{}
-	var request = &instancepb.CreateInstanceRequest{
-		Parent:     formattedParent,
-		InstanceId: instanceId,
-		Instance:   instance,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminCreateInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var instanceId string = "instanceId-2101995259"
-	var instance *instancepb.Instance = &instancepb.Instance{}
-	var request = &instancepb.CreateInstanceRequest{
-		Parent:     formattedParent,
-		InstanceId: instanceId,
-		Instance:   instance,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminUpdateInstance(t *testing.T) {
-	var name string = "name3373707"
-	var config string = "config-1354792126"
-	var displayName string = "displayName1615086568"
-	var nodeCount int32 = 1539922066
-	var expectedResponse = &instancepb.Instance{
-		Name:        name,
-		Config:      config,
-		DisplayName: displayName,
-		NodeCount:   nodeCount,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var instance *instancepb.Instance = &instancepb.Instance{}
-	var fieldMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &instancepb.UpdateInstanceRequest{
-		Instance:  instance,
-		FieldMask: fieldMask,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminUpdateInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var instance *instancepb.Instance = &instancepb.Instance{}
-	var fieldMask *field_maskpb.FieldMask = &field_maskpb.FieldMask{}
-	var request = &instancepb.UpdateInstanceRequest{
-		Instance:  instance,
-		FieldMask: fieldMask,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.UpdateInstance(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminDeleteInstance(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s", "[PROJECT]", "[INSTANCE]")
-	var request = &instancepb.DeleteInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteInstance(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestInstanceAdminDeleteInstanceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s", "[PROJECT]", "[INSTANCE]")
-	var request = &instancepb.DeleteInstanceRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteInstance(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestInstanceAdminSetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var resource string = "resource-341064690"
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: resource,
-		Policy:   policy,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminSetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var resource string = "resource-341064690"
-	var policy *iampb.Policy = &iampb.Policy{}
-	var request = &iampb.SetIamPolicyRequest{
-		Resource: resource,
-		Policy:   policy,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminGetIamPolicy(t *testing.T) {
-	var version int32 = 351608024
-	var etag []byte = []byte("21")
-	var expectedResponse = &iampb.Policy{
-		Version: version,
-		Etag:    etag,
-	}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var resource string = "resource-341064690"
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: resource,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminGetIamPolicyError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var resource string = "resource-341064690"
-	var request = &iampb.GetIamPolicyRequest{
-		Resource: resource,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetIamPolicy(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestInstanceAdminTestIamPermissions(t *testing.T) {
-	var expectedResponse *iampb.TestIamPermissionsResponse = &iampb.TestIamPermissionsResponse{}
-
-	mockInstanceAdmin.err = nil
-	mockInstanceAdmin.reqs = nil
-
-	mockInstanceAdmin.resps = append(mockInstanceAdmin.resps[:0], expectedResponse)
-
-	var resource string = "resource-341064690"
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    resource,
-		Permissions: permissions,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockInstanceAdmin.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestInstanceAdminTestIamPermissionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockInstanceAdmin.err = gstatus.Error(errCode, "test error")
-
-	var resource string = "resource-341064690"
-	var permissions []string = nil
-	var request = &iampb.TestIamPermissionsRequest{
-		Resource:    resource,
-		Permissions: permissions,
-	}
-
-	c, err := NewInstanceAdminClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TestIamPermissions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/spanner/apiv1/mock_test.go b/spanner/apiv1/mock_test.go
deleted file mode 100644
index e66b1113810..00000000000
--- a/spanner/apiv1/mock_test.go
+++ /dev/null
@@ -1,1232 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package spanner
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	spannerpb "google.golang.org/genproto/googleapis/spanner/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockSpannerServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	spannerpb.SpannerServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSpannerServer) CreateSession(ctx context.Context, req *spannerpb.CreateSessionRequest) (*spannerpb.Session, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.Session), nil
-}
-
-func (s *mockSpannerServer) BatchCreateSessions(ctx context.Context, req *spannerpb.BatchCreateSessionsRequest) (*spannerpb.BatchCreateSessionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.BatchCreateSessionsResponse), nil
-}
-
-func (s *mockSpannerServer) GetSession(ctx context.Context, req *spannerpb.GetSessionRequest) (*spannerpb.Session, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.Session), nil
-}
-
-func (s *mockSpannerServer) ListSessions(ctx context.Context, req *spannerpb.ListSessionsRequest) (*spannerpb.ListSessionsResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.ListSessionsResponse), nil
-}
-
-func (s *mockSpannerServer) DeleteSession(ctx context.Context, req *spannerpb.DeleteSessionRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSpannerServer) ExecuteSql(ctx context.Context, req *spannerpb.ExecuteSqlRequest) (*spannerpb.ResultSet, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.ResultSet), nil
-}
-
-func (s *mockSpannerServer) ExecuteStreamingSql(req *spannerpb.ExecuteSqlRequest, stream spannerpb.Spanner_ExecuteStreamingSqlServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*spannerpb.PartialResultSet)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockSpannerServer) ExecuteBatchDml(ctx context.Context, req *spannerpb.ExecuteBatchDmlRequest) (*spannerpb.ExecuteBatchDmlResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.ExecuteBatchDmlResponse), nil
-}
-
-func (s *mockSpannerServer) Read(ctx context.Context, req *spannerpb.ReadRequest) (*spannerpb.ResultSet, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.ResultSet), nil
-}
-
-func (s *mockSpannerServer) StreamingRead(req *spannerpb.ReadRequest, stream spannerpb.Spanner_StreamingReadServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*spannerpb.PartialResultSet)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func (s *mockSpannerServer) BeginTransaction(ctx context.Context, req *spannerpb.BeginTransactionRequest) (*spannerpb.Transaction, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.Transaction), nil
-}
-
-func (s *mockSpannerServer) Commit(ctx context.Context, req *spannerpb.CommitRequest) (*spannerpb.CommitResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.CommitResponse), nil
-}
-
-func (s *mockSpannerServer) Rollback(ctx context.Context, req *spannerpb.RollbackRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockSpannerServer) PartitionQuery(ctx context.Context, req *spannerpb.PartitionQueryRequest) (*spannerpb.PartitionResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.PartitionResponse), nil
-}
-
-func (s *mockSpannerServer) PartitionRead(ctx context.Context, req *spannerpb.PartitionReadRequest) (*spannerpb.PartitionResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*spannerpb.PartitionResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockSpanner mockSpannerServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	spannerpb.RegisterSpannerServer(serv, &mockSpanner)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestSpannerCreateSession(t *testing.T) {
-	var name string = "name3373707"
-	var expectedResponse = &spannerpb.Session{
-		Name: name,
-	}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/instances/%s/databases/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]")
-	var request = &spannerpb.CreateSessionRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSession(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerCreateSessionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/instances/%s/databases/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]")
-	var request = &spannerpb.CreateSessionRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSession(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerBatchCreateSessions(t *testing.T) {
-	var expectedResponse *spannerpb.BatchCreateSessionsResponse = &spannerpb.BatchCreateSessionsResponse{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/instances/%s/databases/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]")
-	var sessionCount int32 = 185691686
-	var request = &spannerpb.BatchCreateSessionsRequest{
-		Database:     formattedDatabase,
-		SessionCount: sessionCount,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateSessions(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerBatchCreateSessionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/instances/%s/databases/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]")
-	var sessionCount int32 = 185691686
-	var request = &spannerpb.BatchCreateSessionsRequest{
-		Database:     formattedDatabase,
-		SessionCount: sessionCount,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchCreateSessions(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerGetSession(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &spannerpb.Session{
-		Name: name2,
-	}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var request = &spannerpb.GetSessionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSession(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerGetSessionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var request = &spannerpb.GetSessionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSession(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerListSessions(t *testing.T) {
-	var nextPageToken string = ""
-	var sessionsElement *spannerpb.Session = &spannerpb.Session{}
-	var sessions = []*spannerpb.Session{sessionsElement}
-	var expectedResponse = &spannerpb.ListSessionsResponse{
-		NextPageToken: nextPageToken,
-		Sessions:      sessions,
-	}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/instances/%s/databases/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]")
-	var request = &spannerpb.ListSessionsRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSessions(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Sessions[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerListSessionsError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedDatabase string = fmt.Sprintf("projects/%s/instances/%s/databases/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]")
-	var request = &spannerpb.ListSessionsRequest{
-		Database: formattedDatabase,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListSessions(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerDeleteSession(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var request = &spannerpb.DeleteSessionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSession(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSpannerDeleteSessionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var request = &spannerpb.DeleteSessionRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.DeleteSession(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSpannerExecuteSql(t *testing.T) {
-	var expectedResponse *spannerpb.ResultSet = &spannerpb.ResultSet{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var sql string = "sql114126"
-	var request = &spannerpb.ExecuteSqlRequest{
-		Session: formattedSession,
-		Sql:     sql,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ExecuteSql(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerExecuteSqlError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var sql string = "sql114126"
-	var request = &spannerpb.ExecuteSqlRequest{
-		Session: formattedSession,
-		Sql:     sql,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ExecuteSql(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerExecuteStreamingSql(t *testing.T) {
-	var chunkedValue bool = true
-	var resumeToken []byte = []byte("103")
-	var expectedResponse = &spannerpb.PartialResultSet{
-		ChunkedValue: chunkedValue,
-		ResumeToken:  resumeToken,
-	}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var sql string = "sql114126"
-	var request = &spannerpb.ExecuteSqlRequest{
-		Session: formattedSession,
-		Sql:     sql,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.ExecuteStreamingSql(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerExecuteStreamingSqlError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var sql string = "sql114126"
-	var request = &spannerpb.ExecuteSqlRequest{
-		Session: formattedSession,
-		Sql:     sql,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.ExecuteStreamingSql(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerExecuteBatchDml(t *testing.T) {
-	var expectedResponse *spannerpb.ExecuteBatchDmlResponse = &spannerpb.ExecuteBatchDmlResponse{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var transaction *spannerpb.TransactionSelector = &spannerpb.TransactionSelector{}
-	var statements []*spannerpb.ExecuteBatchDmlRequest_Statement = nil
-	var seqno int64 = 109325920
-	var request = &spannerpb.ExecuteBatchDmlRequest{
-		Session:     formattedSession,
-		Transaction: transaction,
-		Statements:  statements,
-		Seqno:       seqno,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ExecuteBatchDml(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerExecuteBatchDmlError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var transaction *spannerpb.TransactionSelector = &spannerpb.TransactionSelector{}
-	var statements []*spannerpb.ExecuteBatchDmlRequest_Statement = nil
-	var seqno int64 = 109325920
-	var request = &spannerpb.ExecuteBatchDmlRequest{
-		Session:     formattedSession,
-		Transaction: transaction,
-		Statements:  statements,
-		Seqno:       seqno,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ExecuteBatchDml(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerRead(t *testing.T) {
-	var expectedResponse *spannerpb.ResultSet = &spannerpb.ResultSet{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var table string = "table110115790"
-	var columns []string = nil
-	var keySet *spannerpb.KeySet = &spannerpb.KeySet{}
-	var request = &spannerpb.ReadRequest{
-		Session: formattedSession,
-		Table:   table,
-		Columns: columns,
-		KeySet:  keySet,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Read(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerReadError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var table string = "table110115790"
-	var columns []string = nil
-	var keySet *spannerpb.KeySet = &spannerpb.KeySet{}
-	var request = &spannerpb.ReadRequest{
-		Session: formattedSession,
-		Table:   table,
-		Columns: columns,
-		KeySet:  keySet,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Read(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerStreamingRead(t *testing.T) {
-	var chunkedValue bool = true
-	var resumeToken []byte = []byte("103")
-	var expectedResponse = &spannerpb.PartialResultSet{
-		ChunkedValue: chunkedValue,
-		ResumeToken:  resumeToken,
-	}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var table string = "table110115790"
-	var columns []string = nil
-	var keySet *spannerpb.KeySet = &spannerpb.KeySet{}
-	var request = &spannerpb.ReadRequest{
-		Session: formattedSession,
-		Table:   table,
-		Columns: columns,
-		KeySet:  keySet,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingRead(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerStreamingReadError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var table string = "table110115790"
-	var columns []string = nil
-	var keySet *spannerpb.KeySet = &spannerpb.KeySet{}
-	var request = &spannerpb.ReadRequest{
-		Session: formattedSession,
-		Table:   table,
-		Columns: columns,
-		KeySet:  keySet,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingRead(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerBeginTransaction(t *testing.T) {
-	var id []byte = []byte("27")
-	var expectedResponse = &spannerpb.Transaction{
-		Id: id,
-	}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var options *spannerpb.TransactionOptions = &spannerpb.TransactionOptions{}
-	var request = &spannerpb.BeginTransactionRequest{
-		Session: formattedSession,
-		Options: options,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BeginTransaction(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerBeginTransactionError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var options *spannerpb.TransactionOptions = &spannerpb.TransactionOptions{}
-	var request = &spannerpb.BeginTransactionRequest{
-		Session: formattedSession,
-		Options: options,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BeginTransaction(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerCommit(t *testing.T) {
-	var expectedResponse *spannerpb.CommitResponse = &spannerpb.CommitResponse{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var request = &spannerpb.CommitRequest{
-		Session: formattedSession,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Commit(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerCommitError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var request = &spannerpb.CommitRequest{
-		Session: formattedSession,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Commit(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerRollback(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var transactionId []byte = []byte("28")
-	var request = &spannerpb.RollbackRequest{
-		Session:       formattedSession,
-		TransactionId: transactionId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Rollback(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestSpannerRollbackError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var transactionId []byte = []byte("28")
-	var request = &spannerpb.RollbackRequest{
-		Session:       formattedSession,
-		TransactionId: transactionId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.Rollback(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestSpannerPartitionQuery(t *testing.T) {
-	var expectedResponse *spannerpb.PartitionResponse = &spannerpb.PartitionResponse{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var sql string = "sql114126"
-	var request = &spannerpb.PartitionQueryRequest{
-		Session: formattedSession,
-		Sql:     sql,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PartitionQuery(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerPartitionQueryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var sql string = "sql114126"
-	var request = &spannerpb.PartitionQueryRequest{
-		Session: formattedSession,
-		Sql:     sql,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PartitionQuery(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpannerPartitionRead(t *testing.T) {
-	var expectedResponse *spannerpb.PartitionResponse = &spannerpb.PartitionResponse{}
-
-	mockSpanner.err = nil
-	mockSpanner.reqs = nil
-
-	mockSpanner.resps = append(mockSpanner.resps[:0], expectedResponse)
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var table string = "table110115790"
-	var keySet *spannerpb.KeySet = &spannerpb.KeySet{}
-	var request = &spannerpb.PartitionReadRequest{
-		Session: formattedSession,
-		Table:   table,
-		KeySet:  keySet,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PartitionRead(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpanner.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpannerPartitionReadError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpanner.err = gstatus.Error(errCode, "test error")
-
-	var formattedSession string = fmt.Sprintf("projects/%s/instances/%s/databases/%s/sessions/%s", "[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]")
-	var table string = "table110115790"
-	var keySet *spannerpb.KeySet = &spannerpb.KeySet{}
-	var request = &spannerpb.PartitionReadRequest{
-		Session: formattedSession,
-		Table:   table,
-		KeySet:  keySet,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.PartitionRead(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/speech/apiv1/mock_test.go b/speech/apiv1/mock_test.go
deleted file mode 100644
index 7ef0da0a64e..00000000000
--- a/speech/apiv1/mock_test.go
+++ /dev/null
@@ -1,404 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package speech
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	speechpb "google.golang.org/genproto/googleapis/cloud/speech/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockSpeechServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	speechpb.SpeechServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSpeechServer) Recognize(ctx context.Context, req *speechpb.RecognizeRequest) (*speechpb.RecognizeResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*speechpb.RecognizeResponse), nil
-}
-
-func (s *mockSpeechServer) LongRunningRecognize(ctx context.Context, req *speechpb.LongRunningRecognizeRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockSpeechServer) StreamingRecognize(stream speechpb.Speech_StreamingRecognizeServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*speechpb.StreamingRecognizeResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockSpeech mockSpeechServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	speechpb.RegisterSpeechServer(serv, &mockSpeech)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestSpeechRecognize(t *testing.T) {
-	var expectedResponse *speechpb.RecognizeResponse = &speechpb.RecognizeResponse{}
-
-	mockSpeech.err = nil
-	mockSpeech.reqs = nil
-
-	mockSpeech.resps = append(mockSpeech.resps[:0], expectedResponse)
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.RecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Recognize(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpeechRecognizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpeech.err = gstatus.Error(errCode, "test error")
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.RecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Recognize(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpeechLongRunningRecognize(t *testing.T) {
-	var expectedResponse *speechpb.LongRunningRecognizeResponse = &speechpb.LongRunningRecognizeResponse{}
-
-	mockSpeech.err = nil
-	mockSpeech.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockSpeech.resps = append(mockSpeech.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.LongRunningRecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.LongRunningRecognize(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpeechLongRunningRecognizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpeech.err = nil
-	mockSpeech.resps = append(mockSpeech.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.LongRunningRecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.LongRunningRecognize(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpeechStreamingRecognize(t *testing.T) {
-	var expectedResponse *speechpb.StreamingRecognizeResponse = &speechpb.StreamingRecognizeResponse{}
-
-	mockSpeech.err = nil
-	mockSpeech.reqs = nil
-
-	mockSpeech.resps = append(mockSpeech.resps[:0], expectedResponse)
-
-	var request *speechpb.StreamingRecognizeRequest = &speechpb.StreamingRecognizeRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingRecognize(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpeechStreamingRecognizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpeech.err = gstatus.Error(errCode, "test error")
-
-	var request *speechpb.StreamingRecognizeRequest = &speechpb.StreamingRecognizeRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingRecognize(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/speech/apiv1p1beta1/mock_test.go b/speech/apiv1p1beta1/mock_test.go
deleted file mode 100644
index b30a4490f63..00000000000
--- a/speech/apiv1p1beta1/mock_test.go
+++ /dev/null
@@ -1,404 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package speech
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	speechpb "google.golang.org/genproto/googleapis/cloud/speech/v1p1beta1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockSpeechServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	speechpb.SpeechServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockSpeechServer) Recognize(ctx context.Context, req *speechpb.RecognizeRequest) (*speechpb.RecognizeResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*speechpb.RecognizeResponse), nil
-}
-
-func (s *mockSpeechServer) LongRunningRecognize(ctx context.Context, req *speechpb.LongRunningRecognizeRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockSpeechServer) StreamingRecognize(stream speechpb.Speech_StreamingRecognizeServer) error {
-	md, _ := metadata.FromIncomingContext(stream.Context())
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	for {
-		if req, err := stream.Recv(); err == io.EOF {
-			break
-		} else if err != nil {
-			return err
-		} else {
-			s.reqs = append(s.reqs, req)
-		}
-	}
-	if s.err != nil {
-		return s.err
-	}
-	for _, v := range s.resps {
-		if err := stream.Send(v.(*speechpb.StreamingRecognizeResponse)); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockSpeech mockSpeechServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	speechpb.RegisterSpeechServer(serv, &mockSpeech)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestSpeechRecognize(t *testing.T) {
-	var expectedResponse *speechpb.RecognizeResponse = &speechpb.RecognizeResponse{}
-
-	mockSpeech.err = nil
-	mockSpeech.reqs = nil
-
-	mockSpeech.resps = append(mockSpeech.resps[:0], expectedResponse)
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.RecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Recognize(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpeechRecognizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpeech.err = gstatus.Error(errCode, "test error")
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.RecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.Recognize(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpeechLongRunningRecognize(t *testing.T) {
-	var expectedResponse *speechpb.LongRunningRecognizeResponse = &speechpb.LongRunningRecognizeResponse{}
-
-	mockSpeech.err = nil
-	mockSpeech.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockSpeech.resps = append(mockSpeech.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.LongRunningRecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.LongRunningRecognize(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpeechLongRunningRecognizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpeech.err = nil
-	mockSpeech.resps = append(mockSpeech.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var encoding speechpb.RecognitionConfig_AudioEncoding = speechpb.RecognitionConfig_FLAC
-	var sampleRateHertz int32 = 44100
-	var languageCode string = "en-US"
-	var config = &speechpb.RecognitionConfig{
-		Encoding:        encoding,
-		SampleRateHertz: sampleRateHertz,
-		LanguageCode:    languageCode,
-	}
-	var uri string = "gs://bucket_name/file_name.flac"
-	var audio = &speechpb.RecognitionAudio{
-		AudioSource: &speechpb.RecognitionAudio_Uri{
-			Uri: uri,
-		},
-	}
-	var request = &speechpb.LongRunningRecognizeRequest{
-		Config: config,
-		Audio:  audio,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.LongRunningRecognize(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestSpeechStreamingRecognize(t *testing.T) {
-	var expectedResponse *speechpb.StreamingRecognizeResponse = &speechpb.StreamingRecognizeResponse{}
-
-	mockSpeech.err = nil
-	mockSpeech.reqs = nil
-
-	mockSpeech.resps = append(mockSpeech.resps[:0], expectedResponse)
-
-	var request *speechpb.StreamingRecognizeRequest = &speechpb.StreamingRecognizeRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingRecognize(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestSpeechStreamingRecognizeError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockSpeech.err = gstatus.Error(errCode, "test error")
-
-	var request *speechpb.StreamingRecognizeRequest = &speechpb.StreamingRecognizeRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	stream, err := c.StreamingRecognize(context.Background())
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.Send(request); err != nil {
-		t.Fatal(err)
-	}
-	if err := stream.CloseSend(); err != nil {
-		t.Fatal(err)
-	}
-	resp, err := stream.Recv()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/texttospeech/apiv1/mock_test.go b/texttospeech/apiv1/mock_test.go
deleted file mode 100644
index 59ddea1ec23..00000000000
--- a/texttospeech/apiv1/mock_test.go
+++ /dev/null
@@ -1,231 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package texttospeech
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	texttospeechpb "google.golang.org/genproto/googleapis/cloud/texttospeech/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockTextToSpeechServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	texttospeechpb.TextToSpeechServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockTextToSpeechServer) ListVoices(ctx context.Context, req *texttospeechpb.ListVoicesRequest) (*texttospeechpb.ListVoicesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*texttospeechpb.ListVoicesResponse), nil
-}
-
-func (s *mockTextToSpeechServer) SynthesizeSpeech(ctx context.Context, req *texttospeechpb.SynthesizeSpeechRequest) (*texttospeechpb.SynthesizeSpeechResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*texttospeechpb.SynthesizeSpeechResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockTextToSpeech mockTextToSpeechServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	texttospeechpb.RegisterTextToSpeechServer(serv, &mockTextToSpeech)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestTextToSpeechListVoices(t *testing.T) {
-	var expectedResponse *texttospeechpb.ListVoicesResponse = &texttospeechpb.ListVoicesResponse{}
-
-	mockTextToSpeech.err = nil
-	mockTextToSpeech.reqs = nil
-
-	mockTextToSpeech.resps = append(mockTextToSpeech.resps[:0], expectedResponse)
-
-	var request *texttospeechpb.ListVoicesRequest = &texttospeechpb.ListVoicesRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListVoices(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTextToSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTextToSpeechListVoicesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTextToSpeech.err = gstatus.Error(errCode, "test error")
-
-	var request *texttospeechpb.ListVoicesRequest = &texttospeechpb.ListVoicesRequest{}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListVoices(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTextToSpeechSynthesizeSpeech(t *testing.T) {
-	var audioContent []byte = []byte("16")
-	var expectedResponse = &texttospeechpb.SynthesizeSpeechResponse{
-		AudioContent: audioContent,
-	}
-
-	mockTextToSpeech.err = nil
-	mockTextToSpeech.reqs = nil
-
-	mockTextToSpeech.resps = append(mockTextToSpeech.resps[:0], expectedResponse)
-
-	var input *texttospeechpb.SynthesisInput = &texttospeechpb.SynthesisInput{}
-	var voice *texttospeechpb.VoiceSelectionParams = &texttospeechpb.VoiceSelectionParams{}
-	var audioConfig *texttospeechpb.AudioConfig = &texttospeechpb.AudioConfig{}
-	var request = &texttospeechpb.SynthesizeSpeechRequest{
-		Input:       input,
-		Voice:       voice,
-		AudioConfig: audioConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SynthesizeSpeech(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTextToSpeech.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTextToSpeechSynthesizeSpeechError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTextToSpeech.err = gstatus.Error(errCode, "test error")
-
-	var input *texttospeechpb.SynthesisInput = &texttospeechpb.SynthesisInput{}
-	var voice *texttospeechpb.VoiceSelectionParams = &texttospeechpb.VoiceSelectionParams{}
-	var audioConfig *texttospeechpb.AudioConfig = &texttospeechpb.AudioConfig{}
-	var request = &texttospeechpb.SynthesizeSpeechRequest{
-		Input:       input,
-		Voice:       voice,
-		AudioConfig: audioConfig,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SynthesizeSpeech(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/trace/apiv1/mock_test.go b/trace/apiv1/mock_test.go
deleted file mode 100644
index 0cf5fcc912e..00000000000
--- a/trace/apiv1/mock_test.go
+++ /dev/null
@@ -1,320 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package trace
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	"google.golang.org/api/option"
-	cloudtracepb "google.golang.org/genproto/googleapis/devtools/cloudtrace/v1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockTraceServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	cloudtracepb.TraceServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockTraceServer) ListTraces(ctx context.Context, req *cloudtracepb.ListTracesRequest) (*cloudtracepb.ListTracesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*cloudtracepb.ListTracesResponse), nil
-}
-
-func (s *mockTraceServer) GetTrace(ctx context.Context, req *cloudtracepb.GetTraceRequest) (*cloudtracepb.Trace, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*cloudtracepb.Trace), nil
-}
-
-func (s *mockTraceServer) PatchTraces(ctx context.Context, req *cloudtracepb.PatchTracesRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockTrace mockTraceServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	cloudtracepb.RegisterTraceServiceServer(serv, &mockTrace)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestTraceServicePatchTraces(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockTrace.err = nil
-	mockTrace.reqs = nil
-
-	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var traces *cloudtracepb.Traces = &cloudtracepb.Traces{}
-	var request = &cloudtracepb.PatchTracesRequest{
-		ProjectId: projectId,
-		Traces:    traces,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.PatchTraces(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestTraceServicePatchTracesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTrace.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var traces *cloudtracepb.Traces = &cloudtracepb.Traces{}
-	var request = &cloudtracepb.PatchTracesRequest{
-		ProjectId: projectId,
-		Traces:    traces,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.PatchTraces(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestTraceServiceGetTrace(t *testing.T) {
-	var projectId2 string = "projectId2939242356"
-	var traceId2 string = "traceId2987826376"
-	var expectedResponse = &cloudtracepb.Trace{
-		ProjectId: projectId2,
-		TraceId:   traceId2,
-	}
-
-	mockTrace.err = nil
-	mockTrace.reqs = nil
-
-	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var traceId string = "traceId1270300245"
-	var request = &cloudtracepb.GetTraceRequest{
-		ProjectId: projectId,
-		TraceId:   traceId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTrace(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTraceServiceGetTraceError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTrace.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var traceId string = "traceId1270300245"
-	var request = &cloudtracepb.GetTraceRequest{
-		ProjectId: projectId,
-		TraceId:   traceId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetTrace(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTraceServiceListTraces(t *testing.T) {
-	var nextPageToken string = ""
-	var tracesElement *cloudtracepb.Trace = &cloudtracepb.Trace{}
-	var traces = []*cloudtracepb.Trace{tracesElement}
-	var expectedResponse = &cloudtracepb.ListTracesResponse{
-		NextPageToken: nextPageToken,
-		Traces:        traces,
-	}
-
-	mockTrace.err = nil
-	mockTrace.reqs = nil
-
-	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)
-
-	var projectId string = "projectId-1969970175"
-	var request = &cloudtracepb.ListTracesRequest{
-		ProjectId: projectId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTraces(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Traces[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTraceServiceListTracesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTrace.err = gstatus.Error(errCode, "test error")
-
-	var projectId string = "projectId-1969970175"
-	var request = &cloudtracepb.ListTracesRequest{
-		ProjectId: projectId,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListTraces(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/trace/apiv2/mock_test.go b/trace/apiv2/mock_test.go
deleted file mode 100644
index cb8b1f13e49..00000000000
--- a/trace/apiv2/mock_test.go
+++ /dev/null
@@ -1,251 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package trace
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	emptypb "github.com/golang/protobuf/ptypes/empty"
-	timestamppb "github.com/golang/protobuf/ptypes/timestamp"
-	"google.golang.org/api/option"
-	cloudtracepb "google.golang.org/genproto/googleapis/devtools/cloudtrace/v2"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockTraceServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	cloudtracepb.TraceServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockTraceServer) BatchWriteSpans(ctx context.Context, req *cloudtracepb.BatchWriteSpansRequest) (*emptypb.Empty, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*emptypb.Empty), nil
-}
-
-func (s *mockTraceServer) CreateSpan(ctx context.Context, req *cloudtracepb.Span) (*cloudtracepb.Span, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*cloudtracepb.Span), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockTrace mockTraceServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	cloudtracepb.RegisterTraceServiceServer(serv, &mockTrace)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestTraceServiceBatchWriteSpans(t *testing.T) {
-	var expectedResponse *emptypb.Empty = &emptypb.Empty{}
-
-	mockTrace.err = nil
-	mockTrace.reqs = nil
-
-	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var spans []*cloudtracepb.Span = nil
-	var request = &cloudtracepb.BatchWriteSpansRequest{
-		Name:  formattedName,
-		Spans: spans,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.BatchWriteSpans(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-}
-
-func TestTraceServiceBatchWriteSpansError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTrace.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s", "[PROJECT]")
-	var spans []*cloudtracepb.Span = nil
-	var request = &cloudtracepb.BatchWriteSpansRequest{
-		Name:  formattedName,
-		Spans: spans,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	err = c.BatchWriteSpans(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-}
-func TestTraceServiceCreateSpan(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var spanId2 string = "spanId2-643891741"
-	var parentSpanId string = "parentSpanId-1757797477"
-	var expectedResponse = &cloudtracepb.Span{
-		Name:         name2,
-		SpanId:       spanId2,
-		ParentSpanId: parentSpanId,
-	}
-
-	mockTrace.err = nil
-	mockTrace.reqs = nil
-
-	mockTrace.resps = append(mockTrace.resps[:0], expectedResponse)
-
-	var name string = "name3373707"
-	var spanId string = "spanId-2011840976"
-	var displayName *cloudtracepb.TruncatableString = &cloudtracepb.TruncatableString{}
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var endTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &cloudtracepb.Span{
-		Name:        name,
-		SpanId:      spanId,
-		DisplayName: displayName,
-		StartTime:   startTime,
-		EndTime:     endTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSpan(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTrace.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTraceServiceCreateSpanError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTrace.err = gstatus.Error(errCode, "test error")
-
-	var name string = "name3373707"
-	var spanId string = "spanId-2011840976"
-	var displayName *cloudtracepb.TruncatableString = &cloudtracepb.TruncatableString{}
-	var startTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var endTime *timestamppb.Timestamp = &timestamppb.Timestamp{}
-	var request = &cloudtracepb.Span{
-		Name:        name,
-		SpanId:      spanId,
-		DisplayName: displayName,
-		StartTime:   startTime,
-		EndTime:     endTime,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.CreateSpan(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/translate/apiv3/mock_test.go b/translate/apiv3/mock_test.go
deleted file mode 100644
index 4e01bde6a3e..00000000000
--- a/translate/apiv3/mock_test.go
+++ /dev/null
@@ -1,777 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package translate
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	translatepb "google.golang.org/genproto/googleapis/cloud/translate/v3"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockTranslationServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	translatepb.TranslationServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockTranslationServer) TranslateText(ctx context.Context, req *translatepb.TranslateTextRequest) (*translatepb.TranslateTextResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*translatepb.TranslateTextResponse), nil
-}
-
-func (s *mockTranslationServer) DetectLanguage(ctx context.Context, req *translatepb.DetectLanguageRequest) (*translatepb.DetectLanguageResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*translatepb.DetectLanguageResponse), nil
-}
-
-func (s *mockTranslationServer) GetSupportedLanguages(ctx context.Context, req *translatepb.GetSupportedLanguagesRequest) (*translatepb.SupportedLanguages, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*translatepb.SupportedLanguages), nil
-}
-
-func (s *mockTranslationServer) BatchTranslateText(ctx context.Context, req *translatepb.BatchTranslateTextRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockTranslationServer) CreateGlossary(ctx context.Context, req *translatepb.CreateGlossaryRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-func (s *mockTranslationServer) ListGlossaries(ctx context.Context, req *translatepb.ListGlossariesRequest) (*translatepb.ListGlossariesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*translatepb.ListGlossariesResponse), nil
-}
-
-func (s *mockTranslationServer) GetGlossary(ctx context.Context, req *translatepb.GetGlossaryRequest) (*translatepb.Glossary, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*translatepb.Glossary), nil
-}
-
-func (s *mockTranslationServer) DeleteGlossary(ctx context.Context, req *translatepb.DeleteGlossaryRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockTranslation mockTranslationServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	translatepb.RegisterTranslationServiceServer(serv, &mockTranslation)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestTranslationServiceTranslateText(t *testing.T) {
-	var expectedResponse *translatepb.TranslateTextResponse = &translatepb.TranslateTextResponse{}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	mockTranslation.resps = append(mockTranslation.resps[:0], expectedResponse)
-
-	var contents []string = nil
-	var targetLanguageCode string = "targetLanguageCode1323228230"
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.TranslateTextRequest{
-		Contents:           contents,
-		TargetLanguageCode: targetLanguageCode,
-		Parent:             formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TranslateText(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceTranslateTextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = gstatus.Error(errCode, "test error")
-
-	var contents []string = nil
-	var targetLanguageCode string = "targetLanguageCode1323228230"
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.TranslateTextRequest{
-		Contents:           contents,
-		TargetLanguageCode: targetLanguageCode,
-		Parent:             formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.TranslateText(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceDetectLanguage(t *testing.T) {
-	var expectedResponse *translatepb.DetectLanguageResponse = &translatepb.DetectLanguageResponse{}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	mockTranslation.resps = append(mockTranslation.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.DetectLanguageRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DetectLanguage(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceDetectLanguageError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.DetectLanguageRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.DetectLanguage(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceGetSupportedLanguages(t *testing.T) {
-	var expectedResponse *translatepb.SupportedLanguages = &translatepb.SupportedLanguages{}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	mockTranslation.resps = append(mockTranslation.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.GetSupportedLanguagesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSupportedLanguages(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceGetSupportedLanguagesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.GetSupportedLanguagesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetSupportedLanguages(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceBatchTranslateText(t *testing.T) {
-	var totalCharacters int64 = 1368640955
-	var translatedCharacters int64 = 1337326221
-	var failedCharacters int64 = 1723028396
-	var expectedResponse = &translatepb.BatchTranslateResponse{
-		TotalCharacters:      totalCharacters,
-		TranslatedCharacters: translatedCharacters,
-		FailedCharacters:     failedCharacters,
-	}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockTranslation.resps = append(mockTranslation.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var sourceLanguageCode string = "sourceLanguageCode1687263568"
-	var targetLanguageCodes []string = nil
-	var inputConfigs []*translatepb.InputConfig = nil
-	var outputConfig *translatepb.OutputConfig = &translatepb.OutputConfig{}
-	var request = &translatepb.BatchTranslateTextRequest{
-		Parent:              formattedParent,
-		SourceLanguageCode:  sourceLanguageCode,
-		TargetLanguageCodes: targetLanguageCodes,
-		InputConfigs:        inputConfigs,
-		OutputConfig:        outputConfig,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchTranslateText(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceBatchTranslateTextError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = nil
-	mockTranslation.resps = append(mockTranslation.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var sourceLanguageCode string = "sourceLanguageCode1687263568"
-	var targetLanguageCodes []string = nil
-	var inputConfigs []*translatepb.InputConfig = nil
-	var outputConfig *translatepb.OutputConfig = &translatepb.OutputConfig{}
-	var request = &translatepb.BatchTranslateTextRequest{
-		Parent:              formattedParent,
-		SourceLanguageCode:  sourceLanguageCode,
-		TargetLanguageCodes: targetLanguageCodes,
-		InputConfigs:        inputConfigs,
-		OutputConfig:        outputConfig,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.BatchTranslateText(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceCreateGlossary(t *testing.T) {
-	var name string = "name3373707"
-	var entryCount int32 = 811131134
-	var expectedResponse = &translatepb.Glossary{
-		Name:       name,
-		EntryCount: entryCount,
-	}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockTranslation.resps = append(mockTranslation.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var glossary *translatepb.Glossary = &translatepb.Glossary{}
-	var request = &translatepb.CreateGlossaryRequest{
-		Parent:   formattedParent,
-		Glossary: glossary,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateGlossary(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceCreateGlossaryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = nil
-	mockTranslation.resps = append(mockTranslation.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var glossary *translatepb.Glossary = &translatepb.Glossary{}
-	var request = &translatepb.CreateGlossaryRequest{
-		Parent:   formattedParent,
-		Glossary: glossary,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.CreateGlossary(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceListGlossaries(t *testing.T) {
-	var nextPageToken string = ""
-	var glossariesElement *translatepb.Glossary = &translatepb.Glossary{}
-	var glossaries = []*translatepb.Glossary{glossariesElement}
-	var expectedResponse = &translatepb.ListGlossariesResponse{
-		NextPageToken: nextPageToken,
-		Glossaries:    glossaries,
-	}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	mockTranslation.resps = append(mockTranslation.resps[:0], expectedResponse)
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.ListGlossariesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGlossaries(context.Background(), request).Next()
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	want := (interface{})(expectedResponse.Glossaries[0])
-	got := (interface{})(resp)
-	var ok bool
-
-	switch want := (want).(type) {
-	case proto.Message:
-		ok = proto.Equal(want, got.(proto.Message))
-	default:
-		ok = want == got
-	}
-	if !ok {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceListGlossariesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = gstatus.Error(errCode, "test error")
-
-	var formattedParent string = fmt.Sprintf("projects/%s/locations/%s", "[PROJECT]", "[LOCATION]")
-	var request = &translatepb.ListGlossariesRequest{
-		Parent: formattedParent,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ListGlossaries(context.Background(), request).Next()
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceGetGlossary(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var entryCount int32 = 811131134
-	var expectedResponse = &translatepb.Glossary{
-		Name:       name2,
-		EntryCount: entryCount,
-	}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	mockTranslation.resps = append(mockTranslation.resps[:0], expectedResponse)
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/glossaries/%s", "[PROJECT]", "[LOCATION]", "[GLOSSARY]")
-	var request = &translatepb.GetGlossaryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetGlossary(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceGetGlossaryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = gstatus.Error(errCode, "test error")
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/glossaries/%s", "[PROJECT]", "[LOCATION]", "[GLOSSARY]")
-	var request = &translatepb.GetGlossaryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.GetGlossary(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestTranslationServiceDeleteGlossary(t *testing.T) {
-	var name2 string = "name2-1052831874"
-	var expectedResponse = &translatepb.DeleteGlossaryResponse{
-		Name: name2,
-	}
-
-	mockTranslation.err = nil
-	mockTranslation.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockTranslation.resps = append(mockTranslation.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/glossaries/%s", "[PROJECT]", "[LOCATION]", "[GLOSSARY]")
-	var request = &translatepb.DeleteGlossaryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteGlossary(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockTranslation.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestTranslationServiceDeleteGlossaryError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockTranslation.err = nil
-	mockTranslation.resps = append(mockTranslation.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var formattedName string = fmt.Sprintf("projects/%s/locations/%s/glossaries/%s", "[PROJECT]", "[LOCATION]", "[GLOSSARY]")
-	var request = &translatepb.DeleteGlossaryRequest{
-		Name: formattedName,
-	}
-
-	c, err := NewTranslationClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.DeleteGlossary(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/videointelligence/apiv1/mock_test.go b/videointelligence/apiv1/mock_test.go
deleted file mode 100644
index cf92c59bfaf..00000000000
--- a/videointelligence/apiv1/mock_test.go
+++ /dev/null
@@ -1,191 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package videointelligence
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	videointelligencepb "google.golang.org/genproto/googleapis/cloud/videointelligence/v1"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockVideoIntelligenceServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	videointelligencepb.VideoIntelligenceServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockVideoIntelligenceServer) AnnotateVideo(ctx context.Context, req *videointelligencepb.AnnotateVideoRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockVideoIntelligence mockVideoIntelligenceServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	videointelligencepb.RegisterVideoIntelligenceServiceServer(serv, &mockVideoIntelligence)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestVideoIntelligenceServiceAnnotateVideo(t *testing.T) {
-	var expectedResponse *videointelligencepb.AnnotateVideoResponse = &videointelligencepb.AnnotateVideoResponse{}
-
-	mockVideoIntelligence.err = nil
-	mockVideoIntelligence.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockVideoIntelligence.resps = append(mockVideoIntelligence.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var featuresElement videointelligencepb.Feature = videointelligencepb.Feature_LABEL_DETECTION
-	var features = []videointelligencepb.Feature{featuresElement}
-	var inputUri string = "gs://cloud-samples-data/video/cat.mp4"
-	var request = &videointelligencepb.AnnotateVideoRequest{
-		Features: features,
-		InputUri: inputUri,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.AnnotateVideo(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockVideoIntelligence.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestVideoIntelligenceServiceAnnotateVideoError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockVideoIntelligence.err = nil
-	mockVideoIntelligence.resps = append(mockVideoIntelligence.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var featuresElement videointelligencepb.Feature = videointelligencepb.Feature_LABEL_DETECTION
-	var features = []videointelligencepb.Feature{featuresElement}
-	var inputUri string = "gs://cloud-samples-data/video/cat.mp4"
-	var request = &videointelligencepb.AnnotateVideoRequest{
-		Features: features,
-		InputUri: inputUri,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.AnnotateVideo(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/videointelligence/apiv1beta2/mock_test.go b/videointelligence/apiv1beta2/mock_test.go
deleted file mode 100644
index 53350f2b8fe..00000000000
--- a/videointelligence/apiv1beta2/mock_test.go
+++ /dev/null
@@ -1,191 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package videointelligence
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	videointelligencepb "google.golang.org/genproto/googleapis/cloud/videointelligence/v1beta2"
-	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockVideoIntelligenceServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	videointelligencepb.VideoIntelligenceServiceServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockVideoIntelligenceServer) AnnotateVideo(ctx context.Context, req *videointelligencepb.AnnotateVideoRequest) (*longrunningpb.Operation, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*longrunningpb.Operation), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockVideoIntelligence mockVideoIntelligenceServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	videointelligencepb.RegisterVideoIntelligenceServiceServer(serv, &mockVideoIntelligence)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestVideoIntelligenceServiceAnnotateVideo(t *testing.T) {
-	var expectedResponse *videointelligencepb.AnnotateVideoResponse = &videointelligencepb.AnnotateVideoResponse{}
-
-	mockVideoIntelligence.err = nil
-	mockVideoIntelligence.reqs = nil
-
-	any, err := ptypes.MarshalAny(expectedResponse)
-	if err != nil {
-		t.Fatal(err)
-	}
-	mockVideoIntelligence.resps = append(mockVideoIntelligence.resps[:0], &longrunningpb.Operation{
-		Name:   "longrunning-test",
-		Done:   true,
-		Result: &longrunningpb.Operation_Response{Response: any},
-	})
-
-	var inputUri string = "gs://cloud-samples-data/video/cat.mp4"
-	var featuresElement videointelligencepb.Feature = videointelligencepb.Feature_LABEL_DETECTION
-	var features = []videointelligencepb.Feature{featuresElement}
-	var request = &videointelligencepb.AnnotateVideoRequest{
-		InputUri: inputUri,
-		Features: features,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.AnnotateVideo(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockVideoIntelligence.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestVideoIntelligenceServiceAnnotateVideoError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockVideoIntelligence.err = nil
-	mockVideoIntelligence.resps = append(mockVideoIntelligence.resps[:0], &longrunningpb.Operation{
-		Name: "longrunning-test",
-		Done: true,
-		Result: &longrunningpb.Operation_Error{
-			Error: &status.Status{
-				Code:    int32(errCode),
-				Message: "test error",
-			},
-		},
-	})
-
-	var inputUri string = "gs://cloud-samples-data/video/cat.mp4"
-	var featuresElement videointelligencepb.Feature = videointelligencepb.Feature_LABEL_DETECTION
-	var features = []videointelligencepb.Feature{featuresElement}
-	var request = &videointelligencepb.AnnotateVideoRequest{
-		InputUri: inputUri,
-		Features: features,
-	}
-
-	c, err := NewClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	respLRO, err := c.AnnotateVideo(context.Background(), request)
-	if err != nil {
-		t.Fatal(err)
-	}
-	resp, err := respLRO.Wait(context.Background())
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/vision/apiv1p1beta1/mock_test.go b/vision/apiv1p1beta1/mock_test.go
deleted file mode 100644
index 36bb7b43458..00000000000
--- a/vision/apiv1p1beta1/mock_test.go
+++ /dev/null
@@ -1,158 +0,0 @@
-// Copyright 2020 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package vision
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	visionpb "google.golang.org/genproto/googleapis/cloud/vision/v1p1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockImageAnnotatorServer struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	visionpb.ImageAnnotatorServer
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockImageAnnotatorServer) BatchAnnotateImages(ctx context.Context, req *visionpb.BatchAnnotateImagesRequest) (*visionpb.BatchAnnotateImagesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*visionpb.BatchAnnotateImagesResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockImageAnnotator mockImageAnnotatorServer
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	visionpb.RegisterImageAnnotatorServer(serv, &mockImageAnnotator)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestImageAnnotatorBatchAnnotateImages(t *testing.T) {
-	var expectedResponse *visionpb.BatchAnnotateImagesResponse = &visionpb.BatchAnnotateImagesResponse{}
-
-	mockImageAnnotator.err = nil
-	mockImageAnnotator.reqs = nil
-
-	mockImageAnnotator.resps = append(mockImageAnnotator.resps[:0], expectedResponse)
-
-	var requests []*visionpb.AnnotateImageRequest = nil
-	var request = &visionpb.BatchAnnotateImagesRequest{
-		Requests: requests,
-	}
-
-	c, err := NewImageAnnotatorClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchAnnotateImages(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockImageAnnotator.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestImageAnnotatorBatchAnnotateImagesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockImageAnnotator.err = gstatus.Error(errCode, "test error")
-
-	var requests []*visionpb.AnnotateImageRequest = nil
-	var request = &visionpb.BatchAnnotateImagesRequest{
-		Requests: requests,
-	}
-
-	c, err := NewImageAnnotatorClient(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.BatchAnnotateImages(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
diff --git a/webrisk/apiv1beta1/mock_test.go b/webrisk/apiv1beta1/mock_test.go
deleted file mode 100644
index 8e6dc3f3e1e..00000000000
--- a/webrisk/apiv1beta1/mock_test.go
+++ /dev/null
@@ -1,305 +0,0 @@
-// Copyright 2019 Google LLC
-//
-// Licensed under the Apache License, Version 2.0 (the "License");
-// you may not use this file except in compliance with the License.
-// You may obtain a copy of the License at
-//
-//     https://www.apache.org/licenses/LICENSE-2.0
-//
-// Unless required by applicable law or agreed to in writing, software
-// distributed under the License is distributed on an "AS IS" BASIS,
-// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-// See the License for the specific language governing permissions and
-// limitations under the License.
-
-// Code generated by gapic-generator. DO NOT EDIT.
-
-package webrisk
-
-import (
-	"context"
-	"flag"
-	"fmt"
-	"io"
-	"log"
-	"net"
-	"os"
-	"strings"
-	"testing"
-
-	"github.com/golang/protobuf/proto"
-	"github.com/golang/protobuf/ptypes"
-	"google.golang.org/api/option"
-	webriskpb "google.golang.org/genproto/googleapis/cloud/webrisk/v1beta1"
-
-	status "google.golang.org/genproto/googleapis/rpc/status"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/metadata"
-
-	gstatus "google.golang.org/grpc/status"
-)
-
-var _ = io.EOF
-var _ = ptypes.MarshalAny
-var _ status.Status
-
-type mockWebRiskServiceV1Beta1Server struct {
-	// Embed for forward compatibility.
-	// Tests will keep working if more methods are added
-	// in the future.
-	webriskpb.WebRiskServiceV1Beta1Server
-
-	reqs []proto.Message
-
-	// If set, all calls return this error.
-	err error
-
-	// responses to return if err == nil
-	resps []proto.Message
-}
-
-func (s *mockWebRiskServiceV1Beta1Server) ComputeThreatListDiff(ctx context.Context, req *webriskpb.ComputeThreatListDiffRequest) (*webriskpb.ComputeThreatListDiffResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*webriskpb.ComputeThreatListDiffResponse), nil
-}
-
-func (s *mockWebRiskServiceV1Beta1Server) SearchUris(ctx context.Context, req *webriskpb.SearchUrisRequest) (*webriskpb.SearchUrisResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*webriskpb.SearchUrisResponse), nil
-}
-
-func (s *mockWebRiskServiceV1Beta1Server) SearchHashes(ctx context.Context, req *webriskpb.SearchHashesRequest) (*webriskpb.SearchHashesResponse, error) {
-	md, _ := metadata.FromIncomingContext(ctx)
-	if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
-		return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
-	}
-	s.reqs = append(s.reqs, req)
-	if s.err != nil {
-		return nil, s.err
-	}
-	return s.resps[0].(*webriskpb.SearchHashesResponse), nil
-}
-
-// clientOpt is the option tests should use to connect to the test server.
-// It is initialized by TestMain.
-var clientOpt option.ClientOption
-
-var (
-	mockWebRiskServiceV1Beta1 mockWebRiskServiceV1Beta1Server
-)
-
-func TestMain(m *testing.M) {
-	flag.Parse()
-
-	serv := grpc.NewServer()
-	webriskpb.RegisterWebRiskServiceV1Beta1Server(serv, &mockWebRiskServiceV1Beta1)
-
-	lis, err := net.Listen("tcp", "localhost:0")
-	if err != nil {
-		log.Fatal(err)
-	}
-	go serv.Serve(lis)
-
-	conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
-	if err != nil {
-		log.Fatal(err)
-	}
-	clientOpt = option.WithGRPCConn(conn)
-
-	os.Exit(m.Run())
-}
-
-func TestWebRiskServiceV1Beta1ComputeThreatListDiff(t *testing.T) {
-	var newVersionToken []byte = []byte("115")
-	var expectedResponse = &webriskpb.ComputeThreatListDiffResponse{
-		NewVersionToken: newVersionToken,
-	}
-
-	mockWebRiskServiceV1Beta1.err = nil
-	mockWebRiskServiceV1Beta1.reqs = nil
-
-	mockWebRiskServiceV1Beta1.resps = append(mockWebRiskServiceV1Beta1.resps[:0], expectedResponse)
-
-	var threatType webriskpb.ThreatType = webriskpb.ThreatType_THREAT_TYPE_UNSPECIFIED
-	var constraints *webriskpb.ComputeThreatListDiffRequest_Constraints = &webriskpb.ComputeThreatListDiffRequest_Constraints{}
-	var request = &webriskpb.ComputeThreatListDiffRequest{
-		ThreatType:  threatType,
-		Constraints: constraints,
-	}
-
-	c, err := NewWebRiskServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ComputeThreatListDiff(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWebRiskServiceV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWebRiskServiceV1Beta1ComputeThreatListDiffError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWebRiskServiceV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var threatType webriskpb.ThreatType = webriskpb.ThreatType_THREAT_TYPE_UNSPECIFIED
-	var constraints *webriskpb.ComputeThreatListDiffRequest_Constraints = &webriskpb.ComputeThreatListDiffRequest_Constraints{}
-	var request = &webriskpb.ComputeThreatListDiffRequest{
-		ThreatType:  threatType,
-		Constraints: constraints,
-	}
-
-	c, err := NewWebRiskServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.ComputeThreatListDiff(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestWebRiskServiceV1Beta1SearchUris(t *testing.T) {
-	var expectedResponse *webriskpb.SearchUrisResponse = &webriskpb.SearchUrisResponse{}
-
-	mockWebRiskServiceV1Beta1.err = nil
-	mockWebRiskServiceV1Beta1.reqs = nil
-
-	mockWebRiskServiceV1Beta1.resps = append(mockWebRiskServiceV1Beta1.resps[:0], expectedResponse)
-
-	var uri string = "uri116076"
-	var threatTypes []webriskpb.ThreatType = nil
-	var request = &webriskpb.SearchUrisRequest{
-		Uri:         uri,
-		ThreatTypes: threatTypes,
-	}
-
-	c, err := NewWebRiskServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SearchUris(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWebRiskServiceV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWebRiskServiceV1Beta1SearchUrisError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWebRiskServiceV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var uri string = "uri116076"
-	var threatTypes []webriskpb.ThreatType = nil
-	var request = &webriskpb.SearchUrisRequest{
-		Uri:         uri,
-		ThreatTypes: threatTypes,
-	}
-
-	c, err := NewWebRiskServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SearchUris(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}
-func TestWebRiskServiceV1Beta1SearchHashes(t *testing.T) {
-	var expectedResponse *webriskpb.SearchHashesResponse = &webriskpb.SearchHashesResponse{}
-
-	mockWebRiskServiceV1Beta1.err = nil
-	mockWebRiskServiceV1Beta1.reqs = nil
-
-	mockWebRiskServiceV1Beta1.resps = append(mockWebRiskServiceV1Beta1.resps[:0], expectedResponse)
-
-	var threatTypes []webriskpb.ThreatType = nil
-	var request = &webriskpb.SearchHashesRequest{
-		ThreatTypes: threatTypes,
-	}
-
-	c, err := NewWebRiskServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SearchHashes(context.Background(), request)
-
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	if want, got := request, mockWebRiskServiceV1Beta1.reqs[0]; !proto.Equal(want, got) {
-		t.Errorf("wrong request %q, want %q", got, want)
-	}
-
-	if want, got := expectedResponse, resp; !proto.Equal(want, got) {
-		t.Errorf("wrong response %q, want %q)", got, want)
-	}
-}
-
-func TestWebRiskServiceV1Beta1SearchHashesError(t *testing.T) {
-	errCode := codes.PermissionDenied
-	mockWebRiskServiceV1Beta1.err = gstatus.Error(errCode, "test error")
-
-	var threatTypes []webriskpb.ThreatType = nil
-	var request = &webriskpb.SearchHashesRequest{
-		ThreatTypes: threatTypes,
-	}
-
-	c, err := NewWebRiskServiceV1Beta1Client(context.Background(), clientOpt)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	resp, err := c.SearchHashes(context.Background(), request)
-
-	if st, ok := gstatus.FromError(err); !ok {
-		t.Errorf("got error %v, expected grpc error", err)
-	} else if c := st.Code(); c != errCode {
-		t.Errorf("got error code %q, want %q", c, errCode)
-	}
-	_ = resp
-}