Skip to content

Commit c5f8d8d

Browse files
noahdietznoahdietz
committed
update a unit test golden
1 parent 40fe026 commit c5f8d8d

File tree

1 file changed

+62
-21
lines changed

1 file changed

+62
-21
lines changed

‎gapic-generator-java/src/test/java/com/google/api/generator/gapic/composer/grpc/goldens/SubscriberClientTest.golden‎

Lines changed: 62 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ import com.google.api.gax.rpc.StatusCode;
1717
import com.google.common.collect.Lists;
1818
import com.google.protobuf.AbstractMessage;
1919
import com.google.protobuf.Empty;
20+
import com.google.protobuf.FieldMask;
2021
import google.cloud.ProjectName;
2122
import io.grpc.StatusRuntimeException;
2223
import java.io.IOException;
@@ -79,6 +80,8 @@ public class SubscriberClientTest {
7980
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
8081
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
8182
.setPushConfig(PushConfig.newBuilder().build())
83+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
84+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
8285
.setAckDeadlineSeconds(2135351438)
8386
.setRetainAckedMessages(true)
8487
.putAllLabels(new HashMap<String, String>())
@@ -89,6 +92,10 @@ public class SubscriberClientTest {
8992
.setRetryPolicy(RetryPolicy.newBuilder().build())
9093
.setDetached(true)
9194
.setEnableExactlyOnceDelivery(true)
95+
.setAnalyticsHubSubscriptionInfo(
96+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
97+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
98+
.putAllTags(new HashMap<String, String>())
9299
.build();
93100
mockSubscriber.addResponse(expectedResponse);
94101

@@ -139,6 +146,8 @@ public class SubscriberClientTest {
139146
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
140147
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
141148
.setPushConfig(PushConfig.newBuilder().build())
149+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
150+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
142151
.setAckDeadlineSeconds(2135351438)
143152
.setRetainAckedMessages(true)
144153
.putAllLabels(new HashMap<String, String>())
@@ -149,6 +158,10 @@ public class SubscriberClientTest {
149158
.setRetryPolicy(RetryPolicy.newBuilder().build())
150159
.setDetached(true)
151160
.setEnableExactlyOnceDelivery(true)
161+
.setAnalyticsHubSubscriptionInfo(
162+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
163+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
164+
.putAllTags(new HashMap<String, String>())
152165
.build();
153166
mockSubscriber.addResponse(expectedResponse);
154167

@@ -199,6 +212,8 @@ public class SubscriberClientTest {
199212
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
200213
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
201214
.setPushConfig(PushConfig.newBuilder().build())
215+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
216+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
202217
.setAckDeadlineSeconds(2135351438)
203218
.setRetainAckedMessages(true)
204219
.putAllLabels(new HashMap<String, String>())
@@ -209,6 +224,10 @@ public class SubscriberClientTest {
209224
.setRetryPolicy(RetryPolicy.newBuilder().build())
210225
.setDetached(true)
211226
.setEnableExactlyOnceDelivery(true)
227+
.setAnalyticsHubSubscriptionInfo(
228+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
229+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
230+
.putAllTags(new HashMap<String, String>())
212231
.build();
213232
mockSubscriber.addResponse(expectedResponse);
214233

@@ -259,6 +278,8 @@ public class SubscriberClientTest {
259278
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
260279
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
261280
.setPushConfig(PushConfig.newBuilder().build())
281+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
282+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
262283
.setAckDeadlineSeconds(2135351438)
263284
.setRetainAckedMessages(true)
264285
.putAllLabels(new HashMap<String, String>())
@@ -269,6 +290,10 @@ public class SubscriberClientTest {
269290
.setRetryPolicy(RetryPolicy.newBuilder().build())
270291
.setDetached(true)
271292
.setEnableExactlyOnceDelivery(true)
293+
.setAnalyticsHubSubscriptionInfo(
294+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
295+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
296+
.putAllTags(new HashMap<String, String>())
272297
.build();
273298
mockSubscriber.addResponse(expectedResponse);
274299

@@ -319,6 +344,8 @@ public class SubscriberClientTest {
319344
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
320345
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
321346
.setPushConfig(PushConfig.newBuilder().build())
347+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
348+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
322349
.setAckDeadlineSeconds(2135351438)
323350
.setRetainAckedMessages(true)
324351
.putAllLabels(new HashMap<String, String>())
@@ -329,6 +356,10 @@ public class SubscriberClientTest {
329356
.setRetryPolicy(RetryPolicy.newBuilder().build())
330357
.setDetached(true)
331358
.setEnableExactlyOnceDelivery(true)
359+
.setAnalyticsHubSubscriptionInfo(
360+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
361+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
362+
.putAllTags(new HashMap<String, String>())
332363
.build();
333364
mockSubscriber.addResponse(expectedResponse);
334365

@@ -369,6 +400,8 @@ public class SubscriberClientTest {
369400
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
370401
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
371402
.setPushConfig(PushConfig.newBuilder().build())
403+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
404+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
372405
.setAckDeadlineSeconds(2135351438)
373406
.setRetainAckedMessages(true)
374407
.putAllLabels(new HashMap<String, String>())
@@ -379,6 +412,10 @@ public class SubscriberClientTest {
379412
.setRetryPolicy(RetryPolicy.newBuilder().build())
380413
.setDetached(true)
381414
.setEnableExactlyOnceDelivery(true)
415+
.setAnalyticsHubSubscriptionInfo(
416+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
417+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
418+
.putAllTags(new HashMap<String, String>())
382419
.build();
383420
mockSubscriber.addResponse(expectedResponse);
384421

@@ -419,6 +456,8 @@ public class SubscriberClientTest {
419456
.setName(SubscriptionName.of("[PROJECT]", "[SUBSCRIPTION]").toString())
420457
.setTopic(TopicName.ofProjectTopicName("[PROJECT]", "[TOPIC]").toString())
421458
.setPushConfig(PushConfig.newBuilder().build())
459+
.setBigqueryConfig(BigQueryConfig.newBuilder().build())
460+
.setCloudStorageConfig(CloudStorageConfig.newBuilder().build())
422461
.setAckDeadlineSeconds(2135351438)
423462
.setRetainAckedMessages(true)
424463
.putAllLabels(new HashMap<String, String>())
@@ -429,23 +468,25 @@ public class SubscriberClientTest {
429468
.setRetryPolicy(RetryPolicy.newBuilder().build())
430469
.setDetached(true)
431470
.setEnableExactlyOnceDelivery(true)
471+
.setAnalyticsHubSubscriptionInfo(
472+
Subscription.AnalyticsHubSubscriptionInfo.newBuilder().build())
473+
.addAllMessageTransforms(new ArrayList<MessageTransform>())
474+
.putAllTags(new HashMap<String, String>())
432475
.build();
433476
mockSubscriber.addResponse(expectedResponse);
434477

435-
UpdateSubscriptionRequest request =
436-
UpdateSubscriptionRequest.newBuilder()
437-
.setSubscription(Subscription.newBuilder().build())
438-
.build();
478+
Subscription subscription = Subscription.newBuilder().build();
479+
FieldMask updateMask = FieldMask.newBuilder().build();
439480

440-
Subscription actualResponse = client.updateSubscription(request);
481+
Subscription actualResponse = client.updateSubscription(subscription, updateMask);
441482
Assert.assertEquals(expectedResponse, actualResponse);
442483

443484
List<AbstractMessage> actualRequests = mockSubscriber.getRequests();
444485
Assert.assertEquals(1, actualRequests.size());
445486
UpdateSubscriptionRequest actualRequest = ((UpdateSubscriptionRequest) actualRequests.get(0));
446487

447-
Assert.assertEquals(request.getSubscription(), actualRequest.getSubscription());
448-
Assert.assertEquals(request.getUpdateMask(), actualRequest.getUpdateMask());
488+
Assert.assertEquals(subscription, actualRequest.getSubscription());
489+
Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
449490
Assert.assertTrue(
450491
channelProvider.isHeaderSent(
451492
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -458,11 +499,9 @@ public class SubscriberClientTest {
458499
mockSubscriber.addException(exception);
459500

460501
try {
461-
UpdateSubscriptionRequest request =
462-
UpdateSubscriptionRequest.newBuilder()
463-
.setSubscription(Subscription.newBuilder().build())
464-
.build();
465-
client.updateSubscription(request);
502+
Subscription subscription = Subscription.newBuilder().build();
503+
FieldMask updateMask = FieldMask.newBuilder().build();
504+
client.updateSubscription(subscription, updateMask);
466505
Assert.fail("No exception raised");
467506
} catch (InvalidArgumentException e) {
468507
// Expected exception.
@@ -946,7 +985,7 @@ public class SubscriberClientTest {
946985
StreamingPullResponse expectedResponse =
947986
StreamingPullResponse.newBuilder()
948987
.addAllReceivedMessages(new ArrayList<ReceivedMessage>())
949-
.setAcknowlegeConfirmation(
988+
.setAcknowledgeConfirmation(
950989
StreamingPullResponse.AcknowledgeConfirmation.newBuilder().build())
951990
.setModifyAckDeadlineConfirmation(
952991
StreamingPullResponse.ModifyAckDeadlineConfirmation.newBuilder().build())
@@ -964,6 +1003,7 @@ public class SubscriberClientTest {
9641003
.setClientId("clientId908408390")
9651004
.setMaxOutstandingMessages(-1315266996)
9661005
.setMaxOutstandingBytes(-2103098517)
1006+
.setProtocolVersion(-1161610703)
9671007
.build();
9681008

9691009
MockStreamObserver<StreamingPullResponse> responseObserver = new MockStreamObserver<>();
@@ -995,6 +1035,7 @@ public class SubscriberClientTest {
9951035
.setClientId("clientId908408390")
9961036
.setMaxOutstandingMessages(-1315266996)
9971037
.setMaxOutstandingBytes(-2103098517)
1038+
.setProtocolVersion(-1161610703)
9981039
.build();
9991040

10001041
MockStreamObserver<StreamingPullResponse> responseObserver = new MockStreamObserver<>();
@@ -1440,18 +1481,18 @@ public class SubscriberClientTest {
14401481
.build();
14411482
mockSubscriber.addResponse(expectedResponse);
14421483

1443-
UpdateSnapshotRequest request =
1444-
UpdateSnapshotRequest.newBuilder().setSnapshot(Snapshot.newBuilder().build()).build();
1484+
Snapshot snapshot = Snapshot.newBuilder().build();
1485+
FieldMask updateMask = FieldMask.newBuilder().build();
14451486

1446-
Snapshot actualResponse = client.updateSnapshot(request);
1487+
Snapshot actualResponse = client.updateSnapshot(snapshot, updateMask);
14471488
Assert.assertEquals(expectedResponse, actualResponse);
14481489

14491490
List<AbstractMessage> actualRequests = mockSubscriber.getRequests();
14501491
Assert.assertEquals(1, actualRequests.size());
14511492
UpdateSnapshotRequest actualRequest = ((UpdateSnapshotRequest) actualRequests.get(0));
14521493

1453-
Assert.assertEquals(request.getSnapshot(), actualRequest.getSnapshot());
1454-
Assert.assertEquals(request.getUpdateMask(), actualRequest.getUpdateMask());
1494+
Assert.assertEquals(snapshot, actualRequest.getSnapshot());
1495+
Assert.assertEquals(updateMask, actualRequest.getUpdateMask());
14551496
Assert.assertTrue(
14561497
channelProvider.isHeaderSent(
14571498
ApiClientHeaderProvider.getDefaultApiClientHeaderKey(),
@@ -1464,9 +1505,9 @@ public class SubscriberClientTest {
14641505
mockSubscriber.addException(exception);
14651506

14661507
try {
1467-
UpdateSnapshotRequest request =
1468-
UpdateSnapshotRequest.newBuilder().setSnapshot(Snapshot.newBuilder().build()).build();
1469-
client.updateSnapshot(request);
1508+
Snapshot snapshot = Snapshot.newBuilder().build();
1509+
FieldMask updateMask = FieldMask.newBuilder().build();
1510+
client.updateSnapshot(snapshot, updateMask);
14701511
Assert.fail("No exception raised");
14711512
} catch (InvalidArgumentException e) {
14721513
// Expected exception.

0 commit comments

Comments
 (0)