Skip to content
This repository was archived by the owner on Oct 14, 2023. It is now read-only.

Commit c1fcecb

Browse files
yoshi-automationyoshi-automation
authored
docs(regen): update sample code to set total timeout, add API client header test
1 parent bf84975 commit c1fcecb

File tree

7 files changed

+252
-15
lines changed

7 files changed

+252
-15
lines changed

‎google-cloud-translate/src/main/java/com/google/cloud/translate/v3/TranslationServiceSettings.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,12 @@
5656
* <code>
5757
* TranslationServiceSettings.Builder translationServiceSettingsBuilder =
5858
* TranslationServiceSettings.newBuilder();
59-
* translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
60-
* .setTotalTimeout(Duration.ofSeconds(30));
59+
* translationServiceSettingsBuilder
60+
* .translateTextSettings()
61+
* .setRetrySettings(
62+
* translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
63+
* .setTotalTimeout(Duration.ofSeconds(30))
64+
* .build());
6165
* TranslationServiceSettings translationServiceSettings = translationServiceSettingsBuilder.build();
6266
* </code>
6367
* </pre>

‎google-cloud-translate/src/main/java/com/google/cloud/translate/v3/stub/TranslationServiceStubSettings.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,12 @@
9292
* <code>
9393
* TranslationServiceStubSettings.Builder translationServiceSettingsBuilder =
9494
* TranslationServiceStubSettings.newBuilder();
95-
* translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
96-
* .setTotalTimeout(Duration.ofSeconds(30));
95+
* translationServiceSettingsBuilder
96+
* .translateTextSettings()
97+
* .setRetrySettings(
98+
* translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
99+
* .setTotalTimeout(Duration.ofSeconds(30))
100+
* .build());
97101
* TranslationServiceStubSettings translationServiceSettings = translationServiceSettingsBuilder.build();
98102
* </code>
99103
* </pre>

‎google-cloud-translate/src/main/java/com/google/cloud/translate/v3beta1/TranslationServiceSettings.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,12 @@
5656
* <code>
5757
* TranslationServiceSettings.Builder translationServiceSettingsBuilder =
5858
* TranslationServiceSettings.newBuilder();
59-
* translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
60-
* .setTotalTimeout(Duration.ofSeconds(30));
59+
* translationServiceSettingsBuilder
60+
* .detectLanguageSettings()
61+
* .setRetrySettings(
62+
* translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
63+
* .setTotalTimeout(Duration.ofSeconds(30))
64+
* .build());
6165
* TranslationServiceSettings translationServiceSettings = translationServiceSettingsBuilder.build();
6266
* </code>
6367
* </pre>

‎google-cloud-translate/src/main/java/com/google/cloud/translate/v3beta1/stub/TranslationServiceStubSettings.java

+6-2
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,12 @@
9292
* <code>
9393
* TranslationServiceStubSettings.Builder translationServiceSettingsBuilder =
9494
* TranslationServiceStubSettings.newBuilder();
95-
* translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
96-
* .setTotalTimeout(Duration.ofSeconds(30));
95+
* translationServiceSettingsBuilder
96+
* .detectLanguageSettings()
97+
* .setRetrySettings(
98+
* translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
99+
* .setTotalTimeout(Duration.ofSeconds(30))
100+
* .build());
97101
* TranslationServiceStubSettings translationServiceSettings = translationServiceSettingsBuilder.build();
98102
* </code>
99103
* </pre>

‎google-cloud-translate/src/test/java/com/google/cloud/translate/v3/TranslationServiceClientTest.java

+82
Original file line numberDiff line numberDiff line change
@@ -276,6 +276,88 @@ public void getSupportedLanguagesExceptionTest() throws Exception {
276276
}
277277
}
278278

