docs(regen): update sample code to set total timeout, add API client …

…header test
googleapis · Feb 10, 2020 · c1fcecb · c1fcecb
1 parent bf84975
commit c1fcecb
Show file tree

Hide file tree

Showing 7 changed files with 252 additions and 15 deletions.
diff --git a/...oud-translate/src/main/java/com/google/cloud/translate/v3/TranslationServiceSettings.java b/...oud-translate/src/main/java/com/google/cloud/translate/v3/TranslationServiceSettings.java
@@ -56,8 +56,12 @@
  * <code>
  * TranslationServiceSettings.Builder translationServiceSettingsBuilder =
  *     TranslationServiceSettings.newBuilder();
- * translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * translationServiceSettingsBuilder
+ *     .translateTextSettings()
+ *     .setRetrySettings(
+ *         translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * TranslationServiceSettings translationServiceSettings = translationServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...late/src/main/java/com/google/cloud/translate/v3/stub/TranslationServiceStubSettings.java b/...late/src/main/java/com/google/cloud/translate/v3/stub/TranslationServiceStubSettings.java
@@ -92,8 +92,12 @@
  * <code>
  * TranslationServiceStubSettings.Builder translationServiceSettingsBuilder =
  *     TranslationServiceStubSettings.newBuilder();
- * translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * translationServiceSettingsBuilder
+ *     .translateTextSettings()
+ *     .setRetrySettings(
+ *         translationServiceSettingsBuilder.translateTextSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * TranslationServiceStubSettings translationServiceSettings = translationServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...ranslate/src/main/java/com/google/cloud/translate/v3beta1/TranslationServiceSettings.java b/...ranslate/src/main/java/com/google/cloud/translate/v3beta1/TranslationServiceSettings.java
@@ -56,8 +56,12 @@
  * <code>
  * TranslationServiceSettings.Builder translationServiceSettingsBuilder =
  *     TranslationServiceSettings.newBuilder();
- * translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * translationServiceSettingsBuilder
+ *     .detectLanguageSettings()
+ *     .setRetrySettings(
+ *         translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * TranslationServiceSettings translationServiceSettings = translationServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...src/main/java/com/google/cloud/translate/v3beta1/stub/TranslationServiceStubSettings.java b/...src/main/java/com/google/cloud/translate/v3beta1/stub/TranslationServiceStubSettings.java
@@ -92,8 +92,12 @@
  * <code>
  * TranslationServiceStubSettings.Builder translationServiceSettingsBuilder =
  *     TranslationServiceStubSettings.newBuilder();
- * translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * translationServiceSettingsBuilder
+ *     .detectLanguageSettings()
+ *     .setRetrySettings(
+ *         translationServiceSettingsBuilder.detectLanguageSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * TranslationServiceStubSettings translationServiceSettings = translationServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...d-translate/src/test/java/com/google/cloud/translate/v3/TranslationServiceClientTest.java b/...d-translate/src/test/java/com/google/cloud/translate/v3/TranslationServiceClientTest.java
@@ -276,6 +276,88 @@ public void getSupportedLanguagesExceptionTest() throws Exception {
     }
   }
 
+  @Test
+  @SuppressWarnings("all")
+  public void batchTranslateTextTest() throws Exception {
+    long totalCharacters = 1368640955L;
+    long translatedCharacters = 1337326221L;
+    long failedCharacters = 1723028396L;
+    BatchTranslateResponse expectedResponse =
+        BatchTranslateResponse.newBuilder()
+            .setTotalCharacters(totalCharacters)
+            .setTranslatedCharacters(translatedCharacters)
+            .setFailedCharacters(failedCharacters)
+            .build();
+    Operation resultOperation =
+        Operation.newBuilder()
+            .setName("batchTranslateTextTest")
+            .setDone(true)
+            .setResponse(Any.pack(expectedResponse))
+            .build();
+    mockTranslationService.addResponse(resultOperation);
+
+    LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+    String sourceLanguageCode = "sourceLanguageCode1687263568";
+    List<String> targetLanguageCodes = new ArrayList<>();
+    List<InputConfig> inputConfigs = new ArrayList<>();
+    OutputConfig outputConfig = OutputConfig.newBuilder().build();
+    BatchTranslateTextRequest request =
+        BatchTranslateTextRequest.newBuilder()
+            .setParent(parent.toString())
+            .setSourceLanguageCode(sourceLanguageCode)
+            .addAllTargetLanguageCodes(targetLanguageCodes)
+            .addAllInputConfigs(inputConfigs)
+            .setOutputConfig(outputConfig)
+            .build();
+
+    BatchTranslateResponse actualResponse = client.batchTranslateTextAsync(request).get();
+    Assert.assertEquals(expectedResponse, actualResponse);
+
+    List<AbstractMessage> actualRequests = mockTranslationService.getRequests();
+    Assert.assertEquals(1, actualRequests.size());
+    BatchTranslateTextRequest actualRequest = (BatchTranslateTextRequest) actualRequests.get(0);
+
+    Assert.assertEquals(parent, LocationName.parse(actualRequest.getParent()));
+    Assert.assertEquals(sourceLanguageCode, actualRequest.getSourceLanguageCode());
+    Assert.assertEquals(targetLanguageCodes, actualRequest.getTargetLanguageCodesList());
+    Assert.assertEquals(inputConfigs, actualRequest.getInputConfigsList());
+    Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
+    Assert.assertTrue(
+        channelProvider.isHeaderSent(
+            ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
+            GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
+  }
+
+  @Test
+  @SuppressWarnings("all")
+  public void batchTranslateTextExceptionTest() throws Exception {
+    StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
+    mockTranslationService.addException(exception);
+
+    try {
+      LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+      String sourceLanguageCode = "sourceLanguageCode1687263568";
+      List<String> targetLanguageCodes = new ArrayList<>();
+      List<InputConfig> inputConfigs = new ArrayList<>();
+      OutputConfig outputConfig = OutputConfig.newBuilder().build();
+      BatchTranslateTextRequest request =
+          BatchTranslateTextRequest.newBuilder()
+              .setParent(parent.toString())
+              .setSourceLanguageCode(sourceLanguageCode)
+              .addAllTargetLanguageCodes(targetLanguageCodes)
+              .addAllInputConfigs(inputConfigs)
+              .setOutputConfig(outputConfig)
+              .build();
+
+      client.batchTranslateTextAsync(request).get();
+      Assert.fail("No exception raised");
+    } catch (ExecutionException e) {
+      Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
+      InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
+      Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
+    }
+  }
+
   @Test
   @SuppressWarnings("all")
   public void createGlossaryTest() throws Exception {

diff --git a/...nslate/src/test/java/com/google/cloud/translate/v3beta1/TranslationServiceClientTest.java b/...nslate/src/test/java/com/google/cloud/translate/v3beta1/TranslationServiceClientTest.java
@@ -32,6 +32,7 @@
 import io.grpc.Status;
 import io.grpc.StatusRuntimeException;
 import java.io.IOException;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.UUID;
@@ -81,6 +82,62 @@ public void tearDown() throws Exception {
     client.close();
   }
 
+  @Test
+  @SuppressWarnings("all")
+  public void translateTextTest() {
+    TranslateTextResponse expectedResponse = TranslateTextResponse.newBuilder().build();
+    mockTranslationService.addResponse(expectedResponse);
+
+    List<String> contents = new ArrayList<>();
+    String targetLanguageCode = "targetLanguageCode1323228230";
+    LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+    TranslateTextRequest request =
+        TranslateTextRequest.newBuilder()
+            .addAllContents(contents)
+            .setTargetLanguageCode(targetLanguageCode)
+            .setParent(parent.toString())
+            .build();
+
+    TranslateTextResponse actualResponse = client.translateText(request);
+    Assert.assertEquals(expectedResponse, actualResponse);
+
+    List<AbstractMessage> actualRequests = mockTranslationService.getRequests();
+    Assert.assertEquals(1, actualRequests.size());
+    TranslateTextRequest actualRequest = (TranslateTextRequest) actualRequests.get(0);
+
+    Assert.assertEquals(contents, actualRequest.getContentsList());
+    Assert.assertEquals(targetLanguageCode, actualRequest.getTargetLanguageCode());
+    Assert.assertEquals(parent, LocationName.parse(actualRequest.getParent()));
+    Assert.assertTrue(
+        channelProvider.isHeaderSent(
+            ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
+            GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
+  }
+
+  @Test
+  @SuppressWarnings("all")
+  public void translateTextExceptionTest() throws Exception {
+    StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
+    mockTranslationService.addException(exception);
+
+    try {
+      List<String> contents = new ArrayList<>();
+      String targetLanguageCode = "targetLanguageCode1323228230";
+      LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+      TranslateTextRequest request =
+          TranslateTextRequest.newBuilder()
+              .addAllContents(contents)
+              .setTargetLanguageCode(targetLanguageCode)
+              .setParent(parent.toString())
+              .build();
+
+      client.translateText(request);
+      Assert.fail("No exception raised");
+    } catch (InvalidArgumentException e) {
+      // Expected exception
+    }
+  }
+
   @Test
   @SuppressWarnings("all")
   public void detectLanguageTest() {
@@ -171,6 +228,88 @@ public void getSupportedLanguagesExceptionTest() throws Exception {
     }
   }
 
+  @Test
+  @SuppressWarnings("all")
+  public void batchTranslateTextTest() throws Exception {
+    long totalCharacters = 1368640955L;
+    long translatedCharacters = 1337326221L;
+    long failedCharacters = 1723028396L;
+    BatchTranslateResponse expectedResponse =
+        BatchTranslateResponse.newBuilder()
+            .setTotalCharacters(totalCharacters)
+            .setTranslatedCharacters(translatedCharacters)
+            .setFailedCharacters(failedCharacters)
+            .build();
+    Operation resultOperation =
+        Operation.newBuilder()
+            .setName("batchTranslateTextTest")
+            .setDone(true)
+            .setResponse(Any.pack(expectedResponse))
+            .build();
+    mockTranslationService.addResponse(resultOperation);
+
+    LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+    String sourceLanguageCode = "sourceLanguageCode1687263568";
+    List<String> targetLanguageCodes = new ArrayList<>();
+    List<InputConfig> inputConfigs = new ArrayList<>();
+    OutputConfig outputConfig = OutputConfig.newBuilder().build();
+    BatchTranslateTextRequest request =
+        BatchTranslateTextRequest.newBuilder()
+            .setParent(parent.toString())
+            .setSourceLanguageCode(sourceLanguageCode)
+            .addAllTargetLanguageCodes(targetLanguageCodes)
+            .addAllInputConfigs(inputConfigs)
+            .setOutputConfig(outputConfig)
+            .build();
+
+    BatchTranslateResponse actualResponse = client.batchTranslateTextAsync(request).get();
+    Assert.assertEquals(expectedResponse, actualResponse);
+
+    List<AbstractMessage> actualRequests = mockTranslationService.getRequests();
+    Assert.assertEquals(1, actualRequests.size());
+    BatchTranslateTextRequest actualRequest = (BatchTranslateTextRequest) actualRequests.get(0);
+
+    Assert.assertEquals(parent, LocationName.parse(actualRequest.getParent()));
+    Assert.assertEquals(sourceLanguageCode, actualRequest.getSourceLanguageCode());
+    Assert.assertEquals(targetLanguageCodes, actualRequest.getTargetLanguageCodesList());
+    Assert.assertEquals(inputConfigs, actualRequest.getInputConfigsList());
+    Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
+    Assert.assertTrue(
+        channelProvider.isHeaderSent(
+            ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
+            GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
+  }
+
+  @Test
+  @SuppressWarnings("all")
+  public void batchTranslateTextExceptionTest() throws Exception {
+    StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
+    mockTranslationService.addException(exception);
+
+    try {
+      LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
+      String sourceLanguageCode = "sourceLanguageCode1687263568";
+      List<String> targetLanguageCodes = new ArrayList<>();
+      List<InputConfig> inputConfigs = new ArrayList<>();
+      OutputConfig outputConfig = OutputConfig.newBuilder().build();
+      BatchTranslateTextRequest request =
+          BatchTranslateTextRequest.newBuilder()
+              .setParent(parent.toString())
+              .setSourceLanguageCode(sourceLanguageCode)
+              .addAllTargetLanguageCodes(targetLanguageCodes)
+              .addAllInputConfigs(inputConfigs)
+              .setOutputConfig(outputConfig)
+              .build();
+
+      client.batchTranslateTextAsync(request).get();
+      Assert.fail("No exception raised");
+    } catch (ExecutionException e) {
+      Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
+      InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
+      Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
+    }
+  }
+
   @Test
   @SuppressWarnings("all")
   public void createGlossaryTest() throws Exception {