Search in sources :

Example 1 with PullRequest

use of com.vmware.connectors.github.pr.v3.PullRequest in project google-cloud-java by GoogleCloudPlatform.

the class SubscriptionAdminClientTest method pullTest.

@Test
@SuppressWarnings("all")
public void pullTest() {
    PullResponse expectedResponse = PullResponse.newBuilder().build();
    mockSubscriber.addResponse(expectedResponse);
    SubscriptionName subscription = SubscriptionName.create("[PROJECT]", "[SUBSCRIPTION]");
    boolean returnImmediately = false;
    int maxMessages = 496131527;
    PullResponse actualResponse = client.pull(subscription, returnImmediately, maxMessages);
    Assert.assertEquals(expectedResponse, actualResponse);
    List<GeneratedMessageV3> actualRequests = mockSubscriber.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    PullRequest actualRequest = (PullRequest) actualRequests.get(0);
    Assert.assertEquals(subscription, actualRequest.getSubscriptionAsSubscriptionName());
    Assert.assertEquals(returnImmediately, actualRequest.getReturnImmediately());
    Assert.assertEquals(maxMessages, actualRequest.getMaxMessages());
}
Also used : PullResponse(com.google.pubsub.v1.PullResponse) StreamingPullResponse(com.google.pubsub.v1.StreamingPullResponse) StreamingPullRequest(com.google.pubsub.v1.StreamingPullRequest) PullRequest(com.google.pubsub.v1.PullRequest) SubscriptionName(com.google.pubsub.v1.SubscriptionName) GeneratedMessageV3(com.google.protobuf.GeneratedMessageV3) Test(org.junit.Test)

Example 2 with PullRequest

use of com.vmware.connectors.github.pr.v3.PullRequest in project connectors-workspace-one by vmware.

the class GithubPrController method makeCard.

private Card makeCard(String routingPrefix, Pair<PullRequestId, PullRequest> info, Locale locale, HttpServletRequest request) {
    logger.trace("makeCard called: routingPrefix={}, info={}", routingPrefix, info);
    PullRequestId pullRequestId = info.getLeft();
    PullRequest pullRequest = info.getRight();
    boolean isOpen = OPEN_STATE.equalsIgnoreCase(pullRequest.getState());
    Card.Builder card = new Card.Builder().setName(// TODO - remove this in APF-536
    "GithubPr").setTemplate(routingPrefix + "templates/generic.hbs").setHeader(cardTextAccessor.getHeader(locale), cardTextAccessor.getMessage("subtitle", locale, pullRequestId.getOwner(), pullRequestId.getRepo(), pullRequestId.getNumber())).setBody(createBody(pullRequestId, pullRequest, locale));
    // Set image url.
    CommonUtils.buildConnectorImageUrl(card, request);
    addCloseAction(card, routingPrefix, pullRequestId, isOpen, locale);
    addMergeAction(card, routingPrefix, pullRequestId, pullRequest, isOpen, locale);
    addApproveAction(card, routingPrefix, pullRequestId, isOpen, locale);
    addCommentAction(card, routingPrefix, pullRequestId, locale);
    addRequestChangesAction(card, routingPrefix, pullRequestId, isOpen, locale);
    return card.build();
}
Also used : PullRequest(com.vmware.connectors.github.pr.v3.PullRequest) UriComponentsBuilder(org.springframework.web.util.UriComponentsBuilder)

Example 3 with PullRequest

use of com.vmware.connectors.github.pr.v3.PullRequest in project flink by apache.

the class DefaultPubSubSubscriberFactory method getSubscriber.

@Override
public PubSubSubscriber getSubscriber(Credentials credentials) throws IOException {
    ManagedChannel channel = NettyChannelBuilder.forTarget(SubscriberStubSettings.getDefaultEndpoint()).negotiationType(NegotiationType.TLS).sslContext(GrpcSslContexts.forClient().ciphers(null).build()).build();
    PullRequest pullRequest = PullRequest.newBuilder().setMaxMessages(maxMessagesPerPull).setSubscription(projectSubscriptionName).build();
    SubscriberGrpc.SubscriberBlockingStub stub = SubscriberGrpc.newBlockingStub(channel).withCallCredentials(MoreCallCredentials.from(credentials));
    return new BlockingGrpcPubSubSubscriber(projectSubscriptionName, channel, stub, pullRequest, retries, timeout);
}
Also used : PullRequest(com.google.pubsub.v1.PullRequest) ManagedChannel(io.grpc.ManagedChannel) SubscriberGrpc(com.google.pubsub.v1.SubscriberGrpc)

Example 4 with PullRequest

use of com.vmware.connectors.github.pr.v3.PullRequest in project flink by apache.

the class PubSubSubscriberFactoryForEmulator method getSubscriber.

@Override
public PubSubSubscriber getSubscriber(Credentials credentials) throws IOException {
    ManagedChannel managedChannel = NettyChannelBuilder.forTarget(hostAndPort).usePlaintext().build();
    PullRequest pullRequest = PullRequest.newBuilder().setMaxMessages(maxMessagesPerPull).setSubscription(projectSubscriptionName).build();
    SubscriberGrpc.SubscriberBlockingStub stub = SubscriberGrpc.newBlockingStub(managedChannel);
    return new BlockingGrpcPubSubSubscriber(projectSubscriptionName, managedChannel, stub, pullRequest, retries, timeout);
}
Also used : BlockingGrpcPubSubSubscriber(org.apache.flink.streaming.connectors.gcp.pubsub.BlockingGrpcPubSubSubscriber) PullRequest(com.google.pubsub.v1.PullRequest) ManagedChannel(io.grpc.ManagedChannel) SubscriberGrpc(com.google.pubsub.v1.SubscriberGrpc)

