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

docs(regen): update sample code to set total timeout, add API client header test #85

Merged
merged 1 commit into from
Feb 10, 2020
Merged

docs(regen): update sample code to set total timeout, add API client header test #85

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions ...oud-translate/src/main/java/com/google/cloud/translate/v3/TranslationServiceSettings.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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>
Expand Down
8 changes: 6 additions & 2 deletions ...late/src/main/java/com/google/cloud/translate/v3/stub/TranslationServiceStubSettings.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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>
Expand Down
8 changes: 6 additions & 2 deletions ...ranslate/src/main/java/com/google/cloud/translate/v3beta1/TranslationServiceSettings.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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>
Expand Down
8 changes: 6 additions & 2 deletions ...src/main/java/com/google/cloud/translate/v3beta1/stub/TranslationServiceStubSettings.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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>
Expand Down
82 changes: 82 additions & 0 deletions ...d-translate/src/test/java/com/google/cloud/translate/v3/TranslationServiceClientTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down
139 changes: 139 additions & 0 deletions ...nslate/src/test/java/com/google/cloud/translate/v3beta1/TranslationServiceClientTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -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() {
Expand Down Expand Up @@ -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 {
Expand Down
Loading