1// Copyright 2019 Google LLC 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at 6// 7// https://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14 15// Code generated by gapic-generator. DO NOT EDIT. 16 17package dlp 18 19import ( 20 "context" 21 "flag" 22 "fmt" 23 "io" 24 "log" 25 "net" 26 "os" 27 "strings" 28 "testing" 29 30 "github.com/golang/protobuf/proto" 31 "github.com/golang/protobuf/ptypes" 32 emptypb "github.com/golang/protobuf/ptypes/empty" 33 "google.golang.org/api/option" 34 dlppb "google.golang.org/genproto/googleapis/privacy/dlp/v2" 35 36 status "google.golang.org/genproto/googleapis/rpc/status" 37 "google.golang.org/grpc" 38 "google.golang.org/grpc/codes" 39 "google.golang.org/grpc/metadata" 40 41 gstatus "google.golang.org/grpc/status" 42) 43 44var _ = io.EOF 45var _ = ptypes.MarshalAny 46var _ status.Status 47 48type mockDlpServer struct { 49 // Embed for forward compatibility. 50 // Tests will keep working if more methods are added 51 // in the future. 52 dlppb.DlpServiceServer 53 54 reqs []proto.Message 55 56 // If set, all calls return this error. 57 err error 58 59 // responses to return if err == nil 60 resps []proto.Message 61} 62 63func (s *mockDlpServer) InspectContent(ctx context.Context, req *dlppb.InspectContentRequest) (*dlppb.InspectContentResponse, error) { 64 md, _ := metadata.FromIncomingContext(ctx) 65 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 66 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 67 } 68 s.reqs = append(s.reqs, req) 69 if s.err != nil { 70 return nil, s.err 71 } 72 return s.resps[0].(*dlppb.InspectContentResponse), nil 73} 74 75func (s *mockDlpServer) RedactImage(ctx context.Context, req *dlppb.RedactImageRequest) (*dlppb.RedactImageResponse, error) { 76 md, _ := metadata.FromIncomingContext(ctx) 77 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 78 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 79 } 80 s.reqs = append(s.reqs, req) 81 if s.err != nil { 82 return nil, s.err 83 } 84 return s.resps[0].(*dlppb.RedactImageResponse), nil 85} 86 87func (s *mockDlpServer) DeidentifyContent(ctx context.Context, req *dlppb.DeidentifyContentRequest) (*dlppb.DeidentifyContentResponse, error) { 88 md, _ := metadata.FromIncomingContext(ctx) 89 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 90 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 91 } 92 s.reqs = append(s.reqs, req) 93 if s.err != nil { 94 return nil, s.err 95 } 96 return s.resps[0].(*dlppb.DeidentifyContentResponse), nil 97} 98 99func (s *mockDlpServer) ReidentifyContent(ctx context.Context, req *dlppb.ReidentifyContentRequest) (*dlppb.ReidentifyContentResponse, error) { 100 md, _ := metadata.FromIncomingContext(ctx) 101 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 102 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 103 } 104 s.reqs = append(s.reqs, req) 105 if s.err != nil { 106 return nil, s.err 107 } 108 return s.resps[0].(*dlppb.ReidentifyContentResponse), nil 109} 110 111func (s *mockDlpServer) ListInfoTypes(ctx context.Context, req *dlppb.ListInfoTypesRequest) (*dlppb.ListInfoTypesResponse, error) { 112 md, _ := metadata.FromIncomingContext(ctx) 113 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 114 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 115 } 116 s.reqs = append(s.reqs, req) 117 if s.err != nil { 118 return nil, s.err 119 } 120 return s.resps[0].(*dlppb.ListInfoTypesResponse), nil 121} 122 123func (s *mockDlpServer) CreateInspectTemplate(ctx context.Context, req *dlppb.CreateInspectTemplateRequest) (*dlppb.InspectTemplate, error) { 124 md, _ := metadata.FromIncomingContext(ctx) 125 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 126 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 127 } 128 s.reqs = append(s.reqs, req) 129 if s.err != nil { 130 return nil, s.err 131 } 132 return s.resps[0].(*dlppb.InspectTemplate), nil 133} 134 135func (s *mockDlpServer) UpdateInspectTemplate(ctx context.Context, req *dlppb.UpdateInspectTemplateRequest) (*dlppb.InspectTemplate, error) { 136 md, _ := metadata.FromIncomingContext(ctx) 137 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 138 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 139 } 140 s.reqs = append(s.reqs, req) 141 if s.err != nil { 142 return nil, s.err 143 } 144 return s.resps[0].(*dlppb.InspectTemplate), nil 145} 146 147func (s *mockDlpServer) GetInspectTemplate(ctx context.Context, req *dlppb.GetInspectTemplateRequest) (*dlppb.InspectTemplate, error) { 148 md, _ := metadata.FromIncomingContext(ctx) 149 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 150 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 151 } 152 s.reqs = append(s.reqs, req) 153 if s.err != nil { 154 return nil, s.err 155 } 156 return s.resps[0].(*dlppb.InspectTemplate), nil 157} 158 159func (s *mockDlpServer) ListInspectTemplates(ctx context.Context, req *dlppb.ListInspectTemplatesRequest) (*dlppb.ListInspectTemplatesResponse, error) { 160 md, _ := metadata.FromIncomingContext(ctx) 161 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 162 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 163 } 164 s.reqs = append(s.reqs, req) 165 if s.err != nil { 166 return nil, s.err 167 } 168 return s.resps[0].(*dlppb.ListInspectTemplatesResponse), nil 169} 170 171func (s *mockDlpServer) DeleteInspectTemplate(ctx context.Context, req *dlppb.DeleteInspectTemplateRequest) (*emptypb.Empty, error) { 172 md, _ := metadata.FromIncomingContext(ctx) 173 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 174 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 175 } 176 s.reqs = append(s.reqs, req) 177 if s.err != nil { 178 return nil, s.err 179 } 180 return s.resps[0].(*emptypb.Empty), nil 181} 182 183func (s *mockDlpServer) CreateDeidentifyTemplate(ctx context.Context, req *dlppb.CreateDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) { 184 md, _ := metadata.FromIncomingContext(ctx) 185 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 186 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 187 } 188 s.reqs = append(s.reqs, req) 189 if s.err != nil { 190 return nil, s.err 191 } 192 return s.resps[0].(*dlppb.DeidentifyTemplate), nil 193} 194 195func (s *mockDlpServer) UpdateDeidentifyTemplate(ctx context.Context, req *dlppb.UpdateDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) { 196 md, _ := metadata.FromIncomingContext(ctx) 197 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 198 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 199 } 200 s.reqs = append(s.reqs, req) 201 if s.err != nil { 202 return nil, s.err 203 } 204 return s.resps[0].(*dlppb.DeidentifyTemplate), nil 205} 206 207func (s *mockDlpServer) GetDeidentifyTemplate(ctx context.Context, req *dlppb.GetDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) { 208 md, _ := metadata.FromIncomingContext(ctx) 209 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 210 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 211 } 212 s.reqs = append(s.reqs, req) 213 if s.err != nil { 214 return nil, s.err 215 } 216 return s.resps[0].(*dlppb.DeidentifyTemplate), nil 217} 218 219func (s *mockDlpServer) ListDeidentifyTemplates(ctx context.Context, req *dlppb.ListDeidentifyTemplatesRequest) (*dlppb.ListDeidentifyTemplatesResponse, error) { 220 md, _ := metadata.FromIncomingContext(ctx) 221 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 222 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 223 } 224 s.reqs = append(s.reqs, req) 225 if s.err != nil { 226 return nil, s.err 227 } 228 return s.resps[0].(*dlppb.ListDeidentifyTemplatesResponse), nil 229} 230 231func (s *mockDlpServer) DeleteDeidentifyTemplate(ctx context.Context, req *dlppb.DeleteDeidentifyTemplateRequest) (*emptypb.Empty, error) { 232 md, _ := metadata.FromIncomingContext(ctx) 233 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 234 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 235 } 236 s.reqs = append(s.reqs, req) 237 if s.err != nil { 238 return nil, s.err 239 } 240 return s.resps[0].(*emptypb.Empty), nil 241} 242 243func (s *mockDlpServer) CreateJobTrigger(ctx context.Context, req *dlppb.CreateJobTriggerRequest) (*dlppb.JobTrigger, error) { 244 md, _ := metadata.FromIncomingContext(ctx) 245 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 246 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 247 } 248 s.reqs = append(s.reqs, req) 249 if s.err != nil { 250 return nil, s.err 251 } 252 return s.resps[0].(*dlppb.JobTrigger), nil 253} 254 255func (s *mockDlpServer) UpdateJobTrigger(ctx context.Context, req *dlppb.UpdateJobTriggerRequest) (*dlppb.JobTrigger, error) { 256 md, _ := metadata.FromIncomingContext(ctx) 257 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 258 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 259 } 260 s.reqs = append(s.reqs, req) 261 if s.err != nil { 262 return nil, s.err 263 } 264 return s.resps[0].(*dlppb.JobTrigger), nil 265} 266 267func (s *mockDlpServer) GetJobTrigger(ctx context.Context, req *dlppb.GetJobTriggerRequest) (*dlppb.JobTrigger, error) { 268 md, _ := metadata.FromIncomingContext(ctx) 269 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 270 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 271 } 272 s.reqs = append(s.reqs, req) 273 if s.err != nil { 274 return nil, s.err 275 } 276 return s.resps[0].(*dlppb.JobTrigger), nil 277} 278 279func (s *mockDlpServer) ListJobTriggers(ctx context.Context, req *dlppb.ListJobTriggersRequest) (*dlppb.ListJobTriggersResponse, error) { 280 md, _ := metadata.FromIncomingContext(ctx) 281 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 282 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 283 } 284 s.reqs = append(s.reqs, req) 285 if s.err != nil { 286 return nil, s.err 287 } 288 return s.resps[0].(*dlppb.ListJobTriggersResponse), nil 289} 290 291func (s *mockDlpServer) DeleteJobTrigger(ctx context.Context, req *dlppb.DeleteJobTriggerRequest) (*emptypb.Empty, error) { 292 md, _ := metadata.FromIncomingContext(ctx) 293 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 294 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 295 } 296 s.reqs = append(s.reqs, req) 297 if s.err != nil { 298 return nil, s.err 299 } 300 return s.resps[0].(*emptypb.Empty), nil 301} 302 303func (s *mockDlpServer) CreateDlpJob(ctx context.Context, req *dlppb.CreateDlpJobRequest) (*dlppb.DlpJob, error) { 304 md, _ := metadata.FromIncomingContext(ctx) 305 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 306 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 307 } 308 s.reqs = append(s.reqs, req) 309 if s.err != nil { 310 return nil, s.err 311 } 312 return s.resps[0].(*dlppb.DlpJob), nil 313} 314 315func (s *mockDlpServer) ListDlpJobs(ctx context.Context, req *dlppb.ListDlpJobsRequest) (*dlppb.ListDlpJobsResponse, error) { 316 md, _ := metadata.FromIncomingContext(ctx) 317 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 318 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 319 } 320 s.reqs = append(s.reqs, req) 321 if s.err != nil { 322 return nil, s.err 323 } 324 return s.resps[0].(*dlppb.ListDlpJobsResponse), nil 325} 326 327func (s *mockDlpServer) GetDlpJob(ctx context.Context, req *dlppb.GetDlpJobRequest) (*dlppb.DlpJob, error) { 328 md, _ := metadata.FromIncomingContext(ctx) 329 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 330 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 331 } 332 s.reqs = append(s.reqs, req) 333 if s.err != nil { 334 return nil, s.err 335 } 336 return s.resps[0].(*dlppb.DlpJob), nil 337} 338 339func (s *mockDlpServer) DeleteDlpJob(ctx context.Context, req *dlppb.DeleteDlpJobRequest) (*emptypb.Empty, error) { 340 md, _ := metadata.FromIncomingContext(ctx) 341 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 342 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 343 } 344 s.reqs = append(s.reqs, req) 345 if s.err != nil { 346 return nil, s.err 347 } 348 return s.resps[0].(*emptypb.Empty), nil 349} 350 351func (s *mockDlpServer) CancelDlpJob(ctx context.Context, req *dlppb.CancelDlpJobRequest) (*emptypb.Empty, error) { 352 md, _ := metadata.FromIncomingContext(ctx) 353 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 354 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 355 } 356 s.reqs = append(s.reqs, req) 357 if s.err != nil { 358 return nil, s.err 359 } 360 return s.resps[0].(*emptypb.Empty), nil 361} 362 363func (s *mockDlpServer) CreateStoredInfoType(ctx context.Context, req *dlppb.CreateStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) { 364 md, _ := metadata.FromIncomingContext(ctx) 365 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 366 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 367 } 368 s.reqs = append(s.reqs, req) 369 if s.err != nil { 370 return nil, s.err 371 } 372 return s.resps[0].(*dlppb.StoredInfoType), nil 373} 374 375func (s *mockDlpServer) UpdateStoredInfoType(ctx context.Context, req *dlppb.UpdateStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) { 376 md, _ := metadata.FromIncomingContext(ctx) 377 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 378 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 379 } 380 s.reqs = append(s.reqs, req) 381 if s.err != nil { 382 return nil, s.err 383 } 384 return s.resps[0].(*dlppb.StoredInfoType), nil 385} 386 387func (s *mockDlpServer) GetStoredInfoType(ctx context.Context, req *dlppb.GetStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) { 388 md, _ := metadata.FromIncomingContext(ctx) 389 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 390 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 391 } 392 s.reqs = append(s.reqs, req) 393 if s.err != nil { 394 return nil, s.err 395 } 396 return s.resps[0].(*dlppb.StoredInfoType), nil 397} 398 399func (s *mockDlpServer) ListStoredInfoTypes(ctx context.Context, req *dlppb.ListStoredInfoTypesRequest) (*dlppb.ListStoredInfoTypesResponse, error) { 400 md, _ := metadata.FromIncomingContext(ctx) 401 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 402 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 403 } 404 s.reqs = append(s.reqs, req) 405 if s.err != nil { 406 return nil, s.err 407 } 408 return s.resps[0].(*dlppb.ListStoredInfoTypesResponse), nil 409} 410 411func (s *mockDlpServer) DeleteStoredInfoType(ctx context.Context, req *dlppb.DeleteStoredInfoTypeRequest) (*emptypb.Empty, error) { 412 md, _ := metadata.FromIncomingContext(ctx) 413 if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") { 414 return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg) 415 } 416 s.reqs = append(s.reqs, req) 417 if s.err != nil { 418 return nil, s.err 419 } 420 return s.resps[0].(*emptypb.Empty), nil 421} 422 423// clientOpt is the option tests should use to connect to the test server. 424// It is initialized by TestMain. 425var clientOpt option.ClientOption 426 427var ( 428 mockDlp mockDlpServer 429) 430 431func TestMain(m *testing.M) { 432 flag.Parse() 433 434 serv := grpc.NewServer() 435 dlppb.RegisterDlpServiceServer(serv, &mockDlp) 436 437 lis, err := net.Listen("tcp", "localhost:0") 438 if err != nil { 439 log.Fatal(err) 440 } 441 go serv.Serve(lis) 442 443 conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure()) 444 if err != nil { 445 log.Fatal(err) 446 } 447 clientOpt = option.WithGRPCConn(conn) 448 449 os.Exit(m.Run()) 450} 451 452func TestDlpServiceInspectContent(t *testing.T) { 453 var expectedResponse *dlppb.InspectContentResponse = &dlppb.InspectContentResponse{} 454 455 mockDlp.err = nil 456 mockDlp.reqs = nil 457 458 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 459 460 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 461 var request = &dlppb.InspectContentRequest{ 462 Parent: formattedParent, 463 } 464 465 c, err := NewClient(context.Background(), clientOpt) 466 if err != nil { 467 t.Fatal(err) 468 } 469 470 resp, err := c.InspectContent(context.Background(), request) 471 472 if err != nil { 473 t.Fatal(err) 474 } 475 476 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 477 t.Errorf("wrong request %q, want %q", got, want) 478 } 479 480 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 481 t.Errorf("wrong response %q, want %q)", got, want) 482 } 483} 484 485func TestDlpServiceInspectContentError(t *testing.T) { 486 errCode := codes.PermissionDenied 487 mockDlp.err = gstatus.Error(errCode, "test error") 488 489 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 490 var request = &dlppb.InspectContentRequest{ 491 Parent: formattedParent, 492 } 493 494 c, err := NewClient(context.Background(), clientOpt) 495 if err != nil { 496 t.Fatal(err) 497 } 498 499 resp, err := c.InspectContent(context.Background(), request) 500 501 if st, ok := gstatus.FromError(err); !ok { 502 t.Errorf("got error %v, expected grpc error", err) 503 } else if c := st.Code(); c != errCode { 504 t.Errorf("got error code %q, want %q", c, errCode) 505 } 506 _ = resp 507} 508func TestDlpServiceRedactImage(t *testing.T) { 509 var redactedImage []byte = []byte("28") 510 var extractedText string = "extractedText998260012" 511 var expectedResponse = &dlppb.RedactImageResponse{ 512 RedactedImage: redactedImage, 513 ExtractedText: extractedText, 514 } 515 516 mockDlp.err = nil 517 mockDlp.reqs = nil 518 519 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 520 521 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 522 var request = &dlppb.RedactImageRequest{ 523 Parent: formattedParent, 524 } 525 526 c, err := NewClient(context.Background(), clientOpt) 527 if err != nil { 528 t.Fatal(err) 529 } 530 531 resp, err := c.RedactImage(context.Background(), request) 532 533 if err != nil { 534 t.Fatal(err) 535 } 536 537 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 538 t.Errorf("wrong request %q, want %q", got, want) 539 } 540 541 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 542 t.Errorf("wrong response %q, want %q)", got, want) 543 } 544} 545 546func TestDlpServiceRedactImageError(t *testing.T) { 547 errCode := codes.PermissionDenied 548 mockDlp.err = gstatus.Error(errCode, "test error") 549 550 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 551 var request = &dlppb.RedactImageRequest{ 552 Parent: formattedParent, 553 } 554 555 c, err := NewClient(context.Background(), clientOpt) 556 if err != nil { 557 t.Fatal(err) 558 } 559 560 resp, err := c.RedactImage(context.Background(), request) 561 562 if st, ok := gstatus.FromError(err); !ok { 563 t.Errorf("got error %v, expected grpc error", err) 564 } else if c := st.Code(); c != errCode { 565 t.Errorf("got error code %q, want %q", c, errCode) 566 } 567 _ = resp 568} 569func TestDlpServiceDeidentifyContent(t *testing.T) { 570 var expectedResponse *dlppb.DeidentifyContentResponse = &dlppb.DeidentifyContentResponse{} 571 572 mockDlp.err = nil 573 mockDlp.reqs = nil 574 575 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 576 577 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 578 var request = &dlppb.DeidentifyContentRequest{ 579 Parent: formattedParent, 580 } 581 582 c, err := NewClient(context.Background(), clientOpt) 583 if err != nil { 584 t.Fatal(err) 585 } 586 587 resp, err := c.DeidentifyContent(context.Background(), request) 588 589 if err != nil { 590 t.Fatal(err) 591 } 592 593 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 594 t.Errorf("wrong request %q, want %q", got, want) 595 } 596 597 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 598 t.Errorf("wrong response %q, want %q)", got, want) 599 } 600} 601 602func TestDlpServiceDeidentifyContentError(t *testing.T) { 603 errCode := codes.PermissionDenied 604 mockDlp.err = gstatus.Error(errCode, "test error") 605 606 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 607 var request = &dlppb.DeidentifyContentRequest{ 608 Parent: formattedParent, 609 } 610 611 c, err := NewClient(context.Background(), clientOpt) 612 if err != nil { 613 t.Fatal(err) 614 } 615 616 resp, err := c.DeidentifyContent(context.Background(), request) 617 618 if st, ok := gstatus.FromError(err); !ok { 619 t.Errorf("got error %v, expected grpc error", err) 620 } else if c := st.Code(); c != errCode { 621 t.Errorf("got error code %q, want %q", c, errCode) 622 } 623 _ = resp 624} 625func TestDlpServiceReidentifyContent(t *testing.T) { 626 var expectedResponse *dlppb.ReidentifyContentResponse = &dlppb.ReidentifyContentResponse{} 627 628 mockDlp.err = nil 629 mockDlp.reqs = nil 630 631 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 632 633 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 634 var request = &dlppb.ReidentifyContentRequest{ 635 Parent: formattedParent, 636 } 637 638 c, err := NewClient(context.Background(), clientOpt) 639 if err != nil { 640 t.Fatal(err) 641 } 642 643 resp, err := c.ReidentifyContent(context.Background(), request) 644 645 if err != nil { 646 t.Fatal(err) 647 } 648 649 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 650 t.Errorf("wrong request %q, want %q", got, want) 651 } 652 653 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 654 t.Errorf("wrong response %q, want %q)", got, want) 655 } 656} 657 658func TestDlpServiceReidentifyContentError(t *testing.T) { 659 errCode := codes.PermissionDenied 660 mockDlp.err = gstatus.Error(errCode, "test error") 661 662 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 663 var request = &dlppb.ReidentifyContentRequest{ 664 Parent: formattedParent, 665 } 666 667 c, err := NewClient(context.Background(), clientOpt) 668 if err != nil { 669 t.Fatal(err) 670 } 671 672 resp, err := c.ReidentifyContent(context.Background(), request) 673 674 if st, ok := gstatus.FromError(err); !ok { 675 t.Errorf("got error %v, expected grpc error", err) 676 } else if c := st.Code(); c != errCode { 677 t.Errorf("got error code %q, want %q", c, errCode) 678 } 679 _ = resp 680} 681func TestDlpServiceListInfoTypes(t *testing.T) { 682 var expectedResponse *dlppb.ListInfoTypesResponse = &dlppb.ListInfoTypesResponse{} 683 684 mockDlp.err = nil 685 mockDlp.reqs = nil 686 687 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 688 689 var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{} 690 691 c, err := NewClient(context.Background(), clientOpt) 692 if err != nil { 693 t.Fatal(err) 694 } 695 696 resp, err := c.ListInfoTypes(context.Background(), request) 697 698 if err != nil { 699 t.Fatal(err) 700 } 701 702 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 703 t.Errorf("wrong request %q, want %q", got, want) 704 } 705 706 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 707 t.Errorf("wrong response %q, want %q)", got, want) 708 } 709} 710 711func TestDlpServiceListInfoTypesError(t *testing.T) { 712 errCode := codes.PermissionDenied 713 mockDlp.err = gstatus.Error(errCode, "test error") 714 715 var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{} 716 717 c, err := NewClient(context.Background(), clientOpt) 718 if err != nil { 719 t.Fatal(err) 720 } 721 722 resp, err := c.ListInfoTypes(context.Background(), request) 723 724 if st, ok := gstatus.FromError(err); !ok { 725 t.Errorf("got error %v, expected grpc error", err) 726 } else if c := st.Code(); c != errCode { 727 t.Errorf("got error code %q, want %q", c, errCode) 728 } 729 _ = resp 730} 731func TestDlpServiceCreateInspectTemplate(t *testing.T) { 732 var name string = "name3373707" 733 var displayName string = "displayName1615086568" 734 var description string = "description-1724546052" 735 var expectedResponse = &dlppb.InspectTemplate{ 736 Name: name, 737 DisplayName: displayName, 738 Description: description, 739 } 740 741 mockDlp.err = nil 742 mockDlp.reqs = nil 743 744 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 745 746 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 747 var request = &dlppb.CreateInspectTemplateRequest{ 748 Parent: formattedParent, 749 } 750 751 c, err := NewClient(context.Background(), clientOpt) 752 if err != nil { 753 t.Fatal(err) 754 } 755 756 resp, err := c.CreateInspectTemplate(context.Background(), request) 757 758 if err != nil { 759 t.Fatal(err) 760 } 761 762 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 763 t.Errorf("wrong request %q, want %q", got, want) 764 } 765 766 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 767 t.Errorf("wrong response %q, want %q)", got, want) 768 } 769} 770 771func TestDlpServiceCreateInspectTemplateError(t *testing.T) { 772 errCode := codes.PermissionDenied 773 mockDlp.err = gstatus.Error(errCode, "test error") 774 775 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 776 var request = &dlppb.CreateInspectTemplateRequest{ 777 Parent: formattedParent, 778 } 779 780 c, err := NewClient(context.Background(), clientOpt) 781 if err != nil { 782 t.Fatal(err) 783 } 784 785 resp, err := c.CreateInspectTemplate(context.Background(), request) 786 787 if st, ok := gstatus.FromError(err); !ok { 788 t.Errorf("got error %v, expected grpc error", err) 789 } else if c := st.Code(); c != errCode { 790 t.Errorf("got error code %q, want %q", c, errCode) 791 } 792 _ = resp 793} 794func TestDlpServiceUpdateInspectTemplate(t *testing.T) { 795 var name2 string = "name2-1052831874" 796 var displayName string = "displayName1615086568" 797 var description string = "description-1724546052" 798 var expectedResponse = &dlppb.InspectTemplate{ 799 Name: name2, 800 DisplayName: displayName, 801 Description: description, 802 } 803 804 mockDlp.err = nil 805 mockDlp.reqs = nil 806 807 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 808 809 var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]") 810 var request = &dlppb.UpdateInspectTemplateRequest{ 811 Name: formattedName, 812 } 813 814 c, err := NewClient(context.Background(), clientOpt) 815 if err != nil { 816 t.Fatal(err) 817 } 818 819 resp, err := c.UpdateInspectTemplate(context.Background(), request) 820 821 if err != nil { 822 t.Fatal(err) 823 } 824 825 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 826 t.Errorf("wrong request %q, want %q", got, want) 827 } 828 829 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 830 t.Errorf("wrong response %q, want %q)", got, want) 831 } 832} 833 834func TestDlpServiceUpdateInspectTemplateError(t *testing.T) { 835 errCode := codes.PermissionDenied 836 mockDlp.err = gstatus.Error(errCode, "test error") 837 838 var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]") 839 var request = &dlppb.UpdateInspectTemplateRequest{ 840 Name: formattedName, 841 } 842 843 c, err := NewClient(context.Background(), clientOpt) 844 if err != nil { 845 t.Fatal(err) 846 } 847 848 resp, err := c.UpdateInspectTemplate(context.Background(), request) 849 850 if st, ok := gstatus.FromError(err); !ok { 851 t.Errorf("got error %v, expected grpc error", err) 852 } else if c := st.Code(); c != errCode { 853 t.Errorf("got error code %q, want %q", c, errCode) 854 } 855 _ = resp 856} 857func TestDlpServiceGetInspectTemplate(t *testing.T) { 858 var name string = "name3373707" 859 var displayName string = "displayName1615086568" 860 var description string = "description-1724546052" 861 var expectedResponse = &dlppb.InspectTemplate{ 862 Name: name, 863 DisplayName: displayName, 864 Description: description, 865 } 866 867 mockDlp.err = nil 868 mockDlp.reqs = nil 869 870 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 871 872 var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{} 873 874 c, err := NewClient(context.Background(), clientOpt) 875 if err != nil { 876 t.Fatal(err) 877 } 878 879 resp, err := c.GetInspectTemplate(context.Background(), request) 880 881 if err != nil { 882 t.Fatal(err) 883 } 884 885 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 886 t.Errorf("wrong request %q, want %q", got, want) 887 } 888 889 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 890 t.Errorf("wrong response %q, want %q)", got, want) 891 } 892} 893 894func TestDlpServiceGetInspectTemplateError(t *testing.T) { 895 errCode := codes.PermissionDenied 896 mockDlp.err = gstatus.Error(errCode, "test error") 897 898 var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{} 899 900 c, err := NewClient(context.Background(), clientOpt) 901 if err != nil { 902 t.Fatal(err) 903 } 904 905 resp, err := c.GetInspectTemplate(context.Background(), request) 906 907 if st, ok := gstatus.FromError(err); !ok { 908 t.Errorf("got error %v, expected grpc error", err) 909 } else if c := st.Code(); c != errCode { 910 t.Errorf("got error code %q, want %q", c, errCode) 911 } 912 _ = resp 913} 914func TestDlpServiceListInspectTemplates(t *testing.T) { 915 var nextPageToken string = "" 916 var inspectTemplatesElement *dlppb.InspectTemplate = &dlppb.InspectTemplate{} 917 var inspectTemplates = []*dlppb.InspectTemplate{inspectTemplatesElement} 918 var expectedResponse = &dlppb.ListInspectTemplatesResponse{ 919 NextPageToken: nextPageToken, 920 InspectTemplates: inspectTemplates, 921 } 922 923 mockDlp.err = nil 924 mockDlp.reqs = nil 925 926 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 927 928 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 929 var request = &dlppb.ListInspectTemplatesRequest{ 930 Parent: formattedParent, 931 } 932 933 c, err := NewClient(context.Background(), clientOpt) 934 if err != nil { 935 t.Fatal(err) 936 } 937 938 resp, err := c.ListInspectTemplates(context.Background(), request).Next() 939 940 if err != nil { 941 t.Fatal(err) 942 } 943 944 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 945 t.Errorf("wrong request %q, want %q", got, want) 946 } 947 948 want := (interface{})(expectedResponse.InspectTemplates[0]) 949 got := (interface{})(resp) 950 var ok bool 951 952 switch want := (want).(type) { 953 case proto.Message: 954 ok = proto.Equal(want, got.(proto.Message)) 955 default: 956 ok = want == got 957 } 958 if !ok { 959 t.Errorf("wrong response %q, want %q)", got, want) 960 } 961} 962 963func TestDlpServiceListInspectTemplatesError(t *testing.T) { 964 errCode := codes.PermissionDenied 965 mockDlp.err = gstatus.Error(errCode, "test error") 966 967 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 968 var request = &dlppb.ListInspectTemplatesRequest{ 969 Parent: formattedParent, 970 } 971 972 c, err := NewClient(context.Background(), clientOpt) 973 if err != nil { 974 t.Fatal(err) 975 } 976 977 resp, err := c.ListInspectTemplates(context.Background(), request).Next() 978 979 if st, ok := gstatus.FromError(err); !ok { 980 t.Errorf("got error %v, expected grpc error", err) 981 } else if c := st.Code(); c != errCode { 982 t.Errorf("got error code %q, want %q", c, errCode) 983 } 984 _ = resp 985} 986func TestDlpServiceDeleteInspectTemplate(t *testing.T) { 987 var expectedResponse *emptypb.Empty = &emptypb.Empty{} 988 989 mockDlp.err = nil 990 mockDlp.reqs = nil 991 992 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 993 994 var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]") 995 var request = &dlppb.DeleteInspectTemplateRequest{ 996 Name: formattedName, 997 } 998 999 c, err := NewClient(context.Background(), clientOpt) 1000 if err != nil { 1001 t.Fatal(err) 1002 } 1003 1004 err = c.DeleteInspectTemplate(context.Background(), request) 1005 1006 if err != nil { 1007 t.Fatal(err) 1008 } 1009 1010 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1011 t.Errorf("wrong request %q, want %q", got, want) 1012 } 1013 1014} 1015 1016func TestDlpServiceDeleteInspectTemplateError(t *testing.T) { 1017 errCode := codes.PermissionDenied 1018 mockDlp.err = gstatus.Error(errCode, "test error") 1019 1020 var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]") 1021 var request = &dlppb.DeleteInspectTemplateRequest{ 1022 Name: formattedName, 1023 } 1024 1025 c, err := NewClient(context.Background(), clientOpt) 1026 if err != nil { 1027 t.Fatal(err) 1028 } 1029 1030 err = c.DeleteInspectTemplate(context.Background(), request) 1031 1032 if st, ok := gstatus.FromError(err); !ok { 1033 t.Errorf("got error %v, expected grpc error", err) 1034 } else if c := st.Code(); c != errCode { 1035 t.Errorf("got error code %q, want %q", c, errCode) 1036 } 1037} 1038func TestDlpServiceCreateDeidentifyTemplate(t *testing.T) { 1039 var name string = "name3373707" 1040 var displayName string = "displayName1615086568" 1041 var description string = "description-1724546052" 1042 var expectedResponse = &dlppb.DeidentifyTemplate{ 1043 Name: name, 1044 DisplayName: displayName, 1045 Description: description, 1046 } 1047 1048 mockDlp.err = nil 1049 mockDlp.reqs = nil 1050 1051 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1052 1053 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 1054 var request = &dlppb.CreateDeidentifyTemplateRequest{ 1055 Parent: formattedParent, 1056 } 1057 1058 c, err := NewClient(context.Background(), clientOpt) 1059 if err != nil { 1060 t.Fatal(err) 1061 } 1062 1063 resp, err := c.CreateDeidentifyTemplate(context.Background(), request) 1064 1065 if err != nil { 1066 t.Fatal(err) 1067 } 1068 1069 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1070 t.Errorf("wrong request %q, want %q", got, want) 1071 } 1072 1073 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1074 t.Errorf("wrong response %q, want %q)", got, want) 1075 } 1076} 1077 1078func TestDlpServiceCreateDeidentifyTemplateError(t *testing.T) { 1079 errCode := codes.PermissionDenied 1080 mockDlp.err = gstatus.Error(errCode, "test error") 1081 1082 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 1083 var request = &dlppb.CreateDeidentifyTemplateRequest{ 1084 Parent: formattedParent, 1085 } 1086 1087 c, err := NewClient(context.Background(), clientOpt) 1088 if err != nil { 1089 t.Fatal(err) 1090 } 1091 1092 resp, err := c.CreateDeidentifyTemplate(context.Background(), request) 1093 1094 if st, ok := gstatus.FromError(err); !ok { 1095 t.Errorf("got error %v, expected grpc error", err) 1096 } else if c := st.Code(); c != errCode { 1097 t.Errorf("got error code %q, want %q", c, errCode) 1098 } 1099 _ = resp 1100} 1101func TestDlpServiceUpdateDeidentifyTemplate(t *testing.T) { 1102 var name2 string = "name2-1052831874" 1103 var displayName string = "displayName1615086568" 1104 var description string = "description-1724546052" 1105 var expectedResponse = &dlppb.DeidentifyTemplate{ 1106 Name: name2, 1107 DisplayName: displayName, 1108 Description: description, 1109 } 1110 1111 mockDlp.err = nil 1112 mockDlp.reqs = nil 1113 1114 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1115 1116 var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]") 1117 var request = &dlppb.UpdateDeidentifyTemplateRequest{ 1118 Name: formattedName, 1119 } 1120 1121 c, err := NewClient(context.Background(), clientOpt) 1122 if err != nil { 1123 t.Fatal(err) 1124 } 1125 1126 resp, err := c.UpdateDeidentifyTemplate(context.Background(), request) 1127 1128 if err != nil { 1129 t.Fatal(err) 1130 } 1131 1132 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1133 t.Errorf("wrong request %q, want %q", got, want) 1134 } 1135 1136 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1137 t.Errorf("wrong response %q, want %q)", got, want) 1138 } 1139} 1140 1141func TestDlpServiceUpdateDeidentifyTemplateError(t *testing.T) { 1142 errCode := codes.PermissionDenied 1143 mockDlp.err = gstatus.Error(errCode, "test error") 1144 1145 var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]") 1146 var request = &dlppb.UpdateDeidentifyTemplateRequest{ 1147 Name: formattedName, 1148 } 1149 1150 c, err := NewClient(context.Background(), clientOpt) 1151 if err != nil { 1152 t.Fatal(err) 1153 } 1154 1155 resp, err := c.UpdateDeidentifyTemplate(context.Background(), request) 1156 1157 if st, ok := gstatus.FromError(err); !ok { 1158 t.Errorf("got error %v, expected grpc error", err) 1159 } else if c := st.Code(); c != errCode { 1160 t.Errorf("got error code %q, want %q", c, errCode) 1161 } 1162 _ = resp 1163} 1164func TestDlpServiceGetDeidentifyTemplate(t *testing.T) { 1165 var name2 string = "name2-1052831874" 1166 var displayName string = "displayName1615086568" 1167 var description string = "description-1724546052" 1168 var expectedResponse = &dlppb.DeidentifyTemplate{ 1169 Name: name2, 1170 DisplayName: displayName, 1171 Description: description, 1172 } 1173 1174 mockDlp.err = nil 1175 mockDlp.reqs = nil 1176 1177 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1178 1179 var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]") 1180 var request = &dlppb.GetDeidentifyTemplateRequest{ 1181 Name: formattedName, 1182 } 1183 1184 c, err := NewClient(context.Background(), clientOpt) 1185 if err != nil { 1186 t.Fatal(err) 1187 } 1188 1189 resp, err := c.GetDeidentifyTemplate(context.Background(), request) 1190 1191 if err != nil { 1192 t.Fatal(err) 1193 } 1194 1195 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1196 t.Errorf("wrong request %q, want %q", got, want) 1197 } 1198 1199 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1200 t.Errorf("wrong response %q, want %q)", got, want) 1201 } 1202} 1203 1204func TestDlpServiceGetDeidentifyTemplateError(t *testing.T) { 1205 errCode := codes.PermissionDenied 1206 mockDlp.err = gstatus.Error(errCode, "test error") 1207 1208 var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]") 1209 var request = &dlppb.GetDeidentifyTemplateRequest{ 1210 Name: formattedName, 1211 } 1212 1213 c, err := NewClient(context.Background(), clientOpt) 1214 if err != nil { 1215 t.Fatal(err) 1216 } 1217 1218 resp, err := c.GetDeidentifyTemplate(context.Background(), request) 1219 1220 if st, ok := gstatus.FromError(err); !ok { 1221 t.Errorf("got error %v, expected grpc error", err) 1222 } else if c := st.Code(); c != errCode { 1223 t.Errorf("got error code %q, want %q", c, errCode) 1224 } 1225 _ = resp 1226} 1227func TestDlpServiceListDeidentifyTemplates(t *testing.T) { 1228 var nextPageToken string = "" 1229 var deidentifyTemplatesElement *dlppb.DeidentifyTemplate = &dlppb.DeidentifyTemplate{} 1230 var deidentifyTemplates = []*dlppb.DeidentifyTemplate{deidentifyTemplatesElement} 1231 var expectedResponse = &dlppb.ListDeidentifyTemplatesResponse{ 1232 NextPageToken: nextPageToken, 1233 DeidentifyTemplates: deidentifyTemplates, 1234 } 1235 1236 mockDlp.err = nil 1237 mockDlp.reqs = nil 1238 1239 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1240 1241 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 1242 var request = &dlppb.ListDeidentifyTemplatesRequest{ 1243 Parent: formattedParent, 1244 } 1245 1246 c, err := NewClient(context.Background(), clientOpt) 1247 if err != nil { 1248 t.Fatal(err) 1249 } 1250 1251 resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next() 1252 1253 if err != nil { 1254 t.Fatal(err) 1255 } 1256 1257 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1258 t.Errorf("wrong request %q, want %q", got, want) 1259 } 1260 1261 want := (interface{})(expectedResponse.DeidentifyTemplates[0]) 1262 got := (interface{})(resp) 1263 var ok bool 1264 1265 switch want := (want).(type) { 1266 case proto.Message: 1267 ok = proto.Equal(want, got.(proto.Message)) 1268 default: 1269 ok = want == got 1270 } 1271 if !ok { 1272 t.Errorf("wrong response %q, want %q)", got, want) 1273 } 1274} 1275 1276func TestDlpServiceListDeidentifyTemplatesError(t *testing.T) { 1277 errCode := codes.PermissionDenied 1278 mockDlp.err = gstatus.Error(errCode, "test error") 1279 1280 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 1281 var request = &dlppb.ListDeidentifyTemplatesRequest{ 1282 Parent: formattedParent, 1283 } 1284 1285 c, err := NewClient(context.Background(), clientOpt) 1286 if err != nil { 1287 t.Fatal(err) 1288 } 1289 1290 resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next() 1291 1292 if st, ok := gstatus.FromError(err); !ok { 1293 t.Errorf("got error %v, expected grpc error", err) 1294 } else if c := st.Code(); c != errCode { 1295 t.Errorf("got error code %q, want %q", c, errCode) 1296 } 1297 _ = resp 1298} 1299func TestDlpServiceDeleteDeidentifyTemplate(t *testing.T) { 1300 var expectedResponse *emptypb.Empty = &emptypb.Empty{} 1301 1302 mockDlp.err = nil 1303 mockDlp.reqs = nil 1304 1305 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1306 1307 var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]") 1308 var request = &dlppb.DeleteDeidentifyTemplateRequest{ 1309 Name: formattedName, 1310 } 1311 1312 c, err := NewClient(context.Background(), clientOpt) 1313 if err != nil { 1314 t.Fatal(err) 1315 } 1316 1317 err = c.DeleteDeidentifyTemplate(context.Background(), request) 1318 1319 if err != nil { 1320 t.Fatal(err) 1321 } 1322 1323 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1324 t.Errorf("wrong request %q, want %q", got, want) 1325 } 1326 1327} 1328 1329func TestDlpServiceDeleteDeidentifyTemplateError(t *testing.T) { 1330 errCode := codes.PermissionDenied 1331 mockDlp.err = gstatus.Error(errCode, "test error") 1332 1333 var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]") 1334 var request = &dlppb.DeleteDeidentifyTemplateRequest{ 1335 Name: formattedName, 1336 } 1337 1338 c, err := NewClient(context.Background(), clientOpt) 1339 if err != nil { 1340 t.Fatal(err) 1341 } 1342 1343 err = c.DeleteDeidentifyTemplate(context.Background(), request) 1344 1345 if st, ok := gstatus.FromError(err); !ok { 1346 t.Errorf("got error %v, expected grpc error", err) 1347 } else if c := st.Code(); c != errCode { 1348 t.Errorf("got error code %q, want %q", c, errCode) 1349 } 1350} 1351func TestDlpServiceCreateDlpJob(t *testing.T) { 1352 var name string = "name3373707" 1353 var jobTriggerName string = "jobTriggerName1819490804" 1354 var expectedResponse = &dlppb.DlpJob{ 1355 Name: name, 1356 JobTriggerName: jobTriggerName, 1357 } 1358 1359 mockDlp.err = nil 1360 mockDlp.reqs = nil 1361 1362 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1363 1364 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1365 var request = &dlppb.CreateDlpJobRequest{ 1366 Parent: formattedParent, 1367 } 1368 1369 c, err := NewClient(context.Background(), clientOpt) 1370 if err != nil { 1371 t.Fatal(err) 1372 } 1373 1374 resp, err := c.CreateDlpJob(context.Background(), request) 1375 1376 if err != nil { 1377 t.Fatal(err) 1378 } 1379 1380 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1381 t.Errorf("wrong request %q, want %q", got, want) 1382 } 1383 1384 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1385 t.Errorf("wrong response %q, want %q)", got, want) 1386 } 1387} 1388 1389func TestDlpServiceCreateDlpJobError(t *testing.T) { 1390 errCode := codes.PermissionDenied 1391 mockDlp.err = gstatus.Error(errCode, "test error") 1392 1393 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1394 var request = &dlppb.CreateDlpJobRequest{ 1395 Parent: formattedParent, 1396 } 1397 1398 c, err := NewClient(context.Background(), clientOpt) 1399 if err != nil { 1400 t.Fatal(err) 1401 } 1402 1403 resp, err := c.CreateDlpJob(context.Background(), request) 1404 1405 if st, ok := gstatus.FromError(err); !ok { 1406 t.Errorf("got error %v, expected grpc error", err) 1407 } else if c := st.Code(); c != errCode { 1408 t.Errorf("got error code %q, want %q", c, errCode) 1409 } 1410 _ = resp 1411} 1412func TestDlpServiceListDlpJobs(t *testing.T) { 1413 var nextPageToken string = "" 1414 var jobsElement *dlppb.DlpJob = &dlppb.DlpJob{} 1415 var jobs = []*dlppb.DlpJob{jobsElement} 1416 var expectedResponse = &dlppb.ListDlpJobsResponse{ 1417 NextPageToken: nextPageToken, 1418 Jobs: jobs, 1419 } 1420 1421 mockDlp.err = nil 1422 mockDlp.reqs = nil 1423 1424 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1425 1426 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1427 var request = &dlppb.ListDlpJobsRequest{ 1428 Parent: formattedParent, 1429 } 1430 1431 c, err := NewClient(context.Background(), clientOpt) 1432 if err != nil { 1433 t.Fatal(err) 1434 } 1435 1436 resp, err := c.ListDlpJobs(context.Background(), request).Next() 1437 1438 if err != nil { 1439 t.Fatal(err) 1440 } 1441 1442 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1443 t.Errorf("wrong request %q, want %q", got, want) 1444 } 1445 1446 want := (interface{})(expectedResponse.Jobs[0]) 1447 got := (interface{})(resp) 1448 var ok bool 1449 1450 switch want := (want).(type) { 1451 case proto.Message: 1452 ok = proto.Equal(want, got.(proto.Message)) 1453 default: 1454 ok = want == got 1455 } 1456 if !ok { 1457 t.Errorf("wrong response %q, want %q)", got, want) 1458 } 1459} 1460 1461func TestDlpServiceListDlpJobsError(t *testing.T) { 1462 errCode := codes.PermissionDenied 1463 mockDlp.err = gstatus.Error(errCode, "test error") 1464 1465 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1466 var request = &dlppb.ListDlpJobsRequest{ 1467 Parent: formattedParent, 1468 } 1469 1470 c, err := NewClient(context.Background(), clientOpt) 1471 if err != nil { 1472 t.Fatal(err) 1473 } 1474 1475 resp, err := c.ListDlpJobs(context.Background(), request).Next() 1476 1477 if st, ok := gstatus.FromError(err); !ok { 1478 t.Errorf("got error %v, expected grpc error", err) 1479 } else if c := st.Code(); c != errCode { 1480 t.Errorf("got error code %q, want %q", c, errCode) 1481 } 1482 _ = resp 1483} 1484func TestDlpServiceGetDlpJob(t *testing.T) { 1485 var name2 string = "name2-1052831874" 1486 var jobTriggerName string = "jobTriggerName1819490804" 1487 var expectedResponse = &dlppb.DlpJob{ 1488 Name: name2, 1489 JobTriggerName: jobTriggerName, 1490 } 1491 1492 mockDlp.err = nil 1493 mockDlp.reqs = nil 1494 1495 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1496 1497 var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]") 1498 var request = &dlppb.GetDlpJobRequest{ 1499 Name: formattedName, 1500 } 1501 1502 c, err := NewClient(context.Background(), clientOpt) 1503 if err != nil { 1504 t.Fatal(err) 1505 } 1506 1507 resp, err := c.GetDlpJob(context.Background(), request) 1508 1509 if err != nil { 1510 t.Fatal(err) 1511 } 1512 1513 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1514 t.Errorf("wrong request %q, want %q", got, want) 1515 } 1516 1517 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1518 t.Errorf("wrong response %q, want %q)", got, want) 1519 } 1520} 1521 1522func TestDlpServiceGetDlpJobError(t *testing.T) { 1523 errCode := codes.PermissionDenied 1524 mockDlp.err = gstatus.Error(errCode, "test error") 1525 1526 var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]") 1527 var request = &dlppb.GetDlpJobRequest{ 1528 Name: formattedName, 1529 } 1530 1531 c, err := NewClient(context.Background(), clientOpt) 1532 if err != nil { 1533 t.Fatal(err) 1534 } 1535 1536 resp, err := c.GetDlpJob(context.Background(), request) 1537 1538 if st, ok := gstatus.FromError(err); !ok { 1539 t.Errorf("got error %v, expected grpc error", err) 1540 } else if c := st.Code(); c != errCode { 1541 t.Errorf("got error code %q, want %q", c, errCode) 1542 } 1543 _ = resp 1544} 1545func TestDlpServiceDeleteDlpJob(t *testing.T) { 1546 var expectedResponse *emptypb.Empty = &emptypb.Empty{} 1547 1548 mockDlp.err = nil 1549 mockDlp.reqs = nil 1550 1551 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1552 1553 var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]") 1554 var request = &dlppb.DeleteDlpJobRequest{ 1555 Name: formattedName, 1556 } 1557 1558 c, err := NewClient(context.Background(), clientOpt) 1559 if err != nil { 1560 t.Fatal(err) 1561 } 1562 1563 err = c.DeleteDlpJob(context.Background(), request) 1564 1565 if err != nil { 1566 t.Fatal(err) 1567 } 1568 1569 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1570 t.Errorf("wrong request %q, want %q", got, want) 1571 } 1572 1573} 1574 1575func TestDlpServiceDeleteDlpJobError(t *testing.T) { 1576 errCode := codes.PermissionDenied 1577 mockDlp.err = gstatus.Error(errCode, "test error") 1578 1579 var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]") 1580 var request = &dlppb.DeleteDlpJobRequest{ 1581 Name: formattedName, 1582 } 1583 1584 c, err := NewClient(context.Background(), clientOpt) 1585 if err != nil { 1586 t.Fatal(err) 1587 } 1588 1589 err = c.DeleteDlpJob(context.Background(), request) 1590 1591 if st, ok := gstatus.FromError(err); !ok { 1592 t.Errorf("got error %v, expected grpc error", err) 1593 } else if c := st.Code(); c != errCode { 1594 t.Errorf("got error code %q, want %q", c, errCode) 1595 } 1596} 1597func TestDlpServiceCancelDlpJob(t *testing.T) { 1598 var expectedResponse *emptypb.Empty = &emptypb.Empty{} 1599 1600 mockDlp.err = nil 1601 mockDlp.reqs = nil 1602 1603 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1604 1605 var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]") 1606 var request = &dlppb.CancelDlpJobRequest{ 1607 Name: formattedName, 1608 } 1609 1610 c, err := NewClient(context.Background(), clientOpt) 1611 if err != nil { 1612 t.Fatal(err) 1613 } 1614 1615 err = c.CancelDlpJob(context.Background(), request) 1616 1617 if err != nil { 1618 t.Fatal(err) 1619 } 1620 1621 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1622 t.Errorf("wrong request %q, want %q", got, want) 1623 } 1624 1625} 1626 1627func TestDlpServiceCancelDlpJobError(t *testing.T) { 1628 errCode := codes.PermissionDenied 1629 mockDlp.err = gstatus.Error(errCode, "test error") 1630 1631 var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]") 1632 var request = &dlppb.CancelDlpJobRequest{ 1633 Name: formattedName, 1634 } 1635 1636 c, err := NewClient(context.Background(), clientOpt) 1637 if err != nil { 1638 t.Fatal(err) 1639 } 1640 1641 err = c.CancelDlpJob(context.Background(), request) 1642 1643 if st, ok := gstatus.FromError(err); !ok { 1644 t.Errorf("got error %v, expected grpc error", err) 1645 } else if c := st.Code(); c != errCode { 1646 t.Errorf("got error code %q, want %q", c, errCode) 1647 } 1648} 1649func TestDlpServiceListJobTriggers(t *testing.T) { 1650 var nextPageToken string = "" 1651 var jobTriggersElement *dlppb.JobTrigger = &dlppb.JobTrigger{} 1652 var jobTriggers = []*dlppb.JobTrigger{jobTriggersElement} 1653 var expectedResponse = &dlppb.ListJobTriggersResponse{ 1654 NextPageToken: nextPageToken, 1655 JobTriggers: jobTriggers, 1656 } 1657 1658 mockDlp.err = nil 1659 mockDlp.reqs = nil 1660 1661 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1662 1663 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1664 var request = &dlppb.ListJobTriggersRequest{ 1665 Parent: formattedParent, 1666 } 1667 1668 c, err := NewClient(context.Background(), clientOpt) 1669 if err != nil { 1670 t.Fatal(err) 1671 } 1672 1673 resp, err := c.ListJobTriggers(context.Background(), request).Next() 1674 1675 if err != nil { 1676 t.Fatal(err) 1677 } 1678 1679 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1680 t.Errorf("wrong request %q, want %q", got, want) 1681 } 1682 1683 want := (interface{})(expectedResponse.JobTriggers[0]) 1684 got := (interface{})(resp) 1685 var ok bool 1686 1687 switch want := (want).(type) { 1688 case proto.Message: 1689 ok = proto.Equal(want, got.(proto.Message)) 1690 default: 1691 ok = want == got 1692 } 1693 if !ok { 1694 t.Errorf("wrong response %q, want %q)", got, want) 1695 } 1696} 1697 1698func TestDlpServiceListJobTriggersError(t *testing.T) { 1699 errCode := codes.PermissionDenied 1700 mockDlp.err = gstatus.Error(errCode, "test error") 1701 1702 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1703 var request = &dlppb.ListJobTriggersRequest{ 1704 Parent: formattedParent, 1705 } 1706 1707 c, err := NewClient(context.Background(), clientOpt) 1708 if err != nil { 1709 t.Fatal(err) 1710 } 1711 1712 resp, err := c.ListJobTriggers(context.Background(), request).Next() 1713 1714 if st, ok := gstatus.FromError(err); !ok { 1715 t.Errorf("got error %v, expected grpc error", err) 1716 } else if c := st.Code(); c != errCode { 1717 t.Errorf("got error code %q, want %q", c, errCode) 1718 } 1719 _ = resp 1720} 1721func TestDlpServiceGetJobTrigger(t *testing.T) { 1722 var name2 string = "name2-1052831874" 1723 var displayName string = "displayName1615086568" 1724 var description string = "description-1724546052" 1725 var expectedResponse = &dlppb.JobTrigger{ 1726 Name: name2, 1727 DisplayName: displayName, 1728 Description: description, 1729 } 1730 1731 mockDlp.err = nil 1732 mockDlp.reqs = nil 1733 1734 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1735 1736 var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]") 1737 var request = &dlppb.GetJobTriggerRequest{ 1738 Name: formattedName, 1739 } 1740 1741 c, err := NewClient(context.Background(), clientOpt) 1742 if err != nil { 1743 t.Fatal(err) 1744 } 1745 1746 resp, err := c.GetJobTrigger(context.Background(), request) 1747 1748 if err != nil { 1749 t.Fatal(err) 1750 } 1751 1752 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1753 t.Errorf("wrong request %q, want %q", got, want) 1754 } 1755 1756 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1757 t.Errorf("wrong response %q, want %q)", got, want) 1758 } 1759} 1760 1761func TestDlpServiceGetJobTriggerError(t *testing.T) { 1762 errCode := codes.PermissionDenied 1763 mockDlp.err = gstatus.Error(errCode, "test error") 1764 1765 var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]") 1766 var request = &dlppb.GetJobTriggerRequest{ 1767 Name: formattedName, 1768 } 1769 1770 c, err := NewClient(context.Background(), clientOpt) 1771 if err != nil { 1772 t.Fatal(err) 1773 } 1774 1775 resp, err := c.GetJobTrigger(context.Background(), request) 1776 1777 if st, ok := gstatus.FromError(err); !ok { 1778 t.Errorf("got error %v, expected grpc error", err) 1779 } else if c := st.Code(); c != errCode { 1780 t.Errorf("got error code %q, want %q", c, errCode) 1781 } 1782 _ = resp 1783} 1784func TestDlpServiceDeleteJobTrigger(t *testing.T) { 1785 var expectedResponse *emptypb.Empty = &emptypb.Empty{} 1786 1787 mockDlp.err = nil 1788 mockDlp.reqs = nil 1789 1790 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1791 1792 var name string = "name3373707" 1793 var request = &dlppb.DeleteJobTriggerRequest{ 1794 Name: name, 1795 } 1796 1797 c, err := NewClient(context.Background(), clientOpt) 1798 if err != nil { 1799 t.Fatal(err) 1800 } 1801 1802 err = c.DeleteJobTrigger(context.Background(), request) 1803 1804 if err != nil { 1805 t.Fatal(err) 1806 } 1807 1808 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1809 t.Errorf("wrong request %q, want %q", got, want) 1810 } 1811 1812} 1813 1814func TestDlpServiceDeleteJobTriggerError(t *testing.T) { 1815 errCode := codes.PermissionDenied 1816 mockDlp.err = gstatus.Error(errCode, "test error") 1817 1818 var name string = "name3373707" 1819 var request = &dlppb.DeleteJobTriggerRequest{ 1820 Name: name, 1821 } 1822 1823 c, err := NewClient(context.Background(), clientOpt) 1824 if err != nil { 1825 t.Fatal(err) 1826 } 1827 1828 err = c.DeleteJobTrigger(context.Background(), request) 1829 1830 if st, ok := gstatus.FromError(err); !ok { 1831 t.Errorf("got error %v, expected grpc error", err) 1832 } else if c := st.Code(); c != errCode { 1833 t.Errorf("got error code %q, want %q", c, errCode) 1834 } 1835} 1836func TestDlpServiceUpdateJobTrigger(t *testing.T) { 1837 var name2 string = "name2-1052831874" 1838 var displayName string = "displayName1615086568" 1839 var description string = "description-1724546052" 1840 var expectedResponse = &dlppb.JobTrigger{ 1841 Name: name2, 1842 DisplayName: displayName, 1843 Description: description, 1844 } 1845 1846 mockDlp.err = nil 1847 mockDlp.reqs = nil 1848 1849 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1850 1851 var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]") 1852 var request = &dlppb.UpdateJobTriggerRequest{ 1853 Name: formattedName, 1854 } 1855 1856 c, err := NewClient(context.Background(), clientOpt) 1857 if err != nil { 1858 t.Fatal(err) 1859 } 1860 1861 resp, err := c.UpdateJobTrigger(context.Background(), request) 1862 1863 if err != nil { 1864 t.Fatal(err) 1865 } 1866 1867 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1868 t.Errorf("wrong request %q, want %q", got, want) 1869 } 1870 1871 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1872 t.Errorf("wrong response %q, want %q)", got, want) 1873 } 1874} 1875 1876func TestDlpServiceUpdateJobTriggerError(t *testing.T) { 1877 errCode := codes.PermissionDenied 1878 mockDlp.err = gstatus.Error(errCode, "test error") 1879 1880 var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]") 1881 var request = &dlppb.UpdateJobTriggerRequest{ 1882 Name: formattedName, 1883 } 1884 1885 c, err := NewClient(context.Background(), clientOpt) 1886 if err != nil { 1887 t.Fatal(err) 1888 } 1889 1890 resp, err := c.UpdateJobTrigger(context.Background(), request) 1891 1892 if st, ok := gstatus.FromError(err); !ok { 1893 t.Errorf("got error %v, expected grpc error", err) 1894 } else if c := st.Code(); c != errCode { 1895 t.Errorf("got error code %q, want %q", c, errCode) 1896 } 1897 _ = resp 1898} 1899func TestDlpServiceCreateJobTrigger(t *testing.T) { 1900 var name string = "name3373707" 1901 var displayName string = "displayName1615086568" 1902 var description string = "description-1724546052" 1903 var expectedResponse = &dlppb.JobTrigger{ 1904 Name: name, 1905 DisplayName: displayName, 1906 Description: description, 1907 } 1908 1909 mockDlp.err = nil 1910 mockDlp.reqs = nil 1911 1912 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1913 1914 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1915 var request = &dlppb.CreateJobTriggerRequest{ 1916 Parent: formattedParent, 1917 } 1918 1919 c, err := NewClient(context.Background(), clientOpt) 1920 if err != nil { 1921 t.Fatal(err) 1922 } 1923 1924 resp, err := c.CreateJobTrigger(context.Background(), request) 1925 1926 if err != nil { 1927 t.Fatal(err) 1928 } 1929 1930 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1931 t.Errorf("wrong request %q, want %q", got, want) 1932 } 1933 1934 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1935 t.Errorf("wrong response %q, want %q)", got, want) 1936 } 1937} 1938 1939func TestDlpServiceCreateJobTriggerError(t *testing.T) { 1940 errCode := codes.PermissionDenied 1941 mockDlp.err = gstatus.Error(errCode, "test error") 1942 1943 var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]") 1944 var request = &dlppb.CreateJobTriggerRequest{ 1945 Parent: formattedParent, 1946 } 1947 1948 c, err := NewClient(context.Background(), clientOpt) 1949 if err != nil { 1950 t.Fatal(err) 1951 } 1952 1953 resp, err := c.CreateJobTrigger(context.Background(), request) 1954 1955 if st, ok := gstatus.FromError(err); !ok { 1956 t.Errorf("got error %v, expected grpc error", err) 1957 } else if c := st.Code(); c != errCode { 1958 t.Errorf("got error code %q, want %q", c, errCode) 1959 } 1960 _ = resp 1961} 1962func TestDlpServiceCreateStoredInfoType(t *testing.T) { 1963 var name string = "name3373707" 1964 var expectedResponse = &dlppb.StoredInfoType{ 1965 Name: name, 1966 } 1967 1968 mockDlp.err = nil 1969 mockDlp.reqs = nil 1970 1971 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 1972 1973 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 1974 var request = &dlppb.CreateStoredInfoTypeRequest{ 1975 Parent: formattedParent, 1976 } 1977 1978 c, err := NewClient(context.Background(), clientOpt) 1979 if err != nil { 1980 t.Fatal(err) 1981 } 1982 1983 resp, err := c.CreateStoredInfoType(context.Background(), request) 1984 1985 if err != nil { 1986 t.Fatal(err) 1987 } 1988 1989 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 1990 t.Errorf("wrong request %q, want %q", got, want) 1991 } 1992 1993 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 1994 t.Errorf("wrong response %q, want %q)", got, want) 1995 } 1996} 1997 1998func TestDlpServiceCreateStoredInfoTypeError(t *testing.T) { 1999 errCode := codes.PermissionDenied 2000 mockDlp.err = gstatus.Error(errCode, "test error") 2001 2002 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 2003 var request = &dlppb.CreateStoredInfoTypeRequest{ 2004 Parent: formattedParent, 2005 } 2006 2007 c, err := NewClient(context.Background(), clientOpt) 2008 if err != nil { 2009 t.Fatal(err) 2010 } 2011 2012 resp, err := c.CreateStoredInfoType(context.Background(), request) 2013 2014 if st, ok := gstatus.FromError(err); !ok { 2015 t.Errorf("got error %v, expected grpc error", err) 2016 } else if c := st.Code(); c != errCode { 2017 t.Errorf("got error code %q, want %q", c, errCode) 2018 } 2019 _ = resp 2020} 2021func TestDlpServiceUpdateStoredInfoType(t *testing.T) { 2022 var name2 string = "name2-1052831874" 2023 var expectedResponse = &dlppb.StoredInfoType{ 2024 Name: name2, 2025 } 2026 2027 mockDlp.err = nil 2028 mockDlp.reqs = nil 2029 2030 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 2031 2032 var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]") 2033 var request = &dlppb.UpdateStoredInfoTypeRequest{ 2034 Name: formattedName, 2035 } 2036 2037 c, err := NewClient(context.Background(), clientOpt) 2038 if err != nil { 2039 t.Fatal(err) 2040 } 2041 2042 resp, err := c.UpdateStoredInfoType(context.Background(), request) 2043 2044 if err != nil { 2045 t.Fatal(err) 2046 } 2047 2048 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 2049 t.Errorf("wrong request %q, want %q", got, want) 2050 } 2051 2052 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 2053 t.Errorf("wrong response %q, want %q)", got, want) 2054 } 2055} 2056 2057func TestDlpServiceUpdateStoredInfoTypeError(t *testing.T) { 2058 errCode := codes.PermissionDenied 2059 mockDlp.err = gstatus.Error(errCode, "test error") 2060 2061 var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]") 2062 var request = &dlppb.UpdateStoredInfoTypeRequest{ 2063 Name: formattedName, 2064 } 2065 2066 c, err := NewClient(context.Background(), clientOpt) 2067 if err != nil { 2068 t.Fatal(err) 2069 } 2070 2071 resp, err := c.UpdateStoredInfoType(context.Background(), request) 2072 2073 if st, ok := gstatus.FromError(err); !ok { 2074 t.Errorf("got error %v, expected grpc error", err) 2075 } else if c := st.Code(); c != errCode { 2076 t.Errorf("got error code %q, want %q", c, errCode) 2077 } 2078 _ = resp 2079} 2080func TestDlpServiceGetStoredInfoType(t *testing.T) { 2081 var name2 string = "name2-1052831874" 2082 var expectedResponse = &dlppb.StoredInfoType{ 2083 Name: name2, 2084 } 2085 2086 mockDlp.err = nil 2087 mockDlp.reqs = nil 2088 2089 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 2090 2091 var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]") 2092 var request = &dlppb.GetStoredInfoTypeRequest{ 2093 Name: formattedName, 2094 } 2095 2096 c, err := NewClient(context.Background(), clientOpt) 2097 if err != nil { 2098 t.Fatal(err) 2099 } 2100 2101 resp, err := c.GetStoredInfoType(context.Background(), request) 2102 2103 if err != nil { 2104 t.Fatal(err) 2105 } 2106 2107 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 2108 t.Errorf("wrong request %q, want %q", got, want) 2109 } 2110 2111 if want, got := expectedResponse, resp; !proto.Equal(want, got) { 2112 t.Errorf("wrong response %q, want %q)", got, want) 2113 } 2114} 2115 2116func TestDlpServiceGetStoredInfoTypeError(t *testing.T) { 2117 errCode := codes.PermissionDenied 2118 mockDlp.err = gstatus.Error(errCode, "test error") 2119 2120 var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]") 2121 var request = &dlppb.GetStoredInfoTypeRequest{ 2122 Name: formattedName, 2123 } 2124 2125 c, err := NewClient(context.Background(), clientOpt) 2126 if err != nil { 2127 t.Fatal(err) 2128 } 2129 2130 resp, err := c.GetStoredInfoType(context.Background(), request) 2131 2132 if st, ok := gstatus.FromError(err); !ok { 2133 t.Errorf("got error %v, expected grpc error", err) 2134 } else if c := st.Code(); c != errCode { 2135 t.Errorf("got error code %q, want %q", c, errCode) 2136 } 2137 _ = resp 2138} 2139func TestDlpServiceListStoredInfoTypes(t *testing.T) { 2140 var nextPageToken string = "" 2141 var storedInfoTypesElement *dlppb.StoredInfoType = &dlppb.StoredInfoType{} 2142 var storedInfoTypes = []*dlppb.StoredInfoType{storedInfoTypesElement} 2143 var expectedResponse = &dlppb.ListStoredInfoTypesResponse{ 2144 NextPageToken: nextPageToken, 2145 StoredInfoTypes: storedInfoTypes, 2146 } 2147 2148 mockDlp.err = nil 2149 mockDlp.reqs = nil 2150 2151 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 2152 2153 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 2154 var request = &dlppb.ListStoredInfoTypesRequest{ 2155 Parent: formattedParent, 2156 } 2157 2158 c, err := NewClient(context.Background(), clientOpt) 2159 if err != nil { 2160 t.Fatal(err) 2161 } 2162 2163 resp, err := c.ListStoredInfoTypes(context.Background(), request).Next() 2164 2165 if err != nil { 2166 t.Fatal(err) 2167 } 2168 2169 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 2170 t.Errorf("wrong request %q, want %q", got, want) 2171 } 2172 2173 want := (interface{})(expectedResponse.StoredInfoTypes[0]) 2174 got := (interface{})(resp) 2175 var ok bool 2176 2177 switch want := (want).(type) { 2178 case proto.Message: 2179 ok = proto.Equal(want, got.(proto.Message)) 2180 default: 2181 ok = want == got 2182 } 2183 if !ok { 2184 t.Errorf("wrong response %q, want %q)", got, want) 2185 } 2186} 2187 2188func TestDlpServiceListStoredInfoTypesError(t *testing.T) { 2189 errCode := codes.PermissionDenied 2190 mockDlp.err = gstatus.Error(errCode, "test error") 2191 2192 var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]") 2193 var request = &dlppb.ListStoredInfoTypesRequest{ 2194 Parent: formattedParent, 2195 } 2196 2197 c, err := NewClient(context.Background(), clientOpt) 2198 if err != nil { 2199 t.Fatal(err) 2200 } 2201 2202 resp, err := c.ListStoredInfoTypes(context.Background(), request).Next() 2203 2204 if st, ok := gstatus.FromError(err); !ok { 2205 t.Errorf("got error %v, expected grpc error", err) 2206 } else if c := st.Code(); c != errCode { 2207 t.Errorf("got error code %q, want %q", c, errCode) 2208 } 2209 _ = resp 2210} 2211func TestDlpServiceDeleteStoredInfoType(t *testing.T) { 2212 var expectedResponse *emptypb.Empty = &emptypb.Empty{} 2213 2214 mockDlp.err = nil 2215 mockDlp.reqs = nil 2216 2217 mockDlp.resps = append(mockDlp.resps[:0], expectedResponse) 2218 2219 var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]") 2220 var request = &dlppb.DeleteStoredInfoTypeRequest{ 2221 Name: formattedName, 2222 } 2223 2224 c, err := NewClient(context.Background(), clientOpt) 2225 if err != nil { 2226 t.Fatal(err) 2227 } 2228 2229 err = c.DeleteStoredInfoType(context.Background(), request) 2230 2231 if err != nil { 2232 t.Fatal(err) 2233 } 2234 2235 if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) { 2236 t.Errorf("wrong request %q, want %q", got, want) 2237 } 2238 2239} 2240 2241func TestDlpServiceDeleteStoredInfoTypeError(t *testing.T) { 2242 errCode := codes.PermissionDenied 2243 mockDlp.err = gstatus.Error(errCode, "test error") 2244 2245 var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]") 2246 var request = &dlppb.DeleteStoredInfoTypeRequest{ 2247 Name: formattedName, 2248 } 2249 2250 c, err := NewClient(context.Background(), clientOpt) 2251 if err != nil { 2252 t.Fatal(err) 2253 } 2254 2255 err = c.DeleteStoredInfoType(context.Background(), request) 2256 2257 if st, ok := gstatus.FromError(err); !ok { 2258 t.Errorf("got error %v, expected grpc error", err) 2259 } else if c := st.Code(); c != errCode { 2260 t.Errorf("got error code %q, want %q", c, errCode) 2261 } 2262} 2263