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

Commit 9d7dddc

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

21 files changed

+262
-58
lines changed

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/AutoscalingPolicyServiceSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,12 @@
5555
* <code>
5656
* AutoscalingPolicyServiceSettings.Builder autoscalingPolicyServiceSettingsBuilder =
5757
* AutoscalingPolicyServiceSettings.newBuilder();
58-
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
59-
* .setTotalTimeout(Duration.ofSeconds(30));
58+
* autoscalingPolicyServiceSettingsBuilder
59+
* .createAutoscalingPolicySettings()
60+
* .setRetrySettings(
61+
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
62+
* .setTotalTimeout(Duration.ofSeconds(30))
63+
* .build());
6064
* AutoscalingPolicyServiceSettings autoscalingPolicyServiceSettings = autoscalingPolicyServiceSettingsBuilder.build();
6165
* </code>
6266
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/ClusterControllerSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,12 @@
5757
* <code>
5858
* ClusterControllerSettings.Builder clusterControllerSettingsBuilder =
5959
* ClusterControllerSettings.newBuilder();
60-
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
61-
* .setTotalTimeout(Duration.ofSeconds(30));
60+
* clusterControllerSettingsBuilder
61+
* .getClusterSettings()
62+
* .setRetrySettings(
63+
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
64+
* .setTotalTimeout(Duration.ofSeconds(30))
65+
* .build());
6266
* ClusterControllerSettings clusterControllerSettings = clusterControllerSettingsBuilder.build();
6367
* </code>
6468
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/JobControllerSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,12 @@
5555
* <code>
5656
* JobControllerSettings.Builder jobControllerSettingsBuilder =
5757
* JobControllerSettings.newBuilder();
58-
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
59-
* .setTotalTimeout(Duration.ofSeconds(30));
58+
* jobControllerSettingsBuilder
59+
* .submitJobSettings()
60+
* .setRetrySettings(
61+
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
62+
* .setTotalTimeout(Duration.ofSeconds(30))
63+
* .build());
6064
* JobControllerSettings jobControllerSettings = jobControllerSettingsBuilder.build();
6165
* </code>
6266
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/WorkflowTemplateServiceSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,12 @@
5757
* <code>
5858
* WorkflowTemplateServiceSettings.Builder workflowTemplateServiceSettingsBuilder =
5959
* WorkflowTemplateServiceSettings.newBuilder();
60-
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
61-
* .setTotalTimeout(Duration.ofSeconds(30));
60+
* workflowTemplateServiceSettingsBuilder
61+
* .createWorkflowTemplateSettings()
62+
* .setRetrySettings(
63+
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
64+
* .setTotalTimeout(Duration.ofSeconds(30))
65+
* .build());
6266
* WorkflowTemplateServiceSettings workflowTemplateServiceSettings = workflowTemplateServiceSettingsBuilder.build();
6367
* </code>
6468
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/stub/AutoscalingPolicyServiceStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,12 @@
7777
* <code>
7878
* AutoscalingPolicyServiceStubSettings.Builder autoscalingPolicyServiceSettingsBuilder =
7979
* AutoscalingPolicyServiceStubSettings.newBuilder();
80-
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
81-
* .setTotalTimeout(Duration.ofSeconds(30));
80+
* autoscalingPolicyServiceSettingsBuilder
81+
* .createAutoscalingPolicySettings()
82+
* .setRetrySettings(
83+
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
84+
* .setTotalTimeout(Duration.ofSeconds(30))
85+
* .build());
8286
* AutoscalingPolicyServiceStubSettings autoscalingPolicyServiceSettings = autoscalingPolicyServiceSettingsBuilder.build();
8387
* </code>
8488
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/stub/ClusterControllerStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,12 @@
8585
* <code>
8686
* ClusterControllerStubSettings.Builder clusterControllerSettingsBuilder =
8787
* ClusterControllerStubSettings.newBuilder();
88-
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
89-
* .setTotalTimeout(Duration.ofSeconds(30));
88+
* clusterControllerSettingsBuilder
89+
* .getClusterSettings()
90+
* .setRetrySettings(
91+
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
92+
* .setTotalTimeout(Duration.ofSeconds(30))
93+
* .build());
9094
* ClusterControllerStubSettings clusterControllerSettings = clusterControllerSettingsBuilder.build();
9195
* </code>
9296
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/stub/JobControllerStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,12 @@
7878
* <code>
7979
* JobControllerStubSettings.Builder jobControllerSettingsBuilder =
8080
* JobControllerStubSettings.newBuilder();
81-
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
82-
* .setTotalTimeout(Duration.ofSeconds(30));
81+
* jobControllerSettingsBuilder
82+
* .submitJobSettings()
83+
* .setRetrySettings(
84+
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
85+
* .setTotalTimeout(Duration.ofSeconds(30))
86+
* .build());
8387
* JobControllerStubSettings jobControllerSettings = jobControllerSettingsBuilder.build();
8488
* </code>
8589
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1/stub/WorkflowTemplateServiceStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,12 @@
8585
* <code>
8686
* WorkflowTemplateServiceStubSettings.Builder workflowTemplateServiceSettingsBuilder =
8787
* WorkflowTemplateServiceStubSettings.newBuilder();
88-
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
89-
* .setTotalTimeout(Duration.ofSeconds(30));
88+
* workflowTemplateServiceSettingsBuilder
89+
* .createWorkflowTemplateSettings()
90+
* .setRetrySettings(
91+
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
92+
* .setTotalTimeout(Duration.ofSeconds(30))
93+
* .build());
9094
* WorkflowTemplateServiceStubSettings workflowTemplateServiceSettings = workflowTemplateServiceSettingsBuilder.build();
9195
* </code>
9296
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/AutoscalingPolicyServiceSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,12 @@
5555
* <code>
5656
* AutoscalingPolicyServiceSettings.Builder autoscalingPolicyServiceSettingsBuilder =
5757
* AutoscalingPolicyServiceSettings.newBuilder();
58-
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
59-
* .setTotalTimeout(Duration.ofSeconds(30));
58+
* autoscalingPolicyServiceSettingsBuilder
59+
* .createAutoscalingPolicySettings()
60+
* .setRetrySettings(
61+
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
62+
* .setTotalTimeout(Duration.ofSeconds(30))
63+
* .build());
6064
* AutoscalingPolicyServiceSettings autoscalingPolicyServiceSettings = autoscalingPolicyServiceSettingsBuilder.build();
6165
* </code>
6266
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/ClusterControllerSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,12 @@
5757
* <code>
5858
* ClusterControllerSettings.Builder clusterControllerSettingsBuilder =
5959
* ClusterControllerSettings.newBuilder();
60-
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
61-
* .setTotalTimeout(Duration.ofSeconds(30));
60+
* clusterControllerSettingsBuilder
61+
* .getClusterSettings()
62+
* .setRetrySettings(
63+
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
64+
* .setTotalTimeout(Duration.ofSeconds(30))
65+
* .build());
6266
* ClusterControllerSettings clusterControllerSettings = clusterControllerSettingsBuilder.build();
6367
* </code>
6468
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/JobControllerSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,12 @@
5555
* <code>
5656
* JobControllerSettings.Builder jobControllerSettingsBuilder =
5757
* JobControllerSettings.newBuilder();
58-
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
59-
* .setTotalTimeout(Duration.ofSeconds(30));
58+
* jobControllerSettingsBuilder
59+
* .submitJobSettings()
60+
* .setRetrySettings(
61+
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
62+
* .setTotalTimeout(Duration.ofSeconds(30))
63+
* .build());
6064
* JobControllerSettings jobControllerSettings = jobControllerSettingsBuilder.build();
6165
* </code>
6266
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/WorkflowTemplateServiceSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,12 @@
5757
* <code>
5858
* WorkflowTemplateServiceSettings.Builder workflowTemplateServiceSettingsBuilder =
5959
* WorkflowTemplateServiceSettings.newBuilder();
60-
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
61-
* .setTotalTimeout(Duration.ofSeconds(30));
60+
* workflowTemplateServiceSettingsBuilder
61+
* .createWorkflowTemplateSettings()
62+
* .setRetrySettings(
63+
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
64+
* .setTotalTimeout(Duration.ofSeconds(30))
65+
* .build());
6266
* WorkflowTemplateServiceSettings workflowTemplateServiceSettings = workflowTemplateServiceSettingsBuilder.build();
6367
* </code>
6468
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/stub/AutoscalingPolicyServiceStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,12 @@
7777
* <code>
7878
* AutoscalingPolicyServiceStubSettings.Builder autoscalingPolicyServiceSettingsBuilder =
7979
* AutoscalingPolicyServiceStubSettings.newBuilder();
80-
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
81-
* .setTotalTimeout(Duration.ofSeconds(30));
80+
* autoscalingPolicyServiceSettingsBuilder
81+
* .createAutoscalingPolicySettings()
82+
* .setRetrySettings(
83+
* autoscalingPolicyServiceSettingsBuilder.createAutoscalingPolicySettings().getRetrySettings().toBuilder()
84+
* .setTotalTimeout(Duration.ofSeconds(30))
85+
* .build());
8286
* AutoscalingPolicyServiceStubSettings autoscalingPolicyServiceSettings = autoscalingPolicyServiceSettingsBuilder.build();
8387
* </code>
8488
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/stub/ClusterControllerStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,12 @@
8585
* <code>
8686
* ClusterControllerStubSettings.Builder clusterControllerSettingsBuilder =
8787
* ClusterControllerStubSettings.newBuilder();
88-
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
89-
* .setTotalTimeout(Duration.ofSeconds(30));
88+
* clusterControllerSettingsBuilder
89+
* .getClusterSettings()
90+
* .setRetrySettings(
91+
* clusterControllerSettingsBuilder.getClusterSettings().getRetrySettings().toBuilder()
92+
* .setTotalTimeout(Duration.ofSeconds(30))
93+
* .build());
9094
* ClusterControllerStubSettings clusterControllerSettings = clusterControllerSettingsBuilder.build();
9195
* </code>
9296
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/stub/JobControllerStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,12 @@
7878
* <code>
7979
* JobControllerStubSettings.Builder jobControllerSettingsBuilder =
8080
* JobControllerStubSettings.newBuilder();
81-
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
82-
* .setTotalTimeout(Duration.ofSeconds(30));
81+
* jobControllerSettingsBuilder
82+
* .submitJobSettings()
83+
* .setRetrySettings(
84+
* jobControllerSettingsBuilder.submitJobSettings().getRetrySettings().toBuilder()
85+
* .setTotalTimeout(Duration.ofSeconds(30))
86+
* .build());
8387
* JobControllerStubSettings jobControllerSettings = jobControllerSettingsBuilder.build();
8488
* </code>
8589
* </pre>

