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.
 
 
 

2264 lines
63 KiB

  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. // Code generated by gapic-generator. 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. "context"
  22. "flag"
  23. "fmt"
  24. "io"
  25. "log"
  26. "net"
  27. "os"
  28. "strings"
  29. "testing"
  30. "github.com/golang/protobuf/proto"
  31. "github.com/golang/protobuf/ptypes"
  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. func (s *mockDlpServer) CreateStoredInfoType(ctx context.Context, req *dlppb.CreateStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) {
  329. md, _ := metadata.FromIncomingContext(ctx)
  330. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  331. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  332. }
  333. s.reqs = append(s.reqs, req)
  334. if s.err != nil {
  335. return nil, s.err
  336. }
  337. return s.resps[0].(*dlppb.StoredInfoType), nil
  338. }
  339. func (s *mockDlpServer) UpdateStoredInfoType(ctx context.Context, req *dlppb.UpdateStoredInfoTypeRequest) (*dlppb.StoredInfoType, 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].(*dlppb.StoredInfoType), nil
  349. }
  350. func (s *mockDlpServer) GetStoredInfoType(ctx context.Context, req *dlppb.GetStoredInfoTypeRequest) (*dlppb.StoredInfoType, error) {
  351. md, _ := metadata.FromIncomingContext(ctx)
  352. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  353. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  354. }
  355. s.reqs = append(s.reqs, req)
  356. if s.err != nil {
  357. return nil, s.err
  358. }
  359. return s.resps[0].(*dlppb.StoredInfoType), nil
  360. }
  361. func (s *mockDlpServer) ListStoredInfoTypes(ctx context.Context, req *dlppb.ListStoredInfoTypesRequest) (*dlppb.ListStoredInfoTypesResponse, error) {
  362. md, _ := metadata.FromIncomingContext(ctx)
  363. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  364. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  365. }
  366. s.reqs = append(s.reqs, req)
  367. if s.err != nil {
  368. return nil, s.err
  369. }
  370. return s.resps[0].(*dlppb.ListStoredInfoTypesResponse), nil
  371. }
  372. func (s *mockDlpServer) DeleteStoredInfoType(ctx context.Context, req *dlppb.DeleteStoredInfoTypeRequest) (*emptypb.Empty, error) {
  373. md, _ := metadata.FromIncomingContext(ctx)
  374. if xg := md["x-goog-api-client"]; len(xg) == 0 || !strings.Contains(xg[0], "gl-go/") {
  375. return nil, fmt.Errorf("x-goog-api-client = %v, expected gl-go key", xg)
  376. }
  377. s.reqs = append(s.reqs, req)
  378. if s.err != nil {
  379. return nil, s.err
  380. }
  381. return s.resps[0].(*emptypb.Empty), nil
  382. }
  383. // clientOpt is the option tests should use to connect to the test server.
  384. // It is initialized by TestMain.
  385. var clientOpt option.ClientOption
  386. var (
  387. mockDlp mockDlpServer
  388. )
  389. func TestMain(m *testing.M) {
  390. flag.Parse()
  391. serv := grpc.NewServer()
  392. dlppb.RegisterDlpServiceServer(serv, &mockDlp)
  393. lis, err := net.Listen("tcp", "localhost:0")
  394. if err != nil {
  395. log.Fatal(err)
  396. }
  397. go serv.Serve(lis)
  398. conn, err := grpc.Dial(lis.Addr().String(), grpc.WithInsecure())
  399. if err != nil {
  400. log.Fatal(err)
  401. }
  402. clientOpt = option.WithGRPCConn(conn)
  403. os.Exit(m.Run())
  404. }
  405. func TestDlpServiceInspectContent(t *testing.T) {
  406. var expectedResponse *dlppb.InspectContentResponse = &dlppb.InspectContentResponse{}
  407. mockDlp.err = nil
  408. mockDlp.reqs = nil
  409. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  410. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  411. var request = &dlppb.InspectContentRequest{
  412. Parent: formattedParent,
  413. }
  414. c, err := NewClient(context.Background(), clientOpt)
  415. if err != nil {
  416. t.Fatal(err)
  417. }
  418. resp, err := c.InspectContent(context.Background(), request)
  419. if err != nil {
  420. t.Fatal(err)
  421. }
  422. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  423. t.Errorf("wrong request %q, want %q", got, want)
  424. }
  425. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  426. t.Errorf("wrong response %q, want %q)", got, want)
  427. }
  428. }
  429. func TestDlpServiceInspectContentError(t *testing.T) {
  430. errCode := codes.PermissionDenied
  431. mockDlp.err = gstatus.Error(errCode, "test error")
  432. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  433. var request = &dlppb.InspectContentRequest{
  434. Parent: formattedParent,
  435. }
  436. c, err := NewClient(context.Background(), clientOpt)
  437. if err != nil {
  438. t.Fatal(err)
  439. }
  440. resp, err := c.InspectContent(context.Background(), request)
  441. if st, ok := gstatus.FromError(err); !ok {
  442. t.Errorf("got error %v, expected grpc error", err)
  443. } else if c := st.Code(); c != errCode {
  444. t.Errorf("got error code %q, want %q", c, errCode)
  445. }
  446. _ = resp
  447. }
  448. func TestDlpServiceRedactImage(t *testing.T) {
  449. var redactedImage []byte = []byte("28")
  450. var extractedText string = "extractedText998260012"
  451. var expectedResponse = &dlppb.RedactImageResponse{
  452. RedactedImage: redactedImage,
  453. ExtractedText: extractedText,
  454. }
  455. mockDlp.err = nil
  456. mockDlp.reqs = nil
  457. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  458. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  459. var request = &dlppb.RedactImageRequest{
  460. Parent: formattedParent,
  461. }
  462. c, err := NewClient(context.Background(), clientOpt)
  463. if err != nil {
  464. t.Fatal(err)
  465. }
  466. resp, err := c.RedactImage(context.Background(), request)
  467. if err != nil {
  468. t.Fatal(err)
  469. }
  470. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  471. t.Errorf("wrong request %q, want %q", got, want)
  472. }
  473. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  474. t.Errorf("wrong response %q, want %q)", got, want)
  475. }
  476. }
  477. func TestDlpServiceRedactImageError(t *testing.T) {
  478. errCode := codes.PermissionDenied
  479. mockDlp.err = gstatus.Error(errCode, "test error")
  480. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  481. var request = &dlppb.RedactImageRequest{
  482. Parent: formattedParent,
  483. }
  484. c, err := NewClient(context.Background(), clientOpt)
  485. if err != nil {
  486. t.Fatal(err)
  487. }
  488. resp, err := c.RedactImage(context.Background(), request)
  489. if st, ok := gstatus.FromError(err); !ok {
  490. t.Errorf("got error %v, expected grpc error", err)
  491. } else if c := st.Code(); c != errCode {
  492. t.Errorf("got error code %q, want %q", c, errCode)
  493. }
  494. _ = resp
  495. }
  496. func TestDlpServiceDeidentifyContent(t *testing.T) {
  497. var expectedResponse *dlppb.DeidentifyContentResponse = &dlppb.DeidentifyContentResponse{}
  498. mockDlp.err = nil
  499. mockDlp.reqs = nil
  500. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  501. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  502. var request = &dlppb.DeidentifyContentRequest{
  503. Parent: formattedParent,
  504. }
  505. c, err := NewClient(context.Background(), clientOpt)
  506. if err != nil {
  507. t.Fatal(err)
  508. }
  509. resp, err := c.DeidentifyContent(context.Background(), request)
  510. if err != nil {
  511. t.Fatal(err)
  512. }
  513. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  514. t.Errorf("wrong request %q, want %q", got, want)
  515. }
  516. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  517. t.Errorf("wrong response %q, want %q)", got, want)
  518. }
  519. }
  520. func TestDlpServiceDeidentifyContentError(t *testing.T) {
  521. errCode := codes.PermissionDenied
  522. mockDlp.err = gstatus.Error(errCode, "test error")
  523. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  524. var request = &dlppb.DeidentifyContentRequest{
  525. Parent: formattedParent,
  526. }
  527. c, err := NewClient(context.Background(), clientOpt)
  528. if err != nil {
  529. t.Fatal(err)
  530. }
  531. resp, err := c.DeidentifyContent(context.Background(), request)
  532. if st, ok := gstatus.FromError(err); !ok {
  533. t.Errorf("got error %v, expected grpc error", err)
  534. } else if c := st.Code(); c != errCode {
  535. t.Errorf("got error code %q, want %q", c, errCode)
  536. }
  537. _ = resp
  538. }
  539. func TestDlpServiceReidentifyContent(t *testing.T) {
  540. var expectedResponse *dlppb.ReidentifyContentResponse = &dlppb.ReidentifyContentResponse{}
  541. mockDlp.err = nil
  542. mockDlp.reqs = nil
  543. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  544. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  545. var request = &dlppb.ReidentifyContentRequest{
  546. Parent: formattedParent,
  547. }
  548. c, err := NewClient(context.Background(), clientOpt)
  549. if err != nil {
  550. t.Fatal(err)
  551. }
  552. resp, err := c.ReidentifyContent(context.Background(), request)
  553. if err != nil {
  554. t.Fatal(err)
  555. }
  556. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  557. t.Errorf("wrong request %q, want %q", got, want)
  558. }
  559. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  560. t.Errorf("wrong response %q, want %q)", got, want)
  561. }
  562. }
  563. func TestDlpServiceReidentifyContentError(t *testing.T) {
  564. errCode := codes.PermissionDenied
  565. mockDlp.err = gstatus.Error(errCode, "test error")
  566. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  567. var request = &dlppb.ReidentifyContentRequest{
  568. Parent: formattedParent,
  569. }
  570. c, err := NewClient(context.Background(), clientOpt)
  571. if err != nil {
  572. t.Fatal(err)
  573. }
  574. resp, err := c.ReidentifyContent(context.Background(), request)
  575. if st, ok := gstatus.FromError(err); !ok {
  576. t.Errorf("got error %v, expected grpc error", err)
  577. } else if c := st.Code(); c != errCode {
  578. t.Errorf("got error code %q, want %q", c, errCode)
  579. }
  580. _ = resp
  581. }
  582. func TestDlpServiceListInfoTypes(t *testing.T) {
  583. var expectedResponse *dlppb.ListInfoTypesResponse = &dlppb.ListInfoTypesResponse{}
  584. mockDlp.err = nil
  585. mockDlp.reqs = nil
  586. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  587. var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{}
  588. c, err := NewClient(context.Background(), clientOpt)
  589. if err != nil {
  590. t.Fatal(err)
  591. }
  592. resp, err := c.ListInfoTypes(context.Background(), request)
  593. if err != nil {
  594. t.Fatal(err)
  595. }
  596. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  597. t.Errorf("wrong request %q, want %q", got, want)
  598. }
  599. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  600. t.Errorf("wrong response %q, want %q)", got, want)
  601. }
  602. }
  603. func TestDlpServiceListInfoTypesError(t *testing.T) {
  604. errCode := codes.PermissionDenied
  605. mockDlp.err = gstatus.Error(errCode, "test error")
  606. var request *dlppb.ListInfoTypesRequest = &dlppb.ListInfoTypesRequest{}
  607. c, err := NewClient(context.Background(), clientOpt)
  608. if err != nil {
  609. t.Fatal(err)
  610. }
  611. resp, err := c.ListInfoTypes(context.Background(), request)
  612. if st, ok := gstatus.FromError(err); !ok {
  613. t.Errorf("got error %v, expected grpc error", err)
  614. } else if c := st.Code(); c != errCode {
  615. t.Errorf("got error code %q, want %q", c, errCode)
  616. }
  617. _ = resp
  618. }
  619. func TestDlpServiceCreateInspectTemplate(t *testing.T) {
  620. var name string = "name3373707"
  621. var displayName string = "displayName1615086568"
  622. var description string = "description-1724546052"
  623. var expectedResponse = &dlppb.InspectTemplate{
  624. Name: name,
  625. DisplayName: displayName,
  626. Description: description,
  627. }
  628. mockDlp.err = nil
  629. mockDlp.reqs = nil
  630. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  631. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  632. var request = &dlppb.CreateInspectTemplateRequest{
  633. Parent: formattedParent,
  634. }
  635. c, err := NewClient(context.Background(), clientOpt)
  636. if err != nil {
  637. t.Fatal(err)
  638. }
  639. resp, err := c.CreateInspectTemplate(context.Background(), request)
  640. if err != nil {
  641. t.Fatal(err)
  642. }
  643. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  644. t.Errorf("wrong request %q, want %q", got, want)
  645. }
  646. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  647. t.Errorf("wrong response %q, want %q)", got, want)
  648. }
  649. }
  650. func TestDlpServiceCreateInspectTemplateError(t *testing.T) {
  651. errCode := codes.PermissionDenied
  652. mockDlp.err = gstatus.Error(errCode, "test error")
  653. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  654. var request = &dlppb.CreateInspectTemplateRequest{
  655. Parent: formattedParent,
  656. }
  657. c, err := NewClient(context.Background(), clientOpt)
  658. if err != nil {
  659. t.Fatal(err)
  660. }
  661. resp, err := c.CreateInspectTemplate(context.Background(), request)
  662. if st, ok := gstatus.FromError(err); !ok {
  663. t.Errorf("got error %v, expected grpc error", err)
  664. } else if c := st.Code(); c != errCode {
  665. t.Errorf("got error code %q, want %q", c, errCode)
  666. }
  667. _ = resp
  668. }
  669. func TestDlpServiceUpdateInspectTemplate(t *testing.T) {
  670. var name2 string = "name2-1052831874"
  671. var displayName string = "displayName1615086568"
  672. var description string = "description-1724546052"
  673. var expectedResponse = &dlppb.InspectTemplate{
  674. Name: name2,
  675. DisplayName: displayName,
  676. Description: description,
  677. }
  678. mockDlp.err = nil
  679. mockDlp.reqs = nil
  680. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  681. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  682. var request = &dlppb.UpdateInspectTemplateRequest{
  683. Name: formattedName,
  684. }
  685. c, err := NewClient(context.Background(), clientOpt)
  686. if err != nil {
  687. t.Fatal(err)
  688. }
  689. resp, err := c.UpdateInspectTemplate(context.Background(), request)
  690. if err != nil {
  691. t.Fatal(err)
  692. }
  693. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  694. t.Errorf("wrong request %q, want %q", got, want)
  695. }
  696. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  697. t.Errorf("wrong response %q, want %q)", got, want)
  698. }
  699. }
  700. func TestDlpServiceUpdateInspectTemplateError(t *testing.T) {
  701. errCode := codes.PermissionDenied
  702. mockDlp.err = gstatus.Error(errCode, "test error")
  703. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  704. var request = &dlppb.UpdateInspectTemplateRequest{
  705. Name: formattedName,
  706. }
  707. c, err := NewClient(context.Background(), clientOpt)
  708. if err != nil {
  709. t.Fatal(err)
  710. }
  711. resp, err := c.UpdateInspectTemplate(context.Background(), request)
  712. if st, ok := gstatus.FromError(err); !ok {
  713. t.Errorf("got error %v, expected grpc error", err)
  714. } else if c := st.Code(); c != errCode {
  715. t.Errorf("got error code %q, want %q", c, errCode)
  716. }
  717. _ = resp
  718. }
  719. func TestDlpServiceGetInspectTemplate(t *testing.T) {
  720. var name string = "name3373707"
  721. var displayName string = "displayName1615086568"
  722. var description string = "description-1724546052"
  723. var expectedResponse = &dlppb.InspectTemplate{
  724. Name: name,
  725. DisplayName: displayName,
  726. Description: description,
  727. }
  728. mockDlp.err = nil
  729. mockDlp.reqs = nil
  730. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  731. var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{}
  732. c, err := NewClient(context.Background(), clientOpt)
  733. if err != nil {
  734. t.Fatal(err)
  735. }
  736. resp, err := c.GetInspectTemplate(context.Background(), request)
  737. if err != nil {
  738. t.Fatal(err)
  739. }
  740. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  741. t.Errorf("wrong request %q, want %q", got, want)
  742. }
  743. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  744. t.Errorf("wrong response %q, want %q)", got, want)
  745. }
  746. }
  747. func TestDlpServiceGetInspectTemplateError(t *testing.T) {
  748. errCode := codes.PermissionDenied
  749. mockDlp.err = gstatus.Error(errCode, "test error")
  750. var request *dlppb.GetInspectTemplateRequest = &dlppb.GetInspectTemplateRequest{}
  751. c, err := NewClient(context.Background(), clientOpt)
  752. if err != nil {
  753. t.Fatal(err)
  754. }
  755. resp, err := c.GetInspectTemplate(context.Background(), request)
  756. if st, ok := gstatus.FromError(err); !ok {
  757. t.Errorf("got error %v, expected grpc error", err)
  758. } else if c := st.Code(); c != errCode {
  759. t.Errorf("got error code %q, want %q", c, errCode)
  760. }
  761. _ = resp
  762. }
  763. func TestDlpServiceListInspectTemplates(t *testing.T) {
  764. var nextPageToken string = ""
  765. var inspectTemplatesElement *dlppb.InspectTemplate = &dlppb.InspectTemplate{}
  766. var inspectTemplates = []*dlppb.InspectTemplate{inspectTemplatesElement}
  767. var expectedResponse = &dlppb.ListInspectTemplatesResponse{
  768. NextPageToken: nextPageToken,
  769. InspectTemplates: inspectTemplates,
  770. }
  771. mockDlp.err = nil
  772. mockDlp.reqs = nil
  773. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  774. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  775. var request = &dlppb.ListInspectTemplatesRequest{
  776. Parent: formattedParent,
  777. }
  778. c, err := NewClient(context.Background(), clientOpt)
  779. if err != nil {
  780. t.Fatal(err)
  781. }
  782. resp, err := c.ListInspectTemplates(context.Background(), request).Next()
  783. if err != nil {
  784. t.Fatal(err)
  785. }
  786. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  787. t.Errorf("wrong request %q, want %q", got, want)
  788. }
  789. want := (interface{})(expectedResponse.InspectTemplates[0])
  790. got := (interface{})(resp)
  791. var ok bool
  792. switch want := (want).(type) {
  793. case proto.Message:
  794. ok = proto.Equal(want, got.(proto.Message))
  795. default:
  796. ok = want == got
  797. }
  798. if !ok {
  799. t.Errorf("wrong response %q, want %q)", got, want)
  800. }
  801. }
  802. func TestDlpServiceListInspectTemplatesError(t *testing.T) {
  803. errCode := codes.PermissionDenied
  804. mockDlp.err = gstatus.Error(errCode, "test error")
  805. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  806. var request = &dlppb.ListInspectTemplatesRequest{
  807. Parent: formattedParent,
  808. }
  809. c, err := NewClient(context.Background(), clientOpt)
  810. if err != nil {
  811. t.Fatal(err)
  812. }
  813. resp, err := c.ListInspectTemplates(context.Background(), request).Next()
  814. if st, ok := gstatus.FromError(err); !ok {
  815. t.Errorf("got error %v, expected grpc error", err)
  816. } else if c := st.Code(); c != errCode {
  817. t.Errorf("got error code %q, want %q", c, errCode)
  818. }
  819. _ = resp
  820. }
  821. func TestDlpServiceDeleteInspectTemplate(t *testing.T) {
  822. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  823. mockDlp.err = nil
  824. mockDlp.reqs = nil
  825. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  826. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  827. var request = &dlppb.DeleteInspectTemplateRequest{
  828. Name: formattedName,
  829. }
  830. c, err := NewClient(context.Background(), clientOpt)
  831. if err != nil {
  832. t.Fatal(err)
  833. }
  834. err = c.DeleteInspectTemplate(context.Background(), request)
  835. if err != nil {
  836. t.Fatal(err)
  837. }
  838. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  839. t.Errorf("wrong request %q, want %q", got, want)
  840. }
  841. }
  842. func TestDlpServiceDeleteInspectTemplateError(t *testing.T) {
  843. errCode := codes.PermissionDenied
  844. mockDlp.err = gstatus.Error(errCode, "test error")
  845. var formattedName string = fmt.Sprintf("organizations/%s/inspectTemplates/%s", "[ORGANIZATION]", "[INSPECT_TEMPLATE]")
  846. var request = &dlppb.DeleteInspectTemplateRequest{
  847. Name: formattedName,
  848. }
  849. c, err := NewClient(context.Background(), clientOpt)
  850. if err != nil {
  851. t.Fatal(err)
  852. }
  853. err = c.DeleteInspectTemplate(context.Background(), request)
  854. if st, ok := gstatus.FromError(err); !ok {
  855. t.Errorf("got error %v, expected grpc error", err)
  856. } else if c := st.Code(); c != errCode {
  857. t.Errorf("got error code %q, want %q", c, errCode)
  858. }
  859. }
  860. func TestDlpServiceCreateDeidentifyTemplate(t *testing.T) {
  861. var name string = "name3373707"
  862. var displayName string = "displayName1615086568"
  863. var description string = "description-1724546052"
  864. var expectedResponse = &dlppb.DeidentifyTemplate{
  865. Name: name,
  866. DisplayName: displayName,
  867. Description: description,
  868. }
  869. mockDlp.err = nil
  870. mockDlp.reqs = nil
  871. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  872. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  873. var request = &dlppb.CreateDeidentifyTemplateRequest{
  874. Parent: formattedParent,
  875. }
  876. c, err := NewClient(context.Background(), clientOpt)
  877. if err != nil {
  878. t.Fatal(err)
  879. }
  880. resp, err := c.CreateDeidentifyTemplate(context.Background(), request)
  881. if err != nil {
  882. t.Fatal(err)
  883. }
  884. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  885. t.Errorf("wrong request %q, want %q", got, want)
  886. }
  887. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  888. t.Errorf("wrong response %q, want %q)", got, want)
  889. }
  890. }
  891. func TestDlpServiceCreateDeidentifyTemplateError(t *testing.T) {
  892. errCode := codes.PermissionDenied
  893. mockDlp.err = gstatus.Error(errCode, "test error")
  894. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  895. var request = &dlppb.CreateDeidentifyTemplateRequest{
  896. Parent: formattedParent,
  897. }
  898. c, err := NewClient(context.Background(), clientOpt)
  899. if err != nil {
  900. t.Fatal(err)
  901. }
  902. resp, err := c.CreateDeidentifyTemplate(context.Background(), request)
  903. if st, ok := gstatus.FromError(err); !ok {
  904. t.Errorf("got error %v, expected grpc error", err)
  905. } else if c := st.Code(); c != errCode {
  906. t.Errorf("got error code %q, want %q", c, errCode)
  907. }
  908. _ = resp
  909. }
  910. func TestDlpServiceUpdateDeidentifyTemplate(t *testing.T) {
  911. var name2 string = "name2-1052831874"
  912. var displayName string = "displayName1615086568"
  913. var description string = "description-1724546052"
  914. var expectedResponse = &dlppb.DeidentifyTemplate{
  915. Name: name2,
  916. DisplayName: displayName,
  917. Description: description,
  918. }
  919. mockDlp.err = nil
  920. mockDlp.reqs = nil
  921. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  922. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  923. var request = &dlppb.UpdateDeidentifyTemplateRequest{
  924. Name: formattedName,
  925. }
  926. c, err := NewClient(context.Background(), clientOpt)
  927. if err != nil {
  928. t.Fatal(err)
  929. }
  930. resp, err := c.UpdateDeidentifyTemplate(context.Background(), request)
  931. if err != nil {
  932. t.Fatal(err)
  933. }
  934. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  935. t.Errorf("wrong request %q, want %q", got, want)
  936. }
  937. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  938. t.Errorf("wrong response %q, want %q)", got, want)
  939. }
  940. }
  941. func TestDlpServiceUpdateDeidentifyTemplateError(t *testing.T) {
  942. errCode := codes.PermissionDenied
  943. mockDlp.err = gstatus.Error(errCode, "test error")
  944. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  945. var request = &dlppb.UpdateDeidentifyTemplateRequest{
  946. Name: formattedName,
  947. }
  948. c, err := NewClient(context.Background(), clientOpt)
  949. if err != nil {
  950. t.Fatal(err)
  951. }
  952. resp, err := c.UpdateDeidentifyTemplate(context.Background(), request)
  953. if st, ok := gstatus.FromError(err); !ok {
  954. t.Errorf("got error %v, expected grpc error", err)
  955. } else if c := st.Code(); c != errCode {
  956. t.Errorf("got error code %q, want %q", c, errCode)
  957. }
  958. _ = resp
  959. }
  960. func TestDlpServiceGetDeidentifyTemplate(t *testing.T) {
  961. var name2 string = "name2-1052831874"
  962. var displayName string = "displayName1615086568"
  963. var description string = "description-1724546052"
  964. var expectedResponse = &dlppb.DeidentifyTemplate{
  965. Name: name2,
  966. DisplayName: displayName,
  967. Description: description,
  968. }
  969. mockDlp.err = nil
  970. mockDlp.reqs = nil
  971. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  972. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  973. var request = &dlppb.GetDeidentifyTemplateRequest{
  974. Name: formattedName,
  975. }
  976. c, err := NewClient(context.Background(), clientOpt)
  977. if err != nil {
  978. t.Fatal(err)
  979. }
  980. resp, err := c.GetDeidentifyTemplate(context.Background(), request)
  981. if err != nil {
  982. t.Fatal(err)
  983. }
  984. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  985. t.Errorf("wrong request %q, want %q", got, want)
  986. }
  987. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  988. t.Errorf("wrong response %q, want %q)", got, want)
  989. }
  990. }
  991. func TestDlpServiceGetDeidentifyTemplateError(t *testing.T) {
  992. errCode := codes.PermissionDenied
  993. mockDlp.err = gstatus.Error(errCode, "test error")
  994. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  995. var request = &dlppb.GetDeidentifyTemplateRequest{
  996. Name: formattedName,
  997. }
  998. c, err := NewClient(context.Background(), clientOpt)
  999. if err != nil {
  1000. t.Fatal(err)
  1001. }
  1002. resp, err := c.GetDeidentifyTemplate(context.Background(), request)
  1003. if st, ok := gstatus.FromError(err); !ok {
  1004. t.Errorf("got error %v, expected grpc error", err)
  1005. } else if c := st.Code(); c != errCode {
  1006. t.Errorf("got error code %q, want %q", c, errCode)
  1007. }
  1008. _ = resp
  1009. }
  1010. func TestDlpServiceListDeidentifyTemplates(t *testing.T) {
  1011. var nextPageToken string = ""
  1012. var deidentifyTemplatesElement *dlppb.DeidentifyTemplate = &dlppb.DeidentifyTemplate{}
  1013. var deidentifyTemplates = []*dlppb.DeidentifyTemplate{deidentifyTemplatesElement}
  1014. var expectedResponse = &dlppb.ListDeidentifyTemplatesResponse{
  1015. NextPageToken: nextPageToken,
  1016. DeidentifyTemplates: deidentifyTemplates,
  1017. }
  1018. mockDlp.err = nil
  1019. mockDlp.reqs = nil
  1020. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1021. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  1022. var request = &dlppb.ListDeidentifyTemplatesRequest{
  1023. Parent: formattedParent,
  1024. }
  1025. c, err := NewClient(context.Background(), clientOpt)
  1026. if err != nil {
  1027. t.Fatal(err)
  1028. }
  1029. resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next()
  1030. if err != nil {
  1031. t.Fatal(err)
  1032. }
  1033. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1034. t.Errorf("wrong request %q, want %q", got, want)
  1035. }
  1036. want := (interface{})(expectedResponse.DeidentifyTemplates[0])
  1037. got := (interface{})(resp)
  1038. var ok bool
  1039. switch want := (want).(type) {
  1040. case proto.Message:
  1041. ok = proto.Equal(want, got.(proto.Message))
  1042. default:
  1043. ok = want == got
  1044. }
  1045. if !ok {
  1046. t.Errorf("wrong response %q, want %q)", got, want)
  1047. }
  1048. }
  1049. func TestDlpServiceListDeidentifyTemplatesError(t *testing.T) {
  1050. errCode := codes.PermissionDenied
  1051. mockDlp.err = gstatus.Error(errCode, "test error")
  1052. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  1053. var request = &dlppb.ListDeidentifyTemplatesRequest{
  1054. Parent: formattedParent,
  1055. }
  1056. c, err := NewClient(context.Background(), clientOpt)
  1057. if err != nil {
  1058. t.Fatal(err)
  1059. }
  1060. resp, err := c.ListDeidentifyTemplates(context.Background(), request).Next()
  1061. if st, ok := gstatus.FromError(err); !ok {
  1062. t.Errorf("got error %v, expected grpc error", err)
  1063. } else if c := st.Code(); c != errCode {
  1064. t.Errorf("got error code %q, want %q", c, errCode)
  1065. }
  1066. _ = resp
  1067. }
  1068. func TestDlpServiceDeleteDeidentifyTemplate(t *testing.T) {
  1069. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1070. mockDlp.err = nil
  1071. mockDlp.reqs = nil
  1072. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1073. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  1074. var request = &dlppb.DeleteDeidentifyTemplateRequest{
  1075. Name: formattedName,
  1076. }
  1077. c, err := NewClient(context.Background(), clientOpt)
  1078. if err != nil {
  1079. t.Fatal(err)
  1080. }
  1081. err = c.DeleteDeidentifyTemplate(context.Background(), request)
  1082. if err != nil {
  1083. t.Fatal(err)
  1084. }
  1085. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1086. t.Errorf("wrong request %q, want %q", got, want)
  1087. }
  1088. }
  1089. func TestDlpServiceDeleteDeidentifyTemplateError(t *testing.T) {
  1090. errCode := codes.PermissionDenied
  1091. mockDlp.err = gstatus.Error(errCode, "test error")
  1092. var formattedName string = fmt.Sprintf("organizations/%s/deidentifyTemplates/%s", "[ORGANIZATION]", "[DEIDENTIFY_TEMPLATE]")
  1093. var request = &dlppb.DeleteDeidentifyTemplateRequest{
  1094. Name: formattedName,
  1095. }
  1096. c, err := NewClient(context.Background(), clientOpt)
  1097. if err != nil {
  1098. t.Fatal(err)
  1099. }
  1100. err = c.DeleteDeidentifyTemplate(context.Background(), request)
  1101. if st, ok := gstatus.FromError(err); !ok {
  1102. t.Errorf("got error %v, expected grpc error", err)
  1103. } else if c := st.Code(); c != errCode {
  1104. t.Errorf("got error code %q, want %q", c, errCode)
  1105. }
  1106. }
  1107. func TestDlpServiceCreateDlpJob(t *testing.T) {
  1108. var name string = "name3373707"
  1109. var jobTriggerName string = "jobTriggerName1819490804"
  1110. var expectedResponse = &dlppb.DlpJob{
  1111. Name: name,
  1112. JobTriggerName: jobTriggerName,
  1113. }
  1114. mockDlp.err = nil
  1115. mockDlp.reqs = nil
  1116. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1117. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1118. var request = &dlppb.CreateDlpJobRequest{
  1119. Parent: formattedParent,
  1120. }
  1121. c, err := NewClient(context.Background(), clientOpt)
  1122. if err != nil {
  1123. t.Fatal(err)
  1124. }
  1125. resp, err := c.CreateDlpJob(context.Background(), request)
  1126. if err != nil {
  1127. t.Fatal(err)
  1128. }
  1129. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1130. t.Errorf("wrong request %q, want %q", got, want)
  1131. }
  1132. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1133. t.Errorf("wrong response %q, want %q)", got, want)
  1134. }
  1135. }
  1136. func TestDlpServiceCreateDlpJobError(t *testing.T) {
  1137. errCode := codes.PermissionDenied
  1138. mockDlp.err = gstatus.Error(errCode, "test error")
  1139. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1140. var request = &dlppb.CreateDlpJobRequest{
  1141. Parent: formattedParent,
  1142. }
  1143. c, err := NewClient(context.Background(), clientOpt)
  1144. if err != nil {
  1145. t.Fatal(err)
  1146. }
  1147. resp, err := c.CreateDlpJob(context.Background(), request)
  1148. if st, ok := gstatus.FromError(err); !ok {
  1149. t.Errorf("got error %v, expected grpc error", err)
  1150. } else if c := st.Code(); c != errCode {
  1151. t.Errorf("got error code %q, want %q", c, errCode)
  1152. }
  1153. _ = resp
  1154. }
  1155. func TestDlpServiceListDlpJobs(t *testing.T) {
  1156. var nextPageToken string = ""
  1157. var jobsElement *dlppb.DlpJob = &dlppb.DlpJob{}
  1158. var jobs = []*dlppb.DlpJob{jobsElement}
  1159. var expectedResponse = &dlppb.ListDlpJobsResponse{
  1160. NextPageToken: nextPageToken,
  1161. Jobs: jobs,
  1162. }
  1163. mockDlp.err = nil
  1164. mockDlp.reqs = nil
  1165. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1166. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1167. var request = &dlppb.ListDlpJobsRequest{
  1168. Parent: formattedParent,
  1169. }
  1170. c, err := NewClient(context.Background(), clientOpt)
  1171. if err != nil {
  1172. t.Fatal(err)
  1173. }
  1174. resp, err := c.ListDlpJobs(context.Background(), request).Next()
  1175. if err != nil {
  1176. t.Fatal(err)
  1177. }
  1178. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1179. t.Errorf("wrong request %q, want %q", got, want)
  1180. }
  1181. want := (interface{})(expectedResponse.Jobs[0])
  1182. got := (interface{})(resp)
  1183. var ok bool
  1184. switch want := (want).(type) {
  1185. case proto.Message:
  1186. ok = proto.Equal(want, got.(proto.Message))
  1187. default:
  1188. ok = want == got
  1189. }
  1190. if !ok {
  1191. t.Errorf("wrong response %q, want %q)", got, want)
  1192. }
  1193. }
  1194. func TestDlpServiceListDlpJobsError(t *testing.T) {
  1195. errCode := codes.PermissionDenied
  1196. mockDlp.err = gstatus.Error(errCode, "test error")
  1197. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1198. var request = &dlppb.ListDlpJobsRequest{
  1199. Parent: formattedParent,
  1200. }
  1201. c, err := NewClient(context.Background(), clientOpt)
  1202. if err != nil {
  1203. t.Fatal(err)
  1204. }
  1205. resp, err := c.ListDlpJobs(context.Background(), request).Next()
  1206. if st, ok := gstatus.FromError(err); !ok {
  1207. t.Errorf("got error %v, expected grpc error", err)
  1208. } else if c := st.Code(); c != errCode {
  1209. t.Errorf("got error code %q, want %q", c, errCode)
  1210. }
  1211. _ = resp
  1212. }
  1213. func TestDlpServiceGetDlpJob(t *testing.T) {
  1214. var name2 string = "name2-1052831874"
  1215. var jobTriggerName string = "jobTriggerName1819490804"
  1216. var expectedResponse = &dlppb.DlpJob{
  1217. Name: name2,
  1218. JobTriggerName: jobTriggerName,
  1219. }
  1220. mockDlp.err = nil
  1221. mockDlp.reqs = nil
  1222. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1223. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1224. var request = &dlppb.GetDlpJobRequest{
  1225. Name: formattedName,
  1226. }
  1227. c, err := NewClient(context.Background(), clientOpt)
  1228. if err != nil {
  1229. t.Fatal(err)
  1230. }
  1231. resp, err := c.GetDlpJob(context.Background(), request)
  1232. if err != nil {
  1233. t.Fatal(err)
  1234. }
  1235. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1236. t.Errorf("wrong request %q, want %q", got, want)
  1237. }
  1238. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1239. t.Errorf("wrong response %q, want %q)", got, want)
  1240. }
  1241. }
  1242. func TestDlpServiceGetDlpJobError(t *testing.T) {
  1243. errCode := codes.PermissionDenied
  1244. mockDlp.err = gstatus.Error(errCode, "test error")
  1245. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1246. var request = &dlppb.GetDlpJobRequest{
  1247. Name: formattedName,
  1248. }
  1249. c, err := NewClient(context.Background(), clientOpt)
  1250. if err != nil {
  1251. t.Fatal(err)
  1252. }
  1253. resp, err := c.GetDlpJob(context.Background(), request)
  1254. if st, ok := gstatus.FromError(err); !ok {
  1255. t.Errorf("got error %v, expected grpc error", err)
  1256. } else if c := st.Code(); c != errCode {
  1257. t.Errorf("got error code %q, want %q", c, errCode)
  1258. }
  1259. _ = resp
  1260. }
  1261. func TestDlpServiceDeleteDlpJob(t *testing.T) {
  1262. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1263. mockDlp.err = nil
  1264. mockDlp.reqs = nil
  1265. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1266. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1267. var request = &dlppb.DeleteDlpJobRequest{
  1268. Name: formattedName,
  1269. }
  1270. c, err := NewClient(context.Background(), clientOpt)
  1271. if err != nil {
  1272. t.Fatal(err)
  1273. }
  1274. err = c.DeleteDlpJob(context.Background(), request)
  1275. if err != nil {
  1276. t.Fatal(err)
  1277. }
  1278. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1279. t.Errorf("wrong request %q, want %q", got, want)
  1280. }
  1281. }
  1282. func TestDlpServiceDeleteDlpJobError(t *testing.T) {
  1283. errCode := codes.PermissionDenied
  1284. mockDlp.err = gstatus.Error(errCode, "test error")
  1285. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1286. var request = &dlppb.DeleteDlpJobRequest{
  1287. Name: formattedName,
  1288. }
  1289. c, err := NewClient(context.Background(), clientOpt)
  1290. if err != nil {
  1291. t.Fatal(err)
  1292. }
  1293. err = c.DeleteDlpJob(context.Background(), request)
  1294. if st, ok := gstatus.FromError(err); !ok {
  1295. t.Errorf("got error %v, expected grpc error", err)
  1296. } else if c := st.Code(); c != errCode {
  1297. t.Errorf("got error code %q, want %q", c, errCode)
  1298. }
  1299. }
  1300. func TestDlpServiceCancelDlpJob(t *testing.T) {
  1301. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1302. mockDlp.err = nil
  1303. mockDlp.reqs = nil
  1304. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1305. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1306. var request = &dlppb.CancelDlpJobRequest{
  1307. Name: formattedName,
  1308. }
  1309. c, err := NewClient(context.Background(), clientOpt)
  1310. if err != nil {
  1311. t.Fatal(err)
  1312. }
  1313. err = c.CancelDlpJob(context.Background(), request)
  1314. if err != nil {
  1315. t.Fatal(err)
  1316. }
  1317. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1318. t.Errorf("wrong request %q, want %q", got, want)
  1319. }
  1320. }
  1321. func TestDlpServiceCancelDlpJobError(t *testing.T) {
  1322. errCode := codes.PermissionDenied
  1323. mockDlp.err = gstatus.Error(errCode, "test error")
  1324. var formattedName string = fmt.Sprintf("projects/%s/dlpJobs/%s", "[PROJECT]", "[DLP_JOB]")
  1325. var request = &dlppb.CancelDlpJobRequest{
  1326. Name: formattedName,
  1327. }
  1328. c, err := NewClient(context.Background(), clientOpt)
  1329. if err != nil {
  1330. t.Fatal(err)
  1331. }
  1332. err = c.CancelDlpJob(context.Background(), request)
  1333. if st, ok := gstatus.FromError(err); !ok {
  1334. t.Errorf("got error %v, expected grpc error", err)
  1335. } else if c := st.Code(); c != errCode {
  1336. t.Errorf("got error code %q, want %q", c, errCode)
  1337. }
  1338. }
  1339. func TestDlpServiceListJobTriggers(t *testing.T) {
  1340. var nextPageToken string = ""
  1341. var jobTriggersElement *dlppb.JobTrigger = &dlppb.JobTrigger{}
  1342. var jobTriggers = []*dlppb.JobTrigger{jobTriggersElement}
  1343. var expectedResponse = &dlppb.ListJobTriggersResponse{
  1344. NextPageToken: nextPageToken,
  1345. JobTriggers: jobTriggers,
  1346. }
  1347. mockDlp.err = nil
  1348. mockDlp.reqs = nil
  1349. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1350. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1351. var request = &dlppb.ListJobTriggersRequest{
  1352. Parent: formattedParent,
  1353. }
  1354. c, err := NewClient(context.Background(), clientOpt)
  1355. if err != nil {
  1356. t.Fatal(err)
  1357. }
  1358. resp, err := c.ListJobTriggers(context.Background(), request).Next()
  1359. if err != nil {
  1360. t.Fatal(err)
  1361. }
  1362. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1363. t.Errorf("wrong request %q, want %q", got, want)
  1364. }
  1365. want := (interface{})(expectedResponse.JobTriggers[0])
  1366. got := (interface{})(resp)
  1367. var ok bool
  1368. switch want := (want).(type) {
  1369. case proto.Message:
  1370. ok = proto.Equal(want, got.(proto.Message))
  1371. default:
  1372. ok = want == got
  1373. }
  1374. if !ok {
  1375. t.Errorf("wrong response %q, want %q)", got, want)
  1376. }
  1377. }
  1378. func TestDlpServiceListJobTriggersError(t *testing.T) {
  1379. errCode := codes.PermissionDenied
  1380. mockDlp.err = gstatus.Error(errCode, "test error")
  1381. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1382. var request = &dlppb.ListJobTriggersRequest{
  1383. Parent: formattedParent,
  1384. }
  1385. c, err := NewClient(context.Background(), clientOpt)
  1386. if err != nil {
  1387. t.Fatal(err)
  1388. }
  1389. resp, err := c.ListJobTriggers(context.Background(), request).Next()
  1390. if st, ok := gstatus.FromError(err); !ok {
  1391. t.Errorf("got error %v, expected grpc error", err)
  1392. } else if c := st.Code(); c != errCode {
  1393. t.Errorf("got error code %q, want %q", c, errCode)
  1394. }
  1395. _ = resp
  1396. }
  1397. func TestDlpServiceGetJobTrigger(t *testing.T) {
  1398. var name2 string = "name2-1052831874"
  1399. var displayName string = "displayName1615086568"
  1400. var description string = "description-1724546052"
  1401. var expectedResponse = &dlppb.JobTrigger{
  1402. Name: name2,
  1403. DisplayName: displayName,
  1404. Description: description,
  1405. }
  1406. mockDlp.err = nil
  1407. mockDlp.reqs = nil
  1408. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1409. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1410. var request = &dlppb.GetJobTriggerRequest{
  1411. Name: formattedName,
  1412. }
  1413. c, err := NewClient(context.Background(), clientOpt)
  1414. if err != nil {
  1415. t.Fatal(err)
  1416. }
  1417. resp, err := c.GetJobTrigger(context.Background(), request)
  1418. if err != nil {
  1419. t.Fatal(err)
  1420. }
  1421. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1422. t.Errorf("wrong request %q, want %q", got, want)
  1423. }
  1424. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1425. t.Errorf("wrong response %q, want %q)", got, want)
  1426. }
  1427. }
  1428. func TestDlpServiceGetJobTriggerError(t *testing.T) {
  1429. errCode := codes.PermissionDenied
  1430. mockDlp.err = gstatus.Error(errCode, "test error")
  1431. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1432. var request = &dlppb.GetJobTriggerRequest{
  1433. Name: formattedName,
  1434. }
  1435. c, err := NewClient(context.Background(), clientOpt)
  1436. if err != nil {
  1437. t.Fatal(err)
  1438. }
  1439. resp, err := c.GetJobTrigger(context.Background(), request)
  1440. if st, ok := gstatus.FromError(err); !ok {
  1441. t.Errorf("got error %v, expected grpc error", err)
  1442. } else if c := st.Code(); c != errCode {
  1443. t.Errorf("got error code %q, want %q", c, errCode)
  1444. }
  1445. _ = resp
  1446. }
  1447. func TestDlpServiceDeleteJobTrigger(t *testing.T) {
  1448. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1449. mockDlp.err = nil
  1450. mockDlp.reqs = nil
  1451. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1452. var name string = "name3373707"
  1453. var request = &dlppb.DeleteJobTriggerRequest{
  1454. Name: name,
  1455. }
  1456. c, err := NewClient(context.Background(), clientOpt)
  1457. if err != nil {
  1458. t.Fatal(err)
  1459. }
  1460. err = c.DeleteJobTrigger(context.Background(), request)
  1461. if err != nil {
  1462. t.Fatal(err)
  1463. }
  1464. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1465. t.Errorf("wrong request %q, want %q", got, want)
  1466. }
  1467. }
  1468. func TestDlpServiceDeleteJobTriggerError(t *testing.T) {
  1469. errCode := codes.PermissionDenied
  1470. mockDlp.err = gstatus.Error(errCode, "test error")
  1471. var name string = "name3373707"
  1472. var request = &dlppb.DeleteJobTriggerRequest{
  1473. Name: name,
  1474. }
  1475. c, err := NewClient(context.Background(), clientOpt)
  1476. if err != nil {
  1477. t.Fatal(err)
  1478. }
  1479. err = c.DeleteJobTrigger(context.Background(), request)
  1480. if st, ok := gstatus.FromError(err); !ok {
  1481. t.Errorf("got error %v, expected grpc error", err)
  1482. } else if c := st.Code(); c != errCode {
  1483. t.Errorf("got error code %q, want %q", c, errCode)
  1484. }
  1485. }
  1486. func TestDlpServiceUpdateJobTrigger(t *testing.T) {
  1487. var name2 string = "name2-1052831874"
  1488. var displayName string = "displayName1615086568"
  1489. var description string = "description-1724546052"
  1490. var expectedResponse = &dlppb.JobTrigger{
  1491. Name: name2,
  1492. DisplayName: displayName,
  1493. Description: description,
  1494. }
  1495. mockDlp.err = nil
  1496. mockDlp.reqs = nil
  1497. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1498. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1499. var request = &dlppb.UpdateJobTriggerRequest{
  1500. Name: formattedName,
  1501. }
  1502. c, err := NewClient(context.Background(), clientOpt)
  1503. if err != nil {
  1504. t.Fatal(err)
  1505. }
  1506. resp, err := c.UpdateJobTrigger(context.Background(), request)
  1507. if err != nil {
  1508. t.Fatal(err)
  1509. }
  1510. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1511. t.Errorf("wrong request %q, want %q", got, want)
  1512. }
  1513. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1514. t.Errorf("wrong response %q, want %q)", got, want)
  1515. }
  1516. }
  1517. func TestDlpServiceUpdateJobTriggerError(t *testing.T) {
  1518. errCode := codes.PermissionDenied
  1519. mockDlp.err = gstatus.Error(errCode, "test error")
  1520. var formattedName string = fmt.Sprintf("projects/%s/jobTriggers/%s", "[PROJECT]", "[JOB_TRIGGER]")
  1521. var request = &dlppb.UpdateJobTriggerRequest{
  1522. Name: formattedName,
  1523. }
  1524. c, err := NewClient(context.Background(), clientOpt)
  1525. if err != nil {
  1526. t.Fatal(err)
  1527. }
  1528. resp, err := c.UpdateJobTrigger(context.Background(), request)
  1529. if st, ok := gstatus.FromError(err); !ok {
  1530. t.Errorf("got error %v, expected grpc error", err)
  1531. } else if c := st.Code(); c != errCode {
  1532. t.Errorf("got error code %q, want %q", c, errCode)
  1533. }
  1534. _ = resp
  1535. }
  1536. func TestDlpServiceCreateJobTrigger(t *testing.T) {
  1537. var name string = "name3373707"
  1538. var displayName string = "displayName1615086568"
  1539. var description string = "description-1724546052"
  1540. var expectedResponse = &dlppb.JobTrigger{
  1541. Name: name,
  1542. DisplayName: displayName,
  1543. Description: description,
  1544. }
  1545. mockDlp.err = nil
  1546. mockDlp.reqs = nil
  1547. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1548. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1549. var request = &dlppb.CreateJobTriggerRequest{
  1550. Parent: formattedParent,
  1551. }
  1552. c, err := NewClient(context.Background(), clientOpt)
  1553. if err != nil {
  1554. t.Fatal(err)
  1555. }
  1556. resp, err := c.CreateJobTrigger(context.Background(), request)
  1557. if err != nil {
  1558. t.Fatal(err)
  1559. }
  1560. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1561. t.Errorf("wrong request %q, want %q", got, want)
  1562. }
  1563. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1564. t.Errorf("wrong response %q, want %q)", got, want)
  1565. }
  1566. }
  1567. func TestDlpServiceCreateJobTriggerError(t *testing.T) {
  1568. errCode := codes.PermissionDenied
  1569. mockDlp.err = gstatus.Error(errCode, "test error")
  1570. var formattedParent string = fmt.Sprintf("projects/%s", "[PROJECT]")
  1571. var request = &dlppb.CreateJobTriggerRequest{
  1572. Parent: formattedParent,
  1573. }
  1574. c, err := NewClient(context.Background(), clientOpt)
  1575. if err != nil {
  1576. t.Fatal(err)
  1577. }
  1578. resp, err := c.CreateJobTrigger(context.Background(), request)
  1579. if st, ok := gstatus.FromError(err); !ok {
  1580. t.Errorf("got error %v, expected grpc error", err)
  1581. } else if c := st.Code(); c != errCode {
  1582. t.Errorf("got error code %q, want %q", c, errCode)
  1583. }
  1584. _ = resp
  1585. }
  1586. func TestDlpServiceCreateStoredInfoType(t *testing.T) {
  1587. var name string = "name3373707"
  1588. var expectedResponse = &dlppb.StoredInfoType{
  1589. Name: name,
  1590. }
  1591. mockDlp.err = nil
  1592. mockDlp.reqs = nil
  1593. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1594. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  1595. var request = &dlppb.CreateStoredInfoTypeRequest{
  1596. Parent: formattedParent,
  1597. }
  1598. c, err := NewClient(context.Background(), clientOpt)
  1599. if err != nil {
  1600. t.Fatal(err)
  1601. }
  1602. resp, err := c.CreateStoredInfoType(context.Background(), request)
  1603. if err != nil {
  1604. t.Fatal(err)
  1605. }
  1606. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1607. t.Errorf("wrong request %q, want %q", got, want)
  1608. }
  1609. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1610. t.Errorf("wrong response %q, want %q)", got, want)
  1611. }
  1612. }
  1613. func TestDlpServiceCreateStoredInfoTypeError(t *testing.T) {
  1614. errCode := codes.PermissionDenied
  1615. mockDlp.err = gstatus.Error(errCode, "test error")
  1616. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  1617. var request = &dlppb.CreateStoredInfoTypeRequest{
  1618. Parent: formattedParent,
  1619. }
  1620. c, err := NewClient(context.Background(), clientOpt)
  1621. if err != nil {
  1622. t.Fatal(err)
  1623. }
  1624. resp, err := c.CreateStoredInfoType(context.Background(), request)
  1625. if st, ok := gstatus.FromError(err); !ok {
  1626. t.Errorf("got error %v, expected grpc error", err)
  1627. } else if c := st.Code(); c != errCode {
  1628. t.Errorf("got error code %q, want %q", c, errCode)
  1629. }
  1630. _ = resp
  1631. }
  1632. func TestDlpServiceUpdateStoredInfoType(t *testing.T) {
  1633. var name2 string = "name2-1052831874"
  1634. var expectedResponse = &dlppb.StoredInfoType{
  1635. Name: name2,
  1636. }
  1637. mockDlp.err = nil
  1638. mockDlp.reqs = nil
  1639. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1640. var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
  1641. var request = &dlppb.UpdateStoredInfoTypeRequest{
  1642. Name: formattedName,
  1643. }
  1644. c, err := NewClient(context.Background(), clientOpt)
  1645. if err != nil {
  1646. t.Fatal(err)
  1647. }
  1648. resp, err := c.UpdateStoredInfoType(context.Background(), request)
  1649. if err != nil {
  1650. t.Fatal(err)
  1651. }
  1652. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1653. t.Errorf("wrong request %q, want %q", got, want)
  1654. }
  1655. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1656. t.Errorf("wrong response %q, want %q)", got, want)
  1657. }
  1658. }
  1659. func TestDlpServiceUpdateStoredInfoTypeError(t *testing.T) {
  1660. errCode := codes.PermissionDenied
  1661. mockDlp.err = gstatus.Error(errCode, "test error")
  1662. var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
  1663. var request = &dlppb.UpdateStoredInfoTypeRequest{
  1664. Name: formattedName,
  1665. }
  1666. c, err := NewClient(context.Background(), clientOpt)
  1667. if err != nil {
  1668. t.Fatal(err)
  1669. }
  1670. resp, err := c.UpdateStoredInfoType(context.Background(), request)
  1671. if st, ok := gstatus.FromError(err); !ok {
  1672. t.Errorf("got error %v, expected grpc error", err)
  1673. } else if c := st.Code(); c != errCode {
  1674. t.Errorf("got error code %q, want %q", c, errCode)
  1675. }
  1676. _ = resp
  1677. }
  1678. func TestDlpServiceGetStoredInfoType(t *testing.T) {
  1679. var name2 string = "name2-1052831874"
  1680. var expectedResponse = &dlppb.StoredInfoType{
  1681. Name: name2,
  1682. }
  1683. mockDlp.err = nil
  1684. mockDlp.reqs = nil
  1685. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1686. var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
  1687. var request = &dlppb.GetStoredInfoTypeRequest{
  1688. Name: formattedName,
  1689. }
  1690. c, err := NewClient(context.Background(), clientOpt)
  1691. if err != nil {
  1692. t.Fatal(err)
  1693. }
  1694. resp, err := c.GetStoredInfoType(context.Background(), request)
  1695. if err != nil {
  1696. t.Fatal(err)
  1697. }
  1698. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1699. t.Errorf("wrong request %q, want %q", got, want)
  1700. }
  1701. if want, got := expectedResponse, resp; !proto.Equal(want, got) {
  1702. t.Errorf("wrong response %q, want %q)", got, want)
  1703. }
  1704. }
  1705. func TestDlpServiceGetStoredInfoTypeError(t *testing.T) {
  1706. errCode := codes.PermissionDenied
  1707. mockDlp.err = gstatus.Error(errCode, "test error")
  1708. var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
  1709. var request = &dlppb.GetStoredInfoTypeRequest{
  1710. Name: formattedName,
  1711. }
  1712. c, err := NewClient(context.Background(), clientOpt)
  1713. if err != nil {
  1714. t.Fatal(err)
  1715. }
  1716. resp, err := c.GetStoredInfoType(context.Background(), request)
  1717. if st, ok := gstatus.FromError(err); !ok {
  1718. t.Errorf("got error %v, expected grpc error", err)
  1719. } else if c := st.Code(); c != errCode {
  1720. t.Errorf("got error code %q, want %q", c, errCode)
  1721. }
  1722. _ = resp
  1723. }
  1724. func TestDlpServiceListStoredInfoTypes(t *testing.T) {
  1725. var nextPageToken string = ""
  1726. var storedInfoTypesElement *dlppb.StoredInfoType = &dlppb.StoredInfoType{}
  1727. var storedInfoTypes = []*dlppb.StoredInfoType{storedInfoTypesElement}
  1728. var expectedResponse = &dlppb.ListStoredInfoTypesResponse{
  1729. NextPageToken: nextPageToken,
  1730. StoredInfoTypes: storedInfoTypes,
  1731. }
  1732. mockDlp.err = nil
  1733. mockDlp.reqs = nil
  1734. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1735. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  1736. var request = &dlppb.ListStoredInfoTypesRequest{
  1737. Parent: formattedParent,
  1738. }
  1739. c, err := NewClient(context.Background(), clientOpt)
  1740. if err != nil {
  1741. t.Fatal(err)
  1742. }
  1743. resp, err := c.ListStoredInfoTypes(context.Background(), request).Next()
  1744. if err != nil {
  1745. t.Fatal(err)
  1746. }
  1747. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1748. t.Errorf("wrong request %q, want %q", got, want)
  1749. }
  1750. want := (interface{})(expectedResponse.StoredInfoTypes[0])
  1751. got := (interface{})(resp)
  1752. var ok bool
  1753. switch want := (want).(type) {
  1754. case proto.Message:
  1755. ok = proto.Equal(want, got.(proto.Message))
  1756. default:
  1757. ok = want == got
  1758. }
  1759. if !ok {
  1760. t.Errorf("wrong response %q, want %q)", got, want)
  1761. }
  1762. }
  1763. func TestDlpServiceListStoredInfoTypesError(t *testing.T) {
  1764. errCode := codes.PermissionDenied
  1765. mockDlp.err = gstatus.Error(errCode, "test error")
  1766. var formattedParent string = fmt.Sprintf("organizations/%s", "[ORGANIZATION]")
  1767. var request = &dlppb.ListStoredInfoTypesRequest{
  1768. Parent: formattedParent,
  1769. }
  1770. c, err := NewClient(context.Background(), clientOpt)
  1771. if err != nil {
  1772. t.Fatal(err)
  1773. }
  1774. resp, err := c.ListStoredInfoTypes(context.Background(), request).Next()
  1775. if st, ok := gstatus.FromError(err); !ok {
  1776. t.Errorf("got error %v, expected grpc error", err)
  1777. } else if c := st.Code(); c != errCode {
  1778. t.Errorf("got error code %q, want %q", c, errCode)
  1779. }
  1780. _ = resp
  1781. }
  1782. func TestDlpServiceDeleteStoredInfoType(t *testing.T) {
  1783. var expectedResponse *emptypb.Empty = &emptypb.Empty{}
  1784. mockDlp.err = nil
  1785. mockDlp.reqs = nil
  1786. mockDlp.resps = append(mockDlp.resps[:0], expectedResponse)
  1787. var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
  1788. var request = &dlppb.DeleteStoredInfoTypeRequest{
  1789. Name: formattedName,
  1790. }
  1791. c, err := NewClient(context.Background(), clientOpt)
  1792. if err != nil {
  1793. t.Fatal(err)
  1794. }
  1795. err = c.DeleteStoredInfoType(context.Background(), request)
  1796. if err != nil {
  1797. t.Fatal(err)
  1798. }
  1799. if want, got := request, mockDlp.reqs[0]; !proto.Equal(want, got) {
  1800. t.Errorf("wrong request %q, want %q", got, want)
  1801. }
  1802. }
  1803. func TestDlpServiceDeleteStoredInfoTypeError(t *testing.T) {
  1804. errCode := codes.PermissionDenied
  1805. mockDlp.err = gstatus.Error(errCode, "test error")
  1806. var formattedName string = fmt.Sprintf("organizations/%s/storedInfoTypes/%s", "[ORGANIZATION]", "[STORED_INFO_TYPE]")
  1807. var request = &dlppb.DeleteStoredInfoTypeRequest{
  1808. Name: formattedName,
  1809. }
  1810. c, err := NewClient(context.Background(), clientOpt)
  1811. if err != nil {
  1812. t.Fatal(err)
  1813. }
  1814. err = c.DeleteStoredInfoType(context.Background(), request)
  1815. if st, ok := gstatus.FromError(err); !ok {
  1816. t.Errorf("got error %v, expected grpc error", err)
  1817. } else if c := st.Code(); c != errCode {
  1818. t.Errorf("got error code %q, want %q", c, errCode)
  1819. }
  1820. }