• 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.osconfig.v1;
18 
19 import static com.google.cloud.osconfig.v1.OsConfigZonalServiceClient.ListInventoriesPagedResponse;
20 import static com.google.cloud.osconfig.v1.OsConfigZonalServiceClient.ListOSPolicyAssignmentReportsPagedResponse;
21 import static com.google.cloud.osconfig.v1.OsConfigZonalServiceClient.ListOSPolicyAssignmentRevisionsPagedResponse;
22 import static com.google.cloud.osconfig.v1.OsConfigZonalServiceClient.ListOSPolicyAssignmentsPagedResponse;
23 import static com.google.cloud.osconfig.v1.OsConfigZonalServiceClient.ListVulnerabilityReportsPagedResponse;
24 
25 import com.google.api.gax.core.NoCredentialsProvider;
26 import com.google.api.gax.httpjson.GaxHttpJsonProperties;
27 import com.google.api.gax.httpjson.testing.MockHttpService;
28 import com.google.api.gax.rpc.ApiClientHeaderProvider;
29 import com.google.api.gax.rpc.ApiException;
30 import com.google.api.gax.rpc.ApiExceptionFactory;
31 import com.google.api.gax.rpc.InvalidArgumentException;
32 import com.google.api.gax.rpc.StatusCode;
33 import com.google.api.gax.rpc.testing.FakeStatusCode;
34 import com.google.cloud.osconfig.v1.stub.HttpJsonOsConfigZonalServiceStub;
35 import com.google.common.collect.Lists;
36 import com.google.longrunning.Operation;
37 import com.google.protobuf.Any;
38 import com.google.protobuf.Empty;
39 import com.google.protobuf.FieldMask;
40 import com.google.protobuf.Timestamp;
41 import java.io.IOException;
42 import java.util.ArrayList;
43 import java.util.Arrays;
44 import java.util.HashMap;
45 import java.util.List;
46 import java.util.concurrent.ExecutionException;
47 import javax.annotation.Generated;
48 import org.junit.After;
49 import org.junit.AfterClass;
50 import org.junit.Assert;
51 import org.junit.Before;
52 import org.junit.BeforeClass;
53 import org.junit.Test;
54 
55 @Generated("by gapic-generator-java")
56 public class OsConfigZonalServiceClientHttpJsonTest {
57   private static MockHttpService mockService;
58   private static OsConfigZonalServiceClient client;
59 
60   @BeforeClass
startStaticServer()61   public static void startStaticServer() throws IOException {
62     mockService =
63         new MockHttpService(
64             HttpJsonOsConfigZonalServiceStub.getMethodDescriptors(),
65             OsConfigZonalServiceSettings.getDefaultEndpoint());
66     OsConfigZonalServiceSettings settings =
67         OsConfigZonalServiceSettings.newHttpJsonBuilder()
68             .setTransportChannelProvider(
69                 OsConfigZonalServiceSettings.defaultHttpJsonTransportProviderBuilder()
70                     .setHttpTransport(mockService)
71                     .build())
72             .setCredentialsProvider(NoCredentialsProvider.create())
73             .build();
74     client = OsConfigZonalServiceClient.create(settings);
75   }
76 
77   @AfterClass
stopServer()78   public static void stopServer() {
79     client.close();
80   }
81 
82   @Before
setUp()83   public void setUp() {}
84 
85   @After
tearDown()86   public void tearDown() throws Exception {
87     mockService.reset();
88   }
89 
90   @Test
createOSPolicyAssignmentTest()91   public void createOSPolicyAssignmentTest() throws Exception {
92     OSPolicyAssignment expectedResponse =
93         OSPolicyAssignment.newBuilder()
94             .setName(
95                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
96                     .toString())
97             .setDescription("description-1724546052")
98             .addAllOsPolicies(new ArrayList<OSPolicy>())
99             .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
100             .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
101             .setRevisionId("revisionId-1507445162")
102             .setRevisionCreateTime(Timestamp.newBuilder().build())
103             .setEtag("etag3123477")
104             .setBaseline(true)
105             .setDeleted(true)
106             .setReconciling(true)
107             .setUid("uid115792")
108             .build();
109     Operation resultOperation =
110         Operation.newBuilder()
111             .setName("createOSPolicyAssignmentTest")
112             .setDone(true)
113             .setResponse(Any.pack(expectedResponse))
114             .build();
115     mockService.addResponse(resultOperation);
116 
117     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
118     OSPolicyAssignment osPolicyAssignment = OSPolicyAssignment.newBuilder().build();
119     String osPolicyAssignmentId = "osPolicyAssignmentId-44079074";
120 
121     OSPolicyAssignment actualResponse =
122         client
123             .createOSPolicyAssignmentAsync(parent, osPolicyAssignment, osPolicyAssignmentId)
124             .get();
125     Assert.assertEquals(expectedResponse, actualResponse);
126 
127     List<String> actualRequests = mockService.getRequestPaths();
128     Assert.assertEquals(1, actualRequests.size());
129 
130     String apiClientHeaderKey =
131         mockService
132             .getRequestHeaders()
133             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
134             .iterator()
135             .next();
136     Assert.assertTrue(
137         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
138             .matcher(apiClientHeaderKey)
139             .matches());
140   }
141 
142   @Test
createOSPolicyAssignmentExceptionTest()143   public void createOSPolicyAssignmentExceptionTest() throws Exception {
144     ApiException exception =
145         ApiExceptionFactory.createException(
146             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
147     mockService.addException(exception);
148 
149     try {
150       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
151       OSPolicyAssignment osPolicyAssignment = OSPolicyAssignment.newBuilder().build();
152       String osPolicyAssignmentId = "osPolicyAssignmentId-44079074";
153       client.createOSPolicyAssignmentAsync(parent, osPolicyAssignment, osPolicyAssignmentId).get();
154       Assert.fail("No exception raised");
155     } catch (ExecutionException e) {
156     }
157   }
158 
159   @Test
createOSPolicyAssignmentTest2()160   public void createOSPolicyAssignmentTest2() throws Exception {
161     OSPolicyAssignment expectedResponse =
162         OSPolicyAssignment.newBuilder()
163             .setName(
164                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
165                     .toString())
166             .setDescription("description-1724546052")
167             .addAllOsPolicies(new ArrayList<OSPolicy>())
168             .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
169             .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
170             .setRevisionId("revisionId-1507445162")
171             .setRevisionCreateTime(Timestamp.newBuilder().build())
172             .setEtag("etag3123477")
173             .setBaseline(true)
174             .setDeleted(true)
175             .setReconciling(true)
176             .setUid("uid115792")
177             .build();
178     Operation resultOperation =
179         Operation.newBuilder()
180             .setName("createOSPolicyAssignmentTest")
181             .setDone(true)
182             .setResponse(Any.pack(expectedResponse))
183             .build();
184     mockService.addResponse(resultOperation);
185 
186     String parent = "projects/project-5833/locations/location-5833";
187     OSPolicyAssignment osPolicyAssignment = OSPolicyAssignment.newBuilder().build();
188     String osPolicyAssignmentId = "osPolicyAssignmentId-44079074";
189 
190     OSPolicyAssignment actualResponse =
191         client
192             .createOSPolicyAssignmentAsync(parent, osPolicyAssignment, osPolicyAssignmentId)
193             .get();
194     Assert.assertEquals(expectedResponse, actualResponse);
195 
196     List<String> actualRequests = mockService.getRequestPaths();
197     Assert.assertEquals(1, actualRequests.size());
198 
199     String apiClientHeaderKey =
200         mockService
201             .getRequestHeaders()
202             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
203             .iterator()
204             .next();
205     Assert.assertTrue(
206         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
207             .matcher(apiClientHeaderKey)
208             .matches());
209   }
210 
211   @Test
createOSPolicyAssignmentExceptionTest2()212   public void createOSPolicyAssignmentExceptionTest2() throws Exception {
213     ApiException exception =
214         ApiExceptionFactory.createException(
215             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
216     mockService.addException(exception);
217 
218     try {
219       String parent = "projects/project-5833/locations/location-5833";
220       OSPolicyAssignment osPolicyAssignment = OSPolicyAssignment.newBuilder().build();
221       String osPolicyAssignmentId = "osPolicyAssignmentId-44079074";
222       client.createOSPolicyAssignmentAsync(parent, osPolicyAssignment, osPolicyAssignmentId).get();
223       Assert.fail("No exception raised");
224     } catch (ExecutionException e) {
225     }
226   }
227 
228   @Test
updateOSPolicyAssignmentTest()229   public void updateOSPolicyAssignmentTest() throws Exception {
230     OSPolicyAssignment expectedResponse =
231         OSPolicyAssignment.newBuilder()
232             .setName(
233                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
234                     .toString())
235             .setDescription("description-1724546052")
236             .addAllOsPolicies(new ArrayList<OSPolicy>())
237             .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
238             .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
239             .setRevisionId("revisionId-1507445162")
240             .setRevisionCreateTime(Timestamp.newBuilder().build())
241             .setEtag("etag3123477")
242             .setBaseline(true)
243             .setDeleted(true)
244             .setReconciling(true)
245             .setUid("uid115792")
246             .build();
247     Operation resultOperation =
248         Operation.newBuilder()
249             .setName("updateOSPolicyAssignmentTest")
250             .setDone(true)
251             .setResponse(Any.pack(expectedResponse))
252             .build();
253     mockService.addResponse(resultOperation);
254 
255     OSPolicyAssignment osPolicyAssignment =
256         OSPolicyAssignment.newBuilder()
257             .setName(
258                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
259                     .toString())
260             .setDescription("description-1724546052")
261             .addAllOsPolicies(new ArrayList<OSPolicy>())
262             .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
263             .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
264             .setRevisionId("revisionId-1507445162")
265             .setRevisionCreateTime(Timestamp.newBuilder().build())
266             .setEtag("etag3123477")
267             .setBaseline(true)
268             .setDeleted(true)
269             .setReconciling(true)
270             .setUid("uid115792")
271             .build();
272     FieldMask updateMask = FieldMask.newBuilder().build();
273 
274     OSPolicyAssignment actualResponse =
275         client.updateOSPolicyAssignmentAsync(osPolicyAssignment, updateMask).get();
276     Assert.assertEquals(expectedResponse, actualResponse);
277 
278     List<String> actualRequests = mockService.getRequestPaths();
279     Assert.assertEquals(1, actualRequests.size());
280 
281     String apiClientHeaderKey =
282         mockService
283             .getRequestHeaders()
284             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
285             .iterator()
286             .next();
287     Assert.assertTrue(
288         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
289             .matcher(apiClientHeaderKey)
290             .matches());
291   }
292 
293   @Test
updateOSPolicyAssignmentExceptionTest()294   public void updateOSPolicyAssignmentExceptionTest() throws Exception {
295     ApiException exception =
296         ApiExceptionFactory.createException(
297             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
298     mockService.addException(exception);
299 
300     try {
301       OSPolicyAssignment osPolicyAssignment =
302           OSPolicyAssignment.newBuilder()
303               .setName(
304                   OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
305                       .toString())
306               .setDescription("description-1724546052")
307               .addAllOsPolicies(new ArrayList<OSPolicy>())
308               .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
309               .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
310               .setRevisionId("revisionId-1507445162")
311               .setRevisionCreateTime(Timestamp.newBuilder().build())
312               .setEtag("etag3123477")
313               .setBaseline(true)
314               .setDeleted(true)
315               .setReconciling(true)
316               .setUid("uid115792")
317               .build();
318       FieldMask updateMask = FieldMask.newBuilder().build();
319       client.updateOSPolicyAssignmentAsync(osPolicyAssignment, updateMask).get();
320       Assert.fail("No exception raised");
321     } catch (ExecutionException e) {
322     }
323   }
324 
325   @Test
getOSPolicyAssignmentTest()326   public void getOSPolicyAssignmentTest() throws Exception {
327     OSPolicyAssignment expectedResponse =
328         OSPolicyAssignment.newBuilder()
329             .setName(
330                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
331                     .toString())
332             .setDescription("description-1724546052")
333             .addAllOsPolicies(new ArrayList<OSPolicy>())
334             .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
335             .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
336             .setRevisionId("revisionId-1507445162")
337             .setRevisionCreateTime(Timestamp.newBuilder().build())
338             .setEtag("etag3123477")
339             .setBaseline(true)
340             .setDeleted(true)
341             .setReconciling(true)
342             .setUid("uid115792")
343             .build();
344     mockService.addResponse(expectedResponse);
345 
346     OSPolicyAssignmentName name =
347         OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]");
348 
349     OSPolicyAssignment actualResponse = client.getOSPolicyAssignment(name);
350     Assert.assertEquals(expectedResponse, actualResponse);
351 
352     List<String> actualRequests = mockService.getRequestPaths();
353     Assert.assertEquals(1, actualRequests.size());
354 
355     String apiClientHeaderKey =
356         mockService
357             .getRequestHeaders()
358             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
359             .iterator()
360             .next();
361     Assert.assertTrue(
362         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
363             .matcher(apiClientHeaderKey)
364             .matches());
365   }
366 
367   @Test
getOSPolicyAssignmentExceptionTest()368   public void getOSPolicyAssignmentExceptionTest() throws Exception {
369     ApiException exception =
370         ApiExceptionFactory.createException(
371             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
372     mockService.addException(exception);
373 
374     try {
375       OSPolicyAssignmentName name =
376           OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]");
377       client.getOSPolicyAssignment(name);
378       Assert.fail("No exception raised");
379     } catch (InvalidArgumentException e) {
380       // Expected exception.
381     }
382   }
383 
384   @Test
getOSPolicyAssignmentTest2()385   public void getOSPolicyAssignmentTest2() throws Exception {
386     OSPolicyAssignment expectedResponse =
387         OSPolicyAssignment.newBuilder()
388             .setName(
389                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
390                     .toString())
391             .setDescription("description-1724546052")
392             .addAllOsPolicies(new ArrayList<OSPolicy>())
393             .setInstanceFilter(OSPolicyAssignment.InstanceFilter.newBuilder().build())
394             .setRollout(OSPolicyAssignment.Rollout.newBuilder().build())
395             .setRevisionId("revisionId-1507445162")
396             .setRevisionCreateTime(Timestamp.newBuilder().build())
397             .setEtag("etag3123477")
398             .setBaseline(true)
399             .setDeleted(true)
400             .setReconciling(true)
401             .setUid("uid115792")
402             .build();
403     mockService.addResponse(expectedResponse);
404 
405     String name =
406         "projects/project-1862/locations/location-1862/osPolicyAssignments/osPolicyAssignment-1862";
407 
408     OSPolicyAssignment actualResponse = client.getOSPolicyAssignment(name);
409     Assert.assertEquals(expectedResponse, actualResponse);
410 
411     List<String> actualRequests = mockService.getRequestPaths();
412     Assert.assertEquals(1, actualRequests.size());
413 
414     String apiClientHeaderKey =
415         mockService
416             .getRequestHeaders()
417             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
418             .iterator()
419             .next();
420     Assert.assertTrue(
421         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
422             .matcher(apiClientHeaderKey)
423             .matches());
424   }
425 
426   @Test
getOSPolicyAssignmentExceptionTest2()427   public void getOSPolicyAssignmentExceptionTest2() throws Exception {
428     ApiException exception =
429         ApiExceptionFactory.createException(
430             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
431     mockService.addException(exception);
432 
433     try {
434       String name =
435           "projects/project-1862/locations/location-1862/osPolicyAssignments/osPolicyAssignment-1862";
436       client.getOSPolicyAssignment(name);
437       Assert.fail("No exception raised");
438     } catch (InvalidArgumentException e) {
439       // Expected exception.
440     }
441   }
442 
443   @Test
listOSPolicyAssignmentsTest()444   public void listOSPolicyAssignmentsTest() throws Exception {
445     OSPolicyAssignment responsesElement = OSPolicyAssignment.newBuilder().build();
446     ListOSPolicyAssignmentsResponse expectedResponse =
447         ListOSPolicyAssignmentsResponse.newBuilder()
448             .setNextPageToken("")
449             .addAllOsPolicyAssignments(Arrays.asList(responsesElement))
450             .build();
451     mockService.addResponse(expectedResponse);
452 
453     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
454 
455     ListOSPolicyAssignmentsPagedResponse pagedListResponse = client.listOSPolicyAssignments(parent);
456 
457     List<OSPolicyAssignment> resources = Lists.newArrayList(pagedListResponse.iterateAll());
458 
459     Assert.assertEquals(1, resources.size());
460     Assert.assertEquals(expectedResponse.getOsPolicyAssignmentsList().get(0), resources.get(0));
461 
462     List<String> actualRequests = mockService.getRequestPaths();
463     Assert.assertEquals(1, actualRequests.size());
464 
465     String apiClientHeaderKey =
466         mockService
467             .getRequestHeaders()
468             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
469             .iterator()
470             .next();
471     Assert.assertTrue(
472         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
473             .matcher(apiClientHeaderKey)
474             .matches());
475   }
476 
477   @Test
listOSPolicyAssignmentsExceptionTest()478   public void listOSPolicyAssignmentsExceptionTest() throws Exception {
479     ApiException exception =
480         ApiExceptionFactory.createException(
481             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
482     mockService.addException(exception);
483 
484     try {
485       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
486       client.listOSPolicyAssignments(parent);
487       Assert.fail("No exception raised");
488     } catch (InvalidArgumentException e) {
489       // Expected exception.
490     }
491   }
492 
493   @Test
listOSPolicyAssignmentsTest2()494   public void listOSPolicyAssignmentsTest2() throws Exception {
495     OSPolicyAssignment responsesElement = OSPolicyAssignment.newBuilder().build();
496     ListOSPolicyAssignmentsResponse expectedResponse =
497         ListOSPolicyAssignmentsResponse.newBuilder()
498             .setNextPageToken("")
499             .addAllOsPolicyAssignments(Arrays.asList(responsesElement))
500             .build();
501     mockService.addResponse(expectedResponse);
502 
503     String parent = "projects/project-5833/locations/location-5833";
504 
505     ListOSPolicyAssignmentsPagedResponse pagedListResponse = client.listOSPolicyAssignments(parent);
506 
507     List<OSPolicyAssignment> resources = Lists.newArrayList(pagedListResponse.iterateAll());
508 
509     Assert.assertEquals(1, resources.size());
510     Assert.assertEquals(expectedResponse.getOsPolicyAssignmentsList().get(0), resources.get(0));
511 
512     List<String> actualRequests = mockService.getRequestPaths();
513     Assert.assertEquals(1, actualRequests.size());
514 
515     String apiClientHeaderKey =
516         mockService
517             .getRequestHeaders()
518             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
519             .iterator()
520             .next();
521     Assert.assertTrue(
522         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
523             .matcher(apiClientHeaderKey)
524             .matches());
525   }
526 
527   @Test
listOSPolicyAssignmentsExceptionTest2()528   public void listOSPolicyAssignmentsExceptionTest2() throws Exception {
529     ApiException exception =
530         ApiExceptionFactory.createException(
531             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
532     mockService.addException(exception);
533 
534     try {
535       String parent = "projects/project-5833/locations/location-5833";
536       client.listOSPolicyAssignments(parent);
537       Assert.fail("No exception raised");
538     } catch (InvalidArgumentException e) {
539       // Expected exception.
540     }
541   }
542 
543   @Test
listOSPolicyAssignmentRevisionsTest()544   public void listOSPolicyAssignmentRevisionsTest() throws Exception {
545     OSPolicyAssignment responsesElement = OSPolicyAssignment.newBuilder().build();
546     ListOSPolicyAssignmentRevisionsResponse expectedResponse =
547         ListOSPolicyAssignmentRevisionsResponse.newBuilder()
548             .setNextPageToken("")
549             .addAllOsPolicyAssignments(Arrays.asList(responsesElement))
550             .build();
551     mockService.addResponse(expectedResponse);
552 
553     OSPolicyAssignmentName name =
554         OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]");
555 
556     ListOSPolicyAssignmentRevisionsPagedResponse pagedListResponse =
557         client.listOSPolicyAssignmentRevisions(name);
558 
559     List<OSPolicyAssignment> resources = Lists.newArrayList(pagedListResponse.iterateAll());
560 
561     Assert.assertEquals(1, resources.size());
562     Assert.assertEquals(expectedResponse.getOsPolicyAssignmentsList().get(0), resources.get(0));
563 
564     List<String> actualRequests = mockService.getRequestPaths();
565     Assert.assertEquals(1, actualRequests.size());
566 
567     String apiClientHeaderKey =
568         mockService
569             .getRequestHeaders()
570             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
571             .iterator()
572             .next();
573     Assert.assertTrue(
574         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
575             .matcher(apiClientHeaderKey)
576             .matches());
577   }
578 
579   @Test
listOSPolicyAssignmentRevisionsExceptionTest()580   public void listOSPolicyAssignmentRevisionsExceptionTest() throws Exception {
581     ApiException exception =
582         ApiExceptionFactory.createException(
583             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
584     mockService.addException(exception);
585 
586     try {
587       OSPolicyAssignmentName name =
588           OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]");
589       client.listOSPolicyAssignmentRevisions(name);
590       Assert.fail("No exception raised");
591     } catch (InvalidArgumentException e) {
592       // Expected exception.
593     }
594   }
595 
596   @Test
listOSPolicyAssignmentRevisionsTest2()597   public void listOSPolicyAssignmentRevisionsTest2() throws Exception {
598     OSPolicyAssignment responsesElement = OSPolicyAssignment.newBuilder().build();
599     ListOSPolicyAssignmentRevisionsResponse expectedResponse =
600         ListOSPolicyAssignmentRevisionsResponse.newBuilder()
601             .setNextPageToken("")
602             .addAllOsPolicyAssignments(Arrays.asList(responsesElement))
603             .build();
604     mockService.addResponse(expectedResponse);
605 
606     String name =
607         "projects/project-1862/locations/location-1862/osPolicyAssignments/osPolicyAssignment-1862";
608 
609     ListOSPolicyAssignmentRevisionsPagedResponse pagedListResponse =
610         client.listOSPolicyAssignmentRevisions(name);
611 
612     List<OSPolicyAssignment> resources = Lists.newArrayList(pagedListResponse.iterateAll());
613 
614     Assert.assertEquals(1, resources.size());
615     Assert.assertEquals(expectedResponse.getOsPolicyAssignmentsList().get(0), resources.get(0));
616 
617     List<String> actualRequests = mockService.getRequestPaths();
618     Assert.assertEquals(1, actualRequests.size());
619 
620     String apiClientHeaderKey =
621         mockService
622             .getRequestHeaders()
623             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
624             .iterator()
625             .next();
626     Assert.assertTrue(
627         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
628             .matcher(apiClientHeaderKey)
629             .matches());
630   }
631 
632   @Test
listOSPolicyAssignmentRevisionsExceptionTest2()633   public void listOSPolicyAssignmentRevisionsExceptionTest2() throws Exception {
634     ApiException exception =
635         ApiExceptionFactory.createException(
636             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
637     mockService.addException(exception);
638 
639     try {
640       String name =
641           "projects/project-1862/locations/location-1862/osPolicyAssignments/osPolicyAssignment-1862";
642       client.listOSPolicyAssignmentRevisions(name);
643       Assert.fail("No exception raised");
644     } catch (InvalidArgumentException e) {
645       // Expected exception.
646     }
647   }
648 
649   @Test
deleteOSPolicyAssignmentTest()650   public void deleteOSPolicyAssignmentTest() throws Exception {
651     Empty expectedResponse = Empty.newBuilder().build();
652     Operation resultOperation =
653         Operation.newBuilder()
654             .setName("deleteOSPolicyAssignmentTest")
655             .setDone(true)
656             .setResponse(Any.pack(expectedResponse))
657             .build();
658     mockService.addResponse(resultOperation);
659 
660     OSPolicyAssignmentName name =
661         OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]");
662 
663     client.deleteOSPolicyAssignmentAsync(name).get();
664 
665     List<String> actualRequests = mockService.getRequestPaths();
666     Assert.assertEquals(1, actualRequests.size());
667 
668     String apiClientHeaderKey =
669         mockService
670             .getRequestHeaders()
671             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
672             .iterator()
673             .next();
674     Assert.assertTrue(
675         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
676             .matcher(apiClientHeaderKey)
677             .matches());
678   }
679 
680   @Test
deleteOSPolicyAssignmentExceptionTest()681   public void deleteOSPolicyAssignmentExceptionTest() throws Exception {
682     ApiException exception =
683         ApiExceptionFactory.createException(
684             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
685     mockService.addException(exception);
686 
687     try {
688       OSPolicyAssignmentName name =
689           OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]");
690       client.deleteOSPolicyAssignmentAsync(name).get();
691       Assert.fail("No exception raised");
692     } catch (ExecutionException e) {
693     }
694   }
695 
696   @Test
deleteOSPolicyAssignmentTest2()697   public void deleteOSPolicyAssignmentTest2() throws Exception {
698     Empty expectedResponse = Empty.newBuilder().build();
699     Operation resultOperation =
700         Operation.newBuilder()
701             .setName("deleteOSPolicyAssignmentTest")
702             .setDone(true)
703             .setResponse(Any.pack(expectedResponse))
704             .build();
705     mockService.addResponse(resultOperation);
706 
707     String name =
708         "projects/project-1862/locations/location-1862/osPolicyAssignments/osPolicyAssignment-1862";
709 
710     client.deleteOSPolicyAssignmentAsync(name).get();
711 
712     List<String> actualRequests = mockService.getRequestPaths();
713     Assert.assertEquals(1, actualRequests.size());
714 
715     String apiClientHeaderKey =
716         mockService
717             .getRequestHeaders()
718             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
719             .iterator()
720             .next();
721     Assert.assertTrue(
722         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
723             .matcher(apiClientHeaderKey)
724             .matches());
725   }
726 
727   @Test
deleteOSPolicyAssignmentExceptionTest2()728   public void deleteOSPolicyAssignmentExceptionTest2() throws Exception {
729     ApiException exception =
730         ApiExceptionFactory.createException(
731             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
732     mockService.addException(exception);
733 
734     try {
735       String name =
736           "projects/project-1862/locations/location-1862/osPolicyAssignments/osPolicyAssignment-1862";
737       client.deleteOSPolicyAssignmentAsync(name).get();
738       Assert.fail("No exception raised");
739     } catch (ExecutionException e) {
740     }
741   }
742 
743   @Test
getOSPolicyAssignmentReportTest()744   public void getOSPolicyAssignmentReportTest() throws Exception {
745     OSPolicyAssignmentReport expectedResponse =
746         OSPolicyAssignmentReport.newBuilder()
747             .setName(
748                 OSPolicyAssignmentReportName.of(
749                         "[PROJECT]", "[LOCATION]", "[INSTANCE]", "[ASSIGNMENT]")
750                     .toString())
751             .setInstance("instance555127957")
752             .setOsPolicyAssignment(
753                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
754                     .toString())
755             .addAllOsPolicyCompliances(new ArrayList<OSPolicyAssignmentReport.OSPolicyCompliance>())
756             .setUpdateTime(Timestamp.newBuilder().build())
757             .setLastRunId("lastRunId2001336336")
758             .build();
759     mockService.addResponse(expectedResponse);
760 
761     OSPolicyAssignmentReportName name =
762         OSPolicyAssignmentReportName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]", "[ASSIGNMENT]");
763 
764     OSPolicyAssignmentReport actualResponse = client.getOSPolicyAssignmentReport(name);
765     Assert.assertEquals(expectedResponse, actualResponse);
766 
767     List<String> actualRequests = mockService.getRequestPaths();
768     Assert.assertEquals(1, actualRequests.size());
769 
770     String apiClientHeaderKey =
771         mockService
772             .getRequestHeaders()
773             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
774             .iterator()
775             .next();
776     Assert.assertTrue(
777         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
778             .matcher(apiClientHeaderKey)
779             .matches());
780   }
781 
782   @Test
getOSPolicyAssignmentReportExceptionTest()783   public void getOSPolicyAssignmentReportExceptionTest() throws Exception {
784     ApiException exception =
785         ApiExceptionFactory.createException(
786             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
787     mockService.addException(exception);
788 
789     try {
790       OSPolicyAssignmentReportName name =
791           OSPolicyAssignmentReportName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]", "[ASSIGNMENT]");
792       client.getOSPolicyAssignmentReport(name);
793       Assert.fail("No exception raised");
794     } catch (InvalidArgumentException e) {
795       // Expected exception.
796     }
797   }
798 
799   @Test
getOSPolicyAssignmentReportTest2()800   public void getOSPolicyAssignmentReportTest2() throws Exception {
801     OSPolicyAssignmentReport expectedResponse =
802         OSPolicyAssignmentReport.newBuilder()
803             .setName(
804                 OSPolicyAssignmentReportName.of(
805                         "[PROJECT]", "[LOCATION]", "[INSTANCE]", "[ASSIGNMENT]")
806                     .toString())
807             .setInstance("instance555127957")
808             .setOsPolicyAssignment(
809                 OSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[OS_POLICY_ASSIGNMENT]")
810                     .toString())
811             .addAllOsPolicyCompliances(new ArrayList<OSPolicyAssignmentReport.OSPolicyCompliance>())
812             .setUpdateTime(Timestamp.newBuilder().build())
813             .setLastRunId("lastRunId2001336336")
814             .build();
815     mockService.addResponse(expectedResponse);
816 
817     String name =
818         "projects/project-6771/locations/location-6771/instances/instance-6771/osPolicyAssignments/osPolicyAssignment-6771/report";
819 
820     OSPolicyAssignmentReport actualResponse = client.getOSPolicyAssignmentReport(name);
821     Assert.assertEquals(expectedResponse, actualResponse);
822 
823     List<String> actualRequests = mockService.getRequestPaths();
824     Assert.assertEquals(1, actualRequests.size());
825 
826     String apiClientHeaderKey =
827         mockService
828             .getRequestHeaders()
829             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
830             .iterator()
831             .next();
832     Assert.assertTrue(
833         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
834             .matcher(apiClientHeaderKey)
835             .matches());
836   }
837 
838   @Test
getOSPolicyAssignmentReportExceptionTest2()839   public void getOSPolicyAssignmentReportExceptionTest2() throws Exception {
840     ApiException exception =
841         ApiExceptionFactory.createException(
842             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
843     mockService.addException(exception);
844 
845     try {
846       String name =
847           "projects/project-6771/locations/location-6771/instances/instance-6771/osPolicyAssignments/osPolicyAssignment-6771/report";
848       client.getOSPolicyAssignmentReport(name);
849       Assert.fail("No exception raised");
850     } catch (InvalidArgumentException e) {
851       // Expected exception.
852     }
853   }
854 
855   @Test
listOSPolicyAssignmentReportsTest()856   public void listOSPolicyAssignmentReportsTest() throws Exception {
857     OSPolicyAssignmentReport responsesElement = OSPolicyAssignmentReport.newBuilder().build();
858     ListOSPolicyAssignmentReportsResponse expectedResponse =
859         ListOSPolicyAssignmentReportsResponse.newBuilder()
860             .setNextPageToken("")
861             .addAllOsPolicyAssignmentReports(Arrays.asList(responsesElement))
862             .build();
863     mockService.addResponse(expectedResponse);
864 
865     InstanceOSPolicyAssignmentName parent =
866         InstanceOSPolicyAssignmentName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]", "[ASSIGNMENT]");
867 
868     ListOSPolicyAssignmentReportsPagedResponse pagedListResponse =
869         client.listOSPolicyAssignmentReports(parent);
870 
871     List<OSPolicyAssignmentReport> resources = Lists.newArrayList(pagedListResponse.iterateAll());
872 
873     Assert.assertEquals(1, resources.size());
874     Assert.assertEquals(
875         expectedResponse.getOsPolicyAssignmentReportsList().get(0), resources.get(0));
876 
877     List<String> actualRequests = mockService.getRequestPaths();
878     Assert.assertEquals(1, actualRequests.size());
879 
880     String apiClientHeaderKey =
881         mockService
882             .getRequestHeaders()
883             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
884             .iterator()
885             .next();
886     Assert.assertTrue(
887         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
888             .matcher(apiClientHeaderKey)
889             .matches());
890   }
891 
892   @Test
listOSPolicyAssignmentReportsExceptionTest()893   public void listOSPolicyAssignmentReportsExceptionTest() throws Exception {
894     ApiException exception =
895         ApiExceptionFactory.createException(
896             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
897     mockService.addException(exception);
898 
899     try {
900       InstanceOSPolicyAssignmentName parent =
901           InstanceOSPolicyAssignmentName.of(
902               "[PROJECT]", "[LOCATION]", "[INSTANCE]", "[ASSIGNMENT]");
903       client.listOSPolicyAssignmentReports(parent);
904       Assert.fail("No exception raised");
905     } catch (InvalidArgumentException e) {
906       // Expected exception.
907     }
908   }
909 
910   @Test
listOSPolicyAssignmentReportsTest2()911   public void listOSPolicyAssignmentReportsTest2() throws Exception {
912     OSPolicyAssignmentReport responsesElement = OSPolicyAssignmentReport.newBuilder().build();
913     ListOSPolicyAssignmentReportsResponse expectedResponse =
914         ListOSPolicyAssignmentReportsResponse.newBuilder()
915             .setNextPageToken("")
916             .addAllOsPolicyAssignmentReports(Arrays.asList(responsesElement))
917             .build();
918     mockService.addResponse(expectedResponse);
919 
920     String parent =
921         "projects/project-6863/locations/location-6863/instances/instance-6863/osPolicyAssignments/osPolicyAssignment-6863";
922 
923     ListOSPolicyAssignmentReportsPagedResponse pagedListResponse =
924         client.listOSPolicyAssignmentReports(parent);
925 
926     List<OSPolicyAssignmentReport> resources = Lists.newArrayList(pagedListResponse.iterateAll());
927 
928     Assert.assertEquals(1, resources.size());
929     Assert.assertEquals(
930         expectedResponse.getOsPolicyAssignmentReportsList().get(0), resources.get(0));
931 
932     List<String> actualRequests = mockService.getRequestPaths();
933     Assert.assertEquals(1, actualRequests.size());
934 
935     String apiClientHeaderKey =
936         mockService
937             .getRequestHeaders()
938             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
939             .iterator()
940             .next();
941     Assert.assertTrue(
942         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
943             .matcher(apiClientHeaderKey)
944             .matches());
945   }
946 
947   @Test
listOSPolicyAssignmentReportsExceptionTest2()948   public void listOSPolicyAssignmentReportsExceptionTest2() throws Exception {
949     ApiException exception =
950         ApiExceptionFactory.createException(
951             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
952     mockService.addException(exception);
953 
954     try {
955       String parent =
956           "projects/project-6863/locations/location-6863/instances/instance-6863/osPolicyAssignments/osPolicyAssignment-6863";
957       client.listOSPolicyAssignmentReports(parent);
958       Assert.fail("No exception raised");
959     } catch (InvalidArgumentException e) {
960       // Expected exception.
961     }
962   }
963 
964   @Test
getInventoryTest()965   public void getInventoryTest() throws Exception {
966     Inventory expectedResponse =
967         Inventory.newBuilder()
968             .setName(InventoryName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]").toString())
969             .setOsInfo(Inventory.OsInfo.newBuilder().build())
970             .putAllItems(new HashMap<String, Inventory.Item>())
971             .setUpdateTime(Timestamp.newBuilder().build())
972             .build();
973     mockService.addResponse(expectedResponse);
974 
975     InventoryName name = InventoryName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]");
976 
977     Inventory actualResponse = client.getInventory(name);
978     Assert.assertEquals(expectedResponse, actualResponse);
979 
980     List<String> actualRequests = mockService.getRequestPaths();
981     Assert.assertEquals(1, actualRequests.size());
982 
983     String apiClientHeaderKey =
984         mockService
985             .getRequestHeaders()
986             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
987             .iterator()
988             .next();
989     Assert.assertTrue(
990         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
991             .matcher(apiClientHeaderKey)
992             .matches());
993   }
994 
995   @Test
getInventoryExceptionTest()996   public void getInventoryExceptionTest() throws Exception {
997     ApiException exception =
998         ApiExceptionFactory.createException(
999             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1000     mockService.addException(exception);
1001 
1002     try {
1003       InventoryName name = InventoryName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1004       client.getInventory(name);
1005       Assert.fail("No exception raised");
1006     } catch (InvalidArgumentException e) {
1007       // Expected exception.
1008     }
1009   }
1010 
1011   @Test
getInventoryTest2()1012   public void getInventoryTest2() throws Exception {
1013     Inventory expectedResponse =
1014         Inventory.newBuilder()
1015             .setName(InventoryName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]").toString())
1016             .setOsInfo(Inventory.OsInfo.newBuilder().build())
1017             .putAllItems(new HashMap<String, Inventory.Item>())
1018             .setUpdateTime(Timestamp.newBuilder().build())
1019             .build();
1020     mockService.addResponse(expectedResponse);
1021 
1022     String name = "projects/project-943/locations/location-943/instances/instance-943/inventory";
1023 
1024     Inventory actualResponse = client.getInventory(name);
1025     Assert.assertEquals(expectedResponse, actualResponse);
1026 
1027     List<String> actualRequests = mockService.getRequestPaths();
1028     Assert.assertEquals(1, actualRequests.size());
1029 
1030     String apiClientHeaderKey =
1031         mockService
1032             .getRequestHeaders()
1033             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1034             .iterator()
1035             .next();
1036     Assert.assertTrue(
1037         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1038             .matcher(apiClientHeaderKey)
1039             .matches());
1040   }
1041 
1042   @Test
getInventoryExceptionTest2()1043   public void getInventoryExceptionTest2() throws Exception {
1044     ApiException exception =
1045         ApiExceptionFactory.createException(
1046             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1047     mockService.addException(exception);
1048 
1049     try {
1050       String name = "projects/project-943/locations/location-943/instances/instance-943/inventory";
1051       client.getInventory(name);
1052       Assert.fail("No exception raised");
1053     } catch (InvalidArgumentException e) {
1054       // Expected exception.
1055     }
1056   }
1057 
1058   @Test
listInventoriesTest()1059   public void listInventoriesTest() throws Exception {
1060     Inventory responsesElement = Inventory.newBuilder().build();
1061     ListInventoriesResponse expectedResponse =
1062         ListInventoriesResponse.newBuilder()
1063             .setNextPageToken("")
1064             .addAllInventories(Arrays.asList(responsesElement))
1065             .build();
1066     mockService.addResponse(expectedResponse);
1067 
1068     InstanceName parent =
1069         InstanceName.ofProjectLocationInstanceName("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1070 
1071     ListInventoriesPagedResponse pagedListResponse = client.listInventories(parent);
1072 
1073     List<Inventory> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1074 
1075     Assert.assertEquals(1, resources.size());
1076     Assert.assertEquals(expectedResponse.getInventoriesList().get(0), resources.get(0));
1077 
1078     List<String> actualRequests = mockService.getRequestPaths();
1079     Assert.assertEquals(1, actualRequests.size());
1080 
1081     String apiClientHeaderKey =
1082         mockService
1083             .getRequestHeaders()
1084             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1085             .iterator()
1086             .next();
1087     Assert.assertTrue(
1088         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1089             .matcher(apiClientHeaderKey)
1090             .matches());
1091   }
1092 
1093   @Test
listInventoriesExceptionTest()1094   public void listInventoriesExceptionTest() throws Exception {
1095     ApiException exception =
1096         ApiExceptionFactory.createException(
1097             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1098     mockService.addException(exception);
1099 
1100     try {
1101       InstanceName parent =
1102           InstanceName.ofProjectLocationInstanceName("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1103       client.listInventories(parent);
1104       Assert.fail("No exception raised");
1105     } catch (InvalidArgumentException e) {
1106       // Expected exception.
1107     }
1108   }
1109 
1110   @Test
listInventoriesTest2()1111   public void listInventoriesTest2() throws Exception {
1112     Inventory responsesElement = Inventory.newBuilder().build();
1113     ListInventoriesResponse expectedResponse =
1114         ListInventoriesResponse.newBuilder()
1115             .setNextPageToken("")
1116             .addAllInventories(Arrays.asList(responsesElement))
1117             .build();
1118     mockService.addResponse(expectedResponse);
1119 
1120     String parent = "projects/project-5197/locations/location-5197/instances/instance-5197";
1121 
1122     ListInventoriesPagedResponse pagedListResponse = client.listInventories(parent);
1123 
1124     List<Inventory> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1125 
1126     Assert.assertEquals(1, resources.size());
1127     Assert.assertEquals(expectedResponse.getInventoriesList().get(0), resources.get(0));
1128 
1129     List<String> actualRequests = mockService.getRequestPaths();
1130     Assert.assertEquals(1, actualRequests.size());
1131 
1132     String apiClientHeaderKey =
1133         mockService
1134             .getRequestHeaders()
1135             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1136             .iterator()
1137             .next();
1138     Assert.assertTrue(
1139         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1140             .matcher(apiClientHeaderKey)
1141             .matches());
1142   }
1143 
1144   @Test
listInventoriesExceptionTest2()1145   public void listInventoriesExceptionTest2() throws Exception {
1146     ApiException exception =
1147         ApiExceptionFactory.createException(
1148             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1149     mockService.addException(exception);
1150 
1151     try {
1152       String parent = "projects/project-5197/locations/location-5197/instances/instance-5197";
1153       client.listInventories(parent);
1154       Assert.fail("No exception raised");
1155     } catch (InvalidArgumentException e) {
1156       // Expected exception.
1157     }
1158   }
1159 
1160   @Test
getVulnerabilityReportTest()1161   public void getVulnerabilityReportTest() throws Exception {
1162     VulnerabilityReport expectedResponse =
1163         VulnerabilityReport.newBuilder()
1164             .setName(VulnerabilityReportName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]").toString())
1165             .addAllVulnerabilities(new ArrayList<VulnerabilityReport.Vulnerability>())
1166             .setUpdateTime(Timestamp.newBuilder().build())
1167             .build();
1168     mockService.addResponse(expectedResponse);
1169 
1170     VulnerabilityReportName name =
1171         VulnerabilityReportName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1172 
1173     VulnerabilityReport actualResponse = client.getVulnerabilityReport(name);
1174     Assert.assertEquals(expectedResponse, actualResponse);
1175 
1176     List<String> actualRequests = mockService.getRequestPaths();
1177     Assert.assertEquals(1, actualRequests.size());
1178 
1179     String apiClientHeaderKey =
1180         mockService
1181             .getRequestHeaders()
1182             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1183             .iterator()
1184             .next();
1185     Assert.assertTrue(
1186         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1187             .matcher(apiClientHeaderKey)
1188             .matches());
1189   }
1190 
1191   @Test
getVulnerabilityReportExceptionTest()1192   public void getVulnerabilityReportExceptionTest() throws Exception {
1193     ApiException exception =
1194         ApiExceptionFactory.createException(
1195             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1196     mockService.addException(exception);
1197 
1198     try {
1199       VulnerabilityReportName name =
1200           VulnerabilityReportName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1201       client.getVulnerabilityReport(name);
1202       Assert.fail("No exception raised");
1203     } catch (InvalidArgumentException e) {
1204       // Expected exception.
1205     }
1206   }
1207 
1208   @Test
getVulnerabilityReportTest2()1209   public void getVulnerabilityReportTest2() throws Exception {
1210     VulnerabilityReport expectedResponse =
1211         VulnerabilityReport.newBuilder()
1212             .setName(VulnerabilityReportName.of("[PROJECT]", "[LOCATION]", "[INSTANCE]").toString())
1213             .addAllVulnerabilities(new ArrayList<VulnerabilityReport.Vulnerability>())
1214             .setUpdateTime(Timestamp.newBuilder().build())
1215             .build();
1216     mockService.addResponse(expectedResponse);
1217 
1218     String name =
1219         "projects/project-2379/locations/location-2379/instances/instance-2379/vulnerabilityReport";
1220 
1221     VulnerabilityReport actualResponse = client.getVulnerabilityReport(name);
1222     Assert.assertEquals(expectedResponse, actualResponse);
1223 
1224     List<String> actualRequests = mockService.getRequestPaths();
1225     Assert.assertEquals(1, actualRequests.size());
1226 
1227     String apiClientHeaderKey =
1228         mockService
1229             .getRequestHeaders()
1230             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1231             .iterator()
1232             .next();
1233     Assert.assertTrue(
1234         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1235             .matcher(apiClientHeaderKey)
1236             .matches());
1237   }
1238 
1239   @Test
getVulnerabilityReportExceptionTest2()1240   public void getVulnerabilityReportExceptionTest2() throws Exception {
1241     ApiException exception =
1242         ApiExceptionFactory.createException(
1243             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1244     mockService.addException(exception);
1245 
1246     try {
1247       String name =
1248           "projects/project-2379/locations/location-2379/instances/instance-2379/vulnerabilityReport";
1249       client.getVulnerabilityReport(name);
1250       Assert.fail("No exception raised");
1251     } catch (InvalidArgumentException e) {
1252       // Expected exception.
1253     }
1254   }
1255 
1256   @Test
listVulnerabilityReportsTest()1257   public void listVulnerabilityReportsTest() throws Exception {
1258     VulnerabilityReport responsesElement = VulnerabilityReport.newBuilder().build();
1259     ListVulnerabilityReportsResponse expectedResponse =
1260         ListVulnerabilityReportsResponse.newBuilder()
1261             .setNextPageToken("")
1262             .addAllVulnerabilityReports(Arrays.asList(responsesElement))
1263             .build();
1264     mockService.addResponse(expectedResponse);
1265 
1266     InstanceName parent =
1267         InstanceName.ofProjectLocationInstanceName("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1268 
1269     ListVulnerabilityReportsPagedResponse pagedListResponse =
1270         client.listVulnerabilityReports(parent);
1271 
1272     List<VulnerabilityReport> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1273 
1274     Assert.assertEquals(1, resources.size());
1275     Assert.assertEquals(expectedResponse.getVulnerabilityReportsList().get(0), resources.get(0));
1276 
1277     List<String> actualRequests = mockService.getRequestPaths();
1278     Assert.assertEquals(1, actualRequests.size());
1279 
1280     String apiClientHeaderKey =
1281         mockService
1282             .getRequestHeaders()
1283             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1284             .iterator()
1285             .next();
1286     Assert.assertTrue(
1287         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1288             .matcher(apiClientHeaderKey)
1289             .matches());
1290   }
1291 
1292   @Test
listVulnerabilityReportsExceptionTest()1293   public void listVulnerabilityReportsExceptionTest() throws Exception {
1294     ApiException exception =
1295         ApiExceptionFactory.createException(
1296             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1297     mockService.addException(exception);
1298 
1299     try {
1300       InstanceName parent =
1301           InstanceName.ofProjectLocationInstanceName("[PROJECT]", "[LOCATION]", "[INSTANCE]");
1302       client.listVulnerabilityReports(parent);
1303       Assert.fail("No exception raised");
1304     } catch (InvalidArgumentException e) {
1305       // Expected exception.
1306     }
1307   }
1308 
1309   @Test
listVulnerabilityReportsTest2()1310   public void listVulnerabilityReportsTest2() throws Exception {
1311     VulnerabilityReport responsesElement = VulnerabilityReport.newBuilder().build();
1312     ListVulnerabilityReportsResponse expectedResponse =
1313         ListVulnerabilityReportsResponse.newBuilder()
1314             .setNextPageToken("")
1315             .addAllVulnerabilityReports(Arrays.asList(responsesElement))
1316             .build();
1317     mockService.addResponse(expectedResponse);
1318 
1319     String parent = "projects/project-5197/locations/location-5197/instances/instance-5197";
1320 
1321     ListVulnerabilityReportsPagedResponse pagedListResponse =
1322         client.listVulnerabilityReports(parent);
1323 
1324     List<VulnerabilityReport> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1325 
1326     Assert.assertEquals(1, resources.size());
1327     Assert.assertEquals(expectedResponse.getVulnerabilityReportsList().get(0), resources.get(0));
1328 
1329     List<String> actualRequests = mockService.getRequestPaths();
1330     Assert.assertEquals(1, actualRequests.size());
1331 
1332     String apiClientHeaderKey =
1333         mockService
1334             .getRequestHeaders()
1335             .get(ApiClientHeaderProvider.getDefaultApiClientHeaderKey())
1336             .iterator()
1337             .next();
1338     Assert.assertTrue(
1339         GaxHttpJsonProperties.getDefaultApiClientHeaderPattern()
1340             .matcher(apiClientHeaderKey)
1341             .matches());
1342   }
1343 
1344   @Test
listVulnerabilityReportsExceptionTest2()1345   public void listVulnerabilityReportsExceptionTest2() throws Exception {
1346     ApiException exception =
1347         ApiExceptionFactory.createException(
1348             new Exception(), FakeStatusCode.of(StatusCode.Code.INVALID_ARGUMENT), false);
1349     mockService.addException(exception);
1350 
1351     try {
1352       String parent = "projects/project-5197/locations/location-5197/instances/instance-5197";
1353       client.listVulnerabilityReports(parent);
1354       Assert.fail("No exception raised");
1355     } catch (InvalidArgumentException e) {
1356       // Expected exception.
1357     }
1358   }
1359 }
1360