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