• 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.speech.v1p1beta1;
18 
19 import static com.google.cloud.speech.v1p1beta1.AdaptationClient.ListCustomClassesPagedResponse;
20 import static com.google.cloud.speech.v1p1beta1.AdaptationClient.ListPhraseSetPagedResponse;
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.common.collect.Lists;
30 import com.google.protobuf.AbstractMessage;
31 import com.google.protobuf.Empty;
32 import com.google.protobuf.FieldMask;
33 import io.grpc.StatusRuntimeException;
34 import java.io.IOException;
35 import java.util.ArrayList;
36 import java.util.Arrays;
37 import java.util.List;
38 import java.util.UUID;
39 import javax.annotation.Generated;
40 import org.junit.After;
41 import org.junit.AfterClass;
42 import org.junit.Assert;
43 import org.junit.Before;
44 import org.junit.BeforeClass;
45 import org.junit.Test;
46 
47 @Generated("by gapic-generator-java")
48 public class AdaptationClientTest {
49   private static MockAdaptation mockAdaptation;
50   private static MockServiceHelper mockServiceHelper;
51   private LocalChannelProvider channelProvider;
52   private AdaptationClient client;
53 
54   @BeforeClass
startStaticServer()55   public static void startStaticServer() {
56     mockAdaptation = new MockAdaptation();
57     mockServiceHelper =
58         new MockServiceHelper(
59             UUID.randomUUID().toString(), Arrays.<MockGrpcService>asList(mockAdaptation));
60     mockServiceHelper.start();
61   }
62 
63   @AfterClass
stopServer()64   public static void stopServer() {
65     mockServiceHelper.stop();
66   }
67 
68   @Before
setUp()69   public void setUp() throws IOException {
70     mockServiceHelper.reset();
71     channelProvider = mockServiceHelper.createChannelProvider();
72     AdaptationSettings settings =
73         AdaptationSettings.newBuilder()
74             .setTransportChannelProvider(channelProvider)
75             .setCredentialsProvider(NoCredentialsProvider.create())
76             .build();
77     client = AdaptationClient.create(settings);
78   }
79 
80   @After
tearDown()81   public void tearDown() throws Exception {
82     client.close();
83   }
84 
85   @Test
createPhraseSetTest()86   public void createPhraseSetTest() throws Exception {
87     PhraseSet expectedResponse =
88         PhraseSet.newBuilder()
89             .setName(PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]").toString())
90             .addAllPhrases(new ArrayList<PhraseSet.Phrase>())
91             .setBoost(93922211)
92             .build();
93     mockAdaptation.addResponse(expectedResponse);
94 
95     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
96     PhraseSet phraseSet = PhraseSet.newBuilder().build();
97     String phraseSetId = "phraseSetId959902180";
98 
99     PhraseSet actualResponse = client.createPhraseSet(parent, phraseSet, phraseSetId);
100     Assert.assertEquals(expectedResponse, actualResponse);
101 
102     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
103     Assert.assertEquals(1, actualRequests.size());
104     CreatePhraseSetRequest actualRequest = ((CreatePhraseSetRequest) actualRequests.get(0));
105 
106     Assert.assertEquals(parent.toString(), actualRequest.getParent());
107     Assert.assertEquals(phraseSet, actualRequest.getPhraseSet());
108     Assert.assertEquals(phraseSetId, actualRequest.getPhraseSetId());
109     Assert.assertTrue(
110         channelProvider.isHeaderSent(
111             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
112             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
113   }
114 
115   @Test
createPhraseSetExceptionTest()116   public void createPhraseSetExceptionTest() throws Exception {
117     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
118     mockAdaptation.addException(exception);
119 
120     try {
121       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
122       PhraseSet phraseSet = PhraseSet.newBuilder().build();
123       String phraseSetId = "phraseSetId959902180";
124       client.createPhraseSet(parent, phraseSet, phraseSetId);
125       Assert.fail("No exception raised");
126     } catch (InvalidArgumentException e) {
127       // Expected exception.
128     }
129   }
130 
131   @Test
createPhraseSetTest2()132   public void createPhraseSetTest2() throws Exception {
133     PhraseSet expectedResponse =
134         PhraseSet.newBuilder()
135             .setName(PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]").toString())
136             .addAllPhrases(new ArrayList<PhraseSet.Phrase>())
137             .setBoost(93922211)
138             .build();
139     mockAdaptation.addResponse(expectedResponse);
140 
141     String parent = "parent-995424086";
142     PhraseSet phraseSet = PhraseSet.newBuilder().build();
143     String phraseSetId = "phraseSetId959902180";
144 
145     PhraseSet actualResponse = client.createPhraseSet(parent, phraseSet, phraseSetId);
146     Assert.assertEquals(expectedResponse, actualResponse);
147 
148     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
149     Assert.assertEquals(1, actualRequests.size());
150     CreatePhraseSetRequest actualRequest = ((CreatePhraseSetRequest) actualRequests.get(0));
151 
152     Assert.assertEquals(parent, actualRequest.getParent());
153     Assert.assertEquals(phraseSet, actualRequest.getPhraseSet());
154     Assert.assertEquals(phraseSetId, actualRequest.getPhraseSetId());
155     Assert.assertTrue(
156         channelProvider.isHeaderSent(
157             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
158             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
159   }
160 
161   @Test
createPhraseSetExceptionTest2()162   public void createPhraseSetExceptionTest2() throws Exception {
163     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
164     mockAdaptation.addException(exception);
165 
166     try {
167       String parent = "parent-995424086";
168       PhraseSet phraseSet = PhraseSet.newBuilder().build();
169       String phraseSetId = "phraseSetId959902180";
170       client.createPhraseSet(parent, phraseSet, phraseSetId);
171       Assert.fail("No exception raised");
172     } catch (InvalidArgumentException e) {
173       // Expected exception.
174     }
175   }
176 
177   @Test
getPhraseSetTest()178   public void getPhraseSetTest() throws Exception {
179     PhraseSet expectedResponse =
180         PhraseSet.newBuilder()
181             .setName(PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]").toString())
182             .addAllPhrases(new ArrayList<PhraseSet.Phrase>())
183             .setBoost(93922211)
184             .build();
185     mockAdaptation.addResponse(expectedResponse);
186 
187     PhraseSetName name = PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]");
188 
189     PhraseSet actualResponse = client.getPhraseSet(name);
190     Assert.assertEquals(expectedResponse, actualResponse);
191 
192     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
193     Assert.assertEquals(1, actualRequests.size());
194     GetPhraseSetRequest actualRequest = ((GetPhraseSetRequest) actualRequests.get(0));
195 
196     Assert.assertEquals(name.toString(), actualRequest.getName());
197     Assert.assertTrue(
198         channelProvider.isHeaderSent(
199             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
200             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
201   }
202 
203   @Test
getPhraseSetExceptionTest()204   public void getPhraseSetExceptionTest() throws Exception {
205     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
206     mockAdaptation.addException(exception);
207 
208     try {
209       PhraseSetName name = PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]");
210       client.getPhraseSet(name);
211       Assert.fail("No exception raised");
212     } catch (InvalidArgumentException e) {
213       // Expected exception.
214     }
215   }
216 
217   @Test
getPhraseSetTest2()218   public void getPhraseSetTest2() throws Exception {
219     PhraseSet expectedResponse =
220         PhraseSet.newBuilder()
221             .setName(PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]").toString())
222             .addAllPhrases(new ArrayList<PhraseSet.Phrase>())
223             .setBoost(93922211)
224             .build();
225     mockAdaptation.addResponse(expectedResponse);
226 
227     String name = "name3373707";
228 
229     PhraseSet actualResponse = client.getPhraseSet(name);
230     Assert.assertEquals(expectedResponse, actualResponse);
231 
232     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
233     Assert.assertEquals(1, actualRequests.size());
234     GetPhraseSetRequest actualRequest = ((GetPhraseSetRequest) actualRequests.get(0));
235 
236     Assert.assertEquals(name, actualRequest.getName());
237     Assert.assertTrue(
238         channelProvider.isHeaderSent(
239             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
240             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
241   }
242 
243   @Test
getPhraseSetExceptionTest2()244   public void getPhraseSetExceptionTest2() throws Exception {
245     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
246     mockAdaptation.addException(exception);
247 
248     try {
249       String name = "name3373707";
250       client.getPhraseSet(name);
251       Assert.fail("No exception raised");
252     } catch (InvalidArgumentException e) {
253       // Expected exception.
254     }
255   }
256 
257   @Test
listPhraseSetTest()258   public void listPhraseSetTest() throws Exception {
259     PhraseSet responsesElement = PhraseSet.newBuilder().build();
260     ListPhraseSetResponse expectedResponse =
261         ListPhraseSetResponse.newBuilder()
262             .setNextPageToken("")
263             .addAllPhraseSets(Arrays.asList(responsesElement))
264             .build();
265     mockAdaptation.addResponse(expectedResponse);
266 
267     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
268 
269     ListPhraseSetPagedResponse pagedListResponse = client.listPhraseSet(parent);
270 
271     List<PhraseSet> resources = Lists.newArrayList(pagedListResponse.iterateAll());
272 
273     Assert.assertEquals(1, resources.size());
274     Assert.assertEquals(expectedResponse.getPhraseSetsList().get(0), resources.get(0));
275 
276     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
277     Assert.assertEquals(1, actualRequests.size());
278     ListPhraseSetRequest actualRequest = ((ListPhraseSetRequest) actualRequests.get(0));
279 
280     Assert.assertEquals(parent.toString(), actualRequest.getParent());
281     Assert.assertTrue(
282         channelProvider.isHeaderSent(
283             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
284             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
285   }
286 
287   @Test
listPhraseSetExceptionTest()288   public void listPhraseSetExceptionTest() throws Exception {
289     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
290     mockAdaptation.addException(exception);
291 
292     try {
293       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
294       client.listPhraseSet(parent);
295       Assert.fail("No exception raised");
296     } catch (InvalidArgumentException e) {
297       // Expected exception.
298     }
299   }
300 
301   @Test
listPhraseSetTest2()302   public void listPhraseSetTest2() throws Exception {
303     PhraseSet responsesElement = PhraseSet.newBuilder().build();
304     ListPhraseSetResponse expectedResponse =
305         ListPhraseSetResponse.newBuilder()
306             .setNextPageToken("")
307             .addAllPhraseSets(Arrays.asList(responsesElement))
308             .build();
309     mockAdaptation.addResponse(expectedResponse);
310 
311     String parent = "parent-995424086";
312 
313     ListPhraseSetPagedResponse pagedListResponse = client.listPhraseSet(parent);
314 
315     List<PhraseSet> resources = Lists.newArrayList(pagedListResponse.iterateAll());
316 
317     Assert.assertEquals(1, resources.size());
318     Assert.assertEquals(expectedResponse.getPhraseSetsList().get(0), resources.get(0));
319 
320     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
321     Assert.assertEquals(1, actualRequests.size());
322     ListPhraseSetRequest actualRequest = ((ListPhraseSetRequest) actualRequests.get(0));
323 
324     Assert.assertEquals(parent, actualRequest.getParent());
325     Assert.assertTrue(
326         channelProvider.isHeaderSent(
327             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
328             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
329   }
330 
331   @Test
listPhraseSetExceptionTest2()332   public void listPhraseSetExceptionTest2() throws Exception {
333     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
334     mockAdaptation.addException(exception);
335 
336     try {
337       String parent = "parent-995424086";
338       client.listPhraseSet(parent);
339       Assert.fail("No exception raised");
340     } catch (InvalidArgumentException e) {
341       // Expected exception.
342     }
343   }
344 
345   @Test
updatePhraseSetTest()346   public void updatePhraseSetTest() throws Exception {
347     PhraseSet expectedResponse =
348         PhraseSet.newBuilder()
349             .setName(PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]").toString())
350             .addAllPhrases(new ArrayList<PhraseSet.Phrase>())
351             .setBoost(93922211)
352             .build();
353     mockAdaptation.addResponse(expectedResponse);
354 
355     PhraseSet phraseSet = PhraseSet.newBuilder().build();
356     FieldMask updateMask = FieldMask.newBuilder().build();
357 
358     PhraseSet actualResponse = client.updatePhraseSet(phraseSet, updateMask);
359     Assert.assertEquals(expectedResponse, actualResponse);
360 
361     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
362     Assert.assertEquals(1, actualRequests.size());
363     UpdatePhraseSetRequest actualRequest = ((UpdatePhraseSetRequest) actualRequests.get(0));
364 
365     Assert.assertEquals(phraseSet, actualRequest.getPhraseSet());
366     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
367     Assert.assertTrue(
368         channelProvider.isHeaderSent(
369             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
370             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
371   }
372 
373   @Test
updatePhraseSetExceptionTest()374   public void updatePhraseSetExceptionTest() throws Exception {
375     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
376     mockAdaptation.addException(exception);
377 
378     try {
379       PhraseSet phraseSet = PhraseSet.newBuilder().build();
380       FieldMask updateMask = FieldMask.newBuilder().build();
381       client.updatePhraseSet(phraseSet, updateMask);
382       Assert.fail("No exception raised");
383     } catch (InvalidArgumentException e) {
384       // Expected exception.
385     }
386   }
387 
388   @Test
deletePhraseSetTest()389   public void deletePhraseSetTest() throws Exception {
390     Empty expectedResponse = Empty.newBuilder().build();
391     mockAdaptation.addResponse(expectedResponse);
392 
393     PhraseSetName name = PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]");
394 
395     client.deletePhraseSet(name);
396 
397     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
398     Assert.assertEquals(1, actualRequests.size());
399     DeletePhraseSetRequest actualRequest = ((DeletePhraseSetRequest) actualRequests.get(0));
400 
401     Assert.assertEquals(name.toString(), actualRequest.getName());
402     Assert.assertTrue(
403         channelProvider.isHeaderSent(
404             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
405             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
406   }
407 
408   @Test
deletePhraseSetExceptionTest()409   public void deletePhraseSetExceptionTest() throws Exception {
410     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
411     mockAdaptation.addException(exception);
412 
413     try {
414       PhraseSetName name = PhraseSetName.of("[PROJECT]", "[LOCATION]", "[PHRASE_SET]");
415       client.deletePhraseSet(name);
416       Assert.fail("No exception raised");
417     } catch (InvalidArgumentException e) {
418       // Expected exception.
419     }
420   }
421 
422   @Test
deletePhraseSetTest2()423   public void deletePhraseSetTest2() throws Exception {
424     Empty expectedResponse = Empty.newBuilder().build();
425     mockAdaptation.addResponse(expectedResponse);
426 
427     String name = "name3373707";
428 
429     client.deletePhraseSet(name);
430 
431     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
432     Assert.assertEquals(1, actualRequests.size());
433     DeletePhraseSetRequest actualRequest = ((DeletePhraseSetRequest) actualRequests.get(0));
434 
435     Assert.assertEquals(name, actualRequest.getName());
436     Assert.assertTrue(
437         channelProvider.isHeaderSent(
438             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
439             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
440   }
441 
442   @Test
deletePhraseSetExceptionTest2()443   public void deletePhraseSetExceptionTest2() throws Exception {
444     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
445     mockAdaptation.addException(exception);
446 
447     try {
448       String name = "name3373707";
449       client.deletePhraseSet(name);
450       Assert.fail("No exception raised");
451     } catch (InvalidArgumentException e) {
452       // Expected exception.
453     }
454   }
455 
456   @Test
createCustomClassTest()457   public void createCustomClassTest() throws Exception {
458     CustomClass expectedResponse =
459         CustomClass.newBuilder()
460             .setName(CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]").toString())
461             .setCustomClassId("customClassId1871032322")
462             .addAllItems(new ArrayList<CustomClass.ClassItem>())
463             .build();
464     mockAdaptation.addResponse(expectedResponse);
465 
466     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
467     CustomClass customClass = CustomClass.newBuilder().build();
468     String customClassId = "customClassId1871032322";
469 
470     CustomClass actualResponse = client.createCustomClass(parent, customClass, customClassId);
471     Assert.assertEquals(expectedResponse, actualResponse);
472 
473     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
474     Assert.assertEquals(1, actualRequests.size());
475     CreateCustomClassRequest actualRequest = ((CreateCustomClassRequest) actualRequests.get(0));
476 
477     Assert.assertEquals(parent.toString(), actualRequest.getParent());
478     Assert.assertEquals(customClass, actualRequest.getCustomClass());
479     Assert.assertEquals(customClassId, actualRequest.getCustomClassId());
480     Assert.assertTrue(
481         channelProvider.isHeaderSent(
482             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
483             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
484   }
485 
486   @Test
createCustomClassExceptionTest()487   public void createCustomClassExceptionTest() throws Exception {
488     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
489     mockAdaptation.addException(exception);
490 
491     try {
492       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
493       CustomClass customClass = CustomClass.newBuilder().build();
494       String customClassId = "customClassId1871032322";
495       client.createCustomClass(parent, customClass, customClassId);
496       Assert.fail("No exception raised");
497     } catch (InvalidArgumentException e) {
498       // Expected exception.
499     }
500   }
501 
502   @Test
createCustomClassTest2()503   public void createCustomClassTest2() throws Exception {
504     CustomClass expectedResponse =
505         CustomClass.newBuilder()
506             .setName(CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]").toString())
507             .setCustomClassId("customClassId1871032322")
508             .addAllItems(new ArrayList<CustomClass.ClassItem>())
509             .build();
510     mockAdaptation.addResponse(expectedResponse);
511 
512     String parent = "parent-995424086";
513     CustomClass customClass = CustomClass.newBuilder().build();
514     String customClassId = "customClassId1871032322";
515 
516     CustomClass actualResponse = client.createCustomClass(parent, customClass, customClassId);
517     Assert.assertEquals(expectedResponse, actualResponse);
518 
519     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
520     Assert.assertEquals(1, actualRequests.size());
521     CreateCustomClassRequest actualRequest = ((CreateCustomClassRequest) actualRequests.get(0));
522 
523     Assert.assertEquals(parent, actualRequest.getParent());
524     Assert.assertEquals(customClass, actualRequest.getCustomClass());
525     Assert.assertEquals(customClassId, actualRequest.getCustomClassId());
526     Assert.assertTrue(
527         channelProvider.isHeaderSent(
528             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
529             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
530   }
531 
532   @Test
createCustomClassExceptionTest2()533   public void createCustomClassExceptionTest2() throws Exception {
534     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
535     mockAdaptation.addException(exception);
536 
537     try {
538       String parent = "parent-995424086";
539       CustomClass customClass = CustomClass.newBuilder().build();
540       String customClassId = "customClassId1871032322";
541       client.createCustomClass(parent, customClass, customClassId);
542       Assert.fail("No exception raised");
543     } catch (InvalidArgumentException e) {
544       // Expected exception.
545     }
546   }
547 
548   @Test
getCustomClassTest()549   public void getCustomClassTest() throws Exception {
550     CustomClass expectedResponse =
551         CustomClass.newBuilder()
552             .setName(CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]").toString())
553             .setCustomClassId("customClassId1871032322")
554             .addAllItems(new ArrayList<CustomClass.ClassItem>())
555             .build();
556     mockAdaptation.addResponse(expectedResponse);
557 
558     CustomClassName name = CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]");
559 
560     CustomClass actualResponse = client.getCustomClass(name);
561     Assert.assertEquals(expectedResponse, actualResponse);
562 
563     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
564     Assert.assertEquals(1, actualRequests.size());
565     GetCustomClassRequest actualRequest = ((GetCustomClassRequest) actualRequests.get(0));
566 
567     Assert.assertEquals(name.toString(), actualRequest.getName());
568     Assert.assertTrue(
569         channelProvider.isHeaderSent(
570             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
571             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
572   }
573 
574   @Test
getCustomClassExceptionTest()575   public void getCustomClassExceptionTest() throws Exception {
576     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
577     mockAdaptation.addException(exception);
578 
579     try {
580       CustomClassName name = CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]");
581       client.getCustomClass(name);
582       Assert.fail("No exception raised");
583     } catch (InvalidArgumentException e) {
584       // Expected exception.
585     }
586   }
587 
588   @Test
getCustomClassTest2()589   public void getCustomClassTest2() throws Exception {
590     CustomClass expectedResponse =
591         CustomClass.newBuilder()
592             .setName(CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]").toString())
593             .setCustomClassId("customClassId1871032322")
594             .addAllItems(new ArrayList<CustomClass.ClassItem>())
595             .build();
596     mockAdaptation.addResponse(expectedResponse);
597 
598     String name = "name3373707";
599 
600     CustomClass actualResponse = client.getCustomClass(name);
601     Assert.assertEquals(expectedResponse, actualResponse);
602 
603     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
604     Assert.assertEquals(1, actualRequests.size());
605     GetCustomClassRequest actualRequest = ((GetCustomClassRequest) actualRequests.get(0));
606 
607     Assert.assertEquals(name, actualRequest.getName());
608     Assert.assertTrue(
609         channelProvider.isHeaderSent(
610             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
611             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
612   }
613 
614   @Test
getCustomClassExceptionTest2()615   public void getCustomClassExceptionTest2() throws Exception {
616     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
617     mockAdaptation.addException(exception);
618 
619     try {
620       String name = "name3373707";
621       client.getCustomClass(name);
622       Assert.fail("No exception raised");
623     } catch (InvalidArgumentException e) {
624       // Expected exception.
625     }
626   }
627 
628   @Test
listCustomClassesTest()629   public void listCustomClassesTest() throws Exception {
630     CustomClass responsesElement = CustomClass.newBuilder().build();
631     ListCustomClassesResponse expectedResponse =
632         ListCustomClassesResponse.newBuilder()
633             .setNextPageToken("")
634             .addAllCustomClasses(Arrays.asList(responsesElement))
635             .build();
636     mockAdaptation.addResponse(expectedResponse);
637 
638     LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
639 
640     ListCustomClassesPagedResponse pagedListResponse = client.listCustomClasses(parent);
641 
642     List<CustomClass> resources = Lists.newArrayList(pagedListResponse.iterateAll());
643 
644     Assert.assertEquals(1, resources.size());
645     Assert.assertEquals(expectedResponse.getCustomClassesList().get(0), resources.get(0));
646 
647     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
648     Assert.assertEquals(1, actualRequests.size());
649     ListCustomClassesRequest actualRequest = ((ListCustomClassesRequest) actualRequests.get(0));
650 
651     Assert.assertEquals(parent.toString(), actualRequest.getParent());
652     Assert.assertTrue(
653         channelProvider.isHeaderSent(
654             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
655             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
656   }
657 
658   @Test
listCustomClassesExceptionTest()659   public void listCustomClassesExceptionTest() throws Exception {
660     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
661     mockAdaptation.addException(exception);
662 
663     try {
664       LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
665       client.listCustomClasses(parent);
666       Assert.fail("No exception raised");
667     } catch (InvalidArgumentException e) {
668       // Expected exception.
669     }
670   }
671 
672   @Test
listCustomClassesTest2()673   public void listCustomClassesTest2() throws Exception {
674     CustomClass responsesElement = CustomClass.newBuilder().build();
675     ListCustomClassesResponse expectedResponse =
676         ListCustomClassesResponse.newBuilder()
677             .setNextPageToken("")
678             .addAllCustomClasses(Arrays.asList(responsesElement))
679             .build();
680     mockAdaptation.addResponse(expectedResponse);
681 
682     String parent = "parent-995424086";
683 
684     ListCustomClassesPagedResponse pagedListResponse = client.listCustomClasses(parent);
685 
686     List<CustomClass> resources = Lists.newArrayList(pagedListResponse.iterateAll());
687 
688     Assert.assertEquals(1, resources.size());
689     Assert.assertEquals(expectedResponse.getCustomClassesList().get(0), resources.get(0));
690 
691     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
692     Assert.assertEquals(1, actualRequests.size());
693     ListCustomClassesRequest actualRequest = ((ListCustomClassesRequest) actualRequests.get(0));
694 
695     Assert.assertEquals(parent, actualRequest.getParent());
696     Assert.assertTrue(
697         channelProvider.isHeaderSent(
698             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
699             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
700   }
701 
702   @Test
listCustomClassesExceptionTest2()703   public void listCustomClassesExceptionTest2() throws Exception {
704     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
705     mockAdaptation.addException(exception);
706 
707     try {
708       String parent = "parent-995424086";
709       client.listCustomClasses(parent);
710       Assert.fail("No exception raised");
711     } catch (InvalidArgumentException e) {
712       // Expected exception.
713     }
714   }
715 
716   @Test
updateCustomClassTest()717   public void updateCustomClassTest() throws Exception {
718     CustomClass expectedResponse =
719         CustomClass.newBuilder()
720             .setName(CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]").toString())
721             .setCustomClassId("customClassId1871032322")
722             .addAllItems(new ArrayList<CustomClass.ClassItem>())
723             .build();
724     mockAdaptation.addResponse(expectedResponse);
725 
726     CustomClass customClass = CustomClass.newBuilder().build();
727     FieldMask updateMask = FieldMask.newBuilder().build();
728 
729     CustomClass actualResponse = client.updateCustomClass(customClass, updateMask);
730     Assert.assertEquals(expectedResponse, actualResponse);
731 
732     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
733     Assert.assertEquals(1, actualRequests.size());
734     UpdateCustomClassRequest actualRequest = ((UpdateCustomClassRequest) actualRequests.get(0));
735 
736     Assert.assertEquals(customClass, actualRequest.getCustomClass());
737     Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
738     Assert.assertTrue(
739         channelProvider.isHeaderSent(
740             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
741             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
742   }
743 
744   @Test
updateCustomClassExceptionTest()745   public void updateCustomClassExceptionTest() throws Exception {
746     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
747     mockAdaptation.addException(exception);
748 
749     try {
750       CustomClass customClass = CustomClass.newBuilder().build();
751       FieldMask updateMask = FieldMask.newBuilder().build();
752       client.updateCustomClass(customClass, updateMask);
753       Assert.fail("No exception raised");
754     } catch (InvalidArgumentException e) {
755       // Expected exception.
756     }
757   }
758 
759   @Test
deleteCustomClassTest()760   public void deleteCustomClassTest() throws Exception {
761     Empty expectedResponse = Empty.newBuilder().build();
762     mockAdaptation.addResponse(expectedResponse);
763 
764     CustomClassName name = CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]");
765 
766     client.deleteCustomClass(name);
767 
768     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
769     Assert.assertEquals(1, actualRequests.size());
770     DeleteCustomClassRequest actualRequest = ((DeleteCustomClassRequest) actualRequests.get(0));
771 
772     Assert.assertEquals(name.toString(), actualRequest.getName());
773     Assert.assertTrue(
774         channelProvider.isHeaderSent(
775             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
776             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
777   }
778 
779   @Test
deleteCustomClassExceptionTest()780   public void deleteCustomClassExceptionTest() throws Exception {
781     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
782     mockAdaptation.addException(exception);
783 
784     try {
785       CustomClassName name = CustomClassName.of("[PROJECT]", "[LOCATION]", "[CUSTOM_CLASS]");
786       client.deleteCustomClass(name);
787       Assert.fail("No exception raised");
788     } catch (InvalidArgumentException e) {
789       // Expected exception.
790     }
791   }
792 
793   @Test
deleteCustomClassTest2()794   public void deleteCustomClassTest2() throws Exception {
795     Empty expectedResponse = Empty.newBuilder().build();
796     mockAdaptation.addResponse(expectedResponse);
797 
798     String name = "name3373707";
799 
800     client.deleteCustomClass(name);
801 
802     List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
803     Assert.assertEquals(1, actualRequests.size());
804     DeleteCustomClassRequest actualRequest = ((DeleteCustomClassRequest) actualRequests.get(0));
805 
806     Assert.assertEquals(name, actualRequest.getName());
807     Assert.assertTrue(
808         channelProvider.isHeaderSent(
809             ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
810             GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
811   }
812 
813   @Test
deleteCustomClassExceptionTest2()814   public void deleteCustomClassExceptionTest2() throws Exception {
815     StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
816     mockAdaptation.addException(exception);
817 
818     try {
819       String name = "name3373707";
820       client.deleteCustomClass(name);
821       Assert.fail("No exception raised");
822     } catch (InvalidArgumentException e) {
823       // Expected exception.
824     }
825   }
826 }
827