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

…header test
googleapis · Feb 10, 2020 · 10922f4 · 10922f4
1 parent 4a52acf
commit 10922f4
Show file tree

Hide file tree

Showing 18 changed files with 361 additions and 34 deletions.
diff --git a/...loud-talent/src/main/java/com/google/cloud/talent/v4beta1/ApplicationServiceSettings.java b/...loud-talent/src/main/java/com/google/cloud/talent/v4beta1/ApplicationServiceSettings.java
@@ -55,8 +55,12 @@
  * <code>
  * ApplicationServiceSettings.Builder applicationServiceSettingsBuilder =
  *     ApplicationServiceSettings.newBuilder();
- * applicationServiceSettingsBuilder.createApplicationSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * applicationServiceSettingsBuilder
+ *     .createApplicationSettings()
+ *     .setRetrySettings(
+ *         applicationServiceSettingsBuilder.createApplicationSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * ApplicationServiceSettings applicationServiceSettings = applicationServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...le-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/CompanyServiceSettings.java b/...le-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/CompanyServiceSettings.java
@@ -55,8 +55,12 @@
  * <code>
  * CompanyServiceSettings.Builder companyServiceSettingsBuilder =
  *     CompanyServiceSettings.newBuilder();
- * companyServiceSettingsBuilder.createCompanySettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * companyServiceSettingsBuilder
+ *     .createCompanySettings()
+ *     .setRetrySettings(
+ *         companyServiceSettingsBuilder.createCompanySettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * CompanyServiceSettings companyServiceSettings = companyServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/CompletionSettings.java b/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/CompletionSettings.java
@@ -51,8 +51,12 @@
  * <code>
  * CompletionSettings.Builder completionSettingsBuilder =
  *     CompletionSettings.newBuilder();
- * completionSettingsBuilder.completeQuerySettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * completionSettingsBuilder
+ *     .completeQuerySettings()
+ *     .setRetrySettings(
+ *         completionSettingsBuilder.completeQuerySettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * CompletionSettings completionSettings = completionSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/EventServiceSettings.java b/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/EventServiceSettings.java
@@ -51,8 +51,12 @@
  * <code>
  * EventServiceSettings.Builder eventServiceSettingsBuilder =
  *     EventServiceSettings.newBuilder();
- * eventServiceSettingsBuilder.createClientEventSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * eventServiceSettingsBuilder
+ *     .createClientEventSettings()
+ *     .setRetrySettings(
+ *         eventServiceSettingsBuilder.createClientEventSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * EventServiceSettings eventServiceSettings = eventServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/JobServiceSettings.java b/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/JobServiceSettings.java
@@ -59,8 +59,12 @@
  * <code>
  * JobServiceSettings.Builder jobServiceSettingsBuilder =
  *     JobServiceSettings.newBuilder();
- * jobServiceSettingsBuilder.createJobSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * jobServiceSettingsBuilder
+ *     .createJobSettings()
+ *     .setRetrySettings(
+ *         jobServiceSettingsBuilder.createJobSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * JobServiceSettings jobServiceSettings = jobServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...le-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/ProfileServiceSettings.java b/...le-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/ProfileServiceSettings.java
@@ -56,8 +56,12 @@
  * <code>
  * ProfileServiceSettings.Builder profileServiceSettingsBuilder =
  *     ProfileServiceSettings.newBuilder();
- * profileServiceSettingsBuilder.createProfileSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * profileServiceSettingsBuilder
+ *     .createProfileSettings()
+ *     .setRetrySettings(
+ *         profileServiceSettingsBuilder.createProfileSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * ProfileServiceSettings profileServiceSettings = profileServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/TenantServiceSettings.java b/google-cloud-talent/src/main/java/com/google/cloud/talent/v4beta1/TenantServiceSettings.java
@@ -55,8 +55,12 @@
  * <code>
  * TenantServiceSettings.Builder tenantServiceSettingsBuilder =
  *     TenantServiceSettings.newBuilder();
- * tenantServiceSettingsBuilder.createTenantSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * tenantServiceSettingsBuilder
+ *     .createTenantSettings()
+ *     .setRetrySettings(
+ *         tenantServiceSettingsBuilder.createTenantSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * TenantServiceSettings tenantServiceSettings = tenantServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...nt/src/main/java/com/google/cloud/talent/v4beta1/stub/ApplicationServiceStubSettings.java b/...nt/src/main/java/com/google/cloud/talent/v4beta1/stub/ApplicationServiceStubSettings.java
@@ -77,8 +77,12 @@
  * <code>
  * ApplicationServiceStubSettings.Builder applicationServiceSettingsBuilder =
  *     ApplicationServiceStubSettings.newBuilder();
- * applicationServiceSettingsBuilder.createApplicationSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * applicationServiceSettingsBuilder
+ *     .createApplicationSettings()
+ *     .setRetrySettings(
+ *         applicationServiceSettingsBuilder.createApplicationSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * ApplicationServiceStubSettings applicationServiceSettings = applicationServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...talent/src/main/java/com/google/cloud/talent/v4beta1/stub/CompanyServiceStubSettings.java b/...talent/src/main/java/com/google/cloud/talent/v4beta1/stub/CompanyServiceStubSettings.java
@@ -77,8 +77,12 @@
  * <code>
  * CompanyServiceStubSettings.Builder companyServiceSettingsBuilder =
  *     CompanyServiceStubSettings.newBuilder();
- * companyServiceSettingsBuilder.createCompanySettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * companyServiceSettingsBuilder
+ *     .createCompanySettings()
+ *     .setRetrySettings(
+ *         companyServiceSettingsBuilder.createCompanySettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * CompanyServiceStubSettings companyServiceSettings = companyServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...oud-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/CompletionStubSettings.java b/...oud-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/CompletionStubSettings.java
@@ -62,8 +62,12 @@
  * <code>
  * CompletionStubSettings.Builder completionSettingsBuilder =
  *     CompletionStubSettings.newBuilder();
- * completionSettingsBuilder.completeQuerySettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * completionSettingsBuilder
+ *     .completeQuerySettings()
+ *     .setRetrySettings(
+ *         completionSettingsBuilder.completeQuerySettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * CompletionStubSettings completionSettings = completionSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...d-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/EventServiceStubSettings.java b/...d-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/EventServiceStubSettings.java
@@ -62,8 +62,12 @@
  * <code>
  * EventServiceStubSettings.Builder eventServiceSettingsBuilder =
  *     EventServiceStubSettings.newBuilder();
- * eventServiceSettingsBuilder.createClientEventSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * eventServiceSettingsBuilder
+ *     .createClientEventSettings()
+ *     .setRetrySettings(
+ *         eventServiceSettingsBuilder.createClientEventSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * EventServiceStubSettings eventServiceSettings = eventServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...oud-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/JobServiceStubSettings.java b/...oud-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/JobServiceStubSettings.java
@@ -91,8 +91,12 @@
  * <code>
  * JobServiceStubSettings.Builder jobServiceSettingsBuilder =
  *     JobServiceStubSettings.newBuilder();
- * jobServiceSettingsBuilder.createJobSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * jobServiceSettingsBuilder
+ *     .createJobSettings()
+ *     .setRetrySettings(
+ *         jobServiceSettingsBuilder.createJobSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * JobServiceStubSettings jobServiceSettings = jobServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...talent/src/main/java/com/google/cloud/talent/v4beta1/stub/ProfileServiceStubSettings.java b/...talent/src/main/java/com/google/cloud/talent/v4beta1/stub/ProfileServiceStubSettings.java
@@ -81,8 +81,12 @@
  * <code>
  * ProfileServiceStubSettings.Builder profileServiceSettingsBuilder =
  *     ProfileServiceStubSettings.newBuilder();
- * profileServiceSettingsBuilder.createProfileSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * profileServiceSettingsBuilder
+ *     .createProfileSettings()
+ *     .setRetrySettings(
+ *         profileServiceSettingsBuilder.createProfileSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * ProfileServiceStubSettings profileServiceSettings = profileServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/...-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/TenantServiceStubSettings.java b/...-talent/src/main/java/com/google/cloud/talent/v4beta1/stub/TenantServiceStubSettings.java
@@ -77,8 +77,12 @@
  * <code>
  * TenantServiceStubSettings.Builder tenantServiceSettingsBuilder =
  *     TenantServiceStubSettings.newBuilder();
- * tenantServiceSettingsBuilder.createTenantSettings().getRetrySettings().toBuilder()
- *     .setTotalTimeout(Duration.ofSeconds(30));
+ * tenantServiceSettingsBuilder
+ *     .createTenantSettings()
+ *     .setRetrySettings(
+ *         tenantServiceSettingsBuilder.createTenantSettings().getRetrySettings().toBuilder()
+ *             .setTotalTimeout(Duration.ofSeconds(30))
+ *             .build());
  * TenantServiceStubSettings tenantServiceSettings = tenantServiceSettingsBuilder.build();
  * </code>
  * </pre>

diff --git a/google-cloud-talent/src/test/java/com/google/cloud/talent/v4beta1/CompletionClientTest.java b/google-cloud-talent/src/test/java/com/google/cloud/talent/v4beta1/CompletionClientTest.java
@@ -16,16 +16,25 @@
 package com.google.cloud.talent.v4beta1;
 
 import com.google.api.gax.core.NoCredentialsProvider;
+import com.google.api.gax.grpc.GaxGrpcProperties;
 import com.google.api.gax.grpc.testing.LocalChannelProvider;
 import com.google.api.gax.grpc.testing.MockGrpcService;
 import com.google.api.gax.grpc.testing.MockServiceHelper;
+import com.google.api.gax.rpc.ApiClientHeaderProvider;
+import com.google.api.gax.rpc.InvalidArgumentException;
+import com.google.protobuf.AbstractMessage;
+import io.grpc.Status;
+import io.grpc.StatusRuntimeException;
 import java.io.IOException;
 import java.util.Arrays;
+import java.util.List;
 import java.util.UUID;
 import org.junit.After;
 import org.junit.AfterClass;
+import org.junit.Assert;
 import org.junit.Before;
 import org.junit.BeforeClass;
+import org.junit.Test;
 
 @javax.annotation.Generated("by GAPIC")
 public class CompletionClientTest {
@@ -84,4 +93,60 @@ public void setUp() throws IOException {
   public void tearDown() throws Exception {
     client.close();
   }
+
+  @Test
+  @SuppressWarnings("all")
+  public void completeQueryTest() {
+    CompleteQueryResponse expectedResponse = CompleteQueryResponse.newBuilder().build();
+    mockCompletion.addResponse(expectedResponse);
+
+    TenantOrProjectName parent = TenantName.of("[PROJECT]", "[TENANT]");
+    String query = "query107944136";
+    int pageSize = 883849137;
+    CompleteQueryRequest request =
+        CompleteQueryRequest.newBuilder()
+            .setParent(parent.toString())
+            .setQuery(query)
+            .setPageSize(pageSize)
+            .build();
+
+    CompleteQueryResponse actualResponse = client.completeQuery(request);
+    Assert.assertEquals(expectedResponse, actualResponse);
+
+    List<AbstractMessage> actualRequests = mockCompletion.getRequests();
+    Assert.assertEquals(1, actualRequests.size());
+    CompleteQueryRequest actualRequest = (CompleteQueryRequest) actualRequests.get(0);
+
+    Assert.assertEquals(parent, TenantOrProjectNames.parse(actualRequest.getParent()));
+    Assert.assertEquals(query, actualRequest.getQuery());
+    Assert.assertEquals(pageSize, actualRequest.getPageSize());
+    Assert.assertTrue(
+        channelProvider.isHeaderSent(
+            ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
+            GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
+  }
+
+  @Test
+  @SuppressWarnings("all")
+  public void completeQueryExceptionTest() throws Exception {
+    StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
+    mockCompletion.addException(exception);
+
+    try {
+      TenantOrProjectName parent = TenantName.of("[PROJECT]", "[TENANT]");
+      String query = "query107944136";
+      int pageSize = 883849137;
+      CompleteQueryRequest request =
+          CompleteQueryRequest.newBuilder()
+              .setParent(parent.toString())
+              .setQuery(query)
+              .setPageSize(pageSize)
+              .build();
+
+      client.completeQuery(request);
+      Assert.fail("No exception raised");
+    } catch (InvalidArgumentException e) {
+      // Expected exception
+    }
+  }
 }