• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright 2022 Google LLC
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      https://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.google.cloud.orgpolicy.v2;
18 
19 import static com.google.cloud.orgpolicy.v2.OrgPolicyClient.ListConstraintsPagedResponse;
20 import static com.google.cloud.orgpolicy.v2.OrgPolicyClient.ListPoliciesPagedResponse;
21 
22 import com.google.api.gax.core.NoCredentialsProvider;
23 import com.google.api.gax.grpc.GaxGrpcProperties;
24 import com.google.api.gax.grpc.testing.LocalChannelProvider;
25 import com.google.api.gax.grpc.testing.MockGrpcService;
26 import com.google.api.gax.grpc.testing.MockServiceHelper;
27 import com.google.api.gax.rpc.ApiClientHeaderProvider;
28 import com.google.api.gax.rpc.InvalidArgumentException;
29 import com.google.common.collect.Lists;
30 import com.google.protobuf.AbstractMessage;
31 import com.google.protobuf.Empty;
32 import io.grpc.StatusRuntimeException;
33 import java.io.IOException;
34 import java.util.Arrays;
35 import java.util.List;
36 import java.util.UUID;
37 import javax.annotation.Generated;
38 import org.junit.After;
39 import org.junit.AfterClass;
40 import org.junit.Assert;
41 import org.junit.Before;
42 import org.junit.BeforeClass;
43 import org.junit.Test;
44 
45 @Generated("by gapic-generator-java")
46 public class OrgPolicyClientTest {
47   private static MockOrgPolicy mockOrgPolicy;
48   private static MockServiceHelper mockServiceHelper;
49   private LocalChannelProvider channelProvider;
50   private OrgPolicyClient client;
51 
52   @BeforeClass
startStaticServer()53   public static void startStaticServer() {
54     mockOrgPolicy = new MockOrgPolicy();
55     mockServiceHelper =
56         new MockServiceHelper(
57             UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(mockOrgPolicy));
58     mockServiceHelper.start();
59   }
60 
61   @AfterClass
stopServer()62   public static void stopServer() {
63     mockServiceHelper.stop();
64   }
65 
66   @Before
setUp()67   public void setUp() throws IOException {
68     mockServiceHelper.reset();
69     channelProvider = mockServiceHelper.createChannelProvider();
70     OrgPolicySettings settings =
71         OrgPolicySettings.newBuilder()
72             .setTransportChannelProvider(channelProvider)
73             .setCredentialsProvider(NoCredentialsProvider.create())
74             .build();
75     client = OrgPolicyClient.create(settings);
76   }
77 
78   @After
tearDown()79   public void tearDown() throws Exception {
80     client.close();
81   }
82 
83   @Test
listConstraintsTest()84   public void listConstraintsTest() throws Exception {
85     Constraint responsesElement = Constraint.newBuilder().build();
86     ListConstraintsResponse expectedResponse =
87         ListConstraintsResponse.newBuilder()
88             .setNextPageToken("")
89             .addAllConstraints(Arrays.asList(responsesElement))
90             .build();
91     mockOrgPolicy.addResponse(expectedResponse);
92 
93     FolderName parent = FolderName.of("[FOLDER]");
94 
95     ListConstraintsPagedResponse pagedListResponse = client.listConstraints(parent);
96 
97     List<Constraint> resources = Lists.newArrayList(pagedListResponse.iterateAll());
98 
99     Assert.assertEquals(1, resources.size());
100     Assert.assertEquals(expectedResponse.getConstraintsList().get(0), resources.get(0));
101 
102     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
103     Assert.assertEquals(1, actualRequests.size());
104     ListConstraintsRequest actualRequest = ((ListConstraintsRequest) actualRequests.get(0));
105 
106     Assert.assertEquals(parent.toString(), actualRequest.getParent());
107     Assert.assertTrue(
108         channelProvider.isHeaderSent(
109             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
110             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
111   }
112 
113   @Test
listConstraintsExceptionTest()114   public void listConstraintsExceptionTest() throws Exception {
115     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
116     mockOrgPolicy.addException(exception);
117 
118     try {
119       FolderName parent = FolderName.of("[FOLDER]");
120       client.listConstraints(parent);
121       Assert.fail("No exception raised");
122     } catch (InvalidArgumentException e) {
123       // Expected exception.
124     }
125   }
126 
127   @Test
listConstraintsTest2()128   public void listConstraintsTest2() throws Exception {
129     Constraint responsesElement = Constraint.newBuilder().build();
130     ListConstraintsResponse expectedResponse =
131         ListConstraintsResponse.newBuilder()
132             .setNextPageToken("")
133             .addAllConstraints(Arrays.asList(responsesElement))
134             .build();
135     mockOrgPolicy.addResponse(expectedResponse);
136 
137     OrganizationName parent = OrganizationName.of("[ORGANIZATION]");
138 
139     ListConstraintsPagedResponse pagedListResponse = client.listConstraints(parent);
140 
141     List<Constraint> resources = Lists.newArrayList(pagedListResponse.iterateAll());
142 
143     Assert.assertEquals(1, resources.size());
144     Assert.assertEquals(expectedResponse.getConstraintsList().get(0), resources.get(0));
145 
146     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
147     Assert.assertEquals(1, actualRequests.size());
148     ListConstraintsRequest actualRequest = ((ListConstraintsRequest) actualRequests.get(0));
149 
150     Assert.assertEquals(parent.toString(), actualRequest.getParent());
151     Assert.assertTrue(
152         channelProvider.isHeaderSent(
153             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
154             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
155   }
156 
157   @Test
listConstraintsExceptionTest2()158   public void listConstraintsExceptionTest2() throws Exception {
159     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
160     mockOrgPolicy.addException(exception);
161 
162     try {
163       OrganizationName parent = OrganizationName.of("[ORGANIZATION]");
164       client.listConstraints(parent);
165       Assert.fail("No exception raised");
166     } catch (InvalidArgumentException e) {
167       // Expected exception.
168     }
169   }
170 
171   @Test
listConstraintsTest3()172   public void listConstraintsTest3() throws Exception {
173     Constraint responsesElement = Constraint.newBuilder().build();
174     ListConstraintsResponse expectedResponse =
175         ListConstraintsResponse.newBuilder()
176             .setNextPageToken("")
177             .addAllConstraints(Arrays.asList(responsesElement))
178             .build();
179     mockOrgPolicy.addResponse(expectedResponse);
180 
181     ProjectName parent = ProjectName.of("[PROJECT]");
182 
183     ListConstraintsPagedResponse pagedListResponse = client.listConstraints(parent);
184 
185     List<Constraint> resources = Lists.newArrayList(pagedListResponse.iterateAll());
186 
187     Assert.assertEquals(1, resources.size());
188     Assert.assertEquals(expectedResponse.getConstraintsList().get(0), resources.get(0));
189 
190     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
191     Assert.assertEquals(1, actualRequests.size());
192     ListConstraintsRequest actualRequest = ((ListConstraintsRequest) actualRequests.get(0));
193 
194     Assert.assertEquals(parent.toString(), actualRequest.getParent());
195     Assert.assertTrue(
196         channelProvider.isHeaderSent(
197             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
198             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
199   }
200 
201   @Test
listConstraintsExceptionTest3()202   public void listConstraintsExceptionTest3() throws Exception {
203     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
204     mockOrgPolicy.addException(exception);
205 
206     try {
207       ProjectName parent = ProjectName.of("[PROJECT]");
208       client.listConstraints(parent);
209       Assert.fail("No exception raised");
210     } catch (InvalidArgumentException e) {
211       // Expected exception.
212     }
213   }
214 
215   @Test
listConstraintsTest4()216   public void listConstraintsTest4() throws Exception {
217     Constraint responsesElement = Constraint.newBuilder().build();
218     ListConstraintsResponse expectedResponse =
219         ListConstraintsResponse.newBuilder()
220             .setNextPageToken("")
221             .addAllConstraints(Arrays.asList(responsesElement))
222             .build();
223     mockOrgPolicy.addResponse(expectedResponse);
224 
225     String parent = "parent-995424086";
226 
227     ListConstraintsPagedResponse pagedListResponse = client.listConstraints(parent);
228 
229     List<Constraint> resources = Lists.newArrayList(pagedListResponse.iterateAll());
230 
231     Assert.assertEquals(1, resources.size());
232     Assert.assertEquals(expectedResponse.getConstraintsList().get(0), resources.get(0));
233 
234     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
235     Assert.assertEquals(1, actualRequests.size());
236     ListConstraintsRequest actualRequest = ((ListConstraintsRequest) actualRequests.get(0));
237 
238     Assert.assertEquals(parent, actualRequest.getParent());
239     Assert.assertTrue(
240         channelProvider.isHeaderSent(
241             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
242             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
243   }
244 
245   @Test
listConstraintsExceptionTest4()246   public void listConstraintsExceptionTest4() throws Exception {
247     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
248     mockOrgPolicy.addException(exception);
249 
250     try {
251       String parent = "parent-995424086";
252       client.listConstraints(parent);
253       Assert.fail("No exception raised");
254     } catch (InvalidArgumentException e) {
255       // Expected exception.
256     }
257   }
258 
259   @Test
listPoliciesTest()260   public void listPoliciesTest() throws Exception {
261     Policy responsesElement = Policy.newBuilder().build();
262     ListPoliciesResponse expectedResponse =
263         ListPoliciesResponse.newBuilder()
264             .setNextPageToken("")
265             .addAllPolicies(Arrays.asList(responsesElement))
266             .build();
267     mockOrgPolicy.addResponse(expectedResponse);
268 
269     FolderName parent = FolderName.of("[FOLDER]");
270 
271     ListPoliciesPagedResponse pagedListResponse = client.listPolicies(parent);
272 
273     List<Policy> resources = Lists.newArrayList(pagedListResponse.iterateAll());
274 
275     Assert.assertEquals(1, resources.size());
276     Assert.assertEquals(expectedResponse.getPoliciesList().get(0), resources.get(0));
277 
278     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
279     Assert.assertEquals(1, actualRequests.size());
280     ListPoliciesRequest actualRequest = ((ListPoliciesRequest) actualRequests.get(0));
281 
282     Assert.assertEquals(parent.toString(), actualRequest.getParent());
283     Assert.assertTrue(
284         channelProvider.isHeaderSent(
285             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
286             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
287   }
288 
289   @Test
listPoliciesExceptionTest()290   public void listPoliciesExceptionTest() throws Exception {
291     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
292     mockOrgPolicy.addException(exception);
293 
294     try {
295       FolderName parent = FolderName.of("[FOLDER]");
296       client.listPolicies(parent);
297       Assert.fail("No exception raised");
298     } catch (InvalidArgumentException e) {
299       // Expected exception.
300     }
301   }
302 
303   @Test
listPoliciesTest2()304   public void listPoliciesTest2() throws Exception {
305     Policy responsesElement = Policy.newBuilder().build();
306     ListPoliciesResponse expectedResponse =
307         ListPoliciesResponse.newBuilder()
308             .setNextPageToken("")
309             .addAllPolicies(Arrays.asList(responsesElement))
310             .build();
311     mockOrgPolicy.addResponse(expectedResponse);
312 
313     OrganizationName parent = OrganizationName.of("[ORGANIZATION]");
314 
315     ListPoliciesPagedResponse pagedListResponse = client.listPolicies(parent);
316 
317     List<Policy> resources = Lists.newArrayList(pagedListResponse.iterateAll());
318 
319     Assert.assertEquals(1, resources.size());
320     Assert.assertEquals(expectedResponse.getPoliciesList().get(0), resources.get(0));
321 
322     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
323     Assert.assertEquals(1, actualRequests.size());
324     ListPoliciesRequest actualRequest = ((ListPoliciesRequest) actualRequests.get(0));
325 
326     Assert.assertEquals(parent.toString(), actualRequest.getParent());
327     Assert.assertTrue(
328         channelProvider.isHeaderSent(
329             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
330             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
331   }
332 
333   @Test
listPoliciesExceptionTest2()334   public void listPoliciesExceptionTest2() throws Exception {
335     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
336     mockOrgPolicy.addException(exception);
337 
338     try {
339       OrganizationName parent = OrganizationName.of("[ORGANIZATION]");
340       client.listPolicies(parent);
341       Assert.fail("No exception raised");
342     } catch (InvalidArgumentException e) {
343       // Expected exception.
344     }
345   }
346 
347   @Test
listPoliciesTest3()348   public void listPoliciesTest3() throws Exception {
349     Policy responsesElement = Policy.newBuilder().build();
350     ListPoliciesResponse expectedResponse =
351         ListPoliciesResponse.newBuilder()
352             .setNextPageToken("")
353             .addAllPolicies(Arrays.asList(responsesElement))
354             .build();
355     mockOrgPolicy.addResponse(expectedResponse);
356 
357     ProjectName parent = ProjectName.of("[PROJECT]");
358 
359     ListPoliciesPagedResponse pagedListResponse = client.listPolicies(parent);
360 
361     List<Policy> resources = Lists.newArrayList(pagedListResponse.iterateAll());
362 
363     Assert.assertEquals(1, resources.size());
364     Assert.assertEquals(expectedResponse.getPoliciesList().get(0), resources.get(0));
365 
366     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
367     Assert.assertEquals(1, actualRequests.size());
368     ListPoliciesRequest actualRequest = ((ListPoliciesRequest) actualRequests.get(0));
369 
370     Assert.assertEquals(parent.toString(), actualRequest.getParent());
371     Assert.assertTrue(
372         channelProvider.isHeaderSent(
373             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
374             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
375   }
376 
377   @Test
listPoliciesExceptionTest3()378   public void listPoliciesExceptionTest3() throws Exception {
379     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
380     mockOrgPolicy.addException(exception);
381 
382     try {
383       ProjectName parent = ProjectName.of("[PROJECT]");
384       client.listPolicies(parent);
385       Assert.fail("No exception raised");
386     } catch (InvalidArgumentException e) {
387       // Expected exception.
388     }
389   }
390 
391   @Test
listPoliciesTest4()392   public void listPoliciesTest4() throws Exception {
393     Policy responsesElement = Policy.newBuilder().build();
394     ListPoliciesResponse expectedResponse =
395         ListPoliciesResponse.newBuilder()
396             .setNextPageToken("")
397             .addAllPolicies(Arrays.asList(responsesElement))
398             .build();
399     mockOrgPolicy.addResponse(expectedResponse);
400 
401     String parent = "parent-995424086";
402 
403     ListPoliciesPagedResponse pagedListResponse = client.listPolicies(parent);
404 
405     List<Policy> resources = Lists.newArrayList(pagedListResponse.iterateAll());
406 
407     Assert.assertEquals(1, resources.size());
408     Assert.assertEquals(expectedResponse.getPoliciesList().get(0), resources.get(0));
409 
410     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
411     Assert.assertEquals(1, actualRequests.size());
412     ListPoliciesRequest actualRequest = ((ListPoliciesRequest) actualRequests.get(0));
413 
414     Assert.assertEquals(parent, actualRequest.getParent());
415     Assert.assertTrue(
416         channelProvider.isHeaderSent(
417             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
418             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
419   }
420 
421   @Test
listPoliciesExceptionTest4()422   public void listPoliciesExceptionTest4() throws Exception {
423     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
424     mockOrgPolicy.addException(exception);
425 
426     try {
427       String parent = "parent-995424086";
428       client.listPolicies(parent);
429       Assert.fail("No exception raised");
430     } catch (InvalidArgumentException e) {
431       // Expected exception.
432     }
433   }
434 
435   @Test
getPolicyTest()436   public void getPolicyTest() throws Exception {
437     Policy expectedResponse =
438         Policy.newBuilder()
439             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
440             .setSpec(PolicySpec.newBuilder().build())
441             .setAlternate(AlternatePolicySpec.newBuilder().build())
442             .setDryRunSpec(PolicySpec.newBuilder().build())
443             .build();
444     mockOrgPolicy.addResponse(expectedResponse);
445 
446     PolicyName name = PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]");
447 
448     Policy actualResponse = client.getPolicy(name);
449     Assert.assertEquals(expectedResponse, actualResponse);
450 
451     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
452     Assert.assertEquals(1, actualRequests.size());
453     GetPolicyRequest actualRequest = ((GetPolicyRequest) actualRequests.get(0));
454 
455     Assert.assertEquals(name.toString(), actualRequest.getName());
456     Assert.assertTrue(
457         channelProvider.isHeaderSent(
458             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
459             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
460   }
461 
462   @Test
getPolicyExceptionTest()463   public void getPolicyExceptionTest() throws Exception {
464     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
465     mockOrgPolicy.addException(exception);
466 
467     try {
468       PolicyName name = PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]");
469       client.getPolicy(name);
470       Assert.fail("No exception raised");
471     } catch (InvalidArgumentException e) {
472       // Expected exception.
473     }
474   }
475 
476   @Test
getPolicyTest2()477   public void getPolicyTest2() throws Exception {
478     Policy expectedResponse =
479         Policy.newBuilder()
480             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
481             .setSpec(PolicySpec.newBuilder().build())
482             .setAlternate(AlternatePolicySpec.newBuilder().build())
483             .setDryRunSpec(PolicySpec.newBuilder().build())
484             .build();
485     mockOrgPolicy.addResponse(expectedResponse);
486 
487     String name = "name3373707";
488 
489     Policy actualResponse = client.getPolicy(name);
490     Assert.assertEquals(expectedResponse, actualResponse);
491 
492     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
493     Assert.assertEquals(1, actualRequests.size());
494     GetPolicyRequest actualRequest = ((GetPolicyRequest) actualRequests.get(0));
495 
496     Assert.assertEquals(name, actualRequest.getName());
497     Assert.assertTrue(
498         channelProvider.isHeaderSent(
499             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
500             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
501   }
502 
503   @Test
getPolicyExceptionTest2()504   public void getPolicyExceptionTest2() throws Exception {
505     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
506     mockOrgPolicy.addException(exception);
507 
508     try {
509       String name = "name3373707";
510       client.getPolicy(name);
511       Assert.fail("No exception raised");
512     } catch (InvalidArgumentException e) {
513       // Expected exception.
514     }
515   }
516 
517   @Test
getEffectivePolicyTest()518   public void getEffectivePolicyTest() throws Exception {
519     Policy expectedResponse =
520         Policy.newBuilder()
521             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
522             .setSpec(PolicySpec.newBuilder().build())
523             .setAlternate(AlternatePolicySpec.newBuilder().build())
524             .setDryRunSpec(PolicySpec.newBuilder().build())
525             .build();
526     mockOrgPolicy.addResponse(expectedResponse);
527 
528     PolicyName name = PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]");
529 
530     Policy actualResponse = client.getEffectivePolicy(name);
531     Assert.assertEquals(expectedResponse, actualResponse);
532 
533     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
534     Assert.assertEquals(1, actualRequests.size());
535     GetEffectivePolicyRequest actualRequest = ((GetEffectivePolicyRequest) actualRequests.get(0));
536 
537     Assert.assertEquals(name.toString(), actualRequest.getName());
538     Assert.assertTrue(
539         channelProvider.isHeaderSent(
540             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
541             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
542   }
543 
544   @Test
getEffectivePolicyExceptionTest()545   public void getEffectivePolicyExceptionTest() throws Exception {
546     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
547     mockOrgPolicy.addException(exception);
548 
549     try {
550       PolicyName name = PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]");
551       client.getEffectivePolicy(name);
552       Assert.fail("No exception raised");
553     } catch (InvalidArgumentException e) {
554       // Expected exception.
555     }
556   }
557 
558   @Test
getEffectivePolicyTest2()559   public void getEffectivePolicyTest2() throws Exception {
560     Policy expectedResponse =
561         Policy.newBuilder()
562             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
563             .setSpec(PolicySpec.newBuilder().build())
564             .setAlternate(AlternatePolicySpec.newBuilder().build())
565             .setDryRunSpec(PolicySpec.newBuilder().build())
566             .build();
567     mockOrgPolicy.addResponse(expectedResponse);
568 
569     String name = "name3373707";
570 
571     Policy actualResponse = client.getEffectivePolicy(name);
572     Assert.assertEquals(expectedResponse, actualResponse);
573 
574     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
575     Assert.assertEquals(1, actualRequests.size());
576     GetEffectivePolicyRequest actualRequest = ((GetEffectivePolicyRequest) actualRequests.get(0));
577 
578     Assert.assertEquals(name, actualRequest.getName());
579     Assert.assertTrue(
580         channelProvider.isHeaderSent(
581             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
582             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
583   }
584 
585   @Test
getEffectivePolicyExceptionTest2()586   public void getEffectivePolicyExceptionTest2() throws Exception {
587     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
588     mockOrgPolicy.addException(exception);
589 
590     try {
591       String name = "name3373707";
592       client.getEffectivePolicy(name);
593       Assert.fail("No exception raised");
594     } catch (InvalidArgumentException e) {
595       // Expected exception.
596     }
597   }
598 
599   @Test
createPolicyTest()600   public void createPolicyTest() throws Exception {
601     Policy expectedResponse =
602         Policy.newBuilder()
603             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
604             .setSpec(PolicySpec.newBuilder().build())
605             .setAlternate(AlternatePolicySpec.newBuilder().build())
606             .setDryRunSpec(PolicySpec.newBuilder().build())
607             .build();
608     mockOrgPolicy.addResponse(expectedResponse);
609 
610     FolderName parent = FolderName.of("[FOLDER]");
611     Policy policy = Policy.newBuilder().build();
612 
613     Policy actualResponse = client.createPolicy(parent, policy);
614     Assert.assertEquals(expectedResponse, actualResponse);
615 
616     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
617     Assert.assertEquals(1, actualRequests.size());
618     CreatePolicyRequest actualRequest = ((CreatePolicyRequest) actualRequests.get(0));
619 
620     Assert.assertEquals(parent.toString(), actualRequest.getParent());
621     Assert.assertEquals(policy, actualRequest.getPolicy());
622     Assert.assertTrue(
623         channelProvider.isHeaderSent(
624             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
625             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
626   }
627 
628   @Test
createPolicyExceptionTest()629   public void createPolicyExceptionTest() throws Exception {
630     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
631     mockOrgPolicy.addException(exception);
632 
633     try {
634       FolderName parent = FolderName.of("[FOLDER]");
635       Policy policy = Policy.newBuilder().build();
636       client.createPolicy(parent, policy);
637       Assert.fail("No exception raised");
638     } catch (InvalidArgumentException e) {
639       // Expected exception.
640     }
641   }
642 
643   @Test
createPolicyTest2()644   public void createPolicyTest2() throws Exception {
645     Policy expectedResponse =
646         Policy.newBuilder()
647             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
648             .setSpec(PolicySpec.newBuilder().build())
649             .setAlternate(AlternatePolicySpec.newBuilder().build())
650             .setDryRunSpec(PolicySpec.newBuilder().build())
651             .build();
652     mockOrgPolicy.addResponse(expectedResponse);
653 
654     OrganizationName parent = OrganizationName.of("[ORGANIZATION]");
655     Policy policy = Policy.newBuilder().build();
656 
657     Policy actualResponse = client.createPolicy(parent, policy);
658     Assert.assertEquals(expectedResponse, actualResponse);
659 
660     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
661     Assert.assertEquals(1, actualRequests.size());
662     CreatePolicyRequest actualRequest = ((CreatePolicyRequest) actualRequests.get(0));
663 
664     Assert.assertEquals(parent.toString(), actualRequest.getParent());
665     Assert.assertEquals(policy, actualRequest.getPolicy());
666     Assert.assertTrue(
667         channelProvider.isHeaderSent(
668             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
669             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
670   }
671 
672   @Test
createPolicyExceptionTest2()673   public void createPolicyExceptionTest2() throws Exception {
674     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
675     mockOrgPolicy.addException(exception);
676 
677     try {
678       OrganizationName parent = OrganizationName.of("[ORGANIZATION]");
679       Policy policy = Policy.newBuilder().build();
680       client.createPolicy(parent, policy);
681       Assert.fail("No exception raised");
682     } catch (InvalidArgumentException e) {
683       // Expected exception.
684     }
685   }
686 
687   @Test
createPolicyTest3()688   public void createPolicyTest3() throws Exception {
689     Policy expectedResponse =
690         Policy.newBuilder()
691             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
692             .setSpec(PolicySpec.newBuilder().build())
693             .setAlternate(AlternatePolicySpec.newBuilder().build())
694             .setDryRunSpec(PolicySpec.newBuilder().build())
695             .build();
696     mockOrgPolicy.addResponse(expectedResponse);
697 
698     ProjectName parent = ProjectName.of("[PROJECT]");
699     Policy policy = Policy.newBuilder().build();
700 
701     Policy actualResponse = client.createPolicy(parent, policy);
702     Assert.assertEquals(expectedResponse, actualResponse);
703 
704     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
705     Assert.assertEquals(1, actualRequests.size());
706     CreatePolicyRequest actualRequest = ((CreatePolicyRequest) actualRequests.get(0));
707 
708     Assert.assertEquals(parent.toString(), actualRequest.getParent());
709     Assert.assertEquals(policy, actualRequest.getPolicy());
710     Assert.assertTrue(
711         channelProvider.isHeaderSent(
712             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
713             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
714   }
715 
716   @Test
createPolicyExceptionTest3()717   public void createPolicyExceptionTest3() throws Exception {
718     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
719     mockOrgPolicy.addException(exception);
720 
721     try {
722       ProjectName parent = ProjectName.of("[PROJECT]");
723       Policy policy = Policy.newBuilder().build();
724       client.createPolicy(parent, policy);
725       Assert.fail("No exception raised");
726     } catch (InvalidArgumentException e) {
727       // Expected exception.
728     }
729   }
730 
731   @Test
createPolicyTest4()732   public void createPolicyTest4() throws Exception {
733     Policy expectedResponse =
734         Policy.newBuilder()
735             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
736             .setSpec(PolicySpec.newBuilder().build())
737             .setAlternate(AlternatePolicySpec.newBuilder().build())
738             .setDryRunSpec(PolicySpec.newBuilder().build())
739             .build();
740     mockOrgPolicy.addResponse(expectedResponse);
741 
742     String parent = "parent-995424086";
743     Policy policy = Policy.newBuilder().build();
744 
745     Policy actualResponse = client.createPolicy(parent, policy);
746     Assert.assertEquals(expectedResponse, actualResponse);
747 
748     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
749     Assert.assertEquals(1, actualRequests.size());
750     CreatePolicyRequest actualRequest = ((CreatePolicyRequest) actualRequests.get(0));
751 
752     Assert.assertEquals(parent, actualRequest.getParent());
753     Assert.assertEquals(policy, actualRequest.getPolicy());
754     Assert.assertTrue(
755         channelProvider.isHeaderSent(
756             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
757             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
758   }
759 
760   @Test
createPolicyExceptionTest4()761   public void createPolicyExceptionTest4() throws Exception {
762     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
763     mockOrgPolicy.addException(exception);
764 
765     try {
766       String parent = "parent-995424086";
767       Policy policy = Policy.newBuilder().build();
768       client.createPolicy(parent, policy);
769       Assert.fail("No exception raised");
770     } catch (InvalidArgumentException e) {
771       // Expected exception.
772     }
773   }
774 
775   @Test
updatePolicyTest()776   public void updatePolicyTest() throws Exception {
777     Policy expectedResponse =
778         Policy.newBuilder()
779             .setName(PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]").toString())
780             .setSpec(PolicySpec.newBuilder().build())
781             .setAlternate(AlternatePolicySpec.newBuilder().build())
782             .setDryRunSpec(PolicySpec.newBuilder().build())
783             .build();
784     mockOrgPolicy.addResponse(expectedResponse);
785 
786     Policy policy = Policy.newBuilder().build();
787 
788     Policy actualResponse = client.updatePolicy(policy);
789     Assert.assertEquals(expectedResponse, actualResponse);
790 
791     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
792     Assert.assertEquals(1, actualRequests.size());
793     UpdatePolicyRequest actualRequest = ((UpdatePolicyRequest) actualRequests.get(0));
794 
795     Assert.assertEquals(policy, actualRequest.getPolicy());
796     Assert.assertTrue(
797         channelProvider.isHeaderSent(
798             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
799             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
800   }
801 
802   @Test
updatePolicyExceptionTest()803   public void updatePolicyExceptionTest() throws Exception {
804     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
805     mockOrgPolicy.addException(exception);
806 
807     try {
808       Policy policy = Policy.newBuilder().build();
809       client.updatePolicy(policy);
810       Assert.fail("No exception raised");
811     } catch (InvalidArgumentException e) {
812       // Expected exception.
813     }
814   }
815 
816   @Test
deletePolicyTest()817   public void deletePolicyTest() throws Exception {
818     Empty expectedResponse = Empty.newBuilder().build();
819     mockOrgPolicy.addResponse(expectedResponse);
820 
821     PolicyName name = PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]");
822 
823     client.deletePolicy(name);
824 
825     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
826     Assert.assertEquals(1, actualRequests.size());
827     DeletePolicyRequest actualRequest = ((DeletePolicyRequest) actualRequests.get(0));
828 
829     Assert.assertEquals(name.toString(), actualRequest.getName());
830     Assert.assertTrue(
831         channelProvider.isHeaderSent(
832             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
833             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
834   }
835 
836   @Test
deletePolicyExceptionTest()837   public void deletePolicyExceptionTest() throws Exception {
838     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
839     mockOrgPolicy.addException(exception);
840 
841     try {
842       PolicyName name = PolicyName.ofProjectPolicyName("[PROJECT]", "[POLICY]");
843       client.deletePolicy(name);
844       Assert.fail("No exception raised");
845     } catch (InvalidArgumentException e) {
846       // Expected exception.
847     }
848   }
849 
850   @Test
deletePolicyTest2()851   public void deletePolicyTest2() throws Exception {
852     Empty expectedResponse = Empty.newBuilder().build();
853     mockOrgPolicy.addResponse(expectedResponse);
854 
855     String name = "name3373707";
856 
857     client.deletePolicy(name);
858 
859     List<AbstractMessage> actualRequests = mockOrgPolicy.getRequests();
860     Assert.assertEquals(1, actualRequests.size());
861     DeletePolicyRequest actualRequest = ((DeletePolicyRequest) actualRequests.get(0));
862 
863     Assert.assertEquals(name, actualRequest.getName());
864     Assert.assertTrue(
865         channelProvider.isHeaderSent(
866             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
867             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
868   }
869 
870   @Test
deletePolicyExceptionTest2()871   public void deletePolicyExceptionTest2() throws Exception {
872     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
873     mockOrgPolicy.addException(exception);
874 
875     try {
876       String name = "name3373707";
877       client.deletePolicy(name);
878       Assert.fail("No exception raised");
879     } catch (InvalidArgumentException e) {
880       // Expected exception.
881     }
882   }
883 }
884