‎google-cloud-dataproc/src/main/java/com/google/cloud/dataproc/v1beta2/stub/WorkflowTemplateServiceStubSettings.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,12 @@
8585
* <code>
8686
* WorkflowTemplateServiceStubSettings.Builder workflowTemplateServiceSettingsBuilder =
8787
* WorkflowTemplateServiceStubSettings.newBuilder();
88-
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
89-
* .setTotalTimeout(Duration.ofSeconds(30));
88+
* workflowTemplateServiceSettingsBuilder
89+
* .createWorkflowTemplateSettings()
90+
* .setRetrySettings(
91+
* workflowTemplateServiceSettingsBuilder.createWorkflowTemplateSettings().getRetrySettings().toBuilder()
92+
* .setTotalTimeout(Duration.ofSeconds(30))
93+
* .build());
9094
* WorkflowTemplateServiceStubSettings workflowTemplateServiceSettings = workflowTemplateServiceSettingsBuilder.build();
9195
* </code>
9296
* </pre>

‎google-cloud-dataproc/src/test/java/com/google/cloud/dataproc/v1/ClusterControllerClientTest.java

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -391,9 +391,8 @@ public void listClustersTest2() {
391391

392392
String projectId = "projectId-1969970175";
393393
String region = "region-934795532";
394-
String filter = "filter-1274492040";
395394

396-
ListClustersPagedResponse pagedListResponse = client.listClusters(projectId, region, filter);
395+
ListClustersPagedResponse pagedListResponse = client.listClusters(projectId, region);
397396

398397
List<Cluster> resources = Lists.newArrayList(pagedListResponse.iterateAll());
399398
Assert.assertEquals(1, resources.size());
@@ -405,7 +404,6 @@ public void listClustersTest2() {
405404

406405
Assert.assertEquals(projectId, actualRequest.getProjectId());
407406
Assert.assertEquals(region, actualRequest.getRegion());
408-
Assert.assertEquals(filter, actualRequest.getFilter());
409407
Assert.assertTrue(
410408
channelProvider.isHeaderSent(
411409
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -421,9 +419,8 @@ public void listClustersExceptionTest2() throws Exception {
421419
try {
422420
String projectId = "projectId-1969970175";
423421
String region = "region-934795532";
424-
String filter = "filter-1274492040";
425422

426-
client.listClusters(projectId, region, filter);
423+
client.listClusters(projectId, region);
427424
Assert.fail("No exception raised");
428425
} catch (InvalidArgumentException e) {
429426
// Expected exception

‎google-cloud-dataproc/src/test/java/com/google/cloud/dataproc/v1/JobControllerClientTest.java

Lines changed: 77 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import com.google.common.collect.Lists;
2828
import com.google.protobuf.AbstractMessage;
2929
import com.google.protobuf.Empty;
30+
import com.google.protobuf.FieldMask;
3031
import io.grpc.Status;
3132
import io.grpc.StatusRuntimeException;
3233
import java.io.IOException;
@@ -253,9 +254,8 @@ public void listJobsTest2() {
253254

254255
String projectId = "projectId-1969970175";
255256
String region = "region-934795532";
256-
String filter = "filter-1274492040";
257257

258-
ListJobsPagedResponse pagedListResponse = client.listJobs(projectId, region, filter);
258+
ListJobsPagedResponse pagedListResponse = client.listJobs(projectId, region);
259259

260260
List<Job> resources = Lists.newArrayList(pagedListResponse.iterateAll());
261261
Assert.assertEquals(1, resources.size());
@@ -267,7 +267,6 @@ public void listJobsTest2() {
267267

268268
Assert.assertEquals(projectId, actualRequest.getProjectId());
269269
Assert.assertEquals(region, actualRequest.getRegion());
270-
Assert.assertEquals(filter, actualRequest.getFilter());
271270
Assert.assertTrue(
272271
channelProvider.isHeaderSent(
273272
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -283,9 +282,82 @@ public void listJobsExceptionTest2() throws Exception {
283282
try {
284283
String projectId = "projectId-1969970175";
285284
String region = "region-934795532";
286-
String filter = "filter-1274492040";
287285

288-
client.listJobs(projectId, region, filter);
286+
client.listJobs(projectId, region);
287+
Assert.fail("No exception raised");
288+
} catch (InvalidArgumentException e) {
289+
// Expected exception
290+
}
291+
}
292+
293+
@Test
294+
@SuppressWarnings("all")
295+
public void updateJobTest() {
296+
String driverOutputResourceUri = "driverOutputResourceUri-542229086";
297+
String driverControlFilesUri = "driverControlFilesUri207057643";
298+
String jobUuid = "jobUuid-1615012099";
299+
Job expectedResponse =
300+
Job.newBuilder()
301+
.setDriverOutputResourceUri(driverOutputResourceUri)
302+
.setDriverControlFilesUri(driverControlFilesUri)
303+
.setJobUuid(jobUuid)
304+
.build();
305+
mockJobController.addResponse(expectedResponse);
306+
307+
String projectId = "projectId-1969970175";
308+
String region = "region-934795532";
309+
String jobId = "jobId-1154752291";
310+
Job job = Job.newBuilder().build();
311+
FieldMask updateMask = FieldMask.newBuilder().build();
312+
UpdateJobRequest request =
313+
UpdateJobRequest.newBuilder()
314+
.setProjectId(projectId)
315+
.setRegion(region)
316+
.setJobId(jobId)
317+
.setJob(job)
318+
.setUpdateMask(updateMask)
319+
.build();
320+
321+
Job actualResponse = client.updateJob(request);
322+
Assert.assertEquals(expectedResponse, actualResponse);
323+
324+
List<AbstractMessage> actualRequests = mockJobController.getRequests();
325+
Assert.assertEquals(1, actualRequests.size());
326+
UpdateJobRequest actualRequest = (UpdateJobRequest) actualRequests.get(0);
327+
328+
Assert.assertEquals(projectId, actualRequest.getProjectId());
329+
Assert.assertEquals(region, actualRequest.getRegion());
330+
Assert.assertEquals(jobId, actualRequest.getJobId());
331+
Assert.assertEquals(job, actualRequest.getJob());
332+
Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
333+
Assert.assertTrue(
334+
channelProvider.isHeaderSent(
335+
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
336+
GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
337+
}
338+
339+
@Test
340+
@SuppressWarnings("all")
341+
public void updateJobExceptionTest() throws Exception {
342+
StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
343+
mockJobController.addException(exception);
344+
345+
try {
346+
String projectId = "projectId-1969970175";
347+
String region = "region-934795532";
348+
String jobId = "jobId-1154752291";
349+
Job job = Job.newBuilder().build();
350+
FieldMask updateMask = FieldMask.newBuilder().build();
351+
UpdateJobRequest request =
352+
UpdateJobRequest.newBuilder()
353+
.setProjectId(projectId)
354+
.setRegion(region)
355+
.setJobId(jobId)
356+
.setJob(job)
357+
.setUpdateMask(updateMask)
358+
.build();
359+
360+
client.updateJob(request);
289361
Assert.fail("No exception raised");
290362
} catch (InvalidArgumentException e) {
291363
// Expected exception

0 commit comments

Comments
 (0)