Example 5 with PullRequest

use of com.vmware.connectors.github.pr.v3.PullRequest in project beam by apache.

the class PubsubGrpcClientTest method pullOneMessage.

@Test
public void pullOneMessage() throws IOException {
    initializeClient(null, null);
    String expectedSubscription = SUBSCRIPTION.getPath();
    final PullRequest expectedRequest = PullRequest.newBuilder().setSubscription(expectedSubscription).setReturnImmediately(true).setMaxMessages(10).build();
    Timestamp timestamp = Timestamp.newBuilder().setSeconds(PUB_TIME_MS / 1000).setNanos((int) (PUB_TIME_MS % 1000) * 1000 * 1000).build();
    PubsubMessage expectedPubsubMessage = PubsubMessage.newBuilder().setMessageId(MESSAGE_ID).setData(ByteString.copyFrom(DATA.getBytes(StandardCharsets.UTF_8))).setPublishTime(timestamp).build();
    ReceivedMessage expectedReceivedMessage = ReceivedMessage.newBuilder().setMessage(expectedPubsubMessage).setAckId(ACK_ID).build();
    final PullResponse response = PullResponse.newBuilder().addAllReceivedMessages(ImmutableList.of(expectedReceivedMessage)).build();
    final List<PullRequest> requestsReceived = new ArrayList<>();
    SubscriberImplBase subscriberImplBase = new SubscriberImplBase() {

        @Override
        public void pull(PullRequest request, StreamObserver<PullResponse> responseObserver) {
            requestsReceived.add(request);
            responseObserver.onNext(response);
            responseObserver.onCompleted();
        }
    };
    Server server = InProcessServerBuilder.forName(channelName).addService(subscriberImplBase).build().start();
    try {
        List<IncomingMessage> actualMessages = client.pull(REQ_TIME_MS, SUBSCRIPTION, 10, true);
        assertEquals(1, actualMessages.size());
        IncomingMessage actualMessage = actualMessages.get(0);
        assertEquals(ACK_ID, actualMessage.ackId());
        assertEquals(DATA, actualMessage.message().getData().toStringUtf8());
        assertEquals(MESSAGE_ID, actualMessage.recordId());
        assertEquals(REQ_TIME_MS, actualMessage.requestTimeMsSinceEpoch());
        assertEquals(PUB_TIME_MS, actualMessage.timestampMsSinceEpoch());
        assertEquals(expectedRequest, Iterables.getOnlyElement(requestsReceived));
    } finally {
        server.shutdownNow();
    }
}
Also used : StreamObserver(io.grpc.stub.StreamObserver) SubscriberImplBase(com.google.pubsub.v1.SubscriberGrpc.SubscriberImplBase) Server(io.grpc.Server) PullRequest(com.google.pubsub.v1.PullRequest) ArrayList(java.util.ArrayList) ByteString(com.google.protobuf.ByteString) ReceivedMessage(com.google.pubsub.v1.ReceivedMessage) Timestamp(com.google.protobuf.Timestamp) PubsubMessage(com.google.pubsub.v1.PubsubMessage) PullResponse(com.google.pubsub.v1.PullResponse) IncomingMessage(org.apache.beam.sdk.io.gcp.pubsub.PubsubClient.IncomingMessage) Test(org.junit.Test)

Aggregations

PullRequest (com.google.pubsub.v1.PullRequest)11 PullResponse (com.google.pubsub.v1.PullResponse)5 Test (org.junit.Test)5 PubsubMessage (com.google.pubsub.v1.PubsubMessage)4 ReceivedMessage (com.google.pubsub.v1.ReceivedMessage)4 Timestamp (com.google.protobuf.Timestamp)3 ArrayList (java.util.ArrayList)3 ByteString (com.google.protobuf.ByteString)2 SubscriberGrpc (com.google.pubsub.v1.SubscriberGrpc)2 SubscriberImplBase (com.google.pubsub.v1.SubscriberGrpc.SubscriberImplBase)2 ManagedChannel (io.grpc.ManagedChannel)2 Server (io.grpc.Server)2 StreamObserver (io.grpc.stub.StreamObserver)2 List (java.util.List)2 IncomingMessage (org.apache.beam.sdk.io.gcp.pubsub.PubsubClient.IncomingMessage)2 AcknowledgeablePubsubMessage (org.springframework.cloud.gcp.pubsub.support.AcknowledgeablePubsubMessage)2 BasicAcknowledgeablePubsubMessage (org.springframework.cloud.gcp.pubsub.support.BasicAcknowledgeablePubsubMessage)2 ConvertedAcknowledgeablePubsubMessage (org.springframework.cloud.gcp.pubsub.support.converter.ConvertedAcknowledgeablePubsubMessage)2 ConvertedBasicAcknowledgeablePubsubMessage (org.springframework.cloud.gcp.pubsub.support.converter.ConvertedBasicAcknowledgeablePubsubMessage)2 ServiceOptions (com.google.cloud.ServiceOptions)1