• 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.assuredworkloads.v1beta1;
18 
19 import static com.google.cloud.assuredworkloads.v1beta1.AssuredWorkloadsServiceClient.ListWorkloadsPagedResponse;
20 
21 import com.google.api.gax.core.NoCredentialsProvider;
22 import com.google.api.gax.grpc.GaxGrpcProperties;
23 import com.google.api.gax.grpc.testing.LocalChannelProvider;
24 import com.google.api.gax.grpc.testing.MockGrpcService;
25 import com.google.api.gax.grpc.testing.MockServiceHelper;
26 import com.google.api.gax.rpc.ApiClientHeaderProvider;
27 import com.google.api.gax.rpc.InvalidArgumentException;
28 import com.google.api.gax.rpc.StatusCode;
29 import com.google.common.collect.Lists;
30 import com.google.longrunning.Operation;
31 import com.google.protobuf.AbstractMessage;
32 import com.google.protobuf.Any;
33 import com.google.protobuf.Empty;
34 import com.google.protobuf.FieldMask;
35 import com.google.protobuf.Timestamp;
36 import io.grpc.StatusRuntimeException;
37 import java.io.IOException;
38 import java.util.ArrayList;
39 import java.util.Arrays;
40 import java.util.HashMap;
41 import java.util.List;
42 import java.util.UUID;
43 import java.util.concurrent.ExecutionException;
44 import javax.annotation.Generated;
45 import org.junit.After;
46 import org.junit.AfterClass;
47 import org.junit.Assert;
48 import org.junit.Before;
49 import org.junit.BeforeClass;
50 import org.junit.Test;
51 
52 @Generated("by gapic-generator-java")
53 public class AssuredWorkloadsServiceClientTest {
54   private static MockAssuredWorkloadsService mockAssuredWorkloadsService;
55   private static MockServiceHelper mockServiceHelper;
56   private LocalChannelProvider channelProvider;
57   private AssuredWorkloadsServiceClient client;
58 
59   @BeforeClass
startStaticServer()60   public static void startStaticServer() {
61     mockAssuredWorkloadsService = new MockAssuredWorkloadsService();
62     mockServiceHelper =
63         new MockServiceHelper(
64             UUID.randomUUID().toString(),
65             Arrays.<MockGrpcService>asList(mockAssuredWorkloadsService));
66     mockServiceHelper.start();
67   }
68 
69   @AfterClass
stopServer()70   public static void stopServer() {
71     mockServiceHelper.stop();
72   }
73 
74   @Before
setUp()75   public void setUp() throws IOException {
76     mockServiceHelper.reset();
77     channelProvider = mockServiceHelper.createChannelProvider();
78     AssuredWorkloadsServiceSettings settings =
79         AssuredWorkloadsServiceSettings.newBuilder()
80             .setTransportChannelProvider(channelProvider)
81             .setCredentialsProvider(NoCredentialsProvider.create())
82             .build();
83     client = AssuredWorkloadsServiceClient.create(settings);
84   }
85 
86   @After
tearDown()87   public void tearDown() throws Exception {
88     client.close();
89   }
90 
91   @Test
createWorkloadTest()92   public void createWorkloadTest() throws Exception {
93     Workload expectedResponse =
94         Workload.newBuilder()
95             .setName(WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]").toString())
96             .setDisplayName("displayName1714148973")
97             .addAllResources(new ArrayList<Workload.ResourceInfo>())
98             .setCreateTime(Timestamp.newBuilder().build())
99             .setBillingAccount("billingAccount708726578")
100             .setEtag("etag3123477")
101             .putAllLabels(new HashMap<String, String>())
102             .setProvisionedResourcesParent("provisionedResourcesParent1733249115")
103             .setKmsSettings(Workload.KMSSettings.newBuilder().build())
104             .addAllResourceSettings(new ArrayList<Workload.ResourceSettings>())
105             .setEnableSovereignControls(true)
106             .setSaaEnrollmentResponse(Workload.SaaEnrollmentResponse.newBuilder().build())
107             .addAllCompliantButDisallowedServices(new ArrayList<String>())
108             .build();
109     Operation resultOperation =
110         Operation.newBuilder()
111             .setName("createWorkloadTest")
112             .setDone(true)
113             .setResponse(Any.pack(expectedResponse))
114             .build();
115     mockAssuredWorkloadsService.addResponse(resultOperation);
116 
117     LocationName parent = LocationName.of("[ORGANIZATION]", "[LOCATION]");
118     Workload workload = Workload.newBuilder().build();
119 
120     Workload actualResponse = client.createWorkloadAsync(parent, workload).get();
121     Assert.assertEquals(expectedResponse, actualResponse);
122 
123     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
124     Assert.assertEquals(1, actualRequests.size());
125     CreateWorkloadRequest actualRequest = ((CreateWorkloadRequest) actualRequests.get(0));
126 
127     Assert.assertEquals(parent.toString(), actualRequest.getParent());
128     Assert.assertEquals(workload, actualRequest.getWorkload());
129     Assert.assertTrue(
130         channelProvider.isHeaderSent(
131             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
132             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
133   }
134 
135   @Test
createWorkloadExceptionTest()136   public void createWorkloadExceptionTest() throws Exception {
137     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
138     mockAssuredWorkloadsService.addException(exception);
139 
140     try {
141       LocationName parent = LocationName.of("[ORGANIZATION]", "[LOCATION]");
142       Workload workload = Workload.newBuilder().build();
143       client.createWorkloadAsync(parent, workload).get();
144       Assert.fail("No exception raised");
145     } catch (ExecutionException e) {
146       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
147       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
148       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
149     }
150   }
151 
152   @Test
createWorkloadTest2()153   public void createWorkloadTest2() throws Exception {
154     Workload expectedResponse =
155         Workload.newBuilder()
156             .setName(WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]").toString())
157             .setDisplayName("displayName1714148973")
158             .addAllResources(new ArrayList<Workload.ResourceInfo>())
159             .setCreateTime(Timestamp.newBuilder().build())
160             .setBillingAccount("billingAccount708726578")
161             .setEtag("etag3123477")
162             .putAllLabels(new HashMap<String, String>())
163             .setProvisionedResourcesParent("provisionedResourcesParent1733249115")
164             .setKmsSettings(Workload.KMSSettings.newBuilder().build())
165             .addAllResourceSettings(new ArrayList<Workload.ResourceSettings>())
166             .setEnableSovereignControls(true)
167             .setSaaEnrollmentResponse(Workload.SaaEnrollmentResponse.newBuilder().build())
168             .addAllCompliantButDisallowedServices(new ArrayList<String>())
169             .build();
170     Operation resultOperation =
171         Operation.newBuilder()
172             .setName("createWorkloadTest")
173             .setDone(true)
174             .setResponse(Any.pack(expectedResponse))
175             .build();
176     mockAssuredWorkloadsService.addResponse(resultOperation);
177 
178     String parent = "parent-995424086";
179     Workload workload = Workload.newBuilder().build();
180 
181     Workload actualResponse = client.createWorkloadAsync(parent, workload).get();
182     Assert.assertEquals(expectedResponse, actualResponse);
183 
184     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
185     Assert.assertEquals(1, actualRequests.size());
186     CreateWorkloadRequest actualRequest = ((CreateWorkloadRequest) actualRequests.get(0));
187 
188     Assert.assertEquals(parent, actualRequest.getParent());
189     Assert.assertEquals(workload, actualRequest.getWorkload());
190     Assert.assertTrue(
191         channelProvider.isHeaderSent(
192             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
193             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
194   }
195 
196   @Test
createWorkloadExceptionTest2()197   public void createWorkloadExceptionTest2() throws Exception {
198     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
199     mockAssuredWorkloadsService.addException(exception);
200 
201     try {
202       String parent = "parent-995424086";
203       Workload workload = Workload.newBuilder().build();
204       client.createWorkloadAsync(parent, workload).get();
205       Assert.fail("No exception raised");
206     } catch (ExecutionException e) {
207       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
208       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
209       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
210     }
211   }
212 
213   @Test
updateWorkloadTest()214   public void updateWorkloadTest() throws Exception {
215     Workload expectedResponse =
216         Workload.newBuilder()
217             .setName(WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]").toString())
218             .setDisplayName("displayName1714148973")
219             .addAllResources(new ArrayList<Workload.ResourceInfo>())
220             .setCreateTime(Timestamp.newBuilder().build())
221             .setBillingAccount("billingAccount708726578")
222             .setEtag("etag3123477")
223             .putAllLabels(new HashMap<String, String>())
224             .setProvisionedResourcesParent("provisionedResourcesParent1733249115")
225             .setKmsSettings(Workload.KMSSettings.newBuilder().build())
226             .addAllResourceSettings(new ArrayList<Workload.ResourceSettings>())
227             .setEnableSovereignControls(true)
228             .setSaaEnrollmentResponse(Workload.SaaEnrollmentResponse.newBuilder().build())
229             .addAllCompliantButDisallowedServices(new ArrayList<String>())
230             .build();
231     mockAssuredWorkloadsService.addResponse(expectedResponse);
232 
233     Workload workload = Workload.newBuilder().build();
234     FieldMask updateMask = FieldMask.newBuilder().build();
235 
236     Workload actualResponse = client.updateWorkload(workload, updateMask);
237     Assert.assertEquals(expectedResponse, actualResponse);
238 
239     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
240     Assert.assertEquals(1, actualRequests.size());
241     UpdateWorkloadRequest actualRequest = ((UpdateWorkloadRequest) actualRequests.get(0));
242 
243     Assert.assertEquals(workload, actualRequest.getWorkload());
244     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
245     Assert.assertTrue(
246         channelProvider.isHeaderSent(
247             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
248             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
249   }
250 
251   @Test
updateWorkloadExceptionTest()252   public void updateWorkloadExceptionTest() throws Exception {
253     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
254     mockAssuredWorkloadsService.addException(exception);
255 
256     try {
257       Workload workload = Workload.newBuilder().build();
258       FieldMask updateMask = FieldMask.newBuilder().build();
259       client.updateWorkload(workload, updateMask);
260       Assert.fail("No exception raised");
261     } catch (InvalidArgumentException e) {
262       // Expected exception.
263     }
264   }
265 
266   @Test
restrictAllowedResourcesTest()267   public void restrictAllowedResourcesTest() throws Exception {
268     RestrictAllowedResourcesResponse expectedResponse =
269         RestrictAllowedResourcesResponse.newBuilder().build();
270     mockAssuredWorkloadsService.addResponse(expectedResponse);
271 
272     RestrictAllowedResourcesRequest request =
273         RestrictAllowedResourcesRequest.newBuilder().setName("name3373707").build();
274 
275     RestrictAllowedResourcesResponse actualResponse = client.restrictAllowedResources(request);
276     Assert.assertEquals(expectedResponse, actualResponse);
277 
278     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
279     Assert.assertEquals(1, actualRequests.size());
280     RestrictAllowedResourcesRequest actualRequest =
281         ((RestrictAllowedResourcesRequest) actualRequests.get(0));
282 
283     Assert.assertEquals(request.getName(), actualRequest.getName());
284     Assert.assertEquals(request.getRestrictionType(), actualRequest.getRestrictionType());
285     Assert.assertTrue(
286         channelProvider.isHeaderSent(
287             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
288             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
289   }
290 
291   @Test
restrictAllowedResourcesExceptionTest()292   public void restrictAllowedResourcesExceptionTest() throws Exception {
293     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
294     mockAssuredWorkloadsService.addException(exception);
295 
296     try {
297       RestrictAllowedResourcesRequest request =
298           RestrictAllowedResourcesRequest.newBuilder().setName("name3373707").build();
299       client.restrictAllowedResources(request);
300       Assert.fail("No exception raised");
301     } catch (InvalidArgumentException e) {
302       // Expected exception.
303     }
304   }
305 
306   @Test
deleteWorkloadTest()307   public void deleteWorkloadTest() throws Exception {
308     Empty expectedResponse = Empty.newBuilder().build();
309     mockAssuredWorkloadsService.addResponse(expectedResponse);
310 
311     WorkloadName name = WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]");
312 
313     client.deleteWorkload(name);
314 
315     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
316     Assert.assertEquals(1, actualRequests.size());
317     DeleteWorkloadRequest actualRequest = ((DeleteWorkloadRequest) actualRequests.get(0));
318 
319     Assert.assertEquals(name.toString(), actualRequest.getName());
320     Assert.assertTrue(
321         channelProvider.isHeaderSent(
322             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
323             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
324   }
325 
326   @Test
deleteWorkloadExceptionTest()327   public void deleteWorkloadExceptionTest() throws Exception {
328     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
329     mockAssuredWorkloadsService.addException(exception);
330 
331     try {
332       WorkloadName name = WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]");
333       client.deleteWorkload(name);
334       Assert.fail("No exception raised");
335     } catch (InvalidArgumentException e) {
336       // Expected exception.
337     }
338   }
339 
340   @Test
deleteWorkloadTest2()341   public void deleteWorkloadTest2() throws Exception {
342     Empty expectedResponse = Empty.newBuilder().build();
343     mockAssuredWorkloadsService.addResponse(expectedResponse);
344 
345     String name = "name3373707";
346 
347     client.deleteWorkload(name);
348 
349     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
350     Assert.assertEquals(1, actualRequests.size());
351     DeleteWorkloadRequest actualRequest = ((DeleteWorkloadRequest) actualRequests.get(0));
352 
353     Assert.assertEquals(name, actualRequest.getName());
354     Assert.assertTrue(
355         channelProvider.isHeaderSent(
356             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
357             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
358   }
359 
360   @Test
deleteWorkloadExceptionTest2()361   public void deleteWorkloadExceptionTest2() throws Exception {
362     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
363     mockAssuredWorkloadsService.addException(exception);
364 
365     try {
366       String name = "name3373707";
367       client.deleteWorkload(name);
368       Assert.fail("No exception raised");
369     } catch (InvalidArgumentException e) {
370       // Expected exception.
371     }
372   }
373 
374   @Test
getWorkloadTest()375   public void getWorkloadTest() throws Exception {
376     Workload expectedResponse =
377         Workload.newBuilder()
378             .setName(WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]").toString())
379             .setDisplayName("displayName1714148973")
380             .addAllResources(new ArrayList<Workload.ResourceInfo>())
381             .setCreateTime(Timestamp.newBuilder().build())
382             .setBillingAccount("billingAccount708726578")
383             .setEtag("etag3123477")
384             .putAllLabels(new HashMap<String, String>())
385             .setProvisionedResourcesParent("provisionedResourcesParent1733249115")
386             .setKmsSettings(Workload.KMSSettings.newBuilder().build())
387             .addAllResourceSettings(new ArrayList<Workload.ResourceSettings>())
388             .setEnableSovereignControls(true)
389             .setSaaEnrollmentResponse(Workload.SaaEnrollmentResponse.newBuilder().build())
390             .addAllCompliantButDisallowedServices(new ArrayList<String>())
391             .build();
392     mockAssuredWorkloadsService.addResponse(expectedResponse);
393 
394     WorkloadName name = WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]");
395 
396     Workload actualResponse = client.getWorkload(name);
397     Assert.assertEquals(expectedResponse, actualResponse);
398 
399     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
400     Assert.assertEquals(1, actualRequests.size());
401     GetWorkloadRequest actualRequest = ((GetWorkloadRequest) actualRequests.get(0));
402 
403     Assert.assertEquals(name.toString(), actualRequest.getName());
404     Assert.assertTrue(
405         channelProvider.isHeaderSent(
406             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
407             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
408   }
409 
410   @Test
getWorkloadExceptionTest()411   public void getWorkloadExceptionTest() throws Exception {
412     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
413     mockAssuredWorkloadsService.addException(exception);
414 
415     try {
416       WorkloadName name = WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]");
417       client.getWorkload(name);
418       Assert.fail("No exception raised");
419     } catch (InvalidArgumentException e) {
420       // Expected exception.
421     }
422   }
423 
424   @Test
getWorkloadTest2()425   public void getWorkloadTest2() throws Exception {
426     Workload expectedResponse =
427         Workload.newBuilder()
428             .setName(WorkloadName.of("[ORGANIZATION]", "[LOCATION]", "[WORKLOAD]").toString())
429             .setDisplayName("displayName1714148973")
430             .addAllResources(new ArrayList<Workload.ResourceInfo>())
431             .setCreateTime(Timestamp.newBuilder().build())
432             .setBillingAccount("billingAccount708726578")
433             .setEtag("etag3123477")
434             .putAllLabels(new HashMap<String, String>())
435             .setProvisionedResourcesParent("provisionedResourcesParent1733249115")
436             .setKmsSettings(Workload.KMSSettings.newBuilder().build())
437             .addAllResourceSettings(new ArrayList<Workload.ResourceSettings>())
438             .setEnableSovereignControls(true)
439             .setSaaEnrollmentResponse(Workload.SaaEnrollmentResponse.newBuilder().build())
440             .addAllCompliantButDisallowedServices(new ArrayList<String>())
441             .build();
442     mockAssuredWorkloadsService.addResponse(expectedResponse);
443 
444     String name = "name3373707";
445 
446     Workload actualResponse = client.getWorkload(name);
447     Assert.assertEquals(expectedResponse, actualResponse);
448 
449     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
450     Assert.assertEquals(1, actualRequests.size());
451     GetWorkloadRequest actualRequest = ((GetWorkloadRequest) actualRequests.get(0));
452 
453     Assert.assertEquals(name, actualRequest.getName());
454     Assert.assertTrue(
455         channelProvider.isHeaderSent(
456             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
457             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
458   }
459 
460   @Test
getWorkloadExceptionTest2()461   public void getWorkloadExceptionTest2() throws Exception {
462     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
463     mockAssuredWorkloadsService.addException(exception);
464 
465     try {
466       String name = "name3373707";
467       client.getWorkload(name);
468       Assert.fail("No exception raised");
469     } catch (InvalidArgumentException e) {
470       // Expected exception.
471     }
472   }
473 
474   @Test
analyzeWorkloadMoveTest()475   public void analyzeWorkloadMoveTest() throws Exception {
476     AnalyzeWorkloadMoveResponse expectedResponse =
477         AnalyzeWorkloadMoveResponse.newBuilder().addAllBlockers(new ArrayList<String>()).build();
478     mockAssuredWorkloadsService.addResponse(expectedResponse);
479 
480     String project = "project-309310695";
481     String target = "target-880905839";
482 
483     AnalyzeWorkloadMoveResponse actualResponse = client.analyzeWorkloadMove(project, target);
484     Assert.assertEquals(expectedResponse, actualResponse);
485 
486     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
487     Assert.assertEquals(1, actualRequests.size());
488     AnalyzeWorkloadMoveRequest actualRequest = ((AnalyzeWorkloadMoveRequest) actualRequests.get(0));
489 
490     Assert.assertEquals(project, actualRequest.getProject());
491     Assert.assertEquals(target, actualRequest.getTarget());
492     Assert.assertTrue(
493         channelProvider.isHeaderSent(
494             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
495             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
496   }
497 
498   @Test
analyzeWorkloadMoveExceptionTest()499   public void analyzeWorkloadMoveExceptionTest() throws Exception {
500     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
501     mockAssuredWorkloadsService.addException(exception);
502 
503     try {
504       String project = "project-309310695";
505       String target = "target-880905839";
506       client.analyzeWorkloadMove(project, target);
507       Assert.fail("No exception raised");
508     } catch (InvalidArgumentException e) {
509       // Expected exception.
510     }
511   }
512 
513   @Test
listWorkloadsTest()514   public void listWorkloadsTest() throws Exception {
515     Workload responsesElement = Workload.newBuilder().build();
516     ListWorkloadsResponse expectedResponse =
517         ListWorkloadsResponse.newBuilder()
518             .setNextPageToken("")
519             .addAllWorkloads(Arrays.asList(responsesElement))
520             .build();
521     mockAssuredWorkloadsService.addResponse(expectedResponse);
522 
523     LocationName parent = LocationName.of("[ORGANIZATION]", "[LOCATION]");
524 
525     ListWorkloadsPagedResponse pagedListResponse = client.listWorkloads(parent);
526 
527     List<Workload> resources = Lists.newArrayList(pagedListResponse.iterateAll());
528 
529     Assert.assertEquals(1, resources.size());
530     Assert.assertEquals(expectedResponse.getWorkloadsList().get(0), resources.get(0));
531 
532     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
533     Assert.assertEquals(1, actualRequests.size());
534     ListWorkloadsRequest actualRequest = ((ListWorkloadsRequest) actualRequests.get(0));
535 
536     Assert.assertEquals(parent.toString(), actualRequest.getParent());
537     Assert.assertTrue(
538         channelProvider.isHeaderSent(
539             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
540             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
541   }
542 
543   @Test
listWorkloadsExceptionTest()544   public void listWorkloadsExceptionTest() throws Exception {
545     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
546     mockAssuredWorkloadsService.addException(exception);
547 
548     try {
549       LocationName parent = LocationName.of("[ORGANIZATION]", "[LOCATION]");
550       client.listWorkloads(parent);
551       Assert.fail("No exception raised");
552     } catch (InvalidArgumentException e) {
553       // Expected exception.
554     }
555   }
556 
557   @Test
listWorkloadsTest2()558   public void listWorkloadsTest2() throws Exception {
559     Workload responsesElement = Workload.newBuilder().build();
560     ListWorkloadsResponse expectedResponse =
561         ListWorkloadsResponse.newBuilder()
562             .setNextPageToken("")
563             .addAllWorkloads(Arrays.asList(responsesElement))
564             .build();
565     mockAssuredWorkloadsService.addResponse(expectedResponse);
566 
567     String parent = "parent-995424086";
568 
569     ListWorkloadsPagedResponse pagedListResponse = client.listWorkloads(parent);
570 
571     List<Workload> resources = Lists.newArrayList(pagedListResponse.iterateAll());
572 
573     Assert.assertEquals(1, resources.size());
574     Assert.assertEquals(expectedResponse.getWorkloadsList().get(0), resources.get(0));
575 
576     List<AbstractMessage> actualRequests = mockAssuredWorkloadsService.getRequests();
577     Assert.assertEquals(1, actualRequests.size());
578     ListWorkloadsRequest actualRequest = ((ListWorkloadsRequest) actualRequests.get(0));
579 
580     Assert.assertEquals(parent, actualRequest.getParent());
581     Assert.assertTrue(
582         channelProvider.isHeaderSent(
583             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
584             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
585   }
586 
587   @Test
listWorkloadsExceptionTest2()588   public void listWorkloadsExceptionTest2() throws Exception {
589     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
590     mockAssuredWorkloadsService.addException(exception);
591 
592     try {
593       String parent = "parent-995424086";
594       client.listWorkloads(parent);
595       Assert.fail("No exception raised");
596     } catch (InvalidArgumentException e) {
597       // Expected exception.
598     }
599   }
600 }
601