Search in sources :

Example 21 with Subscription

use of com.google.pubsub.v1.Subscription in project google-cloud-java by GoogleCloudPlatform.

the class SubscriptionAdminClientTest method createSnapshotExceptionTest.

@Test
@SuppressWarnings("all")
public void createSnapshotExceptionTest() throws Exception {
    StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
    mockSubscriber.addException(exception);
    try {
        SnapshotName name = SnapshotName.create("[PROJECT]", "[SNAPSHOT]");
        SubscriptionName subscription = SubscriptionName.create("[PROJECT]", "[SUBSCRIPTION]");
        client.createSnapshot(name, subscription);
        Assert.fail("No exception raised");
    } catch (ApiException e) {
        Assert.assertEquals(Status.INVALID_ARGUMENT.getCode(), e.getStatusCode());
    }
}
Also used : StatusRuntimeException(io.grpc.StatusRuntimeException) SubscriptionName(com.google.pubsub.v1.SubscriptionName) SnapshotName(com.google.pubsub.v1.SnapshotName) ApiException(com.google.api.gax.grpc.ApiException) Test(org.junit.Test)

Example 22 with Subscription

use of com.google.pubsub.v1.Subscription in project google-cloud-java by GoogleCloudPlatform.

the class SubscriptionAdminClientTest method modifyAckDeadlineExceptionTest.

@Test
@SuppressWarnings("all")
public void modifyAckDeadlineExceptionTest() throws Exception {
    StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT);
    mockSubscriber.addException(exception);
    try {
        SubscriptionName subscription = SubscriptionName.create("[PROJECT]", "[SUBSCRIPTION]");
        List<String> ackIds = new ArrayList<>();
        int ackDeadlineSeconds = 2135351438;
        client.modifyAckDeadline(subscription, ackIds, ackDeadlineSeconds);
        Assert.fail("No exception raised");
    } catch (ApiException e) {
        Assert.assertEquals(Status.INVALID_ARGUMENT.getCode(), e.getStatusCode());
    }
}
Also used : StatusRuntimeException(io.grpc.StatusRuntimeException) ArrayList(java.util.ArrayList) SubscriptionName(com.google.pubsub.v1.SubscriptionName) ByteString(com.google.protobuf.ByteString) ApiException(com.google.api.gax.grpc.ApiException) Test(org.junit.Test)

Example 23 with Subscription

use of com.google.pubsub.v1.Subscription in project google-cloud-java by GoogleCloudPlatform.

the class SubscriptionAdminClientTest method acknowledgeTest.

@Test
@SuppressWarnings("all")
public void acknowledgeTest() {
    Empty expectedResponse = Empty.newBuilder().build();
    mockSubscriber.addResponse(expectedResponse);
    SubscriptionName subscription = SubscriptionName.create("[PROJECT]", "[SUBSCRIPTION]");
    List<String> ackIds = new ArrayList<>();
    client.acknowledge(subscription, ackIds);
    List<GeneratedMessageV3> actualRequests = mockSubscriber.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    AcknowledgeRequest actualRequest = (AcknowledgeRequest) actualRequests.get(0);
    Assert.assertEquals(subscription, actualRequest.getSubscriptionAsSubscriptionName());
    Assert.assertEquals(ackIds, actualRequest.getAckIdsList());
}
Also used : Empty(com.google.protobuf.Empty) AcknowledgeRequest(com.google.pubsub.v1.AcknowledgeRequest) ArrayList(java.util.ArrayList) SubscriptionName(com.google.pubsub.v1.SubscriptionName) ByteString(com.google.protobuf.ByteString) GeneratedMessageV3(com.google.protobuf.GeneratedMessageV3) Test(org.junit.Test)

Example 24 with Subscription

use of com.google.pubsub.v1.Subscription in project google-cloud-java by GoogleCloudPlatform.

the class ITSubscriptionAdminClientSnippets method createSubscription.

private Subscription createSubscription(String topicName, String subscriptionName) throws Exception {
    createTopic(topicName);
    Subscription subscription = subscriptionAdminClientSnippets.createSubscription(topicName, subscriptionName);
    assertNotNull(subscription);
    Subscription retrievedSubscription = subscriptionAdminClientSnippets.getSubscription(subscriptionName);
    assertNotNull(retrievedSubscription);
    assertEquals(subscription.getName(), retrievedSubscription.getName());
    return subscription;
}
Also used : Subscription(com.google.pubsub.v1.Subscription)

Example 25 with Subscription

use of com.google.pubsub.v1.Subscription in project google-cloud-java by GoogleCloudPlatform.

the class ITSubscriptionAdminClientSnippets method listSubscriptionsRetrievesAllAddedSubscriptions.

@Test
public void listSubscriptionsRetrievesAllAddedSubscriptions() throws Exception {
    List<Subscription> addedSubscriptions = new ArrayList<>();
    String topicName1 = topics[0];
    String subscriptionName1 = subscriptions[0];
    String topicName2 = topics[1];
    String subscriptionName2 = subscriptions[1];
    addedSubscriptions.add(createSubscription(topicName1, subscriptionName1));
    addedSubscriptions.add(createSubscription(topicName2, subscriptionName2));
    boolean[] subFound = { false, false };
    ListSubscriptionsPagedResponse response = subscriptionAdminClientSnippets.listSubscriptions();
    assertNotNull(response);
    Iterable<Subscription> subscriptions = response.iterateAll();
    for (int i = 0; i < 2; i++) {
        if (!subFound[i]) {
            subFound[i] = Iterables.contains(subscriptions, addedSubscriptions.get(i));
        }
    }
    assertTrue(subFound[0] && subFound[1]);
}
Also used : ListSubscriptionsPagedResponse(com.google.cloud.pubsub.spi.v1.PagedResponseWrappers.ListSubscriptionsPagedResponse) ArrayList(java.util.ArrayList) ByteString(com.google.protobuf.ByteString) Subscription(com.google.pubsub.v1.Subscription) Test(org.junit.Test)

Aggregations

SubscriptionName (com.google.pubsub.v1.SubscriptionName)31 Test (org.junit.Test)26 Subscription (com.google.pubsub.v1.Subscription)14 ByteString (com.google.protobuf.ByteString)11 SubscriptionAdminClient (com.google.cloud.pubsub.spi.v1.SubscriptionAdminClient)10 GeneratedMessageV3 (com.google.protobuf.GeneratedMessageV3)10 TopicName (com.google.pubsub.v1.TopicName)10 StatusRuntimeException (io.grpc.StatusRuntimeException)9 ApiException (com.google.api.gax.grpc.ApiException)8 ArrayList (java.util.ArrayList)8 PushConfig (com.google.pubsub.v1.PushConfig)6 PubsubMessage (com.google.pubsub.v1.PubsubMessage)5 Empty (com.google.protobuf.Empty)4 PullResponse (com.google.pubsub.v1.PullResponse)4 AckReplyConsumer (com.google.cloud.pubsub.spi.v1.AckReplyConsumer)3 MessageReceiver (com.google.cloud.pubsub.spi.v1.MessageReceiver)3 ListSubscriptionsPagedResponse (com.google.cloud.pubsub.spi.v1.PagedResponseWrappers.ListSubscriptionsPagedResponse)3 Subscriber (com.google.cloud.pubsub.spi.v1.Subscriber)3 TopicAdminClient (com.google.cloud.pubsub.spi.v1.TopicAdminClient)3 AcknowledgeRequest (com.google.pubsub.v1.AcknowledgeRequest)3