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

Commit 5c08375

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

File tree

11 files changed

+234
-22
lines changed

11 files changed

+234
-22
lines changed

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1/AutoMlSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,12 @@
5959
* <code>
6060
* AutoMlSettings.Builder autoMlSettingsBuilder =
6161
* AutoMlSettings.newBuilder();
62-
* autoMlSettingsBuilder.updateDatasetSettings().getRetrySettings().toBuilder()
63-
* .setTotalTimeout(Duration.ofSeconds(30));
62+
* autoMlSettingsBuilder
63+
* .updateDatasetSettings()
64+
* .setRetrySettings(
65+
* autoMlSettingsBuilder.updateDatasetSettings().getRetrySettings().toBuilder()
66+
* .setTotalTimeout(Duration.ofSeconds(30))
67+
* .build());
6468
* AutoMlSettings autoMlSettings = autoMlSettingsBuilder.build();
6569
* </code>
6670
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1/PredictionServiceSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,12 @@
5353
* <code>
5454
* PredictionServiceSettings.Builder predictionServiceSettingsBuilder =
5555
* PredictionServiceSettings.newBuilder();
56-
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
57-
* .setTotalTimeout(Duration.ofSeconds(30));
56+
* predictionServiceSettingsBuilder
57+
* .predictSettings()
58+
* .setRetrySettings(
59+
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
60+
* .setTotalTimeout(Duration.ofSeconds(30))
61+
* .build());
5862
* PredictionServiceSettings predictionServiceSettings = predictionServiceSettingsBuilder.build();
5963
* </code>
6064
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1/stub/AutoMlStubSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -103,8 +103,12 @@
103103
* <code>
104104
* AutoMlStubSettings.Builder autoMlSettingsBuilder =
105105
* AutoMlStubSettings.newBuilder();
106-
* autoMlSettingsBuilder.updateDatasetSettings().getRetrySettings().toBuilder()
107-
* .setTotalTimeout(Duration.ofSeconds(30));
106+
* autoMlSettingsBuilder
107+
* .updateDatasetSettings()
108+
* .setRetrySettings(
109+
* autoMlSettingsBuilder.updateDatasetSettings().getRetrySettings().toBuilder()
110+
* .setTotalTimeout(Duration.ofSeconds(30))
111+
* .build());
108112
* AutoMlStubSettings autoMlSettings = autoMlSettingsBuilder.build();
109113
* </code>
110114
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1/stub/PredictionServiceStubSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,12 @@
7070
* <code>
7171
* PredictionServiceStubSettings.Builder predictionServiceSettingsBuilder =
7272
* PredictionServiceStubSettings.newBuilder();
73-
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
74-
* .setTotalTimeout(Duration.ofSeconds(30));
73+
* predictionServiceSettingsBuilder
74+
* .predictSettings()
75+
* .setRetrySettings(
76+
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
77+
* .setTotalTimeout(Duration.ofSeconds(30))
78+
* .build());
7579
* PredictionServiceStubSettings predictionServiceSettings = predictionServiceSettingsBuilder.build();
7680
* </code>
7781
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1beta1/AutoMlSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,12 @@
6161
* <code>
6262
* AutoMlSettings.Builder autoMlSettingsBuilder =
6363
* AutoMlSettings.newBuilder();
64-
* autoMlSettingsBuilder.createDatasetSettings().getRetrySettings().toBuilder()
65-
* .setTotalTimeout(Duration.ofSeconds(30));
64+
* autoMlSettingsBuilder
65+
* .createDatasetSettings()
66+
* .setRetrySettings(
67+
* autoMlSettingsBuilder.createDatasetSettings().getRetrySettings().toBuilder()
68+
* .setTotalTimeout(Duration.ofSeconds(30))
69+
* .build());
6670
* AutoMlSettings autoMlSettings = autoMlSettingsBuilder.build();
6771
* </code>
6872
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1beta1/PredictionServiceSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,12 @@
5353
* <code>
5454
* PredictionServiceSettings.Builder predictionServiceSettingsBuilder =
5555
* PredictionServiceSettings.newBuilder();
56-
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
57-
* .setTotalTimeout(Duration.ofSeconds(30));
56+
* predictionServiceSettingsBuilder
57+
* .predictSettings()
58+
* .setRetrySettings(
59+
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
60+
* .setTotalTimeout(Duration.ofSeconds(30))
61+
* .build());
5862
* PredictionServiceSettings predictionServiceSettings = predictionServiceSettingsBuilder.build();
5963
* </code>
6064
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1beta1/stub/AutoMlStubSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,8 +115,12 @@
115115
* <code>
116116
* AutoMlStubSettings.Builder autoMlSettingsBuilder =
117117
* AutoMlStubSettings.newBuilder();
118-
* autoMlSettingsBuilder.createDatasetSettings().getRetrySettings().toBuilder()
119-
* .setTotalTimeout(Duration.ofSeconds(30));
118+
* autoMlSettingsBuilder
119+
* .createDatasetSettings()
120+
* .setRetrySettings(
121+
* autoMlSettingsBuilder.createDatasetSettings().getRetrySettings().toBuilder()
122+
* .setTotalTimeout(Duration.ofSeconds(30))
123+
* .build());
120124
* AutoMlStubSettings autoMlSettings = autoMlSettingsBuilder.build();
121125
* </code>
122126
* </pre>