279+
@Test
280+
@SuppressWarnings("all")
281+
public void batchTranslateTextTest() throws Exception {
282+
long totalCharacters = 1368640955L;
283+
long translatedCharacters = 1337326221L;
284+
long failedCharacters = 1723028396L;
285+
BatchTranslateResponse expectedResponse =
286+
BatchTranslateResponse.newBuilder()
287+
.setTotalCharacters(totalCharacters)
288+
.setTranslatedCharacters(translatedCharacters)
289+
.setFailedCharacters(failedCharacters)
290+
.build();
291+
Operation resultOperation =
292+
Operation.newBuilder()
293+
.setName("batchTranslateTextTest")
294+
.setDone(true)
295+
.setResponse(Any.pack(expectedResponse))
296+
.build();
297+
mockTranslationService.addResponse(resultOperation);
298+
299+
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
300+
String sourceLanguageCode = "sourceLanguageCode1687263568";
301+
List<String> targetLanguageCodes = new ArrayList<>();
302+
List<InputConfig> inputConfigs = new ArrayList<>();
303+
OutputConfig outputConfig = OutputConfig.newBuilder().build();
304+
BatchTranslateTextRequest request =
305+
BatchTranslateTextRequest.newBuilder()
306+
.setParent(parent.toString())
307+
.setSourceLanguageCode(sourceLanguageCode)
308+
.addAllTargetLanguageCodes(targetLanguageCodes)
309+
.addAllInputConfigs(inputConfigs)
310+
.setOutputConfig(outputConfig)
311+
.build();
312+
313+
BatchTranslateResponse actualResponse = client.batchTranslateTextAsync(request).get();
314+
Assert.assertEquals(expectedResponse, actualResponse);
315+
316+
List<AbstractMessage> actualRequests = mockTranslationService.getRequests();
317+
Assert.assertEquals(1, actualRequests.size());
318+
BatchTranslateTextRequest actualRequest = (BatchTranslateTextRequest) actualRequests.get(0);
319+
320+
Assert.assertEquals(parent, LocationName.parse(actualRequest.getParent()));
321+
Assert.assertEquals(sourceLanguageCode, actualRequest.getSourceLanguageCode());
322+
Assert.assertEquals(targetLanguageCodes, actualRequest.getTargetLanguageCodesList());
323+
Assert.assertEquals(inputConfigs, actualRequest.getInputConfigsList());
324+
Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
325+
Assert.assertTrue(
326+
channelProvider.isHeaderSent(
327+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
328+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
329+
}
330+
331+
@Test
332+
@SuppressWarnings("all")
333+
public void batchTranslateTextExceptionTest() throws Exception {
334+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
335+
mockTranslationService.addException(exception);
336+
337+
try {
338+
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
339+
String sourceLanguageCode = "sourceLanguageCode1687263568";
340+
List<String> targetLanguageCodes = new ArrayList<>();
341+
List<InputConfig> inputConfigs = new ArrayList<>();
342+
OutputConfig outputConfig = OutputConfig.newBuilder().build();
343+
BatchTranslateTextRequest request =
344+
BatchTranslateTextRequest.newBuilder()
345+
.setParent(parent.toString())
346+
.setSourceLanguageCode(sourceLanguageCode)
347+
.addAllTargetLanguageCodes(targetLanguageCodes)
348+
.addAllInputConfigs(inputConfigs)
349+
.setOutputConfig(outputConfig)
350+
.build();
351+
352+
client.batchTranslateTextAsync(request).get();
353+
Assert.fail("No exception raised");
354+
} catch (ExecutionException e) {
355+
Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
356+
InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
357+
Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
358+
}
359+
}
360+
279361
@Test
280362
@SuppressWarnings("all")
281363
public void createGlossaryTest() throws Exception {

‎google-cloud-translate/src/test/java/com/google/cloud/translate/v3beta1/TranslationServiceClientTest.java

+139
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
import io.grpc.Status;
3333
import io.grpc.StatusRuntimeException;
3434
import java.io.IOException;
35+
import java.util.ArrayList;
3536
import java.util.Arrays;
3637
import java.util.List;
3738
import java.util.UUID;
@@ -81,6 +82,62 @@ public void tearDown() throws Exception {
8182
client.close();
8283
}
8384

85+
@Test
86+
@SuppressWarnings("all")
87+
public void translateTextTest() {
88+
TranslateTextResponse expectedResponse = TranslateTextResponse.newBuilder().build();
89+
mockTranslationService.addResponse(expectedResponse);
90+
91+
List<String> contents = new ArrayList<>();
92+
String targetLanguageCode = "targetLanguageCode1323228230";
93+
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
94+
TranslateTextRequest request =
95+
TranslateTextRequest.newBuilder()
96+
.addAllContents(contents)
97+
.setTargetLanguageCode(targetLanguageCode)
98+
.setParent(parent.toString())
99+
.build();
100+
101+
TranslateTextResponse actualResponse = client.translateText(request);
102+
Assert.assertEquals(expectedResponse, actualResponse);
103+
104+
List<AbstractMessage> actualRequests = mockTranslationService.getRequests();
105+
Assert.assertEquals(1, actualRequests.size());
106+
TranslateTextRequest actualRequest = (TranslateTextRequest) actualRequests.get(0);
107+
108+
Assert.assertEquals(contents, actualRequest.getContentsList());
109+
Assert.assertEquals(targetLanguageCode, actualRequest.getTargetLanguageCode());
110+
Assert.assertEquals(parent, LocationName.parse(actualRequest.getParent()));
111+
Assert.assertTrue(
112+
channelProvider.isHeaderSent(
113+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
114+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
115+
}
116+
117+
@Test
118+
@SuppressWarnings("all")
119+
public void translateTextExceptionTest() throws Exception {
120+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
121+
mockTranslationService.addException(exception);
122+
123+
try {
124+
List<String> contents = new ArrayList<>();
125+
String targetLanguageCode = "targetLanguageCode1323228230";
126+
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
127+
TranslateTextRequest request =
128+
TranslateTextRequest.newBuilder()
129+
.addAllContents(contents)
130+
.setTargetLanguageCode(targetLanguageCode)
131+
.setParent(parent.toString())
132+
.build();
133+
134+
client.translateText(request);
135+
Assert.fail("No exception raised");
136+
} catch (InvalidArgumentException e) {
137+
// Expected exception
138+
}
139+
}
140+
84141
@Test
85142
@SuppressWarnings("all")
86143
public void detectLanguageTest() {
@@ -171,6 +228,88 @@ public void getSupportedLanguagesExceptionTest() throws Exception {
171228
}
172229
}
173230

231+
@Test
232+
@SuppressWarnings("all")
233+
public void batchTranslateTextTest() throws Exception {
234+
long totalCharacters = 1368640955L;
235+
long translatedCharacters = 1337326221L;
236+
long failedCharacters = 1723028396L;
237+
BatchTranslateResponse expectedResponse =
238+
BatchTranslateResponse.newBuilder()
239+
.setTotalCharacters(totalCharacters)
240+
.setTranslatedCharacters(translatedCharacters)
241+
.setFailedCharacters(failedCharacters)
242+
.build();
243+
Operation resultOperation =
244+
Operation.newBuilder()
245+
.setName("batchTranslateTextTest")
246+
.setDone(true)
247+
.setResponse(Any.pack(expectedResponse))
248+
.build();
249+
mockTranslationService.addResponse(resultOperation);
250+
251+
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
252+
String sourceLanguageCode = "sourceLanguageCode1687263568";
253+
List<String> targetLanguageCodes = new ArrayList<>();
254+
List<InputConfig> inputConfigs = new ArrayList<>();
255+
OutputConfig outputConfig = OutputConfig.newBuilder().build();
256+
BatchTranslateTextRequest request =
257+
BatchTranslateTextRequest.newBuilder()
258+
.setParent(parent.toString())
259+
.setSourceLanguageCode(sourceLanguageCode)
260+
.addAllTargetLanguageCodes(targetLanguageCodes)
261+
.addAllInputConfigs(inputConfigs)
262+
.setOutputConfig(outputConfig)
263+
.build();
264+
265+
BatchTranslateResponse actualResponse = client.batchTranslateTextAsync(request).get();
266+
Assert.assertEquals(expectedResponse, actualResponse);
267+
268+
List<AbstractMessage> actualRequests = mockTranslationService.getRequests();
269+
Assert.assertEquals(1, actualRequests.size());
270+
BatchTranslateTextRequest actualRequest = (BatchTranslateTextRequest) actualRequests.get(0);
271+
272+
Assert.assertEquals(parent, LocationName.parse(actualRequest.getParent()));
273+
Assert.assertEquals(sourceLanguageCode, actualRequest.getSourceLanguageCode());
274+
Assert.assertEquals(targetLanguageCodes, actualRequest.getTargetLanguageCodesList());
275+
Assert.assertEquals(inputConfigs, actualRequest.getInputConfigsList());
276+
Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
277+
Assert.assertTrue(
278+
channelProvider.isHeaderSent(
279+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
280+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
281+
}
282+
283+
@Test
284+
@SuppressWarnings("all")
285+
public void batchTranslateTextExceptionTest() throws Exception {
286+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
287+
mockTranslationService.addException(exception);
288+
289+
try {
290+
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
291+
String sourceLanguageCode = "sourceLanguageCode1687263568";
292+
List<String> targetLanguageCodes = new ArrayList<>();
293+
List<InputConfig> inputConfigs = new ArrayList<>();
294+
OutputConfig outputConfig = OutputConfig.newBuilder().build();
295+
BatchTranslateTextRequest request =
296+
BatchTranslateTextRequest.newBuilder()
297+
.setParent(parent.toString())
298+
.setSourceLanguageCode(sourceLanguageCode)
299+
.addAllTargetLanguageCodes(targetLanguageCodes)
300+
.addAllInputConfigs(inputConfigs)
301+
.setOutputConfig(outputConfig)
302+
.build();
303+
304+
client.batchTranslateTextAsync(request).get();
305+
Assert.fail("No exception raised");
306+
} catch (ExecutionException e) {
307+
Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
308+
InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
309+
Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
310+
}
311+
}
312+
174313
@Test
175314
@SuppressWarnings("all")
176315
public void createGlossaryTest() throws Exception {

0 commit comments

Comments
 (0)