Search in sources :

Example 1 with TopicSubscriptionBuilderImpl

use of io.zeebe.client.event.impl.TopicSubscriptionBuilderImpl in project zeebe by zeebe-io.

the class PartitionedTopicSubscriptionTest method shouldCloseSuccessfulSubscriptionIfAnotherSubscriptionCannotBeOpened.

@Test
public void shouldCloseSuccessfulSubscriptionIfAnotherSubscriptionCannotBeOpened() throws Exception {
    // given
    final int subscriberKey1 = 456;
    broker1.stubTopicSubscriptionApi(subscriberKey1);
    final ResponseController responseController = broker2.onExecuteCommandRequest(EventType.SUBSCRIBER_EVENT, "SUBSCRIBE").respondWithError().errorCode(ErrorCode.REQUEST_PROCESSING_FAILURE).errorData("foo").registerControlled();
    // assuming that subscription to broker 1 is successful
    final TopicSubscriptionBuilderImpl builder = (TopicSubscriptionBuilderImpl) client.topics().newSubscription(TOPIC).handler(new RecordingEventHandler()).name("hohoho");
    final Future<TopicSubscriberGroup> groupFuture = builder.buildSubscriberGroup();
    waitUntil(() -> getOpenSubscriptionRequests(broker1).size() == 1);
    // when
    // triggering the error response and continuing
    responseController.unblockNextResponse();
    // then
    waitUntil(() -> groupFuture.isDone());
    assertFailed(groupFuture);
    final List<ControlMessageRequest> closeRequestsBroker1 = getCloseSubscriptionRequests(broker1);
    assertThat(closeRequestsBroker1).hasSize(1);
    assertThat(closeRequestsBroker1.get(0).getData().get("subscriberKey")).isEqualTo(subscriberKey1);
}
Also used : ControlMessageRequest(io.zeebe.test.broker.protocol.brokerapi.ControlMessageRequest) TopicSubscriberGroup(io.zeebe.client.event.impl.TopicSubscriberGroup) TopicSubscriptionBuilderImpl(io.zeebe.client.event.impl.TopicSubscriptionBuilderImpl) ResponseController(io.zeebe.test.broker.protocol.brokerapi.ResponseController) Test(org.junit.Test)

Example 2 with TopicSubscriptionBuilderImpl

use of io.zeebe.client.event.impl.TopicSubscriptionBuilderImpl in project zeebe by zeebe-io.

the class TopicSubscriptionTest method shouldCloseSubscriptionWhileOpeningSubscriber.

