• 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.vmwareengine.v1;
18 
19 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListClustersPagedResponse;
20 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListHcxActivationKeysPagedResponse;
21 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListLocationsPagedResponse;
22 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListNetworkPoliciesPagedResponse;
23 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListNodeTypesPagedResponse;
24 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListPrivateCloudsPagedResponse;
25 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListSubnetsPagedResponse;
26 import static com.google.cloud.vmwareengine.v1.VmwareEngineClient.ListVmwareEngineNetworksPagedResponse;
27 
28 import com.google.api.gax.core.NoCredentialsProvider;
29 import com.google.api.gax.grpc.GaxGrpcProperties;
30 import com.google.api.gax.grpc.testing.LocalChannelProvider;
31 import com.google.api.gax.grpc.testing.MockGrpcService;
32 import com.google.api.gax.grpc.testing.MockServiceHelper;
33 import com.google.api.gax.rpc.ApiClientHeaderProvider;
34 import com.google.api.gax.rpc.InvalidArgumentException;
35 import com.google.api.gax.rpc.StatusCode;
36 import com.google.cloud.location.GetLocationRequest;
37 import com.google.cloud.location.ListLocationsRequest;
38 import com.google.cloud.location.ListLocationsResponse;
39 import com.google.cloud.location.Location;
40 import com.google.common.collect.Lists;
41 import com.google.iam.v1.AuditConfig;
42 import com.google.iam.v1.Binding;
43 import com.google.iam.v1.GetIamPolicyRequest;
44 import com.google.iam.v1.GetPolicyOptions;
45 import com.google.iam.v1.Policy;
46 import com.google.iam.v1.SetIamPolicyRequest;
47 import com.google.iam.v1.TestIamPermissionsRequest;
48 import com.google.iam.v1.TestIamPermissionsResponse;
49 import com.google.longrunning.Operation;
50 import com.google.protobuf.AbstractMessage;
51 import com.google.protobuf.Any;
52 import com.google.protobuf.ByteString;
53 import com.google.protobuf.Empty;
54 import com.google.protobuf.FieldMask;
55 import com.google.protobuf.Timestamp;
56 import io.grpc.StatusRuntimeException;
57 import java.io.IOException;
58 import java.util.ArrayList;
59 import java.util.Arrays;
60 import java.util.HashMap;
61 import java.util.List;
62 import java.util.UUID;
63 import java.util.concurrent.ExecutionException;
64 import javax.annotation.Generated;
65 import org.junit.After;
66 import org.junit.AfterClass;
67 import org.junit.Assert;
68 import org.junit.Before;
69 import org.junit.BeforeClass;
70 import org.junit.Test;
71 
72 @Generated("by gapic-generator-java")
73 public class VmwareEngineClientTest {
74   private static MockIAMPolicy mockIAMPolicy;
75   private static MockLocations mockLocations;
76   private static MockServiceHelper mockServiceHelper;
77   private static MockVmwareEngine mockVmwareEngine;
78   private LocalChannelProvider channelProvider;
79   private VmwareEngineClient client;
80 
81   @BeforeClass
startStaticServer()82   public static void startStaticServer() {
83     mockVmwareEngine = new MockVmwareEngine();
84     mockLocations = new MockLocations();
85     mockIAMPolicy = new MockIAMPolicy();
86     mockServiceHelper =
87         new MockServiceHelper(
88             UUID.randomUUID().toString(),
89             Arrays.<MockGrpcService>asList(mockVmwareEngine, mockLocations, mockIAMPolicy));
90     mockServiceHelper.start();
91   }
92 
93   @AfterClass
stopServer()94   public static void stopServer() {
95     mockServiceHelper.stop();
96   }
97 
98   @Before
setUp()99   public void setUp() throws IOException {
100     mockServiceHelper.reset();
101     channelProvider = mockServiceHelper.createChannelProvider();
102     VmwareEngineSettings settings =
103         VmwareEngineSettings.newBuilder()
104             .setTransportChannelProvider(channelProvider)
105             .setCredentialsProvider(NoCredentialsProvider.create())
106             .build();
107     client = VmwareEngineClient.create(settings);
108   }
109 
110   @After
tearDown()111   public void tearDown() throws Exception {
112     client.close();
113   }
114 
115   @Test
listPrivateCloudsTest()116   public void listPrivateCloudsTest() throws Exception {
117     PrivateCloud responsesElement = PrivateCloud.newBuilder().build();
118     ListPrivateCloudsResponse expectedResponse =
119         ListPrivateCloudsResponse.newBuilder()
120             .setNextPageToken("")
121             .addAllPrivateClouds(Arrays.asList(responsesElement))
122             .build();
123     mockVmwareEngine.addResponse(expectedResponse);
124 
125     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
126 
127     ListPrivateCloudsPagedResponse pagedListResponse = client.listPrivateClouds(parent);
128 
129     List<PrivateCloud> resources = Lists.newArrayList(pagedListResponse.iterateAll());
130 
131     Assert.assertEquals(1, resources.size());
132     Assert.assertEquals(expectedResponse.getPrivateCloudsList().get(0), resources.get(0));
133 
134     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
135     Assert.assertEquals(1, actualRequests.size());
136     ListPrivateCloudsRequest actualRequest = ((ListPrivateCloudsRequest) actualRequests.get(0));
137 
138     Assert.assertEquals(parent.toString(), actualRequest.getParent());
139     Assert.assertTrue(
140         channelProvider.isHeaderSent(
141             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
142             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
143   }
144 
145   @Test
listPrivateCloudsExceptionTest()146   public void listPrivateCloudsExceptionTest() throws Exception {
147     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
148     mockVmwareEngine.addException(exception);
149 
150     try {
151       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
152       client.listPrivateClouds(parent);
153       Assert.fail("No exception raised");
154     } catch (InvalidArgumentException e) {
155       // Expected exception.
156     }
157   }
158 
159   @Test
listPrivateCloudsTest2()160   public void listPrivateCloudsTest2() throws Exception {
161     PrivateCloud responsesElement = PrivateCloud.newBuilder().build();
162     ListPrivateCloudsResponse expectedResponse =
163         ListPrivateCloudsResponse.newBuilder()
164             .setNextPageToken("")
165             .addAllPrivateClouds(Arrays.asList(responsesElement))
166             .build();
167     mockVmwareEngine.addResponse(expectedResponse);
168 
169     String parent = "parent-995424086";
170 
171     ListPrivateCloudsPagedResponse pagedListResponse = client.listPrivateClouds(parent);
172 
173     List<PrivateCloud> resources = Lists.newArrayList(pagedListResponse.iterateAll());
174 
175     Assert.assertEquals(1, resources.size());
176     Assert.assertEquals(expectedResponse.getPrivateCloudsList().get(0), resources.get(0));
177 
178     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
179     Assert.assertEquals(1, actualRequests.size());
180     ListPrivateCloudsRequest actualRequest = ((ListPrivateCloudsRequest) actualRequests.get(0));
181 
182     Assert.assertEquals(parent, actualRequest.getParent());
183     Assert.assertTrue(
184         channelProvider.isHeaderSent(
185             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
186             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
187   }
188 
189   @Test
listPrivateCloudsExceptionTest2()190   public void listPrivateCloudsExceptionTest2() throws Exception {
191     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
192     mockVmwareEngine.addException(exception);
193 
194     try {
195       String parent = "parent-995424086";
196       client.listPrivateClouds(parent);
197       Assert.fail("No exception raised");
198     } catch (InvalidArgumentException e) {
199       // Expected exception.
200     }
201   }
202 
203   @Test
getPrivateCloudTest()204   public void getPrivateCloudTest() throws Exception {
205     PrivateCloud expectedResponse =
206         PrivateCloud.newBuilder()
207             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
208             .setCreateTime(Timestamp.newBuilder().build())
209             .setUpdateTime(Timestamp.newBuilder().build())
210             .setDeleteTime(Timestamp.newBuilder().build())
211             .setExpireTime(Timestamp.newBuilder().build())
212             .setNetworkConfig(NetworkConfig.newBuilder().build())
213             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
214             .setDescription("description-1724546052")
215             .setHcx(Hcx.newBuilder().build())
216             .setNsx(Nsx.newBuilder().build())
217             .setVcenter(Vcenter.newBuilder().build())
218             .setUid("uid115792")
219             .build();
220     mockVmwareEngine.addResponse(expectedResponse);
221 
222     PrivateCloudName name = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
223 
224     PrivateCloud actualResponse = client.getPrivateCloud(name);
225     Assert.assertEquals(expectedResponse, actualResponse);
226 
227     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
228     Assert.assertEquals(1, actualRequests.size());
229     GetPrivateCloudRequest actualRequest = ((GetPrivateCloudRequest) actualRequests.get(0));
230 
231     Assert.assertEquals(name.toString(), actualRequest.getName());
232     Assert.assertTrue(
233         channelProvider.isHeaderSent(
234             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
235             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
236   }
237 
238   @Test
getPrivateCloudExceptionTest()239   public void getPrivateCloudExceptionTest() throws Exception {
240     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
241     mockVmwareEngine.addException(exception);
242 
243     try {
244       PrivateCloudName name = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
245       client.getPrivateCloud(name);
246       Assert.fail("No exception raised");
247     } catch (InvalidArgumentException e) {
248       // Expected exception.
249     }
250   }
251 
252   @Test
getPrivateCloudTest2()253   public void getPrivateCloudTest2() throws Exception {
254     PrivateCloud expectedResponse =
255         PrivateCloud.newBuilder()
256             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
257             .setCreateTime(Timestamp.newBuilder().build())
258             .setUpdateTime(Timestamp.newBuilder().build())
259             .setDeleteTime(Timestamp.newBuilder().build())
260             .setExpireTime(Timestamp.newBuilder().build())
261             .setNetworkConfig(NetworkConfig.newBuilder().build())
262             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
263             .setDescription("description-1724546052")
264             .setHcx(Hcx.newBuilder().build())
265             .setNsx(Nsx.newBuilder().build())
266             .setVcenter(Vcenter.newBuilder().build())
267             .setUid("uid115792")
268             .build();
269     mockVmwareEngine.addResponse(expectedResponse);
270 
271     String name = "name3373707";
272 
273     PrivateCloud actualResponse = client.getPrivateCloud(name);
274     Assert.assertEquals(expectedResponse, actualResponse);
275 
276     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
277     Assert.assertEquals(1, actualRequests.size());
278     GetPrivateCloudRequest actualRequest = ((GetPrivateCloudRequest) actualRequests.get(0));
279 
280     Assert.assertEquals(name, actualRequest.getName());
281     Assert.assertTrue(
282         channelProvider.isHeaderSent(
283             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
284             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
285   }
286 
287   @Test
getPrivateCloudExceptionTest2()288   public void getPrivateCloudExceptionTest2() throws Exception {
289     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
290     mockVmwareEngine.addException(exception);
291 
292     try {
293       String name = "name3373707";
294       client.getPrivateCloud(name);
295       Assert.fail("No exception raised");
296     } catch (InvalidArgumentException e) {
297       // Expected exception.
298     }
299   }
300 
301   @Test
createPrivateCloudTest()302   public void createPrivateCloudTest() throws Exception {
303     PrivateCloud expectedResponse =
304         PrivateCloud.newBuilder()
305             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
306             .setCreateTime(Timestamp.newBuilder().build())
307             .setUpdateTime(Timestamp.newBuilder().build())
308             .setDeleteTime(Timestamp.newBuilder().build())
309             .setExpireTime(Timestamp.newBuilder().build())
310             .setNetworkConfig(NetworkConfig.newBuilder().build())
311             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
312             .setDescription("description-1724546052")
313             .setHcx(Hcx.newBuilder().build())
314             .setNsx(Nsx.newBuilder().build())
315             .setVcenter(Vcenter.newBuilder().build())
316             .setUid("uid115792")
317             .build();
318     Operation resultOperation =
319         Operation.newBuilder()
320             .setName("createPrivateCloudTest")
321             .setDone(true)
322             .setResponse(Any.pack(expectedResponse))
323             .build();
324     mockVmwareEngine.addResponse(resultOperation);
325 
326     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
327     PrivateCloud privateCloud = PrivateCloud.newBuilder().build();
328     String privateCloudId = "privateCloudId573896493";
329 
330     PrivateCloud actualResponse =
331         client.createPrivateCloudAsync(parent, privateCloud, privateCloudId).get();
332     Assert.assertEquals(expectedResponse, actualResponse);
333 
334     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
335     Assert.assertEquals(1, actualRequests.size());
336     CreatePrivateCloudRequest actualRequest = ((CreatePrivateCloudRequest) actualRequests.get(0));
337 
338     Assert.assertEquals(parent.toString(), actualRequest.getParent());
339     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
340     Assert.assertEquals(privateCloudId, actualRequest.getPrivateCloudId());
341     Assert.assertTrue(
342         channelProvider.isHeaderSent(
343             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
344             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
345   }
346 
347   @Test
createPrivateCloudExceptionTest()348   public void createPrivateCloudExceptionTest() throws Exception {
349     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
350     mockVmwareEngine.addException(exception);
351 
352     try {
353       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
354       PrivateCloud privateCloud = PrivateCloud.newBuilder().build();
355       String privateCloudId = "privateCloudId573896493";
356       client.createPrivateCloudAsync(parent, privateCloud, privateCloudId).get();
357       Assert.fail("No exception raised");
358     } catch (ExecutionException e) {
359       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
360       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
361       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
362     }
363   }
364 
365   @Test
createPrivateCloudTest2()366   public void createPrivateCloudTest2() throws Exception {
367     PrivateCloud expectedResponse =
368         PrivateCloud.newBuilder()
369             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
370             .setCreateTime(Timestamp.newBuilder().build())
371             .setUpdateTime(Timestamp.newBuilder().build())
372             .setDeleteTime(Timestamp.newBuilder().build())
373             .setExpireTime(Timestamp.newBuilder().build())
374             .setNetworkConfig(NetworkConfig.newBuilder().build())
375             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
376             .setDescription("description-1724546052")
377             .setHcx(Hcx.newBuilder().build())
378             .setNsx(Nsx.newBuilder().build())
379             .setVcenter(Vcenter.newBuilder().build())
380             .setUid("uid115792")
381             .build();
382     Operation resultOperation =
383         Operation.newBuilder()
384             .setName("createPrivateCloudTest")
385             .setDone(true)
386             .setResponse(Any.pack(expectedResponse))
387             .build();
388     mockVmwareEngine.addResponse(resultOperation);
389 
390     String parent = "parent-995424086";
391     PrivateCloud privateCloud = PrivateCloud.newBuilder().build();
392     String privateCloudId = "privateCloudId573896493";
393 
394     PrivateCloud actualResponse =
395         client.createPrivateCloudAsync(parent, privateCloud, privateCloudId).get();
396     Assert.assertEquals(expectedResponse, actualResponse);
397 
398     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
399     Assert.assertEquals(1, actualRequests.size());
400     CreatePrivateCloudRequest actualRequest = ((CreatePrivateCloudRequest) actualRequests.get(0));
401 
402     Assert.assertEquals(parent, actualRequest.getParent());
403     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
404     Assert.assertEquals(privateCloudId, actualRequest.getPrivateCloudId());
405     Assert.assertTrue(
406         channelProvider.isHeaderSent(
407             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
408             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
409   }
410 
411   @Test
createPrivateCloudExceptionTest2()412   public void createPrivateCloudExceptionTest2() throws Exception {
413     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
414     mockVmwareEngine.addException(exception);
415 
416     try {
417       String parent = "parent-995424086";
418       PrivateCloud privateCloud = PrivateCloud.newBuilder().build();
419       String privateCloudId = "privateCloudId573896493";
420       client.createPrivateCloudAsync(parent, privateCloud, privateCloudId).get();
421       Assert.fail("No exception raised");
422     } catch (ExecutionException e) {
423       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
424       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
425       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
426     }
427   }
428 
429   @Test
updatePrivateCloudTest()430   public void updatePrivateCloudTest() throws Exception {
431     PrivateCloud expectedResponse =
432         PrivateCloud.newBuilder()
433             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
434             .setCreateTime(Timestamp.newBuilder().build())
435             .setUpdateTime(Timestamp.newBuilder().build())
436             .setDeleteTime(Timestamp.newBuilder().build())
437             .setExpireTime(Timestamp.newBuilder().build())
438             .setNetworkConfig(NetworkConfig.newBuilder().build())
439             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
440             .setDescription("description-1724546052")
441             .setHcx(Hcx.newBuilder().build())
442             .setNsx(Nsx.newBuilder().build())
443             .setVcenter(Vcenter.newBuilder().build())
444             .setUid("uid115792")
445             .build();
446     Operation resultOperation =
447         Operation.newBuilder()
448             .setName("updatePrivateCloudTest")
449             .setDone(true)
450             .setResponse(Any.pack(expectedResponse))
451             .build();
452     mockVmwareEngine.addResponse(resultOperation);
453 
454     PrivateCloud privateCloud = PrivateCloud.newBuilder().build();
455     FieldMask updateMask = FieldMask.newBuilder().build();
456 
457     PrivateCloud actualResponse = client.updatePrivateCloudAsync(privateCloud, updateMask).get();
458     Assert.assertEquals(expectedResponse, actualResponse);
459 
460     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
461     Assert.assertEquals(1, actualRequests.size());
462     UpdatePrivateCloudRequest actualRequest = ((UpdatePrivateCloudRequest) actualRequests.get(0));
463 
464     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
465     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
466     Assert.assertTrue(
467         channelProvider.isHeaderSent(
468             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
469             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
470   }
471 
472   @Test
updatePrivateCloudExceptionTest()473   public void updatePrivateCloudExceptionTest() throws Exception {
474     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
475     mockVmwareEngine.addException(exception);
476 
477     try {
478       PrivateCloud privateCloud = PrivateCloud.newBuilder().build();
479       FieldMask updateMask = FieldMask.newBuilder().build();
480       client.updatePrivateCloudAsync(privateCloud, updateMask).get();
481       Assert.fail("No exception raised");
482     } catch (ExecutionException e) {
483       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
484       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
485       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
486     }
487   }
488 
489   @Test
deletePrivateCloudTest()490   public void deletePrivateCloudTest() throws Exception {
491     PrivateCloud expectedResponse =
492         PrivateCloud.newBuilder()
493             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
494             .setCreateTime(Timestamp.newBuilder().build())
495             .setUpdateTime(Timestamp.newBuilder().build())
496             .setDeleteTime(Timestamp.newBuilder().build())
497             .setExpireTime(Timestamp.newBuilder().build())
498             .setNetworkConfig(NetworkConfig.newBuilder().build())
499             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
500             .setDescription("description-1724546052")
501             .setHcx(Hcx.newBuilder().build())
502             .setNsx(Nsx.newBuilder().build())
503             .setVcenter(Vcenter.newBuilder().build())
504             .setUid("uid115792")
505             .build();
506     Operation resultOperation =
507         Operation.newBuilder()
508             .setName("deletePrivateCloudTest")
509             .setDone(true)
510             .setResponse(Any.pack(expectedResponse))
511             .build();
512     mockVmwareEngine.addResponse(resultOperation);
513 
514     PrivateCloudName name = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
515 
516     PrivateCloud actualResponse = client.deletePrivateCloudAsync(name).get();
517     Assert.assertEquals(expectedResponse, actualResponse);
518 
519     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
520     Assert.assertEquals(1, actualRequests.size());
521     DeletePrivateCloudRequest actualRequest = ((DeletePrivateCloudRequest) actualRequests.get(0));
522 
523     Assert.assertEquals(name.toString(), actualRequest.getName());
524     Assert.assertTrue(
525         channelProvider.isHeaderSent(
526             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
527             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
528   }
529 
530   @Test
deletePrivateCloudExceptionTest()531   public void deletePrivateCloudExceptionTest() throws Exception {
532     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
533     mockVmwareEngine.addException(exception);
534 
535     try {
536       PrivateCloudName name = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
537       client.deletePrivateCloudAsync(name).get();
538       Assert.fail("No exception raised");
539     } catch (ExecutionException e) {
540       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
541       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
542       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
543     }
544   }
545 
546   @Test
deletePrivateCloudTest2()547   public void deletePrivateCloudTest2() throws Exception {
548     PrivateCloud expectedResponse =
549         PrivateCloud.newBuilder()
550             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
551             .setCreateTime(Timestamp.newBuilder().build())
552             .setUpdateTime(Timestamp.newBuilder().build())
553             .setDeleteTime(Timestamp.newBuilder().build())
554             .setExpireTime(Timestamp.newBuilder().build())
555             .setNetworkConfig(NetworkConfig.newBuilder().build())
556             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
557             .setDescription("description-1724546052")
558             .setHcx(Hcx.newBuilder().build())
559             .setNsx(Nsx.newBuilder().build())
560             .setVcenter(Vcenter.newBuilder().build())
561             .setUid("uid115792")
562             .build();
563     Operation resultOperation =
564         Operation.newBuilder()
565             .setName("deletePrivateCloudTest")
566             .setDone(true)
567             .setResponse(Any.pack(expectedResponse))
568             .build();
569     mockVmwareEngine.addResponse(resultOperation);
570 
571     String name = "name3373707";
572 
573     PrivateCloud actualResponse = client.deletePrivateCloudAsync(name).get();
574     Assert.assertEquals(expectedResponse, actualResponse);
575 
576     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
577     Assert.assertEquals(1, actualRequests.size());
578     DeletePrivateCloudRequest actualRequest = ((DeletePrivateCloudRequest) actualRequests.get(0));
579 
580     Assert.assertEquals(name, actualRequest.getName());
581     Assert.assertTrue(
582         channelProvider.isHeaderSent(
583             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
584             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
585   }
586 
587   @Test
deletePrivateCloudExceptionTest2()588   public void deletePrivateCloudExceptionTest2() throws Exception {
589     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
590     mockVmwareEngine.addException(exception);
591 
592     try {
593       String name = "name3373707";
594       client.deletePrivateCloudAsync(name).get();
595       Assert.fail("No exception raised");
596     } catch (ExecutionException e) {
597       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
598       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
599       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
600     }
601   }
602 
603   @Test
undeletePrivateCloudTest()604   public void undeletePrivateCloudTest() throws Exception {
605     PrivateCloud expectedResponse =
606         PrivateCloud.newBuilder()
607             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
608             .setCreateTime(Timestamp.newBuilder().build())
609             .setUpdateTime(Timestamp.newBuilder().build())
610             .setDeleteTime(Timestamp.newBuilder().build())
611             .setExpireTime(Timestamp.newBuilder().build())
612             .setNetworkConfig(NetworkConfig.newBuilder().build())
613             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
614             .setDescription("description-1724546052")
615             .setHcx(Hcx.newBuilder().build())
616             .setNsx(Nsx.newBuilder().build())
617             .setVcenter(Vcenter.newBuilder().build())
618             .setUid("uid115792")
619             .build();
620     Operation resultOperation =
621         Operation.newBuilder()
622             .setName("undeletePrivateCloudTest")
623             .setDone(true)
624             .setResponse(Any.pack(expectedResponse))
625             .build();
626     mockVmwareEngine.addResponse(resultOperation);
627 
628     PrivateCloudName name = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
629 
630     PrivateCloud actualResponse = client.undeletePrivateCloudAsync(name).get();
631     Assert.assertEquals(expectedResponse, actualResponse);
632 
633     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
634     Assert.assertEquals(1, actualRequests.size());
635     UndeletePrivateCloudRequest actualRequest =
636         ((UndeletePrivateCloudRequest) actualRequests.get(0));
637 
638     Assert.assertEquals(name.toString(), actualRequest.getName());
639     Assert.assertTrue(
640         channelProvider.isHeaderSent(
641             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
642             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
643   }
644 
645   @Test
undeletePrivateCloudExceptionTest()646   public void undeletePrivateCloudExceptionTest() throws Exception {
647     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
648     mockVmwareEngine.addException(exception);
649 
650     try {
651       PrivateCloudName name = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
652       client.undeletePrivateCloudAsync(name).get();
653       Assert.fail("No exception raised");
654     } catch (ExecutionException e) {
655       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
656       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
657       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
658     }
659   }
660 
661   @Test
undeletePrivateCloudTest2()662   public void undeletePrivateCloudTest2() throws Exception {
663     PrivateCloud expectedResponse =
664         PrivateCloud.newBuilder()
665             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
666             .setCreateTime(Timestamp.newBuilder().build())
667             .setUpdateTime(Timestamp.newBuilder().build())
668             .setDeleteTime(Timestamp.newBuilder().build())
669             .setExpireTime(Timestamp.newBuilder().build())
670             .setNetworkConfig(NetworkConfig.newBuilder().build())
671             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
672             .setDescription("description-1724546052")
673             .setHcx(Hcx.newBuilder().build())
674             .setNsx(Nsx.newBuilder().build())
675             .setVcenter(Vcenter.newBuilder().build())
676             .setUid("uid115792")
677             .build();
678     Operation resultOperation =
679         Operation.newBuilder()
680             .setName("undeletePrivateCloudTest")
681             .setDone(true)
682             .setResponse(Any.pack(expectedResponse))
683             .build();
684     mockVmwareEngine.addResponse(resultOperation);
685 
686     String name = "name3373707";
687 
688     PrivateCloud actualResponse = client.undeletePrivateCloudAsync(name).get();
689     Assert.assertEquals(expectedResponse, actualResponse);
690 
691     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
692     Assert.assertEquals(1, actualRequests.size());
693     UndeletePrivateCloudRequest actualRequest =
694         ((UndeletePrivateCloudRequest) actualRequests.get(0));
695 
696     Assert.assertEquals(name, actualRequest.getName());
697     Assert.assertTrue(
698         channelProvider.isHeaderSent(
699             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
700             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
701   }
702 
703   @Test
undeletePrivateCloudExceptionTest2()704   public void undeletePrivateCloudExceptionTest2() throws Exception {
705     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
706     mockVmwareEngine.addException(exception);
707 
708     try {
709       String name = "name3373707";
710       client.undeletePrivateCloudAsync(name).get();
711       Assert.fail("No exception raised");
712     } catch (ExecutionException e) {
713       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
714       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
715       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
716     }
717   }
718 
719   @Test
listClustersTest()720   public void listClustersTest() throws Exception {
721     Cluster responsesElement = Cluster.newBuilder().build();
722     ListClustersResponse expectedResponse =
723         ListClustersResponse.newBuilder()
724             .setNextPageToken("")
725             .addAllClusters(Arrays.asList(responsesElement))
726             .build();
727     mockVmwareEngine.addResponse(expectedResponse);
728 
729     PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
730 
731     ListClustersPagedResponse pagedListResponse = client.listClusters(parent);
732 
733     List<Cluster> resources = Lists.newArrayList(pagedListResponse.iterateAll());
734 
735     Assert.assertEquals(1, resources.size());
736     Assert.assertEquals(expectedResponse.getClustersList().get(0), resources.get(0));
737 
738     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
739     Assert.assertEquals(1, actualRequests.size());
740     ListClustersRequest actualRequest = ((ListClustersRequest) actualRequests.get(0));
741 
742     Assert.assertEquals(parent.toString(), actualRequest.getParent());
743     Assert.assertTrue(
744         channelProvider.isHeaderSent(
745             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
746             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
747   }
748 
749   @Test
listClustersExceptionTest()750   public void listClustersExceptionTest() throws Exception {
751     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
752     mockVmwareEngine.addException(exception);
753 
754     try {
755       PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
756       client.listClusters(parent);
757       Assert.fail("No exception raised");
758     } catch (InvalidArgumentException e) {
759       // Expected exception.
760     }
761   }
762 
763   @Test
listClustersTest2()764   public void listClustersTest2() throws Exception {
765     Cluster responsesElement = Cluster.newBuilder().build();
766     ListClustersResponse expectedResponse =
767         ListClustersResponse.newBuilder()
768             .setNextPageToken("")
769             .addAllClusters(Arrays.asList(responsesElement))
770             .build();
771     mockVmwareEngine.addResponse(expectedResponse);
772 
773     String parent = "parent-995424086";
774 
775     ListClustersPagedResponse pagedListResponse = client.listClusters(parent);
776 
777     List<Cluster> resources = Lists.newArrayList(pagedListResponse.iterateAll());
778 
779     Assert.assertEquals(1, resources.size());
780     Assert.assertEquals(expectedResponse.getClustersList().get(0), resources.get(0));
781 
782     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
783     Assert.assertEquals(1, actualRequests.size());
784     ListClustersRequest actualRequest = ((ListClustersRequest) actualRequests.get(0));
785 
786     Assert.assertEquals(parent, actualRequest.getParent());
787     Assert.assertTrue(
788         channelProvider.isHeaderSent(
789             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
790             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
791   }
792 
793   @Test
listClustersExceptionTest2()794   public void listClustersExceptionTest2() throws Exception {
795     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
796     mockVmwareEngine.addException(exception);
797 
798     try {
799       String parent = "parent-995424086";
800       client.listClusters(parent);
801       Assert.fail("No exception raised");
802     } catch (InvalidArgumentException e) {
803       // Expected exception.
804     }
805   }
806 
807   @Test
getClusterTest()808   public void getClusterTest() throws Exception {
809     Cluster expectedResponse =
810         Cluster.newBuilder()
811             .setName(
812                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
813                     .toString())
814             .setCreateTime(Timestamp.newBuilder().build())
815             .setUpdateTime(Timestamp.newBuilder().build())
816             .setManagement(true)
817             .setUid("uid115792")
818             .putAllNodeTypeConfigs(new HashMap<String, NodeTypeConfig>())
819             .build();
820     mockVmwareEngine.addResponse(expectedResponse);
821 
822     ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]");
823 
824     Cluster actualResponse = client.getCluster(name);
825     Assert.assertEquals(expectedResponse, actualResponse);
826 
827     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
828     Assert.assertEquals(1, actualRequests.size());
829     GetClusterRequest actualRequest = ((GetClusterRequest) actualRequests.get(0));
830 
831     Assert.assertEquals(name.toString(), actualRequest.getName());
832     Assert.assertTrue(
833         channelProvider.isHeaderSent(
834             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
835             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
836   }
837 
838   @Test
getClusterExceptionTest()839   public void getClusterExceptionTest() throws Exception {
840     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
841     mockVmwareEngine.addException(exception);
842 
843     try {
844       ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]");
845       client.getCluster(name);
846       Assert.fail("No exception raised");
847     } catch (InvalidArgumentException e) {
848       // Expected exception.
849     }
850   }
851 
852   @Test
getClusterTest2()853   public void getClusterTest2() throws Exception {
854     Cluster expectedResponse =
855         Cluster.newBuilder()
856             .setName(
857                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
858                     .toString())
859             .setCreateTime(Timestamp.newBuilder().build())
860             .setUpdateTime(Timestamp.newBuilder().build())
861             .setManagement(true)
862             .setUid("uid115792")
863             .putAllNodeTypeConfigs(new HashMap<String, NodeTypeConfig>())
864             .build();
865     mockVmwareEngine.addResponse(expectedResponse);
866 
867     String name = "name3373707";
868 
869     Cluster actualResponse = client.getCluster(name);
870     Assert.assertEquals(expectedResponse, actualResponse);
871 
872     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
873     Assert.assertEquals(1, actualRequests.size());
874     GetClusterRequest actualRequest = ((GetClusterRequest) actualRequests.get(0));
875 
876     Assert.assertEquals(name, actualRequest.getName());
877     Assert.assertTrue(
878         channelProvider.isHeaderSent(
879             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
880             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
881   }
882 
883   @Test
getClusterExceptionTest2()884   public void getClusterExceptionTest2() throws Exception {
885     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
886     mockVmwareEngine.addException(exception);
887 
888     try {
889       String name = "name3373707";
890       client.getCluster(name);
891       Assert.fail("No exception raised");
892     } catch (InvalidArgumentException e) {
893       // Expected exception.
894     }
895   }
896 
897   @Test
createClusterTest()898   public void createClusterTest() throws Exception {
899     Cluster expectedResponse =
900         Cluster.newBuilder()
901             .setName(
902                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
903                     .toString())
904             .setCreateTime(Timestamp.newBuilder().build())
905             .setUpdateTime(Timestamp.newBuilder().build())
906             .setManagement(true)
907             .setUid("uid115792")
908             .putAllNodeTypeConfigs(new HashMap<String, NodeTypeConfig>())
909             .build();
910     Operation resultOperation =
911         Operation.newBuilder()
912             .setName("createClusterTest")
913             .setDone(true)
914             .setResponse(Any.pack(expectedResponse))
915             .build();
916     mockVmwareEngine.addResponse(resultOperation);
917 
918     PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
919     Cluster cluster = Cluster.newBuilder().build();
920     String clusterId = "clusterId561939637";
921 
922     Cluster actualResponse = client.createClusterAsync(parent, cluster, clusterId).get();
923     Assert.assertEquals(expectedResponse, actualResponse);
924 
925     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
926     Assert.assertEquals(1, actualRequests.size());
927     CreateClusterRequest actualRequest = ((CreateClusterRequest) actualRequests.get(0));
928 
929     Assert.assertEquals(parent.toString(), actualRequest.getParent());
930     Assert.assertEquals(cluster, actualRequest.getCluster());
931     Assert.assertEquals(clusterId, actualRequest.getClusterId());
932     Assert.assertTrue(
933         channelProvider.isHeaderSent(
934             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
935             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
936   }
937 
938   @Test
createClusterExceptionTest()939   public void createClusterExceptionTest() throws Exception {
940     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
941     mockVmwareEngine.addException(exception);
942 
943     try {
944       PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
945       Cluster cluster = Cluster.newBuilder().build();
946       String clusterId = "clusterId561939637";
947       client.createClusterAsync(parent, cluster, clusterId).get();
948       Assert.fail("No exception raised");
949     } catch (ExecutionException e) {
950       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
951       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
952       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
953     }
954   }
955 
956   @Test
createClusterTest2()957   public void createClusterTest2() throws Exception {
958     Cluster expectedResponse =
959         Cluster.newBuilder()
960             .setName(
961                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
962                     .toString())
963             .setCreateTime(Timestamp.newBuilder().build())
964             .setUpdateTime(Timestamp.newBuilder().build())
965             .setManagement(true)
966             .setUid("uid115792")
967             .putAllNodeTypeConfigs(new HashMap<String, NodeTypeConfig>())
968             .build();
969     Operation resultOperation =
970         Operation.newBuilder()
971             .setName("createClusterTest")
972             .setDone(true)
973             .setResponse(Any.pack(expectedResponse))
974             .build();
975     mockVmwareEngine.addResponse(resultOperation);
976 
977     String parent = "parent-995424086";
978     Cluster cluster = Cluster.newBuilder().build();
979     String clusterId = "clusterId561939637";
980 
981     Cluster actualResponse = client.createClusterAsync(parent, cluster, clusterId).get();
982     Assert.assertEquals(expectedResponse, actualResponse);
983 
984     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
985     Assert.assertEquals(1, actualRequests.size());
986     CreateClusterRequest actualRequest = ((CreateClusterRequest) actualRequests.get(0));
987 
988     Assert.assertEquals(parent, actualRequest.getParent());
989     Assert.assertEquals(cluster, actualRequest.getCluster());
990     Assert.assertEquals(clusterId, actualRequest.getClusterId());
991     Assert.assertTrue(
992         channelProvider.isHeaderSent(
993             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
994             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
995   }
996 
997   @Test
createClusterExceptionTest2()998   public void createClusterExceptionTest2() throws Exception {
999     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1000     mockVmwareEngine.addException(exception);
1001 
1002     try {
1003       String parent = "parent-995424086";
1004       Cluster cluster = Cluster.newBuilder().build();
1005       String clusterId = "clusterId561939637";
1006       client.createClusterAsync(parent, cluster, clusterId).get();
1007       Assert.fail("No exception raised");
1008     } catch (ExecutionException e) {
1009       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1010       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1011       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1012     }
1013   }
1014 
1015   @Test
updateClusterTest()1016   public void updateClusterTest() throws Exception {
1017     Cluster expectedResponse =
1018         Cluster.newBuilder()
1019             .setName(
1020                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
1021                     .toString())
1022             .setCreateTime(Timestamp.newBuilder().build())
1023             .setUpdateTime(Timestamp.newBuilder().build())
1024             .setManagement(true)
1025             .setUid("uid115792")
1026             .putAllNodeTypeConfigs(new HashMap<String, NodeTypeConfig>())
1027             .build();
1028     Operation resultOperation =
1029         Operation.newBuilder()
1030             .setName("updateClusterTest")
1031             .setDone(true)
1032             .setResponse(Any.pack(expectedResponse))
1033             .build();
1034     mockVmwareEngine.addResponse(resultOperation);
1035 
1036     Cluster cluster = Cluster.newBuilder().build();
1037     FieldMask updateMask = FieldMask.newBuilder().build();
1038 
1039     Cluster actualResponse = client.updateClusterAsync(cluster, updateMask).get();
1040     Assert.assertEquals(expectedResponse, actualResponse);
1041 
1042     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1043     Assert.assertEquals(1, actualRequests.size());
1044     UpdateClusterRequest actualRequest = ((UpdateClusterRequest) actualRequests.get(0));
1045 
1046     Assert.assertEquals(cluster, actualRequest.getCluster());
1047     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
1048     Assert.assertTrue(
1049         channelProvider.isHeaderSent(
1050             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1051             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1052   }
1053 
1054   @Test
updateClusterExceptionTest()1055   public void updateClusterExceptionTest() throws Exception {
1056     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1057     mockVmwareEngine.addException(exception);
1058 
1059     try {
1060       Cluster cluster = Cluster.newBuilder().build();
1061       FieldMask updateMask = FieldMask.newBuilder().build();
1062       client.updateClusterAsync(cluster, updateMask).get();
1063       Assert.fail("No exception raised");
1064     } catch (ExecutionException e) {
1065       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1066       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1067       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1068     }
1069   }
1070 
1071   @Test
deleteClusterTest()1072   public void deleteClusterTest() throws Exception {
1073     Empty expectedResponse = Empty.newBuilder().build();
1074     Operation resultOperation =
1075         Operation.newBuilder()
1076             .setName("deleteClusterTest")
1077             .setDone(true)
1078             .setResponse(Any.pack(expectedResponse))
1079             .build();
1080     mockVmwareEngine.addResponse(resultOperation);
1081 
1082     ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]");
1083 
1084     client.deleteClusterAsync(name).get();
1085 
1086     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1087     Assert.assertEquals(1, actualRequests.size());
1088     DeleteClusterRequest actualRequest = ((DeleteClusterRequest) actualRequests.get(0));
1089 
1090     Assert.assertEquals(name.toString(), actualRequest.getName());
1091     Assert.assertTrue(
1092         channelProvider.isHeaderSent(
1093             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1094             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1095   }
1096 
1097   @Test
deleteClusterExceptionTest()1098   public void deleteClusterExceptionTest() throws Exception {
1099     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1100     mockVmwareEngine.addException(exception);
1101 
1102     try {
1103       ClusterName name = ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]");
1104       client.deleteClusterAsync(name).get();
1105       Assert.fail("No exception raised");
1106     } catch (ExecutionException e) {
1107       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1108       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1109       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1110     }
1111   }
1112 
1113   @Test
deleteClusterTest2()1114   public void deleteClusterTest2() throws Exception {
1115     Empty expectedResponse = Empty.newBuilder().build();
1116     Operation resultOperation =
1117         Operation.newBuilder()
1118             .setName("deleteClusterTest")
1119             .setDone(true)
1120             .setResponse(Any.pack(expectedResponse))
1121             .build();
1122     mockVmwareEngine.addResponse(resultOperation);
1123 
1124     String name = "name3373707";
1125 
1126     client.deleteClusterAsync(name).get();
1127 
1128     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1129     Assert.assertEquals(1, actualRequests.size());
1130     DeleteClusterRequest actualRequest = ((DeleteClusterRequest) actualRequests.get(0));
1131 
1132     Assert.assertEquals(name, actualRequest.getName());
1133     Assert.assertTrue(
1134         channelProvider.isHeaderSent(
1135             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1136             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1137   }
1138 
1139   @Test
deleteClusterExceptionTest2()1140   public void deleteClusterExceptionTest2() throws Exception {
1141     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1142     mockVmwareEngine.addException(exception);
1143 
1144     try {
1145       String name = "name3373707";
1146       client.deleteClusterAsync(name).get();
1147       Assert.fail("No exception raised");
1148     } catch (ExecutionException e) {
1149       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1150       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1151       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1152     }
1153   }
1154 
1155   @Test
listSubnetsTest()1156   public void listSubnetsTest() throws Exception {
1157     Subnet responsesElement = Subnet.newBuilder().build();
1158     ListSubnetsResponse expectedResponse =
1159         ListSubnetsResponse.newBuilder()
1160             .setNextPageToken("")
1161             .addAllSubnets(Arrays.asList(responsesElement))
1162             .build();
1163     mockVmwareEngine.addResponse(expectedResponse);
1164 
1165     PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1166 
1167     ListSubnetsPagedResponse pagedListResponse = client.listSubnets(parent);
1168 
1169     List<Subnet> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1170 
1171     Assert.assertEquals(1, resources.size());
1172     Assert.assertEquals(expectedResponse.getSubnetsList().get(0), resources.get(0));
1173 
1174     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1175     Assert.assertEquals(1, actualRequests.size());
1176     ListSubnetsRequest actualRequest = ((ListSubnetsRequest) actualRequests.get(0));
1177 
1178     Assert.assertEquals(parent.toString(), actualRequest.getParent());
1179     Assert.assertTrue(
1180         channelProvider.isHeaderSent(
1181             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1182             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1183   }
1184 
1185   @Test
listSubnetsExceptionTest()1186   public void listSubnetsExceptionTest() throws Exception {
1187     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1188     mockVmwareEngine.addException(exception);
1189 
1190     try {
1191       PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1192       client.listSubnets(parent);
1193       Assert.fail("No exception raised");
1194     } catch (InvalidArgumentException e) {
1195       // Expected exception.
1196     }
1197   }
1198 
1199   @Test
listSubnetsTest2()1200   public void listSubnetsTest2() throws Exception {
1201     Subnet responsesElement = Subnet.newBuilder().build();
1202     ListSubnetsResponse expectedResponse =
1203         ListSubnetsResponse.newBuilder()
1204             .setNextPageToken("")
1205             .addAllSubnets(Arrays.asList(responsesElement))
1206             .build();
1207     mockVmwareEngine.addResponse(expectedResponse);
1208 
1209     String parent = "parent-995424086";
1210 
1211     ListSubnetsPagedResponse pagedListResponse = client.listSubnets(parent);
1212 
1213     List<Subnet> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1214 
1215     Assert.assertEquals(1, resources.size());
1216     Assert.assertEquals(expectedResponse.getSubnetsList().get(0), resources.get(0));
1217 
1218     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1219     Assert.assertEquals(1, actualRequests.size());
1220     ListSubnetsRequest actualRequest = ((ListSubnetsRequest) actualRequests.get(0));
1221 
1222     Assert.assertEquals(parent, actualRequest.getParent());
1223     Assert.assertTrue(
1224         channelProvider.isHeaderSent(
1225             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1226             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1227   }
1228 
1229   @Test
listSubnetsExceptionTest2()1230   public void listSubnetsExceptionTest2() throws Exception {
1231     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1232     mockVmwareEngine.addException(exception);
1233 
1234     try {
1235       String parent = "parent-995424086";
1236       client.listSubnets(parent);
1237       Assert.fail("No exception raised");
1238     } catch (InvalidArgumentException e) {
1239       // Expected exception.
1240     }
1241   }
1242 
1243   @Test
listNodeTypesTest()1244   public void listNodeTypesTest() throws Exception {
1245     NodeType responsesElement = NodeType.newBuilder().build();
1246     ListNodeTypesResponse expectedResponse =
1247         ListNodeTypesResponse.newBuilder()
1248             .setNextPageToken("")
1249             .addAllNodeTypes(Arrays.asList(responsesElement))
1250             .build();
1251     mockVmwareEngine.addResponse(expectedResponse);
1252 
1253     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
1254 
1255     ListNodeTypesPagedResponse pagedListResponse = client.listNodeTypes(parent);
1256 
1257     List<NodeType> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1258 
1259     Assert.assertEquals(1, resources.size());
1260     Assert.assertEquals(expectedResponse.getNodeTypesList().get(0), resources.get(0));
1261 
1262     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1263     Assert.assertEquals(1, actualRequests.size());
1264     ListNodeTypesRequest actualRequest = ((ListNodeTypesRequest) actualRequests.get(0));
1265 
1266     Assert.assertEquals(parent.toString(), actualRequest.getParent());
1267     Assert.assertTrue(
1268         channelProvider.isHeaderSent(
1269             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1270             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1271   }
1272 
1273   @Test
listNodeTypesExceptionTest()1274   public void listNodeTypesExceptionTest() throws Exception {
1275     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1276     mockVmwareEngine.addException(exception);
1277 
1278     try {
1279       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
1280       client.listNodeTypes(parent);
1281       Assert.fail("No exception raised");
1282     } catch (InvalidArgumentException e) {
1283       // Expected exception.
1284     }
1285   }
1286 
1287   @Test
listNodeTypesTest2()1288   public void listNodeTypesTest2() throws Exception {
1289     NodeType responsesElement = NodeType.newBuilder().build();
1290     ListNodeTypesResponse expectedResponse =
1291         ListNodeTypesResponse.newBuilder()
1292             .setNextPageToken("")
1293             .addAllNodeTypes(Arrays.asList(responsesElement))
1294             .build();
1295     mockVmwareEngine.addResponse(expectedResponse);
1296 
1297     String parent = "parent-995424086";
1298 
1299     ListNodeTypesPagedResponse pagedListResponse = client.listNodeTypes(parent);
1300 
1301     List<NodeType> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1302 
1303     Assert.assertEquals(1, resources.size());
1304     Assert.assertEquals(expectedResponse.getNodeTypesList().get(0), resources.get(0));
1305 
1306     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1307     Assert.assertEquals(1, actualRequests.size());
1308     ListNodeTypesRequest actualRequest = ((ListNodeTypesRequest) actualRequests.get(0));
1309 
1310     Assert.assertEquals(parent, actualRequest.getParent());
1311     Assert.assertTrue(
1312         channelProvider.isHeaderSent(
1313             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1314             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1315   }
1316 
1317   @Test
listNodeTypesExceptionTest2()1318   public void listNodeTypesExceptionTest2() throws Exception {
1319     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1320     mockVmwareEngine.addException(exception);
1321 
1322     try {
1323       String parent = "parent-995424086";
1324       client.listNodeTypes(parent);
1325       Assert.fail("No exception raised");
1326     } catch (InvalidArgumentException e) {
1327       // Expected exception.
1328     }
1329   }
1330 
1331   @Test
getNodeTypeTest()1332   public void getNodeTypeTest() throws Exception {
1333     NodeType expectedResponse =
1334         NodeType.newBuilder()
1335             .setName(NodeTypeName.of("[PROJECT]", "[LOCATION]", "[NODE_TYPE]").toString())
1336             .setNodeTypeId("nodeTypeId1245332119")
1337             .setDisplayName("displayName1714148973")
1338             .setVirtualCpuCount(289038212)
1339             .setTotalCoreCount(-282953654)
1340             .setMemoryGb(1726613721)
1341             .setDiskSizeGb(-757478089)
1342             .addAllAvailableCustomCoreCounts(new ArrayList<Integer>())
1343             .build();
1344     mockVmwareEngine.addResponse(expectedResponse);
1345 
1346     NodeTypeName name = NodeTypeName.of("[PROJECT]", "[LOCATION]", "[NODE_TYPE]");
1347 
1348     NodeType actualResponse = client.getNodeType(name);
1349     Assert.assertEquals(expectedResponse, actualResponse);
1350 
1351     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1352     Assert.assertEquals(1, actualRequests.size());
1353     GetNodeTypeRequest actualRequest = ((GetNodeTypeRequest) actualRequests.get(0));
1354 
1355     Assert.assertEquals(name.toString(), actualRequest.getName());
1356     Assert.assertTrue(
1357         channelProvider.isHeaderSent(
1358             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1359             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1360   }
1361 
1362   @Test
getNodeTypeExceptionTest()1363   public void getNodeTypeExceptionTest() throws Exception {
1364     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1365     mockVmwareEngine.addException(exception);
1366 
1367     try {
1368       NodeTypeName name = NodeTypeName.of("[PROJECT]", "[LOCATION]", "[NODE_TYPE]");
1369       client.getNodeType(name);
1370       Assert.fail("No exception raised");
1371     } catch (InvalidArgumentException e) {
1372       // Expected exception.
1373     }
1374   }
1375 
1376   @Test
getNodeTypeTest2()1377   public void getNodeTypeTest2() throws Exception {
1378     NodeType expectedResponse =
1379         NodeType.newBuilder()
1380             .setName(NodeTypeName.of("[PROJECT]", "[LOCATION]", "[NODE_TYPE]").toString())
1381             .setNodeTypeId("nodeTypeId1245332119")
1382             .setDisplayName("displayName1714148973")
1383             .setVirtualCpuCount(289038212)
1384             .setTotalCoreCount(-282953654)
1385             .setMemoryGb(1726613721)
1386             .setDiskSizeGb(-757478089)
1387             .addAllAvailableCustomCoreCounts(new ArrayList<Integer>())
1388             .build();
1389     mockVmwareEngine.addResponse(expectedResponse);
1390 
1391     String name = "name3373707";
1392 
1393     NodeType actualResponse = client.getNodeType(name);
1394     Assert.assertEquals(expectedResponse, actualResponse);
1395 
1396     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1397     Assert.assertEquals(1, actualRequests.size());
1398     GetNodeTypeRequest actualRequest = ((GetNodeTypeRequest) actualRequests.get(0));
1399 
1400     Assert.assertEquals(name, actualRequest.getName());
1401     Assert.assertTrue(
1402         channelProvider.isHeaderSent(
1403             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1404             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1405   }
1406 
1407   @Test
getNodeTypeExceptionTest2()1408   public void getNodeTypeExceptionTest2() throws Exception {
1409     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1410     mockVmwareEngine.addException(exception);
1411 
1412     try {
1413       String name = "name3373707";
1414       client.getNodeType(name);
1415       Assert.fail("No exception raised");
1416     } catch (InvalidArgumentException e) {
1417       // Expected exception.
1418     }
1419   }
1420 
1421   @Test
showNsxCredentialsTest()1422   public void showNsxCredentialsTest() throws Exception {
1423     Credentials expectedResponse =
1424         Credentials.newBuilder()
1425             .setUsername("username-265713450")
1426             .setPassword("password1216985755")
1427             .build();
1428     mockVmwareEngine.addResponse(expectedResponse);
1429 
1430     PrivateCloudName privateCloud =
1431         PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1432 
1433     Credentials actualResponse = client.showNsxCredentials(privateCloud);
1434     Assert.assertEquals(expectedResponse, actualResponse);
1435 
1436     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1437     Assert.assertEquals(1, actualRequests.size());
1438     ShowNsxCredentialsRequest actualRequest = ((ShowNsxCredentialsRequest) actualRequests.get(0));
1439 
1440     Assert.assertEquals(privateCloud.toString(), actualRequest.getPrivateCloud());
1441     Assert.assertTrue(
1442         channelProvider.isHeaderSent(
1443             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1444             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1445   }
1446 
1447   @Test
showNsxCredentialsExceptionTest()1448   public void showNsxCredentialsExceptionTest() throws Exception {
1449     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1450     mockVmwareEngine.addException(exception);
1451 
1452     try {
1453       PrivateCloudName privateCloud =
1454           PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1455       client.showNsxCredentials(privateCloud);
1456       Assert.fail("No exception raised");
1457     } catch (InvalidArgumentException e) {
1458       // Expected exception.
1459     }
1460   }
1461 
1462   @Test
showNsxCredentialsTest2()1463   public void showNsxCredentialsTest2() throws Exception {
1464     Credentials expectedResponse =
1465         Credentials.newBuilder()
1466             .setUsername("username-265713450")
1467             .setPassword("password1216985755")
1468             .build();
1469     mockVmwareEngine.addResponse(expectedResponse);
1470 
1471     String privateCloud = "privateCloud974897778";
1472 
1473     Credentials actualResponse = client.showNsxCredentials(privateCloud);
1474     Assert.assertEquals(expectedResponse, actualResponse);
1475 
1476     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1477     Assert.assertEquals(1, actualRequests.size());
1478     ShowNsxCredentialsRequest actualRequest = ((ShowNsxCredentialsRequest) actualRequests.get(0));
1479 
1480     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
1481     Assert.assertTrue(
1482         channelProvider.isHeaderSent(
1483             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1484             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1485   }
1486 
1487   @Test
showNsxCredentialsExceptionTest2()1488   public void showNsxCredentialsExceptionTest2() throws Exception {
1489     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1490     mockVmwareEngine.addException(exception);
1491 
1492     try {
1493       String privateCloud = "privateCloud974897778";
1494       client.showNsxCredentials(privateCloud);
1495       Assert.fail("No exception raised");
1496     } catch (InvalidArgumentException e) {
1497       // Expected exception.
1498     }
1499   }
1500 
1501   @Test
showVcenterCredentialsTest()1502   public void showVcenterCredentialsTest() throws Exception {
1503     Credentials expectedResponse =
1504         Credentials.newBuilder()
1505             .setUsername("username-265713450")
1506             .setPassword("password1216985755")
1507             .build();
1508     mockVmwareEngine.addResponse(expectedResponse);
1509 
1510     PrivateCloudName privateCloud =
1511         PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1512 
1513     Credentials actualResponse = client.showVcenterCredentials(privateCloud);
1514     Assert.assertEquals(expectedResponse, actualResponse);
1515 
1516     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1517     Assert.assertEquals(1, actualRequests.size());
1518     ShowVcenterCredentialsRequest actualRequest =
1519         ((ShowVcenterCredentialsRequest) actualRequests.get(0));
1520 
1521     Assert.assertEquals(privateCloud.toString(), actualRequest.getPrivateCloud());
1522     Assert.assertTrue(
1523         channelProvider.isHeaderSent(
1524             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1525             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1526   }
1527 
1528   @Test
showVcenterCredentialsExceptionTest()1529   public void showVcenterCredentialsExceptionTest() throws Exception {
1530     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1531     mockVmwareEngine.addException(exception);
1532 
1533     try {
1534       PrivateCloudName privateCloud =
1535           PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1536       client.showVcenterCredentials(privateCloud);
1537       Assert.fail("No exception raised");
1538     } catch (InvalidArgumentException e) {
1539       // Expected exception.
1540     }
1541   }
1542 
1543   @Test
showVcenterCredentialsTest2()1544   public void showVcenterCredentialsTest2() throws Exception {
1545     Credentials expectedResponse =
1546         Credentials.newBuilder()
1547             .setUsername("username-265713450")
1548             .setPassword("password1216985755")
1549             .build();
1550     mockVmwareEngine.addResponse(expectedResponse);
1551 
1552     String privateCloud = "privateCloud974897778";
1553 
1554     Credentials actualResponse = client.showVcenterCredentials(privateCloud);
1555     Assert.assertEquals(expectedResponse, actualResponse);
1556 
1557     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1558     Assert.assertEquals(1, actualRequests.size());
1559     ShowVcenterCredentialsRequest actualRequest =
1560         ((ShowVcenterCredentialsRequest) actualRequests.get(0));
1561 
1562     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
1563     Assert.assertTrue(
1564         channelProvider.isHeaderSent(
1565             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1566             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1567   }
1568 
1569   @Test
showVcenterCredentialsExceptionTest2()1570   public void showVcenterCredentialsExceptionTest2() throws Exception {
1571     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1572     mockVmwareEngine.addException(exception);
1573 
1574     try {
1575       String privateCloud = "privateCloud974897778";
1576       client.showVcenterCredentials(privateCloud);
1577       Assert.fail("No exception raised");
1578     } catch (InvalidArgumentException e) {
1579       // Expected exception.
1580     }
1581   }
1582 
1583   @Test
resetNsxCredentialsTest()1584   public void resetNsxCredentialsTest() throws Exception {
1585     PrivateCloud expectedResponse =
1586         PrivateCloud.newBuilder()
1587             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
1588             .setCreateTime(Timestamp.newBuilder().build())
1589             .setUpdateTime(Timestamp.newBuilder().build())
1590             .setDeleteTime(Timestamp.newBuilder().build())
1591             .setExpireTime(Timestamp.newBuilder().build())
1592             .setNetworkConfig(NetworkConfig.newBuilder().build())
1593             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
1594             .setDescription("description-1724546052")
1595             .setHcx(Hcx.newBuilder().build())
1596             .setNsx(Nsx.newBuilder().build())
1597             .setVcenter(Vcenter.newBuilder().build())
1598             .setUid("uid115792")
1599             .build();
1600     Operation resultOperation =
1601         Operation.newBuilder()
1602             .setName("resetNsxCredentialsTest")
1603             .setDone(true)
1604             .setResponse(Any.pack(expectedResponse))
1605             .build();
1606     mockVmwareEngine.addResponse(resultOperation);
1607 
1608     PrivateCloudName privateCloud =
1609         PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1610 
1611     PrivateCloud actualResponse = client.resetNsxCredentialsAsync(privateCloud).get();
1612     Assert.assertEquals(expectedResponse, actualResponse);
1613 
1614     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1615     Assert.assertEquals(1, actualRequests.size());
1616     ResetNsxCredentialsRequest actualRequest = ((ResetNsxCredentialsRequest) actualRequests.get(0));
1617 
1618     Assert.assertEquals(privateCloud.toString(), actualRequest.getPrivateCloud());
1619     Assert.assertTrue(
1620         channelProvider.isHeaderSent(
1621             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1622             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1623   }
1624 
1625   @Test
resetNsxCredentialsExceptionTest()1626   public void resetNsxCredentialsExceptionTest() throws Exception {
1627     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1628     mockVmwareEngine.addException(exception);
1629 
1630     try {
1631       PrivateCloudName privateCloud =
1632           PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1633       client.resetNsxCredentialsAsync(privateCloud).get();
1634       Assert.fail("No exception raised");
1635     } catch (ExecutionException e) {
1636       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1637       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1638       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1639     }
1640   }
1641 
1642   @Test
resetNsxCredentialsTest2()1643   public void resetNsxCredentialsTest2() throws Exception {
1644     PrivateCloud expectedResponse =
1645         PrivateCloud.newBuilder()
1646             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
1647             .setCreateTime(Timestamp.newBuilder().build())
1648             .setUpdateTime(Timestamp.newBuilder().build())
1649             .setDeleteTime(Timestamp.newBuilder().build())
1650             .setExpireTime(Timestamp.newBuilder().build())
1651             .setNetworkConfig(NetworkConfig.newBuilder().build())
1652             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
1653             .setDescription("description-1724546052")
1654             .setHcx(Hcx.newBuilder().build())
1655             .setNsx(Nsx.newBuilder().build())
1656             .setVcenter(Vcenter.newBuilder().build())
1657             .setUid("uid115792")
1658             .build();
1659     Operation resultOperation =
1660         Operation.newBuilder()
1661             .setName("resetNsxCredentialsTest")
1662             .setDone(true)
1663             .setResponse(Any.pack(expectedResponse))
1664             .build();
1665     mockVmwareEngine.addResponse(resultOperation);
1666 
1667     String privateCloud = "privateCloud974897778";
1668 
1669     PrivateCloud actualResponse = client.resetNsxCredentialsAsync(privateCloud).get();
1670     Assert.assertEquals(expectedResponse, actualResponse);
1671 
1672     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1673     Assert.assertEquals(1, actualRequests.size());
1674     ResetNsxCredentialsRequest actualRequest = ((ResetNsxCredentialsRequest) actualRequests.get(0));
1675 
1676     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
1677     Assert.assertTrue(
1678         channelProvider.isHeaderSent(
1679             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1680             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1681   }
1682 
1683   @Test
resetNsxCredentialsExceptionTest2()1684   public void resetNsxCredentialsExceptionTest2() throws Exception {
1685     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1686     mockVmwareEngine.addException(exception);
1687 
1688     try {
1689       String privateCloud = "privateCloud974897778";
1690       client.resetNsxCredentialsAsync(privateCloud).get();
1691       Assert.fail("No exception raised");
1692     } catch (ExecutionException e) {
1693       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1694       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1695       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1696     }
1697   }
1698 
1699   @Test
resetVcenterCredentialsTest()1700   public void resetVcenterCredentialsTest() throws Exception {
1701     PrivateCloud expectedResponse =
1702         PrivateCloud.newBuilder()
1703             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
1704             .setCreateTime(Timestamp.newBuilder().build())
1705             .setUpdateTime(Timestamp.newBuilder().build())
1706             .setDeleteTime(Timestamp.newBuilder().build())
1707             .setExpireTime(Timestamp.newBuilder().build())
1708             .setNetworkConfig(NetworkConfig.newBuilder().build())
1709             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
1710             .setDescription("description-1724546052")
1711             .setHcx(Hcx.newBuilder().build())
1712             .setNsx(Nsx.newBuilder().build())
1713             .setVcenter(Vcenter.newBuilder().build())
1714             .setUid("uid115792")
1715             .build();
1716     Operation resultOperation =
1717         Operation.newBuilder()
1718             .setName("resetVcenterCredentialsTest")
1719             .setDone(true)
1720             .setResponse(Any.pack(expectedResponse))
1721             .build();
1722     mockVmwareEngine.addResponse(resultOperation);
1723 
1724     PrivateCloudName privateCloud =
1725         PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1726 
1727     PrivateCloud actualResponse = client.resetVcenterCredentialsAsync(privateCloud).get();
1728     Assert.assertEquals(expectedResponse, actualResponse);
1729 
1730     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1731     Assert.assertEquals(1, actualRequests.size());
1732     ResetVcenterCredentialsRequest actualRequest =
1733         ((ResetVcenterCredentialsRequest) actualRequests.get(0));
1734 
1735     Assert.assertEquals(privateCloud.toString(), actualRequest.getPrivateCloud());
1736     Assert.assertTrue(
1737         channelProvider.isHeaderSent(
1738             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1739             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1740   }
1741 
1742   @Test
resetVcenterCredentialsExceptionTest()1743   public void resetVcenterCredentialsExceptionTest() throws Exception {
1744     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1745     mockVmwareEngine.addException(exception);
1746 
1747     try {
1748       PrivateCloudName privateCloud =
1749           PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1750       client.resetVcenterCredentialsAsync(privateCloud).get();
1751       Assert.fail("No exception raised");
1752     } catch (ExecutionException e) {
1753       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1754       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1755       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1756     }
1757   }
1758 
1759   @Test
resetVcenterCredentialsTest2()1760   public void resetVcenterCredentialsTest2() throws Exception {
1761     PrivateCloud expectedResponse =
1762         PrivateCloud.newBuilder()
1763             .setName(PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]").toString())
1764             .setCreateTime(Timestamp.newBuilder().build())
1765             .setUpdateTime(Timestamp.newBuilder().build())
1766             .setDeleteTime(Timestamp.newBuilder().build())
1767             .setExpireTime(Timestamp.newBuilder().build())
1768             .setNetworkConfig(NetworkConfig.newBuilder().build())
1769             .setManagementCluster(PrivateCloud.ManagementCluster.newBuilder().build())
1770             .setDescription("description-1724546052")
1771             .setHcx(Hcx.newBuilder().build())
1772             .setNsx(Nsx.newBuilder().build())
1773             .setVcenter(Vcenter.newBuilder().build())
1774             .setUid("uid115792")
1775             .build();
1776     Operation resultOperation =
1777         Operation.newBuilder()
1778             .setName("resetVcenterCredentialsTest")
1779             .setDone(true)
1780             .setResponse(Any.pack(expectedResponse))
1781             .build();
1782     mockVmwareEngine.addResponse(resultOperation);
1783 
1784     String privateCloud = "privateCloud974897778";
1785 
1786     PrivateCloud actualResponse = client.resetVcenterCredentialsAsync(privateCloud).get();
1787     Assert.assertEquals(expectedResponse, actualResponse);
1788 
1789     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1790     Assert.assertEquals(1, actualRequests.size());
1791     ResetVcenterCredentialsRequest actualRequest =
1792         ((ResetVcenterCredentialsRequest) actualRequests.get(0));
1793 
1794     Assert.assertEquals(privateCloud, actualRequest.getPrivateCloud());
1795     Assert.assertTrue(
1796         channelProvider.isHeaderSent(
1797             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1798             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1799   }
1800 
1801   @Test
resetVcenterCredentialsExceptionTest2()1802   public void resetVcenterCredentialsExceptionTest2() throws Exception {
1803     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1804     mockVmwareEngine.addException(exception);
1805 
1806     try {
1807       String privateCloud = "privateCloud974897778";
1808       client.resetVcenterCredentialsAsync(privateCloud).get();
1809       Assert.fail("No exception raised");
1810     } catch (ExecutionException e) {
1811       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1812       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1813       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1814     }
1815   }
1816 
1817   @Test
createHcxActivationKeyTest()1818   public void createHcxActivationKeyTest() throws Exception {
1819     HcxActivationKey expectedResponse =
1820         HcxActivationKey.newBuilder()
1821             .setName(
1822                 HcxActivationKeyName.of(
1823                         "[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[HCX_ACTIVATION_KEY]")
1824                     .toString())
1825             .setCreateTime(Timestamp.newBuilder().build())
1826             .setActivationKey("activationKey1757742441")
1827             .setUid("uid115792")
1828             .build();
1829     Operation resultOperation =
1830         Operation.newBuilder()
1831             .setName("createHcxActivationKeyTest")
1832             .setDone(true)
1833             .setResponse(Any.pack(expectedResponse))
1834             .build();
1835     mockVmwareEngine.addResponse(resultOperation);
1836 
1837     PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1838     HcxActivationKey hcxActivationKey = HcxActivationKey.newBuilder().build();
1839     String hcxActivationKeyId = "hcxActivationKeyId-645871673";
1840 
1841     HcxActivationKey actualResponse =
1842         client.createHcxActivationKeyAsync(parent, hcxActivationKey, hcxActivationKeyId).get();
1843     Assert.assertEquals(expectedResponse, actualResponse);
1844 
1845     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1846     Assert.assertEquals(1, actualRequests.size());
1847     CreateHcxActivationKeyRequest actualRequest =
1848         ((CreateHcxActivationKeyRequest) actualRequests.get(0));
1849 
1850     Assert.assertEquals(parent.toString(), actualRequest.getParent());
1851     Assert.assertEquals(hcxActivationKey, actualRequest.getHcxActivationKey());
1852     Assert.assertEquals(hcxActivationKeyId, actualRequest.getHcxActivationKeyId());
1853     Assert.assertTrue(
1854         channelProvider.isHeaderSent(
1855             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1856             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1857   }
1858 
1859   @Test
createHcxActivationKeyExceptionTest()1860   public void createHcxActivationKeyExceptionTest() throws Exception {
1861     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1862     mockVmwareEngine.addException(exception);
1863 
1864     try {
1865       PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1866       HcxActivationKey hcxActivationKey = HcxActivationKey.newBuilder().build();
1867       String hcxActivationKeyId = "hcxActivationKeyId-645871673";
1868       client.createHcxActivationKeyAsync(parent, hcxActivationKey, hcxActivationKeyId).get();
1869       Assert.fail("No exception raised");
1870     } catch (ExecutionException e) {
1871       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1872       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1873       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1874     }
1875   }
1876 
1877   @Test
createHcxActivationKeyTest2()1878   public void createHcxActivationKeyTest2() throws Exception {
1879     HcxActivationKey expectedResponse =
1880         HcxActivationKey.newBuilder()
1881             .setName(
1882                 HcxActivationKeyName.of(
1883                         "[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[HCX_ACTIVATION_KEY]")
1884                     .toString())
1885             .setCreateTime(Timestamp.newBuilder().build())
1886             .setActivationKey("activationKey1757742441")
1887             .setUid("uid115792")
1888             .build();
1889     Operation resultOperation =
1890         Operation.newBuilder()
1891             .setName("createHcxActivationKeyTest")
1892             .setDone(true)
1893             .setResponse(Any.pack(expectedResponse))
1894             .build();
1895     mockVmwareEngine.addResponse(resultOperation);
1896 
1897     String parent = "parent-995424086";
1898     HcxActivationKey hcxActivationKey = HcxActivationKey.newBuilder().build();
1899     String hcxActivationKeyId = "hcxActivationKeyId-645871673";
1900 
1901     HcxActivationKey actualResponse =
1902         client.createHcxActivationKeyAsync(parent, hcxActivationKey, hcxActivationKeyId).get();
1903     Assert.assertEquals(expectedResponse, actualResponse);
1904 
1905     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1906     Assert.assertEquals(1, actualRequests.size());
1907     CreateHcxActivationKeyRequest actualRequest =
1908         ((CreateHcxActivationKeyRequest) actualRequests.get(0));
1909 
1910     Assert.assertEquals(parent, actualRequest.getParent());
1911     Assert.assertEquals(hcxActivationKey, actualRequest.getHcxActivationKey());
1912     Assert.assertEquals(hcxActivationKeyId, actualRequest.getHcxActivationKeyId());
1913     Assert.assertTrue(
1914         channelProvider.isHeaderSent(
1915             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1916             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1917   }
1918 
1919   @Test
createHcxActivationKeyExceptionTest2()1920   public void createHcxActivationKeyExceptionTest2() throws Exception {
1921     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1922     mockVmwareEngine.addException(exception);
1923 
1924     try {
1925       String parent = "parent-995424086";
1926       HcxActivationKey hcxActivationKey = HcxActivationKey.newBuilder().build();
1927       String hcxActivationKeyId = "hcxActivationKeyId-645871673";
1928       client.createHcxActivationKeyAsync(parent, hcxActivationKey, hcxActivationKeyId).get();
1929       Assert.fail("No exception raised");
1930     } catch (ExecutionException e) {
1931       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
1932       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
1933       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
1934     }
1935   }
1936 
1937   @Test
listHcxActivationKeysTest()1938   public void listHcxActivationKeysTest() throws Exception {
1939     HcxActivationKey responsesElement = HcxActivationKey.newBuilder().build();
1940     ListHcxActivationKeysResponse expectedResponse =
1941         ListHcxActivationKeysResponse.newBuilder()
1942             .setNextPageToken("")
1943             .addAllHcxActivationKeys(Arrays.asList(responsesElement))
1944             .build();
1945     mockVmwareEngine.addResponse(expectedResponse);
1946 
1947     PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1948 
1949     ListHcxActivationKeysPagedResponse pagedListResponse = client.listHcxActivationKeys(parent);
1950 
1951     List<HcxActivationKey> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1952 
1953     Assert.assertEquals(1, resources.size());
1954     Assert.assertEquals(expectedResponse.getHcxActivationKeysList().get(0), resources.get(0));
1955 
1956     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
1957     Assert.assertEquals(1, actualRequests.size());
1958     ListHcxActivationKeysRequest actualRequest =
1959         ((ListHcxActivationKeysRequest) actualRequests.get(0));
1960 
1961     Assert.assertEquals(parent.toString(), actualRequest.getParent());
1962     Assert.assertTrue(
1963         channelProvider.isHeaderSent(
1964             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
1965             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
1966   }
1967 
1968   @Test
listHcxActivationKeysExceptionTest()1969   public void listHcxActivationKeysExceptionTest() throws Exception {
1970     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
1971     mockVmwareEngine.addException(exception);
1972 
1973     try {
1974       PrivateCloudName parent = PrivateCloudName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]");
1975       client.listHcxActivationKeys(parent);
1976       Assert.fail("No exception raised");
1977     } catch (InvalidArgumentException e) {
1978       // Expected exception.
1979     }
1980   }
1981 
1982   @Test
listHcxActivationKeysTest2()1983   public void listHcxActivationKeysTest2() throws Exception {
1984     HcxActivationKey responsesElement = HcxActivationKey.newBuilder().build();
1985     ListHcxActivationKeysResponse expectedResponse =
1986         ListHcxActivationKeysResponse.newBuilder()
1987             .setNextPageToken("")
1988             .addAllHcxActivationKeys(Arrays.asList(responsesElement))
1989             .build();
1990     mockVmwareEngine.addResponse(expectedResponse);
1991 
1992     String parent = "parent-995424086";
1993 
1994     ListHcxActivationKeysPagedResponse pagedListResponse = client.listHcxActivationKeys(parent);
1995 
1996     List<HcxActivationKey> resources = Lists.newArrayList(pagedListResponse.iterateAll());
1997 
1998     Assert.assertEquals(1, resources.size());
1999     Assert.assertEquals(expectedResponse.getHcxActivationKeysList().get(0), resources.get(0));
2000 
2001     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2002     Assert.assertEquals(1, actualRequests.size());
2003     ListHcxActivationKeysRequest actualRequest =
2004         ((ListHcxActivationKeysRequest) actualRequests.get(0));
2005 
2006     Assert.assertEquals(parent, actualRequest.getParent());
2007     Assert.assertTrue(
2008         channelProvider.isHeaderSent(
2009             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2010             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2011   }
2012 
2013   @Test
listHcxActivationKeysExceptionTest2()2014   public void listHcxActivationKeysExceptionTest2() throws Exception {
2015     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2016     mockVmwareEngine.addException(exception);
2017 
2018     try {
2019       String parent = "parent-995424086";
2020       client.listHcxActivationKeys(parent);
2021       Assert.fail("No exception raised");
2022     } catch (InvalidArgumentException e) {
2023       // Expected exception.
2024     }
2025   }
2026 
2027   @Test
getHcxActivationKeyTest()2028   public void getHcxActivationKeyTest() throws Exception {
2029     HcxActivationKey expectedResponse =
2030         HcxActivationKey.newBuilder()
2031             .setName(
2032                 HcxActivationKeyName.of(
2033                         "[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[HCX_ACTIVATION_KEY]")
2034                     .toString())
2035             .setCreateTime(Timestamp.newBuilder().build())
2036             .setActivationKey("activationKey1757742441")
2037             .setUid("uid115792")
2038             .build();
2039     mockVmwareEngine.addResponse(expectedResponse);
2040 
2041     HcxActivationKeyName name =
2042         HcxActivationKeyName.of(
2043             "[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[HCX_ACTIVATION_KEY]");
2044 
2045     HcxActivationKey actualResponse = client.getHcxActivationKey(name);
2046     Assert.assertEquals(expectedResponse, actualResponse);
2047 
2048     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2049     Assert.assertEquals(1, actualRequests.size());
2050     GetHcxActivationKeyRequest actualRequest = ((GetHcxActivationKeyRequest) actualRequests.get(0));
2051 
2052     Assert.assertEquals(name.toString(), actualRequest.getName());
2053     Assert.assertTrue(
2054         channelProvider.isHeaderSent(
2055             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2056             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2057   }
2058 
2059   @Test
getHcxActivationKeyExceptionTest()2060   public void getHcxActivationKeyExceptionTest() throws Exception {
2061     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2062     mockVmwareEngine.addException(exception);
2063 
2064     try {
2065       HcxActivationKeyName name =
2066           HcxActivationKeyName.of(
2067               "[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[HCX_ACTIVATION_KEY]");
2068       client.getHcxActivationKey(name);
2069       Assert.fail("No exception raised");
2070     } catch (InvalidArgumentException e) {
2071       // Expected exception.
2072     }
2073   }
2074 
2075   @Test
getHcxActivationKeyTest2()2076   public void getHcxActivationKeyTest2() throws Exception {
2077     HcxActivationKey expectedResponse =
2078         HcxActivationKey.newBuilder()
2079             .setName(
2080                 HcxActivationKeyName.of(
2081                         "[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[HCX_ACTIVATION_KEY]")
2082                     .toString())
2083             .setCreateTime(Timestamp.newBuilder().build())
2084             .setActivationKey("activationKey1757742441")
2085             .setUid("uid115792")
2086             .build();
2087     mockVmwareEngine.addResponse(expectedResponse);
2088 
2089     String name = "name3373707";
2090 
2091     HcxActivationKey actualResponse = client.getHcxActivationKey(name);
2092     Assert.assertEquals(expectedResponse, actualResponse);
2093 
2094     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2095     Assert.assertEquals(1, actualRequests.size());
2096     GetHcxActivationKeyRequest actualRequest = ((GetHcxActivationKeyRequest) actualRequests.get(0));
2097 
2098     Assert.assertEquals(name, actualRequest.getName());
2099     Assert.assertTrue(
2100         channelProvider.isHeaderSent(
2101             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2102             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2103   }
2104 
2105   @Test
getHcxActivationKeyExceptionTest2()2106   public void getHcxActivationKeyExceptionTest2() throws Exception {
2107     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2108     mockVmwareEngine.addException(exception);
2109 
2110     try {
2111       String name = "name3373707";
2112       client.getHcxActivationKey(name);
2113       Assert.fail("No exception raised");
2114     } catch (InvalidArgumentException e) {
2115       // Expected exception.
2116     }
2117   }
2118 
2119   @Test
getNetworkPolicyTest()2120   public void getNetworkPolicyTest() throws Exception {
2121     NetworkPolicy expectedResponse =
2122         NetworkPolicy.newBuilder()
2123             .setName(NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]").toString())
2124             .setCreateTime(Timestamp.newBuilder().build())
2125             .setUpdateTime(Timestamp.newBuilder().build())
2126             .setInternetAccess(NetworkPolicy.NetworkService.newBuilder().build())
2127             .setExternalIp(NetworkPolicy.NetworkService.newBuilder().build())
2128             .setEdgeServicesCidr("edgeServicesCidr846515023")
2129             .setUid("uid115792")
2130             .setVmwareEngineNetwork(
2131                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2132                     .toString())
2133             .setDescription("description-1724546052")
2134             .setVmwareEngineNetworkCanonical(
2135                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2136                     .toString())
2137             .build();
2138     mockVmwareEngine.addResponse(expectedResponse);
2139 
2140     NetworkPolicyName name = NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]");
2141 
2142     NetworkPolicy actualResponse = client.getNetworkPolicy(name);
2143     Assert.assertEquals(expectedResponse, actualResponse);
2144 
2145     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2146     Assert.assertEquals(1, actualRequests.size());
2147     GetNetworkPolicyRequest actualRequest = ((GetNetworkPolicyRequest) actualRequests.get(0));
2148 
2149     Assert.assertEquals(name.toString(), actualRequest.getName());
2150     Assert.assertTrue(
2151         channelProvider.isHeaderSent(
2152             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2153             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2154   }
2155 
2156   @Test
getNetworkPolicyExceptionTest()2157   public void getNetworkPolicyExceptionTest() throws Exception {
2158     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2159     mockVmwareEngine.addException(exception);
2160 
2161     try {
2162       NetworkPolicyName name = NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]");
2163       client.getNetworkPolicy(name);
2164       Assert.fail("No exception raised");
2165     } catch (InvalidArgumentException e) {
2166       // Expected exception.
2167     }
2168   }
2169 
2170   @Test
getNetworkPolicyTest2()2171   public void getNetworkPolicyTest2() throws Exception {
2172     NetworkPolicy expectedResponse =
2173         NetworkPolicy.newBuilder()
2174             .setName(NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]").toString())
2175             .setCreateTime(Timestamp.newBuilder().build())
2176             .setUpdateTime(Timestamp.newBuilder().build())
2177             .setInternetAccess(NetworkPolicy.NetworkService.newBuilder().build())
2178             .setExternalIp(NetworkPolicy.NetworkService.newBuilder().build())
2179             .setEdgeServicesCidr("edgeServicesCidr846515023")
2180             .setUid("uid115792")
2181             .setVmwareEngineNetwork(
2182                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2183                     .toString())
2184             .setDescription("description-1724546052")
2185             .setVmwareEngineNetworkCanonical(
2186                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2187                     .toString())
2188             .build();
2189     mockVmwareEngine.addResponse(expectedResponse);
2190 
2191     String name = "name3373707";
2192 
2193     NetworkPolicy actualResponse = client.getNetworkPolicy(name);
2194     Assert.assertEquals(expectedResponse, actualResponse);
2195 
2196     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2197     Assert.assertEquals(1, actualRequests.size());
2198     GetNetworkPolicyRequest actualRequest = ((GetNetworkPolicyRequest) actualRequests.get(0));
2199 
2200     Assert.assertEquals(name, actualRequest.getName());
2201     Assert.assertTrue(
2202         channelProvider.isHeaderSent(
2203             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2204             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2205   }
2206 
2207   @Test
getNetworkPolicyExceptionTest2()2208   public void getNetworkPolicyExceptionTest2() throws Exception {
2209     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2210     mockVmwareEngine.addException(exception);
2211 
2212     try {
2213       String name = "name3373707";
2214       client.getNetworkPolicy(name);
2215       Assert.fail("No exception raised");
2216     } catch (InvalidArgumentException e) {
2217       // Expected exception.
2218     }
2219   }
2220 
2221   @Test
listNetworkPoliciesTest()2222   public void listNetworkPoliciesTest() throws Exception {
2223     NetworkPolicy responsesElement = NetworkPolicy.newBuilder().build();
2224     ListNetworkPoliciesResponse expectedResponse =
2225         ListNetworkPoliciesResponse.newBuilder()
2226             .setNextPageToken("")
2227             .addAllNetworkPolicies(Arrays.asList(responsesElement))
2228             .build();
2229     mockVmwareEngine.addResponse(expectedResponse);
2230 
2231     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2232 
2233     ListNetworkPoliciesPagedResponse pagedListResponse = client.listNetworkPolicies(parent);
2234 
2235     List<NetworkPolicy> resources = Lists.newArrayList(pagedListResponse.iterateAll());
2236 
2237     Assert.assertEquals(1, resources.size());
2238     Assert.assertEquals(expectedResponse.getNetworkPoliciesList().get(0), resources.get(0));
2239 
2240     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2241     Assert.assertEquals(1, actualRequests.size());
2242     ListNetworkPoliciesRequest actualRequest = ((ListNetworkPoliciesRequest) actualRequests.get(0));
2243 
2244     Assert.assertEquals(parent.toString(), actualRequest.getParent());
2245     Assert.assertTrue(
2246         channelProvider.isHeaderSent(
2247             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2248             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2249   }
2250 
2251   @Test
listNetworkPoliciesExceptionTest()2252   public void listNetworkPoliciesExceptionTest() throws Exception {
2253     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2254     mockVmwareEngine.addException(exception);
2255 
2256     try {
2257       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2258       client.listNetworkPolicies(parent);
2259       Assert.fail("No exception raised");
2260     } catch (InvalidArgumentException e) {
2261       // Expected exception.
2262     }
2263   }
2264 
2265   @Test
listNetworkPoliciesTest2()2266   public void listNetworkPoliciesTest2() throws Exception {
2267     NetworkPolicy responsesElement = NetworkPolicy.newBuilder().build();
2268     ListNetworkPoliciesResponse expectedResponse =
2269         ListNetworkPoliciesResponse.newBuilder()
2270             .setNextPageToken("")
2271             .addAllNetworkPolicies(Arrays.asList(responsesElement))
2272             .build();
2273     mockVmwareEngine.addResponse(expectedResponse);
2274 
2275     String parent = "parent-995424086";
2276 
2277     ListNetworkPoliciesPagedResponse pagedListResponse = client.listNetworkPolicies(parent);
2278 
2279     List<NetworkPolicy> resources = Lists.newArrayList(pagedListResponse.iterateAll());
2280 
2281     Assert.assertEquals(1, resources.size());
2282     Assert.assertEquals(expectedResponse.getNetworkPoliciesList().get(0), resources.get(0));
2283 
2284     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2285     Assert.assertEquals(1, actualRequests.size());
2286     ListNetworkPoliciesRequest actualRequest = ((ListNetworkPoliciesRequest) actualRequests.get(0));
2287 
2288     Assert.assertEquals(parent, actualRequest.getParent());
2289     Assert.assertTrue(
2290         channelProvider.isHeaderSent(
2291             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2292             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2293   }
2294 
2295   @Test
listNetworkPoliciesExceptionTest2()2296   public void listNetworkPoliciesExceptionTest2() throws Exception {
2297     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2298     mockVmwareEngine.addException(exception);
2299 
2300     try {
2301       String parent = "parent-995424086";
2302       client.listNetworkPolicies(parent);
2303       Assert.fail("No exception raised");
2304     } catch (InvalidArgumentException e) {
2305       // Expected exception.
2306     }
2307   }
2308 
2309   @Test
createNetworkPolicyTest()2310   public void createNetworkPolicyTest() throws Exception {
2311     NetworkPolicy expectedResponse =
2312         NetworkPolicy.newBuilder()
2313             .setName(NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]").toString())
2314             .setCreateTime(Timestamp.newBuilder().build())
2315             .setUpdateTime(Timestamp.newBuilder().build())
2316             .setInternetAccess(NetworkPolicy.NetworkService.newBuilder().build())
2317             .setExternalIp(NetworkPolicy.NetworkService.newBuilder().build())
2318             .setEdgeServicesCidr("edgeServicesCidr846515023")
2319             .setUid("uid115792")
2320             .setVmwareEngineNetwork(
2321                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2322                     .toString())
2323             .setDescription("description-1724546052")
2324             .setVmwareEngineNetworkCanonical(
2325                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2326                     .toString())
2327             .build();
2328     Operation resultOperation =
2329         Operation.newBuilder()
2330             .setName("createNetworkPolicyTest")
2331             .setDone(true)
2332             .setResponse(Any.pack(expectedResponse))
2333             .build();
2334     mockVmwareEngine.addResponse(resultOperation);
2335 
2336     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2337     NetworkPolicy networkPolicy = NetworkPolicy.newBuilder().build();
2338     String networkPolicyId = "networkPolicyId-727262661";
2339 
2340     NetworkPolicy actualResponse =
2341         client.createNetworkPolicyAsync(parent, networkPolicy, networkPolicyId).get();
2342     Assert.assertEquals(expectedResponse, actualResponse);
2343 
2344     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2345     Assert.assertEquals(1, actualRequests.size());
2346     CreateNetworkPolicyRequest actualRequest = ((CreateNetworkPolicyRequest) actualRequests.get(0));
2347 
2348     Assert.assertEquals(parent.toString(), actualRequest.getParent());
2349     Assert.assertEquals(networkPolicy, actualRequest.getNetworkPolicy());
2350     Assert.assertEquals(networkPolicyId, actualRequest.getNetworkPolicyId());
2351     Assert.assertTrue(
2352         channelProvider.isHeaderSent(
2353             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2354             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2355   }
2356 
2357   @Test
createNetworkPolicyExceptionTest()2358   public void createNetworkPolicyExceptionTest() throws Exception {
2359     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2360     mockVmwareEngine.addException(exception);
2361 
2362     try {
2363       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2364       NetworkPolicy networkPolicy = NetworkPolicy.newBuilder().build();
2365       String networkPolicyId = "networkPolicyId-727262661";
2366       client.createNetworkPolicyAsync(parent, networkPolicy, networkPolicyId).get();
2367       Assert.fail("No exception raised");
2368     } catch (ExecutionException e) {
2369       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2370       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2371       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2372     }
2373   }
2374 
2375   @Test
createNetworkPolicyTest2()2376   public void createNetworkPolicyTest2() throws Exception {
2377     NetworkPolicy expectedResponse =
2378         NetworkPolicy.newBuilder()
2379             .setName(NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]").toString())
2380             .setCreateTime(Timestamp.newBuilder().build())
2381             .setUpdateTime(Timestamp.newBuilder().build())
2382             .setInternetAccess(NetworkPolicy.NetworkService.newBuilder().build())
2383             .setExternalIp(NetworkPolicy.NetworkService.newBuilder().build())
2384             .setEdgeServicesCidr("edgeServicesCidr846515023")
2385             .setUid("uid115792")
2386             .setVmwareEngineNetwork(
2387                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2388                     .toString())
2389             .setDescription("description-1724546052")
2390             .setVmwareEngineNetworkCanonical(
2391                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2392                     .toString())
2393             .build();
2394     Operation resultOperation =
2395         Operation.newBuilder()
2396             .setName("createNetworkPolicyTest")
2397             .setDone(true)
2398             .setResponse(Any.pack(expectedResponse))
2399             .build();
2400     mockVmwareEngine.addResponse(resultOperation);
2401 
2402     String parent = "parent-995424086";
2403     NetworkPolicy networkPolicy = NetworkPolicy.newBuilder().build();
2404     String networkPolicyId = "networkPolicyId-727262661";
2405 
2406     NetworkPolicy actualResponse =
2407         client.createNetworkPolicyAsync(parent, networkPolicy, networkPolicyId).get();
2408     Assert.assertEquals(expectedResponse, actualResponse);
2409 
2410     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2411     Assert.assertEquals(1, actualRequests.size());
2412     CreateNetworkPolicyRequest actualRequest = ((CreateNetworkPolicyRequest) actualRequests.get(0));
2413 
2414     Assert.assertEquals(parent, actualRequest.getParent());
2415     Assert.assertEquals(networkPolicy, actualRequest.getNetworkPolicy());
2416     Assert.assertEquals(networkPolicyId, actualRequest.getNetworkPolicyId());
2417     Assert.assertTrue(
2418         channelProvider.isHeaderSent(
2419             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2420             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2421   }
2422 
2423   @Test
createNetworkPolicyExceptionTest2()2424   public void createNetworkPolicyExceptionTest2() throws Exception {
2425     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2426     mockVmwareEngine.addException(exception);
2427 
2428     try {
2429       String parent = "parent-995424086";
2430       NetworkPolicy networkPolicy = NetworkPolicy.newBuilder().build();
2431       String networkPolicyId = "networkPolicyId-727262661";
2432       client.createNetworkPolicyAsync(parent, networkPolicy, networkPolicyId).get();
2433       Assert.fail("No exception raised");
2434     } catch (ExecutionException e) {
2435       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2436       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2437       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2438     }
2439   }
2440 
2441   @Test
updateNetworkPolicyTest()2442   public void updateNetworkPolicyTest() throws Exception {
2443     NetworkPolicy expectedResponse =
2444         NetworkPolicy.newBuilder()
2445             .setName(NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]").toString())
2446             .setCreateTime(Timestamp.newBuilder().build())
2447             .setUpdateTime(Timestamp.newBuilder().build())
2448             .setInternetAccess(NetworkPolicy.NetworkService.newBuilder().build())
2449             .setExternalIp(NetworkPolicy.NetworkService.newBuilder().build())
2450             .setEdgeServicesCidr("edgeServicesCidr846515023")
2451             .setUid("uid115792")
2452             .setVmwareEngineNetwork(
2453                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2454                     .toString())
2455             .setDescription("description-1724546052")
2456             .setVmwareEngineNetworkCanonical(
2457                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2458                     .toString())
2459             .build();
2460     Operation resultOperation =
2461         Operation.newBuilder()
2462             .setName("updateNetworkPolicyTest")
2463             .setDone(true)
2464             .setResponse(Any.pack(expectedResponse))
2465             .build();
2466     mockVmwareEngine.addResponse(resultOperation);
2467 
2468     NetworkPolicy networkPolicy = NetworkPolicy.newBuilder().build();
2469     FieldMask updateMask = FieldMask.newBuilder().build();
2470 
2471     NetworkPolicy actualResponse = client.updateNetworkPolicyAsync(networkPolicy, updateMask).get();
2472     Assert.assertEquals(expectedResponse, actualResponse);
2473 
2474     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2475     Assert.assertEquals(1, actualRequests.size());
2476     UpdateNetworkPolicyRequest actualRequest = ((UpdateNetworkPolicyRequest) actualRequests.get(0));
2477 
2478     Assert.assertEquals(networkPolicy, actualRequest.getNetworkPolicy());
2479     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
2480     Assert.assertTrue(
2481         channelProvider.isHeaderSent(
2482             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2483             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2484   }
2485 
2486   @Test
updateNetworkPolicyExceptionTest()2487   public void updateNetworkPolicyExceptionTest() throws Exception {
2488     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2489     mockVmwareEngine.addException(exception);
2490 
2491     try {
2492       NetworkPolicy networkPolicy = NetworkPolicy.newBuilder().build();
2493       FieldMask updateMask = FieldMask.newBuilder().build();
2494       client.updateNetworkPolicyAsync(networkPolicy, updateMask).get();
2495       Assert.fail("No exception raised");
2496     } catch (ExecutionException e) {
2497       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2498       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2499       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2500     }
2501   }
2502 
2503   @Test
deleteNetworkPolicyTest()2504   public void deleteNetworkPolicyTest() throws Exception {
2505     Empty expectedResponse = Empty.newBuilder().build();
2506     Operation resultOperation =
2507         Operation.newBuilder()
2508             .setName("deleteNetworkPolicyTest")
2509             .setDone(true)
2510             .setResponse(Any.pack(expectedResponse))
2511             .build();
2512     mockVmwareEngine.addResponse(resultOperation);
2513 
2514     NetworkPolicyName name = NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]");
2515 
2516     client.deleteNetworkPolicyAsync(name).get();
2517 
2518     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2519     Assert.assertEquals(1, actualRequests.size());
2520     DeleteNetworkPolicyRequest actualRequest = ((DeleteNetworkPolicyRequest) actualRequests.get(0));
2521 
2522     Assert.assertEquals(name.toString(), actualRequest.getName());
2523     Assert.assertTrue(
2524         channelProvider.isHeaderSent(
2525             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2526             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2527   }
2528 
2529   @Test
deleteNetworkPolicyExceptionTest()2530   public void deleteNetworkPolicyExceptionTest() throws Exception {
2531     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2532     mockVmwareEngine.addException(exception);
2533 
2534     try {
2535       NetworkPolicyName name = NetworkPolicyName.of("[PROJECT]", "[LOCATION]", "[NETWORK_POLICY]");
2536       client.deleteNetworkPolicyAsync(name).get();
2537       Assert.fail("No exception raised");
2538     } catch (ExecutionException e) {
2539       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2540       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2541       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2542     }
2543   }
2544 
2545   @Test
deleteNetworkPolicyTest2()2546   public void deleteNetworkPolicyTest2() throws Exception {
2547     Empty expectedResponse = Empty.newBuilder().build();
2548     Operation resultOperation =
2549         Operation.newBuilder()
2550             .setName("deleteNetworkPolicyTest")
2551             .setDone(true)
2552             .setResponse(Any.pack(expectedResponse))
2553             .build();
2554     mockVmwareEngine.addResponse(resultOperation);
2555 
2556     String name = "name3373707";
2557 
2558     client.deleteNetworkPolicyAsync(name).get();
2559 
2560     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2561     Assert.assertEquals(1, actualRequests.size());
2562     DeleteNetworkPolicyRequest actualRequest = ((DeleteNetworkPolicyRequest) actualRequests.get(0));
2563 
2564     Assert.assertEquals(name, actualRequest.getName());
2565     Assert.assertTrue(
2566         channelProvider.isHeaderSent(
2567             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2568             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2569   }
2570 
2571   @Test
deleteNetworkPolicyExceptionTest2()2572   public void deleteNetworkPolicyExceptionTest2() throws Exception {
2573     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2574     mockVmwareEngine.addException(exception);
2575 
2576     try {
2577       String name = "name3373707";
2578       client.deleteNetworkPolicyAsync(name).get();
2579       Assert.fail("No exception raised");
2580     } catch (ExecutionException e) {
2581       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2582       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2583       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2584     }
2585   }
2586 
2587   @Test
createVmwareEngineNetworkTest()2588   public void createVmwareEngineNetworkTest() throws Exception {
2589     VmwareEngineNetwork expectedResponse =
2590         VmwareEngineNetwork.newBuilder()
2591             .setName(
2592                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2593                     .toString())
2594             .setCreateTime(Timestamp.newBuilder().build())
2595             .setUpdateTime(Timestamp.newBuilder().build())
2596             .setDescription("description-1724546052")
2597             .addAllVpcNetworks(new ArrayList<VmwareEngineNetwork.VpcNetwork>())
2598             .setUid("uid115792")
2599             .setEtag("etag3123477")
2600             .build();
2601     Operation resultOperation =
2602         Operation.newBuilder()
2603             .setName("createVmwareEngineNetworkTest")
2604             .setDone(true)
2605             .setResponse(Any.pack(expectedResponse))
2606             .build();
2607     mockVmwareEngine.addResponse(resultOperation);
2608 
2609     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2610     VmwareEngineNetwork vmwareEngineNetwork = VmwareEngineNetwork.newBuilder().build();
2611     String vmwareEngineNetworkId = "vmwareEngineNetworkId532532083";
2612 
2613     VmwareEngineNetwork actualResponse =
2614         client
2615             .createVmwareEngineNetworkAsync(parent, vmwareEngineNetwork, vmwareEngineNetworkId)
2616             .get();
2617     Assert.assertEquals(expectedResponse, actualResponse);
2618 
2619     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2620     Assert.assertEquals(1, actualRequests.size());
2621     CreateVmwareEngineNetworkRequest actualRequest =
2622         ((CreateVmwareEngineNetworkRequest) actualRequests.get(0));
2623 
2624     Assert.assertEquals(parent.toString(), actualRequest.getParent());
2625     Assert.assertEquals(vmwareEngineNetwork, actualRequest.getVmwareEngineNetwork());
2626     Assert.assertEquals(vmwareEngineNetworkId, actualRequest.getVmwareEngineNetworkId());
2627     Assert.assertTrue(
2628         channelProvider.isHeaderSent(
2629             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2630             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2631   }
2632 
2633   @Test
createVmwareEngineNetworkExceptionTest()2634   public void createVmwareEngineNetworkExceptionTest() throws Exception {
2635     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2636     mockVmwareEngine.addException(exception);
2637 
2638     try {
2639       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2640       VmwareEngineNetwork vmwareEngineNetwork = VmwareEngineNetwork.newBuilder().build();
2641       String vmwareEngineNetworkId = "vmwareEngineNetworkId532532083";
2642       client
2643           .createVmwareEngineNetworkAsync(parent, vmwareEngineNetwork, vmwareEngineNetworkId)
2644           .get();
2645       Assert.fail("No exception raised");
2646     } catch (ExecutionException e) {
2647       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2648       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2649       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2650     }
2651   }
2652 
2653   @Test
createVmwareEngineNetworkTest2()2654   public void createVmwareEngineNetworkTest2() throws Exception {
2655     VmwareEngineNetwork expectedResponse =
2656         VmwareEngineNetwork.newBuilder()
2657             .setName(
2658                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2659                     .toString())
2660             .setCreateTime(Timestamp.newBuilder().build())
2661             .setUpdateTime(Timestamp.newBuilder().build())
2662             .setDescription("description-1724546052")
2663             .addAllVpcNetworks(new ArrayList<VmwareEngineNetwork.VpcNetwork>())
2664             .setUid("uid115792")
2665             .setEtag("etag3123477")
2666             .build();
2667     Operation resultOperation =
2668         Operation.newBuilder()
2669             .setName("createVmwareEngineNetworkTest")
2670             .setDone(true)
2671             .setResponse(Any.pack(expectedResponse))
2672             .build();
2673     mockVmwareEngine.addResponse(resultOperation);
2674 
2675     String parent = "parent-995424086";
2676     VmwareEngineNetwork vmwareEngineNetwork = VmwareEngineNetwork.newBuilder().build();
2677     String vmwareEngineNetworkId = "vmwareEngineNetworkId532532083";
2678 
2679     VmwareEngineNetwork actualResponse =
2680         client
2681             .createVmwareEngineNetworkAsync(parent, vmwareEngineNetwork, vmwareEngineNetworkId)
2682             .get();
2683     Assert.assertEquals(expectedResponse, actualResponse);
2684 
2685     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2686     Assert.assertEquals(1, actualRequests.size());
2687     CreateVmwareEngineNetworkRequest actualRequest =
2688         ((CreateVmwareEngineNetworkRequest) actualRequests.get(0));
2689 
2690     Assert.assertEquals(parent, actualRequest.getParent());
2691     Assert.assertEquals(vmwareEngineNetwork, actualRequest.getVmwareEngineNetwork());
2692     Assert.assertEquals(vmwareEngineNetworkId, actualRequest.getVmwareEngineNetworkId());
2693     Assert.assertTrue(
2694         channelProvider.isHeaderSent(
2695             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2696             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2697   }
2698 
2699   @Test
createVmwareEngineNetworkExceptionTest2()2700   public void createVmwareEngineNetworkExceptionTest2() throws Exception {
2701     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2702     mockVmwareEngine.addException(exception);
2703 
2704     try {
2705       String parent = "parent-995424086";
2706       VmwareEngineNetwork vmwareEngineNetwork = VmwareEngineNetwork.newBuilder().build();
2707       String vmwareEngineNetworkId = "vmwareEngineNetworkId532532083";
2708       client
2709           .createVmwareEngineNetworkAsync(parent, vmwareEngineNetwork, vmwareEngineNetworkId)
2710           .get();
2711       Assert.fail("No exception raised");
2712     } catch (ExecutionException e) {
2713       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2714       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2715       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2716     }
2717   }
2718 
2719   @Test
updateVmwareEngineNetworkTest()2720   public void updateVmwareEngineNetworkTest() throws Exception {
2721     VmwareEngineNetwork expectedResponse =
2722         VmwareEngineNetwork.newBuilder()
2723             .setName(
2724                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2725                     .toString())
2726             .setCreateTime(Timestamp.newBuilder().build())
2727             .setUpdateTime(Timestamp.newBuilder().build())
2728             .setDescription("description-1724546052")
2729             .addAllVpcNetworks(new ArrayList<VmwareEngineNetwork.VpcNetwork>())
2730             .setUid("uid115792")
2731             .setEtag("etag3123477")
2732             .build();
2733     Operation resultOperation =
2734         Operation.newBuilder()
2735             .setName("updateVmwareEngineNetworkTest")
2736             .setDone(true)
2737             .setResponse(Any.pack(expectedResponse))
2738             .build();
2739     mockVmwareEngine.addResponse(resultOperation);
2740 
2741     VmwareEngineNetwork vmwareEngineNetwork = VmwareEngineNetwork.newBuilder().build();
2742     FieldMask updateMask = FieldMask.newBuilder().build();
2743 
2744     VmwareEngineNetwork actualResponse =
2745         client.updateVmwareEngineNetworkAsync(vmwareEngineNetwork, updateMask).get();
2746     Assert.assertEquals(expectedResponse, actualResponse);
2747 
2748     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2749     Assert.assertEquals(1, actualRequests.size());
2750     UpdateVmwareEngineNetworkRequest actualRequest =
2751         ((UpdateVmwareEngineNetworkRequest) actualRequests.get(0));
2752 
2753     Assert.assertEquals(vmwareEngineNetwork, actualRequest.getVmwareEngineNetwork());
2754     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
2755     Assert.assertTrue(
2756         channelProvider.isHeaderSent(
2757             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2758             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2759   }
2760 
2761   @Test
updateVmwareEngineNetworkExceptionTest()2762   public void updateVmwareEngineNetworkExceptionTest() throws Exception {
2763     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2764     mockVmwareEngine.addException(exception);
2765 
2766     try {
2767       VmwareEngineNetwork vmwareEngineNetwork = VmwareEngineNetwork.newBuilder().build();
2768       FieldMask updateMask = FieldMask.newBuilder().build();
2769       client.updateVmwareEngineNetworkAsync(vmwareEngineNetwork, updateMask).get();
2770       Assert.fail("No exception raised");
2771     } catch (ExecutionException e) {
2772       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2773       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2774       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2775     }
2776   }
2777 
2778   @Test
deleteVmwareEngineNetworkTest()2779   public void deleteVmwareEngineNetworkTest() throws Exception {
2780     Empty expectedResponse = Empty.newBuilder().build();
2781     Operation resultOperation =
2782         Operation.newBuilder()
2783             .setName("deleteVmwareEngineNetworkTest")
2784             .setDone(true)
2785             .setResponse(Any.pack(expectedResponse))
2786             .build();
2787     mockVmwareEngine.addResponse(resultOperation);
2788 
2789     VmwareEngineNetworkName name =
2790         VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]");
2791 
2792     client.deleteVmwareEngineNetworkAsync(name).get();
2793 
2794     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2795     Assert.assertEquals(1, actualRequests.size());
2796     DeleteVmwareEngineNetworkRequest actualRequest =
2797         ((DeleteVmwareEngineNetworkRequest) actualRequests.get(0));
2798 
2799     Assert.assertEquals(name.toString(), actualRequest.getName());
2800     Assert.assertTrue(
2801         channelProvider.isHeaderSent(
2802             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2803             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2804   }
2805 
2806   @Test
deleteVmwareEngineNetworkExceptionTest()2807   public void deleteVmwareEngineNetworkExceptionTest() throws Exception {
2808     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2809     mockVmwareEngine.addException(exception);
2810 
2811     try {
2812       VmwareEngineNetworkName name =
2813           VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]");
2814       client.deleteVmwareEngineNetworkAsync(name).get();
2815       Assert.fail("No exception raised");
2816     } catch (ExecutionException e) {
2817       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2818       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2819       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2820     }
2821   }
2822 
2823   @Test
deleteVmwareEngineNetworkTest2()2824   public void deleteVmwareEngineNetworkTest2() throws Exception {
2825     Empty expectedResponse = Empty.newBuilder().build();
2826     Operation resultOperation =
2827         Operation.newBuilder()
2828             .setName("deleteVmwareEngineNetworkTest")
2829             .setDone(true)
2830             .setResponse(Any.pack(expectedResponse))
2831             .build();
2832     mockVmwareEngine.addResponse(resultOperation);
2833 
2834     String name = "name3373707";
2835 
2836     client.deleteVmwareEngineNetworkAsync(name).get();
2837 
2838     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2839     Assert.assertEquals(1, actualRequests.size());
2840     DeleteVmwareEngineNetworkRequest actualRequest =
2841         ((DeleteVmwareEngineNetworkRequest) actualRequests.get(0));
2842 
2843     Assert.assertEquals(name, actualRequest.getName());
2844     Assert.assertTrue(
2845         channelProvider.isHeaderSent(
2846             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2847             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2848   }
2849 
2850   @Test
deleteVmwareEngineNetworkExceptionTest2()2851   public void deleteVmwareEngineNetworkExceptionTest2() throws Exception {
2852     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2853     mockVmwareEngine.addException(exception);
2854 
2855     try {
2856       String name = "name3373707";
2857       client.deleteVmwareEngineNetworkAsync(name).get();
2858       Assert.fail("No exception raised");
2859     } catch (ExecutionException e) {
2860       Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
2861       InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause());
2862       Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
2863     }
2864   }
2865 
2866   @Test
getVmwareEngineNetworkTest()2867   public void getVmwareEngineNetworkTest() throws Exception {
2868     VmwareEngineNetwork expectedResponse =
2869         VmwareEngineNetwork.newBuilder()
2870             .setName(
2871                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2872                     .toString())
2873             .setCreateTime(Timestamp.newBuilder().build())
2874             .setUpdateTime(Timestamp.newBuilder().build())
2875             .setDescription("description-1724546052")
2876             .addAllVpcNetworks(new ArrayList<VmwareEngineNetwork.VpcNetwork>())
2877             .setUid("uid115792")
2878             .setEtag("etag3123477")
2879             .build();
2880     mockVmwareEngine.addResponse(expectedResponse);
2881 
2882     VmwareEngineNetworkName name =
2883         VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]");
2884 
2885     VmwareEngineNetwork actualResponse = client.getVmwareEngineNetwork(name);
2886     Assert.assertEquals(expectedResponse, actualResponse);
2887 
2888     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2889     Assert.assertEquals(1, actualRequests.size());
2890     GetVmwareEngineNetworkRequest actualRequest =
2891         ((GetVmwareEngineNetworkRequest) actualRequests.get(0));
2892 
2893     Assert.assertEquals(name.toString(), actualRequest.getName());
2894     Assert.assertTrue(
2895         channelProvider.isHeaderSent(
2896             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2897             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2898   }
2899 
2900   @Test
getVmwareEngineNetworkExceptionTest()2901   public void getVmwareEngineNetworkExceptionTest() throws Exception {
2902     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2903     mockVmwareEngine.addException(exception);
2904 
2905     try {
2906       VmwareEngineNetworkName name =
2907           VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]");
2908       client.getVmwareEngineNetwork(name);
2909       Assert.fail("No exception raised");
2910     } catch (InvalidArgumentException e) {
2911       // Expected exception.
2912     }
2913   }
2914 
2915   @Test
getVmwareEngineNetworkTest2()2916   public void getVmwareEngineNetworkTest2() throws Exception {
2917     VmwareEngineNetwork expectedResponse =
2918         VmwareEngineNetwork.newBuilder()
2919             .setName(
2920                 VmwareEngineNetworkName.of("[PROJECT]", "[LOCATION]", "[VMWARE_ENGINE_NETWORK]")
2921                     .toString())
2922             .setCreateTime(Timestamp.newBuilder().build())
2923             .setUpdateTime(Timestamp.newBuilder().build())
2924             .setDescription("description-1724546052")
2925             .addAllVpcNetworks(new ArrayList<VmwareEngineNetwork.VpcNetwork>())
2926             .setUid("uid115792")
2927             .setEtag("etag3123477")
2928             .build();
2929     mockVmwareEngine.addResponse(expectedResponse);
2930 
2931     String name = "name3373707";
2932 
2933     VmwareEngineNetwork actualResponse = client.getVmwareEngineNetwork(name);
2934     Assert.assertEquals(expectedResponse, actualResponse);
2935 
2936     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2937     Assert.assertEquals(1, actualRequests.size());
2938     GetVmwareEngineNetworkRequest actualRequest =
2939         ((GetVmwareEngineNetworkRequest) actualRequests.get(0));
2940 
2941     Assert.assertEquals(name, actualRequest.getName());
2942     Assert.assertTrue(
2943         channelProvider.isHeaderSent(
2944             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2945             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2946   }
2947 
2948   @Test
getVmwareEngineNetworkExceptionTest2()2949   public void getVmwareEngineNetworkExceptionTest2() throws Exception {
2950     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2951     mockVmwareEngine.addException(exception);
2952 
2953     try {
2954       String name = "name3373707";
2955       client.getVmwareEngineNetwork(name);
2956       Assert.fail("No exception raised");
2957     } catch (InvalidArgumentException e) {
2958       // Expected exception.
2959     }
2960   }
2961 
2962   @Test
listVmwareEngineNetworksTest()2963   public void listVmwareEngineNetworksTest() throws Exception {
2964     VmwareEngineNetwork responsesElement = VmwareEngineNetwork.newBuilder().build();
2965     ListVmwareEngineNetworksResponse expectedResponse =
2966         ListVmwareEngineNetworksResponse.newBuilder()
2967             .setNextPageToken("")
2968             .addAllVmwareEngineNetworks(Arrays.asList(responsesElement))
2969             .build();
2970     mockVmwareEngine.addResponse(expectedResponse);
2971 
2972     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
2973 
2974     ListVmwareEngineNetworksPagedResponse pagedListResponse =
2975         client.listVmwareEngineNetworks(parent);
2976 
2977     List<VmwareEngineNetwork> resources = Lists.newArrayList(pagedListResponse.iterateAll());
2978 
2979     Assert.assertEquals(1, resources.size());
2980     Assert.assertEquals(expectedResponse.getVmwareEngineNetworksList().get(0), resources.get(0));
2981 
2982     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
2983     Assert.assertEquals(1, actualRequests.size());
2984     ListVmwareEngineNetworksRequest actualRequest =
2985         ((ListVmwareEngineNetworksRequest) actualRequests.get(0));
2986 
2987     Assert.assertEquals(parent.toString(), actualRequest.getParent());
2988     Assert.assertTrue(
2989         channelProvider.isHeaderSent(
2990             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
2991             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
2992   }
2993 
2994   @Test
listVmwareEngineNetworksExceptionTest()2995   public void listVmwareEngineNetworksExceptionTest() throws Exception {
2996     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
2997     mockVmwareEngine.addException(exception);
2998 
2999     try {
3000       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
3001       client.listVmwareEngineNetworks(parent);
3002       Assert.fail("No exception raised");
3003     } catch (InvalidArgumentException e) {
3004       // Expected exception.
3005     }
3006   }
3007 
3008   @Test
listVmwareEngineNetworksTest2()3009   public void listVmwareEngineNetworksTest2() throws Exception {
3010     VmwareEngineNetwork responsesElement = VmwareEngineNetwork.newBuilder().build();
3011     ListVmwareEngineNetworksResponse expectedResponse =
3012         ListVmwareEngineNetworksResponse.newBuilder()
3013             .setNextPageToken("")
3014             .addAllVmwareEngineNetworks(Arrays.asList(responsesElement))
3015             .build();
3016     mockVmwareEngine.addResponse(expectedResponse);
3017 
3018     String parent = "parent-995424086";
3019 
3020     ListVmwareEngineNetworksPagedResponse pagedListResponse =
3021         client.listVmwareEngineNetworks(parent);
3022 
3023     List<VmwareEngineNetwork> resources = Lists.newArrayList(pagedListResponse.iterateAll());
3024 
3025     Assert.assertEquals(1, resources.size());
3026     Assert.assertEquals(expectedResponse.getVmwareEngineNetworksList().get(0), resources.get(0));
3027 
3028     List<AbstractMessage> actualRequests = mockVmwareEngine.getRequests();
3029     Assert.assertEquals(1, actualRequests.size());
3030     ListVmwareEngineNetworksRequest actualRequest =
3031         ((ListVmwareEngineNetworksRequest) actualRequests.get(0));
3032 
3033     Assert.assertEquals(parent, actualRequest.getParent());
3034     Assert.assertTrue(
3035         channelProvider.isHeaderSent(
3036             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
3037             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
3038   }
3039 
3040   @Test
listVmwareEngineNetworksExceptionTest2()3041   public void listVmwareEngineNetworksExceptionTest2() throws Exception {
3042     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
3043     mockVmwareEngine.addException(exception);
3044 
3045     try {
3046       String parent = "parent-995424086";
3047       client.listVmwareEngineNetworks(parent);
3048       Assert.fail("No exception raised");
3049     } catch (InvalidArgumentException e) {
3050       // Expected exception.
3051     }
3052   }
3053 
3054   @Test
listLocationsTest()3055   public void listLocationsTest() throws Exception {
3056     Location responsesElement = Location.newBuilder().build();
3057     ListLocationsResponse expectedResponse =
3058         ListLocationsResponse.newBuilder()
3059             .setNextPageToken("")
3060             .addAllLocations(Arrays.asList(responsesElement))
3061             .build();
3062     mockLocations.addResponse(expectedResponse);
3063 
3064     ListLocationsRequest request =
3065         ListLocationsRequest.newBuilder()
3066             .setName("name3373707")
3067             .setFilter("filter-1274492040")
3068             .setPageSize(883849137)
3069             .setPageToken("pageToken873572522")
3070             .build();
3071 
3072     ListLocationsPagedResponse pagedListResponse = client.listLocations(request);
3073 
3074     List<Location> resources = Lists.newArrayList(pagedListResponse.iterateAll());
3075 
3076     Assert.assertEquals(1, resources.size());
3077     Assert.assertEquals(expectedResponse.getLocationsList().get(0), resources.get(0));
3078 
3079     List<AbstractMessage> actualRequests = mockLocations.getRequests();
3080     Assert.assertEquals(1, actualRequests.size());
3081     ListLocationsRequest actualRequest = ((ListLocationsRequest) actualRequests.get(0));
3082 
3083     Assert.assertEquals(request.getName(), actualRequest.getName());
3084     Assert.assertEquals(request.getFilter(), actualRequest.getFilter());
3085     Assert.assertEquals(request.getPageSize(), actualRequest.getPageSize());
3086     Assert.assertEquals(request.getPageToken(), actualRequest.getPageToken());
3087     Assert.assertTrue(
3088         channelProvider.isHeaderSent(
3089             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
3090             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
3091   }
3092 
3093   @Test
listLocationsExceptionTest()3094   public void listLocationsExceptionTest() throws Exception {
3095     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
3096     mockLocations.addException(exception);
3097 
3098     try {
3099       ListLocationsRequest request =
3100           ListLocationsRequest.newBuilder()
3101               .setName("name3373707")
3102               .setFilter("filter-1274492040")
3103               .setPageSize(883849137)
3104               .setPageToken("pageToken873572522")
3105               .build();
3106       client.listLocations(request);
3107       Assert.fail("No exception raised");
3108     } catch (InvalidArgumentException e) {
3109       // Expected exception.
3110     }
3111   }
3112 
3113   @Test
getLocationTest()3114   public void getLocationTest() throws Exception {
3115     Location expectedResponse =
3116         Location.newBuilder()
3117             .setName("name3373707")
3118             .setLocationId("locationId1541836720")
3119             .setDisplayName("displayName1714148973")
3120             .putAllLabels(new HashMap<String, String>())
3121             .setMetadata(Any.newBuilder().build())
3122             .build();
3123     mockLocations.addResponse(expectedResponse);
3124 
3125     GetLocationRequest request = GetLocationRequest.newBuilder().setName("name3373707").build();
3126 
3127     Location actualResponse = client.getLocation(request);
3128     Assert.assertEquals(expectedResponse, actualResponse);
3129 
3130     List<AbstractMessage> actualRequests = mockLocations.getRequests();
3131     Assert.assertEquals(1, actualRequests.size());
3132     GetLocationRequest actualRequest = ((GetLocationRequest) actualRequests.get(0));
3133 
3134     Assert.assertEquals(request.getName(), actualRequest.getName());
3135     Assert.assertTrue(
3136         channelProvider.isHeaderSent(
3137             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
3138             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
3139   }
3140 
3141   @Test
getLocationExceptionTest()3142   public void getLocationExceptionTest() throws Exception {
3143     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
3144     mockLocations.addException(exception);
3145 
3146     try {
3147       GetLocationRequest request = GetLocationRequest.newBuilder().setName("name3373707").build();
3148       client.getLocation(request);
3149       Assert.fail("No exception raised");
3150     } catch (InvalidArgumentException e) {
3151       // Expected exception.
3152     }
3153   }
3154 
3155   @Test
setIamPolicyTest()3156   public void setIamPolicyTest() throws Exception {
3157     Policy expectedResponse =
3158         Policy.newBuilder()
3159             .setVersion(351608024)
3160             .addAllBindings(new ArrayList<Binding>())
3161             .addAllAuditConfigs(new ArrayList<AuditConfig>())
3162             .setEtag(ByteString.EMPTY)
3163             .build();
3164     mockIAMPolicy.addResponse(expectedResponse);
3165 
3166     SetIamPolicyRequest request =
3167         SetIamPolicyRequest.newBuilder()
3168             .setResource(
3169                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
3170                     .toString())
3171             .setPolicy(Policy.newBuilder().build())
3172             .setUpdateMask(FieldMask.newBuilder().build())
3173             .build();
3174 
3175     Policy actualResponse = client.setIamPolicy(request);
3176     Assert.assertEquals(expectedResponse, actualResponse);
3177 
3178     List<AbstractMessage> actualRequests = mockIAMPolicy.getRequests();
3179     Assert.assertEquals(1, actualRequests.size());
3180     SetIamPolicyRequest actualRequest = ((SetIamPolicyRequest) actualRequests.get(0));
3181 
3182     Assert.assertEquals(request.getResource(), actualRequest.getResource());
3183     Assert.assertEquals(request.getPolicy(), actualRequest.getPolicy());
3184     Assert.assertEquals(request.getUpdateMask(), actualRequest.getUpdateMask());
3185     Assert.assertTrue(
3186         channelProvider.isHeaderSent(
3187             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
3188             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
3189   }
3190 
3191   @Test
setIamPolicyExceptionTest()3192   public void setIamPolicyExceptionTest() throws Exception {
3193     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
3194     mockIAMPolicy.addException(exception);
3195 
3196     try {
3197       SetIamPolicyRequest request =
3198           SetIamPolicyRequest.newBuilder()
3199               .setResource(
3200                   ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
3201                       .toString())
3202               .setPolicy(Policy.newBuilder().build())
3203               .setUpdateMask(FieldMask.newBuilder().build())
3204               .build();
3205       client.setIamPolicy(request);
3206       Assert.fail("No exception raised");
3207     } catch (InvalidArgumentException e) {
3208       // Expected exception.
3209     }
3210   }
3211 
3212   @Test
getIamPolicyTest()3213   public void getIamPolicyTest() throws Exception {
3214     Policy expectedResponse =
3215         Policy.newBuilder()
3216             .setVersion(351608024)
3217             .addAllBindings(new ArrayList<Binding>())
3218             .addAllAuditConfigs(new ArrayList<AuditConfig>())
3219             .setEtag(ByteString.EMPTY)
3220             .build();
3221     mockIAMPolicy.addResponse(expectedResponse);
3222 
3223     GetIamPolicyRequest request =
3224         GetIamPolicyRequest.newBuilder()
3225             .setResource(
3226                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
3227                     .toString())
3228             .setOptions(GetPolicyOptions.newBuilder().build())
3229             .build();
3230 
3231     Policy actualResponse = client.getIamPolicy(request);
3232     Assert.assertEquals(expectedResponse, actualResponse);
3233 
3234     List<AbstractMessage> actualRequests = mockIAMPolicy.getRequests();
3235     Assert.assertEquals(1, actualRequests.size());
3236     GetIamPolicyRequest actualRequest = ((GetIamPolicyRequest) actualRequests.get(0));
3237 
3238     Assert.assertEquals(request.getResource(), actualRequest.getResource());
3239     Assert.assertEquals(request.getOptions(), actualRequest.getOptions());
3240     Assert.assertTrue(
3241         channelProvider.isHeaderSent(
3242             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
3243             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
3244   }
3245 
3246   @Test
getIamPolicyExceptionTest()3247   public void getIamPolicyExceptionTest() throws Exception {
3248     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
3249     mockIAMPolicy.addException(exception);
3250 
3251     try {
3252       GetIamPolicyRequest request =
3253           GetIamPolicyRequest.newBuilder()
3254               .setResource(
3255                   ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
3256                       .toString())
3257               .setOptions(GetPolicyOptions.newBuilder().build())
3258               .build();
3259       client.getIamPolicy(request);
3260       Assert.fail("No exception raised");
3261     } catch (InvalidArgumentException e) {
3262       // Expected exception.
3263     }
3264   }
3265 
3266   @Test
testIamPermissionsTest()3267   public void testIamPermissionsTest() throws Exception {
3268     TestIamPermissionsResponse expectedResponse =
3269         TestIamPermissionsResponse.newBuilder().addAllPermissions(new ArrayList<String>()).build();
3270     mockIAMPolicy.addResponse(expectedResponse);
3271 
3272     TestIamPermissionsRequest request =
3273         TestIamPermissionsRequest.newBuilder()
3274             .setResource(
3275                 ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
3276                     .toString())
3277             .addAllPermissions(new ArrayList<String>())
3278             .build();
3279 
3280     TestIamPermissionsResponse actualResponse = client.testIamPermissions(request);
3281     Assert.assertEquals(expectedResponse, actualResponse);
3282 
3283     List<AbstractMessage> actualRequests = mockIAMPolicy.getRequests();
3284     Assert.assertEquals(1, actualRequests.size());
3285     TestIamPermissionsRequest actualRequest = ((TestIamPermissionsRequest) actualRequests.get(0));
3286 
3287     Assert.assertEquals(request.getResource(), actualRequest.getResource());
3288     Assert.assertEquals(request.getPermissionsList(), actualRequest.getPermissionsList());
3289     Assert.assertTrue(
3290         channelProvider.isHeaderSent(
3291             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
3292             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
3293   }
3294 
3295   @Test
testIamPermissionsExceptionTest()3296   public void testIamPermissionsExceptionTest() throws Exception {
3297     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
3298     mockIAMPolicy.addException(exception);
3299 
3300     try {
3301       TestIamPermissionsRequest request =
3302           TestIamPermissionsRequest.newBuilder()
3303               .setResource(
3304                   ClusterName.of("[PROJECT]", "[LOCATION]", "[PRIVATE_CLOUD]", "[CLUSTER]")
3305                       .toString())
3306               .addAllPermissions(new ArrayList<String>())
3307               .build();
3308       client.testIamPermissions(request);
3309       Assert.fail("No exception raised");
3310     } catch (InvalidArgumentException e) {
3311       // Expected exception.
3312     }
3313   }
3314 }
3315