‎google-cloud-automl/src/main/java/com/google/cloud/automl/v1beta1/stub/PredictionServiceStubSettings.java‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,12 @@
7070
* <code>
7171
* PredictionServiceStubSettings.Builder predictionServiceSettingsBuilder =
7272
* PredictionServiceStubSettings.newBuilder();
73-
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
74-
* .setTotalTimeout(Duration.ofSeconds(30));
73+
* predictionServiceSettingsBuilder
74+
* .predictSettings()
75+
* .setRetrySettings(
76+
* predictionServiceSettingsBuilder.predictSettings().getRetrySettings().toBuilder()
77+
* .setTotalTimeout(Duration.ofSeconds(30))
78+
* .build());
7579
* PredictionServiceStubSettings predictionServiceSettings = predictionServiceSettingsBuilder.build();
7680
* </code>
7781
* </pre>

‎google-cloud-automl/src/test/java/com/google/cloud/automl/v1/AutoMlClientTest.java‎

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -833,6 +833,65 @@ public void undeployModelExceptionTest() throws Exception {
833833
}
834834
}
835835

836+
@Test
837+
@SuppressWarnings("all")
838+
public void exportModelTest() throws Exception {
839+
Empty expectedResponse = Empty.newBuilder().build();
840+
Operation resultOperation =
841+
Operation.newBuilder()
842+
.setName("exportModelTest")
843+
.setDone(true)
844+
.setResponse(Any.pack(expectedResponse))
845+
.build();
846+
mockAutoMl.addResponse(resultOperation);
847+
848+
ModelName name = ModelName.of("[PROJECT]", "[LOCATION]", "[MODEL]");
849+
ModelExportOutputConfig outputConfig = ModelExportOutputConfig.newBuilder().build();
850+
ExportModelRequest request =
851+
ExportModelRequest.newBuilder()
852+
.setName(name.toString())
853+
.setOutputConfig(outputConfig)
854+
.build();
855+
856+
Empty actualResponse = client.exportModelAsync(request).get();
857+
Assert.assertEquals(expectedResponse, actualResponse);
858+
859+
List<AbstractMessage> actualRequests = mockAutoMl.getRequests();
860+
Assert.assertEquals(1, actualRequests.size());
861+
ExportModelRequest actualRequest = (ExportModelRequest) actualRequests.get(0);
862+
863+
Assert.assertEquals(name, ModelName.parse(actualRequest.getName()));
864+
Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
865+
Assert.assertTrue(
866+
channelProvider.isHeaderSent(
867+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
868+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
869+
}
870+
871+
@Test
872+
@SuppressWarnings("all")
873+
public void exportModelExceptionTest() throws Exception {
874+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
875+
mockAutoMl.addException(exception);
876+
877+
try {
878+
ModelName name = ModelName.of("[PROJECT]", "[LOCATION]", "[MODEL]");
879+
ModelExportOutputConfig outputConfig = ModelExportOutputConfig.newBuilder().build();
880+
ExportModelRequest request =
881+
ExportModelRequest.newBuilder()
882+
.setName(name.toString())
883+
.setOutputConfig(outputConfig)
884+
.build();
885+
886+
client.exportModelAsync(request).get();
887+
Assert.fail("No exception raised");
888+
} catch (ExecutionException e) {
889+
Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
890+
InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
891+
Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
892+
}
893+
}
894+
836895
@Test
837896
@SuppressWarnings("all")
838897
public void getModelEvaluationTest() {

‎google-cloud-automl/src/test/java/com/google/cloud/automl/v1beta1/AutoMlClientTest.java‎

Lines changed: 121 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -770,6 +770,127 @@ public void getModelEvaluationExceptionTest() throws Exception {
770770
}
771771
}
772772

773+
@Test
774+
@SuppressWarnings("all")
775+
public void exportModelTest() throws Exception {
776+
Empty expectedResponse = Empty.newBuilder().build();
777+
Operation resultOperation =
778+
Operation.newBuilder()
779+
.setName("exportModelTest")
780+
.setDone(true)
781+
.setResponse(Any.pack(expectedResponse))
782+
.build();
783+
mockAutoMl.addResponse(resultOperation);
784+
785+
ModelName name = ModelName.of("[PROJECT]", "[LOCATION]", "[MODEL]");
786+
ModelExportOutputConfig outputConfig = ModelExportOutputConfig.newBuilder().build();
787+
ExportModelRequest request =
788+
ExportModelRequest.newBuilder()
789+
.setName(name.toString())
790+
.setOutputConfig(outputConfig)
791+
.build();
792+
793+
Empty actualResponse = client.exportModelAsync(request).get();
794+
Assert.assertEquals(expectedResponse, actualResponse);
795+
796+
List<AbstractMessage> actualRequests = mockAutoMl.getRequests();
797+
Assert.assertEquals(1, actualRequests.size());
798+
ExportModelRequest actualRequest = (ExportModelRequest) actualRequests.get(0);
799+
800+
Assert.assertEquals(name, ModelName.parse(actualRequest.getName()));
801+
Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
802+
Assert.assertTrue(
803+
channelProvider.isHeaderSent(
804+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
805+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
806+
}
807+
808+
@Test
809+
@SuppressWarnings("all")
810+
public void exportModelExceptionTest() throws Exception {
811+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
812+
mockAutoMl.addException(exception);
813+
814+
try {
815+
ModelName name = ModelName.of("[PROJECT]", "[LOCATION]", "[MODEL]");
816+
ModelExportOutputConfig outputConfig = ModelExportOutputConfig.newBuilder().build();
817+
ExportModelRequest request =
818+
ExportModelRequest.newBuilder()
819+
.setName(name.toString())
820+
.setOutputConfig(outputConfig)
821+
.build();
822+
823+
client.exportModelAsync(request).get();
824+
Assert.fail("No exception raised");
825+
} catch (ExecutionException e) {
826+
Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
827+
InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
828+
Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
829+
}
830+
}
831+
832+
@Test
833+
@SuppressWarnings("all")
834+
public void exportEvaluatedExamplesTest() throws Exception {
835+
Empty expectedResponse = Empty.newBuilder().build();
836+
Operation resultOperation =
837+
Operation.newBuilder()
838+
.setName("exportEvaluatedExamplesTest")
839+
.setDone(true)
840+
.setResponse(Any.pack(expectedResponse))
841+
.build();
842+
mockAutoMl.addResponse(resultOperation);
843+
844+
ModelName name = ModelName.of("[PROJECT]", "[LOCATION]", "[MODEL]");
845+
ExportEvaluatedExamplesOutputConfig outputConfig =
846+
ExportEvaluatedExamplesOutputConfig.newBuilder().build();
847+
ExportEvaluatedExamplesRequest request =
848+
ExportEvaluatedExamplesRequest.newBuilder()
849+
.setName(name.toString())
850+
.setOutputConfig(outputConfig)
851+
.build();
852+
853+
Empty actualResponse = client.exportEvaluatedExamplesAsync(request).get();
854+
Assert.assertEquals(expectedResponse, actualResponse);
855+
856+
List<AbstractMessage> actualRequests = mockAutoMl.getRequests();
857+
Assert.assertEquals(1, actualRequests.size());
858+
ExportEvaluatedExamplesRequest actualRequest =
859+
(ExportEvaluatedExamplesRequest) actualRequests.get(0);
860+
861+
Assert.assertEquals(name, ModelName.parse(actualRequest.getName()));
862+
Assert.assertEquals(outputConfig, actualRequest.getOutputConfig());
863+
Assert.assertTrue(
864+
channelProvider.isHeaderSent(
865+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
866+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
867+
}
868+
869+
@Test
870+
@SuppressWarnings("all")
871+
public void exportEvaluatedExamplesExceptionTest() throws Exception {
872+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
873+
mockAutoMl.addException(exception);
874+
875+
try {
876+
ModelName name = ModelName.of("[PROJECT]", "[LOCATION]", "[MODEL]");
877+
ExportEvaluatedExamplesOutputConfig outputConfig =
878+
ExportEvaluatedExamplesOutputConfig.newBuilder().build();
879+
ExportEvaluatedExamplesRequest request =
880+
ExportEvaluatedExamplesRequest.newBuilder()
881+
.setName(name.toString())
882+
.setOutputConfig(outputConfig)
883+
.build();
884+
885+
client.exportEvaluatedExamplesAsync(request).get();
886+
Assert.fail("No exception raised");
887+
} catch (ExecutionException e) {
888+
Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass());
889+
InvalidArgumentException apiException = (InvalidArgumentException) e.getCause();
890+
Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode());
891+
}
892+
}
893+
773894
@Test
774895
@SuppressWarnings("all")
775896
public void listModelEvaluationsTest() {

0 commit comments

Comments
 (0)