@Test
public void shouldCloseSubscriptionWhileOpeningSubscriber() {
    // given
    final int subscriberKey = 123;
    broker.stubTopicSubscriptionApi(0L);
    final ResponseController responseController = broker.onExecuteCommandRequest(EventType.SUBSCRIBER_EVENT, "SUBSCRIBE").respondWith().key(subscriberKey).event().allOf((r) -> r.getCommand()).put("state", "SUBSCRIBED").done().registerControlled();
    final TopicSubscriptionBuilderImpl builder = (TopicSubscriptionBuilderImpl) client.topics().newSubscription(clientRule.getDefaultTopicName()).handler(DO_NOTHING).name("foo");
    final Future<TopicSubscriberGroup> future = builder.buildSubscriberGroup();
    waitUntil(() -> broker.getReceivedCommandRequests().stream().filter(r -> r.eventType() == EventType.SUBSCRIBER_EVENT && "SUBSCRIBE".equals(r.getCommand().get("state"))).count() == 1);
    final Thread closingThread = new Thread(client::close);
    closingThread.start();
    final SubscriptionManager subscriptionManager = ((ZeebeClientImpl) client).getSubscriptionManager();
    waitUntil(() -> subscriptionManager.isClosing());
    // when
    responseController.unblockNextResponse();
    // then
    waitUntil(() -> future.isDone());
    assertThat(future).isDone();
    final Optional<ControlMessageRequest> closeRequest = broker.getReceivedControlMessageRequests().stream().filter(c -> c.messageType() == ControlMessageType.REMOVE_TOPIC_SUBSCRIPTION).findFirst();
    assertThat(closeRequest).isPresent();
    final ControlMessageRequest request = closeRequest.get();
    assertThat(request.getData()).containsEntry("subscriberKey", subscriberKey);
}
Also used : SubscriptionManager(io.zeebe.client.task.impl.subscription.SubscriptionManager) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) ControlMessageType(io.zeebe.protocol.clientapi.ControlMessageType) TimeoutException(java.util.concurrent.TimeoutException) ZeebeClientImpl(io.zeebe.client.impl.ZeebeClientImpl) ControlMessageRequest(io.zeebe.test.broker.protocol.brokerapi.ControlMessageRequest) ZeebeClient(io.zeebe.client.ZeebeClient) StubBrokerRule(io.zeebe.test.broker.protocol.brokerapi.StubBrokerRule) Future(java.util.concurrent.Future) ErrorCode(io.zeebe.protocol.clientapi.ErrorCode) RemoteAddress(io.zeebe.transport.RemoteAddress) ClientRule(io.zeebe.client.util.ClientRule) TopicSubscriberGroup(io.zeebe.client.event.impl.TopicSubscriberGroup) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) Duration(java.time.Duration) ExpectedException(org.junit.rules.ExpectedException) Before(org.junit.Before) TopicSubscriptionBuilderImpl(io.zeebe.client.event.impl.TopicSubscriptionBuilderImpl) TestUtil.waitUntil(io.zeebe.test.util.TestUtil.waitUntil) Set(java.util.Set) Test(org.junit.Test) Collectors(java.util.stream.Collectors) ResponseController(io.zeebe.test.broker.protocol.brokerapi.ResponseController) ExecutionException(java.util.concurrent.ExecutionException) TimeUnit(java.util.concurrent.TimeUnit) RuleChain(org.junit.rules.RuleChain) ActorFuture(io.zeebe.util.sched.future.ActorFuture) List(java.util.List) Stream(java.util.stream.Stream) Rule(org.junit.Rule) EventType(io.zeebe.protocol.clientapi.EventType) Conditions(io.zeebe.test.util.Conditions) Optional(java.util.Optional) ExecuteCommandRequest(io.zeebe.test.broker.protocol.brokerapi.ExecuteCommandRequest) CopyOnWriteArrayList(java.util.concurrent.CopyOnWriteArrayList) TestUtil(io.zeebe.test.util.TestUtil) ControlMessageRequest(io.zeebe.test.broker.protocol.brokerapi.ControlMessageRequest) TopicSubscriberGroup(io.zeebe.client.event.impl.TopicSubscriberGroup) TopicSubscriptionBuilderImpl(io.zeebe.client.event.impl.TopicSubscriptionBuilderImpl) SubscriptionManager(io.zeebe.client.task.impl.subscription.SubscriptionManager) ZeebeClientImpl(io.zeebe.client.impl.ZeebeClientImpl) ResponseController(io.zeebe.test.broker.protocol.brokerapi.ResponseController) Test(org.junit.Test)

Aggregations

TopicSubscriberGroup (io.zeebe.client.event.impl.TopicSubscriberGroup)2 TopicSubscriptionBuilderImpl (io.zeebe.client.event.impl.TopicSubscriptionBuilderImpl)2 ControlMessageRequest (io.zeebe.test.broker.protocol.brokerapi.ControlMessageRequest)2 ResponseController (io.zeebe.test.broker.protocol.brokerapi.ResponseController)2 Test (org.junit.Test)2 ZeebeClient (io.zeebe.client.ZeebeClient)1 ZeebeClientImpl (io.zeebe.client.impl.ZeebeClientImpl)1 SubscriptionManager (io.zeebe.client.task.impl.subscription.SubscriptionManager)1 ClientRule (io.zeebe.client.util.ClientRule)1 ControlMessageType (io.zeebe.protocol.clientapi.ControlMessageType)1 ErrorCode (io.zeebe.protocol.clientapi.ErrorCode)1 EventType (io.zeebe.protocol.clientapi.EventType)1 ExecuteCommandRequest (io.zeebe.test.broker.protocol.brokerapi.ExecuteCommandRequest)1 StubBrokerRule (io.zeebe.test.broker.protocol.brokerapi.StubBrokerRule)1 Conditions (io.zeebe.test.util.Conditions)1 TestUtil (io.zeebe.test.util.TestUtil)1 TestUtil.waitUntil (io.zeebe.test.util.TestUtil.waitUntil)1 RemoteAddress (io.zeebe.transport.RemoteAddress)1 ActorFuture (io.zeebe.util.sched.future.ActorFuture)1 Duration (java.time.Duration)1