@@ -17,6 +17,7 @@ import com.google.api.gax.rpc.StatusCode;
1717import com.google.common.collect.Lists;
1818import com.google.protobuf.AbstractMessage;
1919import com.google.protobuf.Empty;
20+ import com.google.protobuf.FieldMask;
2021import google.cloud.ProjectName;
2122import io.grpc.StatusRuntimeException;
2223import 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