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