You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

1903 lines
53 KiB

  1. // Copyright 2018 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. // AUTO-GENERATED CODE. DO NOT EDIT.
  15. package dlp
  16. import (
  17. emptypb "github.com/golang/protobuf/ptypes/empty"
  18. dlppb "google.golang.org/genproto/googleapis/privacy/dlp/v2"
  19. )
  20. import (
  21. "flag"
  22. "fmt"
  23. "io"
  24. "log"
  25. "net"
  26. "os"
  27. "strings"
  28. "testing"
  29. "github.com/golang/protobuf/proto"
  30. "github.com/golang/protobuf/ptypes"
  31. "golang.org/x/net/context"
  32. "google.golang.org/api/option"
  33. status "google.golang.org/genproto/googleapis/rpc/status"
  34. "google.golang.org/grpc"
  35. "google.golang.org/grpc/codes"
  36. "google.golang.org/grpc/metadata"
  37. gstatus "google.golang.org/grpc/status"
  38. )
  39. var _ = io.EOF
  40. var _ = ptypes.MarshalAny
  41. var _ status.Status
  42. type mockDlpServer struct {
  43. // Embed for forward compatibility.
  44. // Tests will keep working if more methods are added
  45. // in the future.
  46. dlppb.DlpServiceServer
  47. reqs []proto.Message
  48. // If set, all calls return this error.
  49. err error
  50. // responses to return if err == nil
  51. resps []proto.Message
  52. }
  53. func (s *mockDlpServer) InspectContent(ctx context.Context, req *dlppb.InspectContentRequest) (*dlppb.InspectContentResponse, error) {
  54. md, _ := metadata.FromIncomingContext(ctx)
  55. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  56. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  57. }
  58. s.reqs = append(s.reqs, req)
  59. if s.err != nil {
  60. return nil, s.err
  61. }
  62. return s.resps[0].(*dlppb.InspectContentResponse), nil
  63. }
  64. func (s *mockDlpServer) RedactImage(ctx context.Context, req *dlppb.RedactImageRequest) (*dlppb.RedactImageResponse, error) {
  65. md, _ := metadata.FromIncomingContext(ctx)
  66. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  67. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  68. }
  69. s.reqs = append(s.reqs, req)
  70. if s.err != nil {
  71. return nil, s.err
  72. }
  73. return s.resps[0].(*dlppb.RedactImageResponse), nil
  74. }
  75. func (s *mockDlpServer) DeidentifyContent(ctx context.Context, req *dlppb.DeidentifyContentRequest) (*dlppb.DeidentifyContentResponse, 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.DeidentifyContentResponse), nil
  85. }
  86. func (s *mockDlpServer) ReidentifyContent(ctx context.Context, req *dlppb.ReidentifyContentRequest) (*dlppb.ReidentifyContentResponse, error) {
  87. md, _ := metadata.FromIncomingContext(ctx)
  88. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  89. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  90. }
  91. s.reqs = append(s.reqs, req)
  92. if s.err != nil {
  93. return nil, s.err
  94. }
  95. return s.resps[0].(*dlppb.ReidentifyContentResponse), nil
  96. }
  97. func (s *mockDlpServer) ListInfoTypes(ctx context.Context, req *dlppb.ListInfoTypesRequest) (*dlppb.ListInfoTypesResponse, error) {
  98. md, _ := metadata.FromIncomingContext(ctx)
  99. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  100. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  101. }
  102. s.reqs = append(s.reqs, req)
  103. if s.err != nil {
  104. return nil, s.err
  105. }
  106. return s.resps[0].(*dlppb.ListInfoTypesResponse), nil
  107. }
  108. func (s *mockDlpServer) CreateInspectTemplate(ctx context.Context, req *dlppb.CreateInspectTemplateRequest) (*dlppb.InspectTemplate, error) {
  109. md, _ := metadata.FromIncomingContext(ctx)
  110. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  111. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  112. }
  113. s.reqs = append(s.reqs, req)
  114. if s.err != nil {
  115. return nil, s.err
  116. }
  117. return s.resps[0].(*dlppb.InspectTemplate), nil
  118. }
  119. func (s *mockDlpServer) UpdateInspectTemplate(ctx context.Context, req *dlppb.UpdateInspectTemplateRequest) (*dlppb.InspectTemplate, error) {
  120. md, _ := metadata.FromIncomingContext(ctx)
  121. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  122. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  123. }
  124. s.reqs = append(s.reqs, req)
  125. if s.err != nil {
  126. return nil, s.err
  127. }
  128. return s.resps[0].(*dlppb.InspectTemplate), nil
  129. }
  130. func (s *mockDlpServer) GetInspectTemplate(ctx context.Context, req *dlppb.GetInspectTemplateRequest) (*dlppb.InspectTemplate, error) {
  131. md, _ := metadata.FromIncomingContext(ctx)
  132. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  133. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  134. }
  135. s.reqs = append(s.reqs, req)
  136. if s.err != nil {
  137. return nil, s.err
  138. }
  139. return s.resps[0].(*dlppb.InspectTemplate), nil
  140. }
  141. func (s *mockDlpServer) ListInspectTemplates(ctx context.Context, req *dlppb.ListInspectTemplatesRequest) (*dlppb.ListInspectTemplatesResponse, error) {
  142. md, _ := metadata.FromIncomingContext(ctx)
  143. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  144. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  145. }
  146. s.reqs = append(s.reqs, req)
  147. if s.err != nil {
  148. return nil, s.err
  149. }
  150. return s.resps[0].(*dlppb.ListInspectTemplatesResponse), nil
  151. }
  152. func (s *mockDlpServer) DeleteInspectTemplate(ctx context.Context, req *dlppb.DeleteInspectTemplateRequest) (*emptypb.Empty, error) {
  153. md, _ := metadata.FromIncomingContext(ctx)
  154. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  155. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  156. }
  157. s.reqs = append(s.reqs, req)
  158. if s.err != nil {
  159. return nil, s.err
  160. }
  161. return s.resps[0].(*emptypb.Empty), nil
  162. }
  163. func (s *mockDlpServer) CreateDeidentifyTemplate(ctx context.Context, req *dlppb.CreateDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) {
  164. md, _ := metadata.FromIncomingContext(ctx)
  165. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  166. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  167. }
  168. s.reqs = append(s.reqs, req)
  169. if s.err != nil {
  170. return nil, s.err
  171. }
  172. return s.resps[0].(*dlppb.DeidentifyTemplate), nil
  173. }
  174. func (s *mockDlpServer) UpdateDeidentifyTemplate(ctx context.Context, req *dlppb.UpdateDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) {
  175. md, _ := metadata.FromIncomingContext(ctx)
  176. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  177. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  178. }
  179. s.reqs = append(s.reqs, req)
  180. if s.err != nil {
  181. return nil, s.err
  182. }
  183. return s.resps[0].(*dlppb.DeidentifyTemplate), nil
  184. }
  185. func (s *mockDlpServer) GetDeidentifyTemplate(ctx context.Context, req *dlppb.GetDeidentifyTemplateRequest) (*dlppb.DeidentifyTemplate, error) {
  186. md, _ := metadata.FromIncomingContext(ctx)
  187. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  188. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  189. }
  190. s.reqs = append(s.reqs, req)
  191. if s.err != nil {
  192. return nil, s.err
  193. }
  194. return s.resps[0].(*dlppb.DeidentifyTemplate), nil
  195. }
  196. func (s *mockDlpServer) ListDeidentifyTemplates(ctx context.Context, req *dlppb.ListDeidentifyTemplatesRequest) (*dlppb.ListDeidentifyTemplatesResponse, error) {
  197. md, _ := metadata.FromIncomingContext(ctx)
  198. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  199. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  200. }
  201. s.reqs = append(s.reqs, req)
  202. if s.err != nil {
  203. return nil, s.err
  204. }
  205. return s.resps[0].(*dlppb.ListDeidentifyTemplatesResponse), nil
  206. }
  207. func (s *mockDlpServer) DeleteDeidentifyTemplate(ctx context.Context, req *dlppb.DeleteDeidentifyTemplateRequest) (*emptypb.Empty, 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].(*emptypb.Empty), nil
  217. }
  218. func (s *mockDlpServer) CreateJobTrigger(ctx context.Context, req *dlppb.CreateJobTriggerRequest) (*dlppb.JobTrigger, error) {
  219. md, _ := metadata.FromIncomingContext(ctx)
  220. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  221. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  222. }
  223. s.reqs = append(s.reqs, req)
  224. if s.err != nil {
  225. return nil, s.err
  226. }
  227. return s.resps[0].(*dlppb.JobTrigger), nil
  228. }
  229. func (s *mockDlpServer) UpdateJobTrigger(ctx context.Context, req *dlppb.UpdateJobTriggerRequest) (*dlppb.JobTrigger, error) {
  230. md, _ := metadata.FromIncomingContext(ctx)
  231. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  232. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  233. }
  234. s.reqs = append(s.reqs, req)
  235. if s.err != nil {
  236. return nil, s.err
  237. }
  238. return s.resps[0].(*dlppb.JobTrigger), nil
  239. }
  240. func (s *mockDlpServer) GetJobTrigger(ctx context.Context, req *dlppb.GetJobTriggerRequest) (*dlppb.JobTrigger, error) {
  241. md, _ := metadata.FromIncomingContext(ctx)
  242. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  243. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  244. }
  245. s.reqs = append(s.reqs, req)
  246. if s.err != nil {
  247. return nil, s.err
  248. }
  249. return s.resps[0].(*dlppb.JobTrigger), nil
  250. }
  251. func (s *mockDlpServer) ListJobTriggers(ctx context.Context, req *dlppb.ListJobTriggersRequest) (*dlppb.ListJobTriggersResponse, error) {
  252. md, _ := metadata.FromIncomingContext(ctx)
  253. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  254. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  255. }
  256. s.reqs = append(s.reqs, req)
  257. if s.err != nil {
  258. return nil, s.err
  259. }
  260. return s.resps[0].(*dlppb.ListJobTriggersResponse), nil
  261. }
  262. func (s *mockDlpServer) DeleteJobTrigger(ctx context.Context, req *dlppb.DeleteJobTriggerRequest) (*emptypb.Empty, error) {
  263. md, _ := metadata.FromIncomingContext(ctx)
  264. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  265. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  266. }
  267. s.reqs = append(s.reqs, req)
  268. if s.err != nil {
  269. return nil, s.err
  270. }
  271. return s.resps[0].(*emptypb.Empty), nil
  272. }
  273. func (s *mockDlpServer) CreateDlpJob(ctx context.Context, req *dlppb.CreateDlpJobRequest) (*dlppb.DlpJob, error) {
  274. md, _ := metadata.FromIncomingContext(ctx)
  275. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  276. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  277. }
  278. s.reqs = append(s.reqs, req)
  279. if s.err != nil {
  280. return nil, s.err
  281. }
  282. return s.resps[0].(*dlppb.DlpJob), nil
  283. }
  284. func (s *mockDlpServer) ListDlpJobs(ctx context.Context, req *dlppb.ListDlpJobsRequest) (*dlppb.ListDlpJobsResponse, error) {
  285. md, _ := metadata.FromIncomingContext(ctx)
  286. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  287. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  288. }
  289. s.reqs = append(s.reqs, req)
  290. if s.err != nil {
  291. return nil, s.err
  292. }
  293. return s.resps[0].(*dlppb.ListDlpJobsResponse), nil
  294. }
  295. func (s *mockDlpServer) GetDlpJob(ctx context.Context, req *dlppb.GetDlpJobRequest) (*dlppb.DlpJob, error) {
  296. md, _ := metadata.FromIncomingContext(ctx)
  297. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  298. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  299. }
  300. s.reqs = append(s.reqs, req)
  301. if s.err != nil {
  302. return nil, s.err
  303. }
  304. return s.resps[0].(*dlppb.DlpJob), nil
  305. }
  306. func (s *mockDlpServer) DeleteDlpJob(ctx context.Context, req *dlppb.DeleteDlpJobRequest) (*emptypb.Empty, error) {
  307. md, _ := metadata.FromIncomingContext(ctx)
  308. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  309. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  310. }
  311. s.reqs = append(s.reqs, req)
  312. if s.err != nil {
  313. return nil, s.err
  314. }
  315. return s.resps[0].(*emptypb.Empty), nil
  316. }
  317. func (s *mockDlpServer) CancelDlpJob(ctx context.Context, req *dlppb.CancelDlpJobRequest) (*emptypb.Empty, error) {
  318. md, _ := metadata.FromIncomingContext(ctx)
  319. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  320. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  321. }
  322. s.reqs = append(s.reqs, req)
  323. if s.err != nil {
  324. return nil, s.err
  325. }
  326. return s.resps[0].(*emptypb.Empty), nil
  327. }
  328. // clientOpt is the option tests should use to connect to the test server.
  329. // It is initialized by TestMain.
  330. var clientOpt option.ClientOption
  331. var (
  332. mockDlp mockDlpServer
  333. )
  334. func TestMain(m *testing.M) {
  335. flag.Parse()
  336. serv := grpc.NewServer()
  337. dlppb.RegisterDlpServiceServer(serv, &mockDlp)
  338. lis, err := net.Listen("tcp", "localhost:0")
  339. if err != nil {
  340. log.Fatal(err)
  341. }
  342. go serv.Serve(lis)
  343. conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
  344. if err != nil {
  345. log.Fatal(err)
  346. }
  347. clientOpt = option.WithGRPCConn(conn)
  348. os.Exit(m.Run())
  349. }
  350. func TestDlpServiceInspectContent(t *testing.T) {
  351. var expectedResponse *dlppb.InspectContentResponse = &dlppb.InspectContentResponse{}
  352. mockDlp.err = nil
  353. mockDlp.reqs = nil
  354. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  355. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  356. var request = &dlppb.InspectContentRequest{
  357. Parent: formattedParent,
  358. }
  359. c, err := NewClient(context.Background(), clientOpt)
  360. if err != nil {
  361. t.Fatal(err)
  362. }
  363. resp, err := c.InspectContent(context.Background(), request)
  364. if err != nil {
  365. t.Fatal(err)
  366. }
  367. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  368. t.Errorf("wrong request %q, want %q", got, want)
  369. }
  370. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  371. t.Errorf("wrong response %q, want %q)", got, want)
  372. }
  373. }
  374. func TestDlpServiceInspectContentError(t *testing.T) {
  375. errCode := codes.PermissionDenied
  376. mockDlp.err = gstatus.Error(errCode, "test error")
  377. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  378. var request = &dlppb.InspectContentRequest{
  379. Parent: formattedParent,
  380. }
  381. c, err := NewClient(context.Background(), clientOpt)
  382. if err != nil {
  383. t.Fatal(err)
  384. }
  385. resp, err := c.InspectContent(context.Background(), request)
  386. if st, ok := gstatus.FromError(err); !ok {
  387. t.Errorf("got error %v, expected grpc error", err)
  388. } else if c := st.Code(); c != errCode {
  389. t.Errorf("got error code %q, want %q", c, errCode)
  390. }
  391. _ = resp
  392. }
  393. func TestDlpServiceRedactImage(t *testing.T) {
  394. var redactedImage []byte = []byte("28")
  395. var extractedText string = "extractedText998260012"
  396. var expectedResponse = &dlppb.RedactImageResponse{
  397. RedactedImage: redactedImage,
  398. ExtractedText: extractedText,
  399. }
  400. mockDlp.err = nil
  401. mockDlp.reqs = nil
  402. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  403. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  404. var request = &dlppb.RedactImageRequest{
  405. Parent: formattedParent,
  406. }
  407. c, err := NewClient(context.Background(), clientOpt)
  408. if err != nil {
  409. t.Fatal(err)
  410. }
  411. resp, err := c.RedactImage(context.Background(), request)
  412. if err != nil {
  413. t.Fatal(err)
  414. }
  415. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  416. t.Errorf("wrong request %q, want %q", got, want)
  417. }
  418. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  419. t.Errorf("wrong response %q, want %q)", got, want)
  420. }
  421. }
  422. func TestDlpServiceRedactImageError(t *testing.T) {
  423. errCode := codes.PermissionDenied
  424. mockDlp.err = gstatus.Error(errCode, "test error")
  425. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  426. var request = &dlppb.RedactImageRequest{
  427. Parent: formattedParent,
  428. }
  429. c, err := NewClient(context.Background(), clientOpt)
  430. if err != nil {
  431. t.Fatal(err)
  432. }
  433. resp, err := c.RedactImage(context.Background(), request)
  434. if st, ok := gstatus.FromError(err); !ok {
  435. t.Errorf("got error %v, expected grpc error", err)
  436. } else if c := st.Code(); c != errCode {
  437. t.Errorf("got error code %q, want %q", c, errCode)
  438. }
  439. _ = resp
  440. }
  441. func TestDlpServiceDeidentifyContent(t *testing.T) {
  442. var expectedResponse *dlppb.DeidentifyContentResponse = &dlppb.DeidentifyContentResponse{}
  443. mockDlp.err = nil
  444. mockDlp.reqs = nil
  445. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  446. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  447. var request = &dlppb.DeidentifyContentRequest{
  448. Parent: formattedParent,
  449. }
  450. c, err := NewClient(context.Background(), clientOpt)
  451. if err != nil {
  452. t.Fatal(err)
  453. }
  454. resp, err := c.DeidentifyContent(context.Background(), request)
  455. if err != nil {
  456. t.Fatal(err)
  457. }
  458. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  459. t.Errorf("wrong request %q, want %q", got, want)
  460. }
  461. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  462. t.Errorf("wrong response %q, want %q)", got, want)
  463. }
  464. }
  465. func TestDlpServiceDeidentifyContentError(t *testing.T) {
  466. errCode := codes.PermissionDenied
  467. mockDlp.err = gstatus.Error(errCode, "test error")
  468. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  469. var request = &dlppb.DeidentifyContentRequest{
  470. Parent: formattedParent,
  471. }
  472. c, err := NewClient(context.Background(), clientOpt)
  473. if err != nil {
  474. t.Fatal(err)
  475. }
  476. resp, err := c.DeidentifyContent(context.Background(), request)
  477. if st, ok := gstatus.FromError(err); !ok {
  478. t.Errorf("got error %v, expected grpc error", err)
  479. } else if c := st.Code(); c != errCode {
  480. t.Errorf("got error code %q, want %q", c, errCode)
  481. }
  482. _ = resp
  483. }
  484. func TestDlpServiceReidentifyContent(t *testing.T) {
  485. var expectedResponse *dlppb.ReidentifyContentResponse = &dlppb.ReidentifyContentResponse{}
  486. mockDlp.err = nil
  487. mockDlp.reqs = nil
  488. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  489. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  490. var request = &dlppb.ReidentifyContentRequest{
  491. Parent: formattedParent,
  492. }
  493. c, err := NewClient(context.Background(), clientOpt)
  494. if err != nil {
  495. t.Fatal(err)
  496. }
  497. resp, err := c.ReidentifyContent(context.Background(), request)
  498. if err != nil {
  499. t.Fatal(err)
  500. }
  501. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  502. t.Errorf("wrong request %q, want %q", got, want)
  503. }
  504. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  505. t.Errorf("wrong response %q, want %q)", got, want)
  506. }
  507. }
  508. func TestDlpServiceReidentifyContentError(t *testing.T) {
  509. errCode := codes.PermissionDenied
  510. mockDlp.err = gstatus.Error(errCode, "test error")
  511. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  512. var request = &dlppb.ReidentifyContentRequest{
  513. Parent: formattedParent,
  514. }
  515. c, err := NewClient(context.Background(), clientOpt)
  516. if err != nil {
  517. t.Fatal(err)
  518. }
  519. resp, err := c.ReidentifyContent(context.Background(), request)
  520. if st, ok := gstatus.FromError(err); !ok {
  521. t.Errorf("got error %v, expected grpc error", err)
  522. } else if c := st.Code(); c != errCode {
  523. t.Errorf("got error code %q, want %q", c, errCode)
  524. }
  525. _ = resp
  526. }
  527. func TestDlpServiceListInfoTypes(t *testing.T) {
  528. var expectedResponse *dlppb.ListInfoTypesResponse = &dlppb.ListInfoTypesResponse{}
  529. mockDlp.err = nil
  530. mockDlp.reqs = nil
  531. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  532. var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{}
  533. c, err := NewClient(context.Background(), clientOpt)
  534. if err != nil {
  535. t.Fatal(err)
  536. }
  537. resp, err := c.ListInfoTypes(context.Background(), request)
  538. if err != nil {
  539. t.Fatal(err)
  540. }
  541. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  542. t.Errorf("wrong request %q, want %q", got, want)
  543. }
  544. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  545. t.Errorf("wrong response %q, want %q)", got, want)
  546. }
  547. }
  548. func TestDlpServiceListInfoTypesError(t *testing.T) {
  549. errCode := codes.PermissionDenied
  550. mockDlp.err = gstatus.Error(errCode, "test error")
  551. var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{}
  552. c, err := NewClient(context.Background(), clientOpt)
  553. if err != nil {
  554. t.Fatal(err)
  555. }
  556. resp, err := c.ListInfoTypes(context.Background(), request)
  557. if st, ok := gstatus.FromError(err); !ok {
  558. t.Errorf("got error %v, expected grpc error", err)
  559. } else if c := st.Code(); c != errCode {
  560. t.Errorf("got error code %q, want %q", c, errCode)
  561. }
  562. _ = resp
  563. }
  564. func TestDlpServiceCreateInspectTemplate(t *testing.T) {
  565. var name string = "name3373707"
  566. var displayName string = "displayName1615086568"
  567. var description string = "description-1724546052"
  568. var expectedResponse = &dlppb.InspectTemplate{
  569. Name: name,
  570. DisplayName: displayName,
  571. Description: description,
  572. }
  573. mockDlp.err = nil
  574. mockDlp.reqs = nil
  575. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  576. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  577. var request = &dlppb.CreateInspectTemplateRequest{
  578. Parent: formattedParent,
  579. }
  580. c, err := NewClient(context.Background(), clientOpt)
  581. if err != nil {
  582. t.Fatal(err)
  583. }
  584. resp, err := c.CreateInspectTemplate(context.Background(), request)
  585. if err != nil {
  586. t.Fatal(err)
  587. }
  588. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  589. t.Errorf("wrong request %q, want %q", got, want)
  590. }
  591. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  592. t.Errorf("wrong response %q, want %q)", got, want)
  593. }
  594. }
  595. func TestDlpServiceCreateInspectTemplateError(t *testing.T) {
  596. errCode := codes.PermissionDenied
  597. mockDlp.err = gstatus.Error(errCode, "test error")
  598. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  599. var request = &dlppb.CreateInspectTemplateRequest{
  600. Parent: formattedParent,
  601. }
  602. c, err := NewClient(context.Background(), clientOpt)
  603. if err != nil {
  604. t.Fatal(err)
  605. }
  606. resp, err := c.CreateInspectTemplate(context.Background(), request)
  607. if st, ok := gstatus.FromError(err); !ok {
  608. t.Errorf("got error %v, expected grpc error", err)
  609. } else if c := st.Code(); c != errCode {
  610. t.Errorf("got error code %q, want %q", c, errCode)
  611. }
  612. _ = resp
  613. }
  614. func TestDlpServiceUpdateInspectTemplate(t *testing.T) {
  615. var name2 string = "name2-1052831874"
  616. var displayName string = "displayName1615086568"
  617. var description string = "description-1724546052"
  618. var expectedResponse = &dlppb.InspectTemplate{
  619. Name: name2,
  620. DisplayName: displayName,
  621. Description: description,
  622. }
  623. mockDlp.err = nil
  624. mockDlp.reqs = nil
  625. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  626. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  627. var request = &dlppb.UpdateInspectTemplateRequest{
  628. Name: formattedName,
  629. }
  630. c, err := NewClient(context.Background(), clientOpt)
  631. if err != nil {
  632. t.Fatal(err)
  633. }
  634. resp, err := c.UpdateInspectTemplate(context.Background(), request)
  635. if err != nil {
  636. t.Fatal(err)
  637. }
  638. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  639. t.Errorf("wrong request %q, want %q", got, want)
  640. }
  641. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  642. t.Errorf("wrong response %q, want %q)", got, want)
  643. }
  644. }
  645. func TestDlpServiceUpdateInspectTemplateError(t *testing.T) {
  646. errCode := codes.PermissionDenied
  647. mockDlp.err = gstatus.Error(errCode, "test error")
  648. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  649. var request = &dlppb.UpdateInspectTemplateRequest{
  650. Name: formattedName,
  651. }
  652. c, err := NewClient(context.Background(), clientOpt)
  653. if err != nil {
  654. t.Fatal(err)
  655. }
  656. resp, err := c.UpdateInspectTemplate(context.Background(), request)
  657. if st, ok := gstatus.FromError(err); !ok {
  658. t.Errorf("got error %v, expected grpc error", err)
  659. } else if c := st.Code(); c != errCode {
  660. t.Errorf("got error code %q, want %q", c, errCode)
  661. }
  662. _ = resp
  663. }
  664. func TestDlpServiceGetInspectTemplate(t *testing.T) {
  665. var name string = "name3373707"
  666. var displayName string = "displayName1615086568"
  667. var description string = "description-1724546052"
  668. var expectedResponse = &dlppb.InspectTemplate{
  669. Name: name,
  670. DisplayName: displayName,
  671. Description: description,
  672. }
  673. mockDlp.err = nil
  674. mockDlp.reqs = nil
  675. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  676. var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{}
  677. c, err := NewClient(context.Background(), clientOpt)
  678. if err != nil {
  679. t.Fatal(err)
  680. }
  681. resp, err := c.GetInspectTemplate(context.Background(), request)
  682. if err != nil {
  683. t.Fatal(err)
  684. }
  685. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  686. t.Errorf("wrong request %q, want %q", got, want)
  687. }
  688. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  689. t.Errorf("wrong response %q, want %q)", got, want)
  690. }
  691. }
  692. func TestDlpServiceGetInspectTemplateError(t *testing.T) {
  693. errCode := codes.PermissionDenied
  694. mockDlp.err = gstatus.Error(errCode, "test error")
  695. var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{}
  696. c, err := NewClient(context.Background(), clientOpt)
  697. if err != nil {
  698. t.Fatal(err)
  699. }
  700. resp, err := c.GetInspectTemplate(context.Background(), request)
  701. if st, ok := gstatus.FromError(err); !ok {
  702. t.Errorf("got error %v, expected grpc error", err)
  703. } else if c := st.Code(); c != errCode {
  704. t.Errorf("got error code %q, want %q", c, errCode)
  705. }
  706. _ = resp
  707. }
  708. func TestDlpServiceListInspectTemplates(t *testing.T) {
  709. var nextPageToken string = ""
  710. var inspectTemplatesElement *dlppb.InspectTemplate = &dlppb.InspectTemplate{}
  711. var inspectTemplates = []*dlppb.InspectTemplate{inspectTemplatesElement}
  712. var expectedResponse = &dlppb.ListInspectTemplatesResponse{
  713. NextPageToken: nextPageToken,
  714. InspectTemplates: inspectTemplates,
  715. }
  716. mockDlp.err = nil
  717. mockDlp.reqs = nil
  718. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  719. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  720. var request = &dlppb.ListInspectTemplatesRequest{
  721. Parent: formattedParent,
  722. }
  723. c, err := NewClient(context.Background(), clientOpt)
  724. if err != nil {
  725. t.Fatal(err)
  726. }
  727. resp, err := c.ListInspectTemplates(context.Background(), request).Next()
  728. if err != nil {
  729. t.Fatal(err)
  730. }
  731. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  732. t.Errorf("wrong request %q, want %q", got, want)
  733. }
  734. want := (interface{})(expectedResponse.InspectTemplates[0])
  735. got := (interface{})(resp)
  736. var ok bool
  737. switch want := (want).(type) {
  738. case proto.Message:
  739. ok = proto.Equal(want, got.(proto.Message))
  740. default:
  741. ok = want == got
  742. }
  743. if !ok {
  744. t.Errorf("wrong response %q, want %q)", got, want)
  745. }
  746. }
  747. func TestDlpServiceListInspectTemplatesError(t *testing.T) {
  748. errCode := codes.PermissionDenied
  749. mockDlp.err = gstatus.Error(errCode, "test error")
  750. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  751. var request = &dlppb.ListInspectTemplatesRequest{
  752. Parent: formattedParent,
  753. }
  754. c, err := NewClient(context.Background(), clientOpt)
  755. if err != nil {
  756. t.Fatal(err)
  757. }
  758. resp, err := c.ListInspectTemplates(context.Background(), request).Next()
  759. if st, ok := gstatus.FromError(err); !ok {
  760. t.Errorf("got error %v, expected grpc error", err)
  761. } else if c := st.Code(); c != errCode {
  762. t.Errorf("got error code %q, want %q", c, errCode)
  763. }
  764. _ = resp
  765. }
  766. func TestDlpServiceDeleteInspectTemplate(t *testing.T) {
  767. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  768. mockDlp.err = nil
  769. mockDlp.reqs = nil
  770. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  771. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  772. var request = &dlppb.DeleteInspectTemplateRequest{
  773. Name: formattedName,
  774. }
  775. c, err := NewClient(context.Background(), clientOpt)
  776. if err != nil {
  777. t.Fatal(err)
  778. }
  779. err = c.DeleteInspectTemplate(context.Background(), request)
  780. if err != nil {
  781. t.Fatal(err)
  782. }
  783. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  784. t.Errorf("wrong request %q, want %q", got, want)
  785. }
  786. }
  787. func TestDlpServiceDeleteInspectTemplateError(t *testing.T) {
  788. errCode := codes.PermissionDenied
  789. mockDlp.err = gstatus.Error(errCode, "test error")
  790. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  791. var request = &dlppb.DeleteInspectTemplateRequest{
  792. Name: formattedName,
  793. }
  794. c, err := NewClient(context.Background(), clientOpt)
  795. if err != nil {
  796. t.Fatal(err)
  797. }
  798. err = c.DeleteInspectTemplate(context.Background(), request)
  799. if st, ok := gstatus.FromError(err); !ok {
  800. t.Errorf("got error %v, expected grpc error", err)
  801. } else if c := st.Code(); c != errCode {
  802. t.Errorf("got error code %q, want %q", c, errCode)
  803. }
  804. }
  805. func TestDlpServiceCreateDeidentifyTemplate(t *testing.T) {
  806. var name string = "name3373707"
  807. var displayName string = "displayName1615086568"
  808. var description string = "description-1724546052"
  809. var expectedResponse = &dlppb.DeidentifyTemplate{
  810. Name: name,
  811. DisplayName: displayName,
  812. Description: description,
  813. }
  814. mockDlp.err = nil
  815. mockDlp.reqs = nil
  816. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  817. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  818. var request = &dlppb.CreateDeidentifyTemplateRequest{
  819. Parent: formattedParent,
  820. }
  821. c, err := NewClient(context.Background(), clientOpt)
  822. if err != nil {
  823. t.Fatal(err)
  824. }
  825. resp, err := c.CreateDeidentifyTemplate(context.Background(), request)
  826. if err != nil {
  827. t.Fatal(err)
  828. }
  829. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  830. t.Errorf("wrong request %q, want %q", got, want)
  831. }
  832. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  833. t.Errorf("wrong response %q, want %q)", got, want)
  834. }
  835. }
  836. func TestDlpServiceCreateDeidentifyTemplateError(t *testing.T) {
  837. errCode := codes.PermissionDenied
  838. mockDlp.err = gstatus.Error(errCode, "test error")
  839. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  840. var request = &dlppb.CreateDeidentifyTemplateRequest{
  841. Parent: formattedParent,
  842. }
  843. c, err := NewClient(context.Background(), clientOpt)
  844. if err != nil {
  845. t.Fatal(err)
  846. }
  847. resp, err := c.CreateDeidentifyTemplate(context.Background(), request)
  848. if st, ok := gstatus.FromError(err); !ok {
  849. t.Errorf("got error %v, expected grpc error", err)
  850. } else if c := st.Code(); c != errCode {
  851. t.Errorf("got error code %q, want %q", c, errCode)
  852. }
  853. _ = resp
  854. }
  855. func TestDlpServiceUpdateDeidentifyTemplate(t *testing.T) {
  856. var name2 string = "name2-1052831874"
  857. var displayName string = "displayName1615086568"
  858. var description string = "description-1724546052"
  859. var expectedResponse = &dlppb.DeidentifyTemplate{
  860. Name: name2,
  861. DisplayName: displayName,
  862. Description: description,
  863. }
  864. mockDlp.err = nil
  865. mockDlp.reqs = nil
  866. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  867. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  868. var request = &dlppb.UpdateDeidentifyTemplateRequest{
  869. Name: formattedName,
  870. }
  871. c, err := NewClient(context.Background(), clientOpt)
  872. if err != nil {
  873. t.Fatal(err)
  874. }
  875. resp, err := c.UpdateDeidentifyTemplate(context.Background(), request)
  876. if err != nil {
  877. t.Fatal(err)
  878. }
  879. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  880. t.Errorf("wrong request %q, want %q", got, want)
  881. }
  882. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  883. t.Errorf("wrong response %q, want %q)", got, want)
  884. }
  885. }
  886. func TestDlpServiceUpdateDeidentifyTemplateError(t *testing.T) {
  887. errCode := codes.PermissionDenied
  888. mockDlp.err = gstatus.Error(errCode, "test error")
  889. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  890. var request = &dlppb.UpdateDeidentifyTemplateRequest{
  891. Name: formattedName,
  892. }
  893. c, err := NewClient(context.Background(), clientOpt)
  894. if err != nil {
  895. t.Fatal(err)
  896. }
  897. resp, err := c.UpdateDeidentifyTemplate(context.Background(), request)
  898. if st, ok := gstatus.FromError(err); !ok {
  899. t.Errorf("got error %v, expected grpc error", err)
  900. } else if c := st.Code(); c != errCode {
  901. t.Errorf("got error code %q, want %q", c, errCode)
  902. }
  903. _ = resp
  904. }
  905. func TestDlpServiceGetDeidentifyTemplate(t *testing.T) {
  906. var name2 string = "name2-1052831874"
  907. var displayName string = "displayName1615086568"
  908. var description string = "description-1724546052"
  909. var expectedResponse = &dlppb.DeidentifyTemplate{
  910. Name: name2,
  911. DisplayName: displayName,
  912. Description: description,
  913. }
  914. mockDlp.err = nil
  915. mockDlp.reqs = nil
  916. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  917. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  918. var request = &dlppb.GetDeidentifyTemplateRequest{
  919. Name: formattedName,
  920. }
  921. c, err := NewClient(context.Background(), clientOpt)
  922. if err != nil {
  923. t.Fatal(err)
  924. }
  925. resp, err := c.GetDeidentifyTemplate(context.Background(), request)
  926. if err != nil {
  927. t.Fatal(err)
  928. }
  929. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  930. t.Errorf("wrong request %q, want %q", got, want)
  931. }
  932. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  933. t.Errorf("wrong response %q, want %q)", got, want)
  934. }
  935. }
  936. func TestDlpServiceGetDeidentifyTemplateError(t *testing.T) {
  937. errCode := codes.PermissionDenied
  938. mockDlp.err = gstatus.Error(errCode, "test error")
  939. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  940. var request = &dlppb.GetDeidentifyTemplateRequest{
  941. Name: formattedName,
  942. }
  943. c, err := NewClient(context.Background(), clientOpt)
  944. if err != nil {
  945. t.Fatal(err)
  946. }
  947. resp, err := c.GetDeidentifyTemplate(context.Background(), request)
  948. if st, ok := gstatus.FromError(err); !ok {
  949. t.Errorf("got error %v, expected grpc error", err)
  950. } else if c := st.Code(); c != errCode {
  951. t.Errorf("got error code %q, want %q", c, errCode)
  952. }
  953. _ = resp
  954. }
  955. func TestDlpServiceListDeidentifyTemplates(t *testing.T) {
  956. var nextPageToken string = ""
  957. var deidentifyTemplatesElement *dlppb.DeidentifyTemplate = &dlppb.DeidentifyTemplate{}
  958. var deidentifyTemplates = []*dlppb.DeidentifyTemplate{deidentifyTemplatesElement}
  959. var expectedResponse = &dlppb.ListDeidentifyTemplatesResponse{
  960. NextPageToken: nextPageToken,
  961. DeidentifyTemplates: deidentifyTemplates,
  962. }
  963. mockDlp.err = nil
  964. mockDlp.reqs = nil
  965. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  966. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  967. var request = &dlppb.ListDeidentifyTemplatesRequest{
  968. Parent: formattedParent,
  969. }
  970. c, err := NewClient(context.Background(), clientOpt)
  971. if err != nil {
  972. t.Fatal(err)
  973. }
  974. resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next()
  975. if err != nil {
  976. t.Fatal(err)
  977. }
  978. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  979. t.Errorf("wrong request %q, want %q", got, want)
  980. }
  981. want := (interface{})(expectedResponse.DeidentifyTemplates[0])
  982. got := (interface{})(resp)
  983. var ok bool
  984. switch want := (want).(type) {
  985. case proto.Message:
  986. ok = proto.Equal(want, got.(proto.Message))
  987. default:
  988. ok = want == got
  989. }
  990. if !ok {
  991. t.Errorf("wrong response %q, want %q)", got, want)
  992. }
  993. }
  994. func TestDlpServiceListDeidentifyTemplatesError(t *testing.T) {
  995. errCode := codes.PermissionDenied
  996. mockDlp.err = gstatus.Error(errCode, "test error")
  997. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  998. var request = &dlppb.ListDeidentifyTemplatesRequest{
  999. Parent: formattedParent,
  1000. }
  1001. c, err := NewClient(context.Background(), clientOpt)
  1002. if err != nil {
  1003. t.Fatal(err)
  1004. }
  1005. resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next()
  1006. if st, ok := gstatus.FromError(err); !ok {
  1007. t.Errorf("got error %v, expected grpc error", err)
  1008. } else if c := st.Code(); c != errCode {
  1009. t.Errorf("got error code %q, want %q", c, errCode)
  1010. }
  1011. _ = resp
  1012. }
  1013. func TestDlpServiceDeleteDeidentifyTemplate(t *testing.T) {
  1014. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1015. mockDlp.err = nil
  1016. mockDlp.reqs = nil
  1017. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1018. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  1019. var request = &dlppb.DeleteDeidentifyTemplateRequest{
  1020. Name: formattedName,
  1021. }
  1022. c, err := NewClient(context.Background(), clientOpt)
  1023. if err != nil {
  1024. t.Fatal(err)
  1025. }
  1026. err = c.DeleteDeidentifyTemplate(context.Background(), request)
  1027. if err != nil {
  1028. t.Fatal(err)
  1029. }
  1030. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1031. t.Errorf("wrong request %q, want %q", got, want)
  1032. }
  1033. }
  1034. func TestDlpServiceDeleteDeidentifyTemplateError(t *testing.T) {
  1035. errCode := codes.PermissionDenied
  1036. mockDlp.err = gstatus.Error(errCode, "test error")
  1037. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  1038. var request = &dlppb.DeleteDeidentifyTemplateRequest{
  1039. Name: formattedName,
  1040. }
  1041. c, err := NewClient(context.Background(), clientOpt)
  1042. if err != nil {
  1043. t.Fatal(err)
  1044. }
  1045. err = c.DeleteDeidentifyTemplate(context.Background(), request)
  1046. if st, ok := gstatus.FromError(err); !ok {
  1047. t.Errorf("got error %v, expected grpc error", err)
  1048. } else if c := st.Code(); c != errCode {
  1049. t.Errorf("got error code %q, want %q", c, errCode)
  1050. }
  1051. }
  1052. func TestDlpServiceCreateDlpJob(t *testing.T) {
  1053. var name string = "name3373707"
  1054. var jobTriggerName string = "jobTriggerName1819490804"
  1055. var expectedResponse = &dlppb.DlpJob{
  1056. Name: name,
  1057. JobTriggerName: jobTriggerName,
  1058. }
  1059. mockDlp.err = nil
  1060. mockDlp.reqs = nil
  1061. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1062. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1063. var request = &dlppb.CreateDlpJobRequest{
  1064. Parent: formattedParent,
  1065. }
  1066. c, err := NewClient(context.Background(), clientOpt)
  1067. if err != nil {
  1068. t.Fatal(err)
  1069. }
  1070. resp, err := c.CreateDlpJob(context.Background(), request)
  1071. if err != nil {
  1072. t.Fatal(err)
  1073. }
  1074. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1075. t.Errorf("wrong request %q, want %q", got, want)
  1076. }
  1077. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1078. t.Errorf("wrong response %q, want %q)", got, want)
  1079. }
  1080. }
  1081. func TestDlpServiceCreateDlpJobError(t *testing.T) {
  1082. errCode := codes.PermissionDenied
  1083. mockDlp.err = gstatus.Error(errCode, "test error")
  1084. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1085. var request = &dlppb.CreateDlpJobRequest{
  1086. Parent: formattedParent,
  1087. }
  1088. c, err := NewClient(context.Background(), clientOpt)
  1089. if err != nil {
  1090. t.Fatal(err)
  1091. }
  1092. resp, err := c.CreateDlpJob(context.Background(), request)
  1093. if st, ok := gstatus.FromError(err); !ok {
  1094. t.Errorf("got error %v, expected grpc error", err)
  1095. } else if c := st.Code(); c != errCode {
  1096. t.Errorf("got error code %q, want %q", c, errCode)
  1097. }
  1098. _ = resp
  1099. }
  1100. func TestDlpServiceListDlpJobs(t *testing.T) {
  1101. var nextPageToken string = ""
  1102. var jobsElement *dlppb.DlpJob = &dlppb.DlpJob{}
  1103. var jobs = []*dlppb.DlpJob{jobsElement}
  1104. var expectedResponse = &dlppb.ListDlpJobsResponse{
  1105. NextPageToken: nextPageToken,
  1106. Jobs: jobs,
  1107. }
  1108. mockDlp.err = nil
  1109. mockDlp.reqs = nil
  1110. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1111. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1112. var request = &dlppb.ListDlpJobsRequest{
  1113. Parent: formattedParent,
  1114. }
  1115. c, err := NewClient(context.Background(), clientOpt)
  1116. if err != nil {
  1117. t.Fatal(err)
  1118. }
  1119. resp, err := c.ListDlpJobs(context.Background(), request).Next()
  1120. if err != nil {
  1121. t.Fatal(err)
  1122. }
  1123. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1124. t.Errorf("wrong request %q, want %q", got, want)
  1125. }
  1126. want := (interface{})(expectedResponse.Jobs[0])
  1127. got := (interface{})(resp)
  1128. var ok bool
  1129. switch want := (want).(type) {
  1130. case proto.Message:
  1131. ok = proto.Equal(want, got.(proto.Message))
  1132. default:
  1133. ok = want == got
  1134. }
  1135. if !ok {
  1136. t.Errorf("wrong response %q, want %q)", got, want)
  1137. }
  1138. }
  1139. func TestDlpServiceListDlpJobsError(t *testing.T) {
  1140. errCode := codes.PermissionDenied
  1141. mockDlp.err = gstatus.Error(errCode, "test error")
  1142. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1143. var request = &dlppb.ListDlpJobsRequest{
  1144. Parent: formattedParent,
  1145. }
  1146. c, err := NewClient(context.Background(), clientOpt)
  1147. if err != nil {
  1148. t.Fatal(err)
  1149. }
  1150. resp, err := c.ListDlpJobs(context.Background(), request).Next()
  1151. if st, ok := gstatus.FromError(err); !ok {
  1152. t.Errorf("got error %v, expected grpc error", err)
  1153. } else if c := st.Code(); c != errCode {
  1154. t.Errorf("got error code %q, want %q", c, errCode)
  1155. }
  1156. _ = resp
  1157. }
  1158. func TestDlpServiceGetDlpJob(t *testing.T) {
  1159. var name2 string = "name2-1052831874"
  1160. var jobTriggerName string = "jobTriggerName1819490804"
  1161. var expectedResponse = &dlppb.DlpJob{
  1162. Name: name2,
  1163. JobTriggerName: jobTriggerName,
  1164. }
  1165. mockDlp.err = nil
  1166. mockDlp.reqs = nil
  1167. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1168. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1169. var request = &dlppb.GetDlpJobRequest{
  1170. Name: formattedName,
  1171. }
  1172. c, err := NewClient(context.Background(), clientOpt)
  1173. if err != nil {
  1174. t.Fatal(err)
  1175. }
  1176. resp, err := c.GetDlpJob(context.Background(), request)
  1177. if err != nil {
  1178. t.Fatal(err)
  1179. }
  1180. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1181. t.Errorf("wrong request %q, want %q", got, want)
  1182. }
  1183. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1184. t.Errorf("wrong response %q, want %q)", got, want)
  1185. }
  1186. }
  1187. func TestDlpServiceGetDlpJobError(t *testing.T) {
  1188. errCode := codes.PermissionDenied
  1189. mockDlp.err = gstatus.Error(errCode, "test error")
  1190. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1191. var request = &dlppb.GetDlpJobRequest{
  1192. Name: formattedName,
  1193. }
  1194. c, err := NewClient(context.Background(), clientOpt)
  1195. if err != nil {
  1196. t.Fatal(err)
  1197. }
  1198. resp, err := c.GetDlpJob(context.Background(), request)
  1199. if st, ok := gstatus.FromError(err); !ok {
  1200. t.Errorf("got error %v, expected grpc error", err)
  1201. } else if c := st.Code(); c != errCode {
  1202. t.Errorf("got error code %q, want %q", c, errCode)
  1203. }
  1204. _ = resp
  1205. }
  1206. func TestDlpServiceDeleteDlpJob(t *testing.T) {
  1207. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1208. mockDlp.err = nil
  1209. mockDlp.reqs = nil
  1210. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1211. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1212. var request = &dlppb.DeleteDlpJobRequest{
  1213. Name: formattedName,
  1214. }
  1215. c, err := NewClient(context.Background(), clientOpt)
  1216. if err != nil {
  1217. t.Fatal(err)
  1218. }
  1219. err = c.DeleteDlpJob(context.Background(), request)
  1220. if err != nil {
  1221. t.Fatal(err)
  1222. }
  1223. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1224. t.Errorf("wrong request %q, want %q", got, want)
  1225. }
  1226. }
  1227. func TestDlpServiceDeleteDlpJobError(t *testing.T) {
  1228. errCode := codes.PermissionDenied
  1229. mockDlp.err = gstatus.Error(errCode, "test error")
  1230. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1231. var request = &dlppb.DeleteDlpJobRequest{
  1232. Name: formattedName,
  1233. }
  1234. c, err := NewClient(context.Background(), clientOpt)
  1235. if err != nil {
  1236. t.Fatal(err)
  1237. }
  1238. err = c.DeleteDlpJob(context.Background(), request)
  1239. if st, ok := gstatus.FromError(err); !ok {
  1240. t.Errorf("got error %v, expected grpc error", err)
  1241. } else if c := st.Code(); c != errCode {
  1242. t.Errorf("got error code %q, want %q", c, errCode)
  1243. }
  1244. }
  1245. func TestDlpServiceCancelDlpJob(t *testing.T) {
  1246. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1247. mockDlp.err = nil
  1248. mockDlp.reqs = nil
  1249. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1250. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1251. var request = &dlppb.CancelDlpJobRequest{
  1252. Name: formattedName,
  1253. }
  1254. c, err := NewClient(context.Background(), clientOpt)
  1255. if err != nil {
  1256. t.Fatal(err)
  1257. }
  1258. err = c.CancelDlpJob(context.Background(), request)
  1259. if err != nil {
  1260. t.Fatal(err)
  1261. }
  1262. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1263. t.Errorf("wrong request %q, want %q", got, want)
  1264. }
  1265. }
  1266. func TestDlpServiceCancelDlpJobError(t *testing.T) {
  1267. errCode := codes.PermissionDenied
  1268. mockDlp.err = gstatus.Error(errCode, "test error")
  1269. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1270. var request = &dlppb.CancelDlpJobRequest{
  1271. Name: formattedName,
  1272. }
  1273. c, err := NewClient(context.Background(), clientOpt)
  1274. if err != nil {
  1275. t.Fatal(err)
  1276. }
  1277. err = c.CancelDlpJob(context.Background(), request)
  1278. if st, ok := gstatus.FromError(err); !ok {
  1279. t.Errorf("got error %v, expected grpc error", err)
  1280. } else if c := st.Code(); c != errCode {
  1281. t.Errorf("got error code %q, want %q", c, errCode)
  1282. }
  1283. }
  1284. func TestDlpServiceListJobTriggers(t *testing.T) {
  1285. var nextPageToken string = ""
  1286. var jobTriggersElement *dlppb.JobTrigger = &dlppb.JobTrigger{}
  1287. var jobTriggers = []*dlppb.JobTrigger{jobTriggersElement}
  1288. var expectedResponse = &dlppb.ListJobTriggersResponse{
  1289. NextPageToken: nextPageToken,
  1290. JobTriggers: jobTriggers,
  1291. }
  1292. mockDlp.err = nil
  1293. mockDlp.reqs = nil
  1294. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1295. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1296. var request = &dlppb.ListJobTriggersRequest{
  1297. Parent: formattedParent,
  1298. }
  1299. c, err := NewClient(context.Background(), clientOpt)
  1300. if err != nil {
  1301. t.Fatal(err)
  1302. }
  1303. resp, err := c.ListJobTriggers(context.Background(), request).Next()
  1304. if err != nil {
  1305. t.Fatal(err)
  1306. }
  1307. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1308. t.Errorf("wrong request %q, want %q", got, want)
  1309. }
  1310. want := (interface{})(expectedResponse.JobTriggers[0])
  1311. got := (interface{})(resp)
  1312. var ok bool
  1313. switch want := (want).(type) {
  1314. case proto.Message:
  1315. ok = proto.Equal(want, got.(proto.Message))
  1316. default:
  1317. ok = want == got
  1318. }
  1319. if !ok {
  1320. t.Errorf("wrong response %q, want %q)", got, want)
  1321. }
  1322. }
  1323. func TestDlpServiceListJobTriggersError(t *testing.T) {
  1324. errCode := codes.PermissionDenied
  1325. mockDlp.err = gstatus.Error(errCode, "test error")
  1326. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1327. var request = &dlppb.ListJobTriggersRequest{
  1328. Parent: formattedParent,
  1329. }
  1330. c, err := NewClient(context.Background(), clientOpt)
  1331. if err != nil {
  1332. t.Fatal(err)
  1333. }
  1334. resp, err := c.ListJobTriggers(context.Background(), request).Next()
  1335. if st, ok := gstatus.FromError(err); !ok {
  1336. t.Errorf("got error %v, expected grpc error", err)
  1337. } else if c := st.Code(); c != errCode {
  1338. t.Errorf("got error code %q, want %q", c, errCode)
  1339. }
  1340. _ = resp
  1341. }
  1342. func TestDlpServiceGetJobTrigger(t *testing.T) {
  1343. var name2 string = "name2-1052831874"
  1344. var displayName string = "displayName1615086568"
  1345. var description string = "description-1724546052"
  1346. var expectedResponse = &dlppb.JobTrigger{
  1347. Name: name2,
  1348. DisplayName: displayName,
  1349. Description: description,
  1350. }
  1351. mockDlp.err = nil
  1352. mockDlp.reqs = nil
  1353. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1354. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1355. var request = &dlppb.GetJobTriggerRequest{
  1356. Name: formattedName,
  1357. }
  1358. c, err := NewClient(context.Background(), clientOpt)
  1359. if err != nil {
  1360. t.Fatal(err)
  1361. }
  1362. resp, err := c.GetJobTrigger(context.Background(), request)
  1363. if err != nil {
  1364. t.Fatal(err)
  1365. }
  1366. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1367. t.Errorf("wrong request %q, want %q", got, want)
  1368. }
  1369. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1370. t.Errorf("wrong response %q, want %q)", got, want)
  1371. }
  1372. }
  1373. func TestDlpServiceGetJobTriggerError(t *testing.T) {
  1374. errCode := codes.PermissionDenied
  1375. mockDlp.err = gstatus.Error(errCode, "test error")
  1376. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1377. var request = &dlppb.GetJobTriggerRequest{
  1378. Name: formattedName,
  1379. }
  1380. c, err := NewClient(context.Background(), clientOpt)
  1381. if err != nil {
  1382. t.Fatal(err)
  1383. }
  1384. resp, err := c.GetJobTrigger(context.Background(), request)
  1385. if st, ok := gstatus.FromError(err); !ok {
  1386. t.Errorf("got error %v, expected grpc error", err)
  1387. } else if c := st.Code(); c != errCode {
  1388. t.Errorf("got error code %q, want %q", c, errCode)
  1389. }
  1390. _ = resp
  1391. }
  1392. func TestDlpServiceDeleteJobTrigger(t *testing.T) {
  1393. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1394. mockDlp.err = nil
  1395. mockDlp.reqs = nil
  1396. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1397. var name string = "name3373707"
  1398. var request = &dlppb.DeleteJobTriggerRequest{
  1399. Name: name,
  1400. }
  1401. c, err := NewClient(context.Background(), clientOpt)
  1402. if err != nil {
  1403. t.Fatal(err)
  1404. }
  1405. err = c.DeleteJobTrigger(context.Background(), request)
  1406. if err != nil {
  1407. t.Fatal(err)
  1408. }
  1409. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1410. t.Errorf("wrong request %q, want %q", got, want)
  1411. }
  1412. }
  1413. func TestDlpServiceDeleteJobTriggerError(t *testing.T) {
  1414. errCode := codes.PermissionDenied
  1415. mockDlp.err = gstatus.Error(errCode, "test error")
  1416. var name string = "name3373707"
  1417. var request = &dlppb.DeleteJobTriggerRequest{
  1418. Name: name,
  1419. }
  1420. c, err := NewClient(context.Background(), clientOpt)
  1421. if err != nil {
  1422. t.Fatal(err)
  1423. }
  1424. err = c.DeleteJobTrigger(context.Background(), request)
  1425. if st, ok := gstatus.FromError(err); !ok {
  1426. t.Errorf("got error %v, expected grpc error", err)
  1427. } else if c := st.Code(); c != errCode {
  1428. t.Errorf("got error code %q, want %q", c, errCode)
  1429. }
  1430. }
  1431. func TestDlpServiceUpdateJobTrigger(t *testing.T) {
  1432. var name2 string = "name2-1052831874"
  1433. var displayName string = "displayName1615086568"
  1434. var description string = "description-1724546052"
  1435. var expectedResponse = &dlppb.JobTrigger{
  1436. Name: name2,
  1437. DisplayName: displayName,
  1438. Description: description,
  1439. }
  1440. mockDlp.err = nil
  1441. mockDlp.reqs = nil
  1442. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1443. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1444. var request = &dlppb.UpdateJobTriggerRequest{
  1445. Name: formattedName,
  1446. }
  1447. c, err := NewClient(context.Background(), clientOpt)
  1448. if err != nil {
  1449. t.Fatal(err)
  1450. }
  1451. resp, err := c.UpdateJobTrigger(context.Background(), request)
  1452. if err != nil {
  1453. t.Fatal(err)
  1454. }
  1455. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1456. t.Errorf("wrong request %q, want %q", got, want)
  1457. }
  1458. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1459. t.Errorf("wrong response %q, want %q)", got, want)
  1460. }
  1461. }
  1462. func TestDlpServiceUpdateJobTriggerError(t *testing.T) {
  1463. errCode := codes.PermissionDenied
  1464. mockDlp.err = gstatus.Error(errCode, "test error")
  1465. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1466. var request = &dlppb.UpdateJobTriggerRequest{
  1467. Name: formattedName,
  1468. }
  1469. c, err := NewClient(context.Background(), clientOpt)
  1470. if err != nil {
  1471. t.Fatal(err)
  1472. }
  1473. resp, err := c.UpdateJobTrigger(context.Background(), request)
  1474. if st, ok := gstatus.FromError(err); !ok {
  1475. t.Errorf("got error %v, expected grpc error", err)
  1476. } else if c := st.Code(); c != errCode {
  1477. t.Errorf("got error code %q, want %q", c, errCode)
  1478. }
  1479. _ = resp
  1480. }
  1481. func TestDlpServiceCreateJobTrigger(t *testing.T) {
  1482. var name string = "name3373707"
  1483. var displayName string = "displayName1615086568"
  1484. var description string = "description-1724546052"
  1485. var expectedResponse = &dlppb.JobTrigger{
  1486. Name: name,
  1487. DisplayName: displayName,
  1488. Description: description,
  1489. }
  1490. mockDlp.err = nil
  1491. mockDlp.reqs = nil
  1492. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1493. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1494. var request = &dlppb.CreateJobTriggerRequest{
  1495. Parent: formattedParent,
  1496. }
  1497. c, err := NewClient(context.Background(), clientOpt)
  1498. if err != nil {
  1499. t.Fatal(err)
  1500. }
  1501. resp, err := c.CreateJobTrigger(context.Background(), request)
  1502. if err != nil {
  1503. t.Fatal(err)
  1504. }
  1505. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1506. t.Errorf("wrong request %q, want %q", got, want)
  1507. }
  1508. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1509. t.Errorf("wrong response %q, want %q)", got, want)
  1510. }
  1511. }
  1512. func TestDlpServiceCreateJobTriggerError(t *testing.T) {
  1513. errCode := codes.PermissionDenied
  1514. mockDlp.err = gstatus.Error(errCode, "test error")
  1515. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1516. var request = &dlppb.CreateJobTriggerRequest{
  1517. Parent: formattedParent,
  1518. }
  1519. c, err := NewClient(context.Background(), clientOpt)
  1520. if err != nil {
  1521. t.Fatal(err)
  1522. }
  1523. resp, err := c.CreateJobTrigger(context.Background(), request)
  1524. if st, ok := gstatus.FromError(err); !ok {
  1525. t.Errorf("got error %v, expected grpc error", err)
  1526. } else if c := st.Code(); c != errCode {
  1527. t.Errorf("got error code %q, want %q", c, errCode)
  1528. }
  1529. _ = resp
  1530. }