Examples with NakadiCursor org.zalando.nakadi.domain.NakadiCursor used on opensource projects

Example 1 with NakadiCursor

use of org.zalando.nakadi.domain.NakadiCursor in project nakadi by zalando.

the class StreamingStateTest method ensureOffsetsSubscriptionsAreRefreshedAndClosed.

@Test
public void ensureOffsetsSubscriptionsAreRefreshedAndClosed() throws InternalNakadiException, NoSuchEventTypeException, ServiceUnavailableException, InvalidCursorException {
    final ZkSubscription<SubscriptionCursorWithoutToken> offsetSubscription = mock(ZkSubscription.class);
    final EventTypePartition pk = new EventTypePartition("t", "0");
    Mockito.when(zkMock.subscribeForOffsetChanges(Mockito.eq(pk), Mockito.any())).thenReturn(offsetSubscription);
    final EventConsumer.ReassignableEventConsumer consumer = mock(EventConsumer.ReassignableEventConsumer.class);
    when(consumer.getAssignment()).thenReturn(Collections.emptySet());
    when(timelineService.createEventConsumer(any())).thenReturn(consumer);
    when(subscription.getEventTypes()).thenReturn(Collections.singleton("t"));
    final Storage storage = mock(Storage.class);
    when(storage.getType()).thenReturn(Storage.Type.KAFKA);
    final Timeline timeline = new Timeline("t", 0, storage, "t", new Date());
    when(timelineService.getActiveTimelinesOrdered(eq("t"))).thenReturn(Collections.singletonList(timeline));
    final TopicRepository topicRepository = mock(TopicRepository.class);
    when(timelineService.getTopicRepository(eq(timeline))).thenReturn(topicRepository);
    final PartitionStatistics stats = mock(PartitionStatistics.class);
    final NakadiCursor beforeFirstCursor = NakadiCursor.of(timeline, "0", "0");
    when(stats.getBeforeFirst()).thenReturn(beforeFirstCursor);
    when(topicRepository.loadTopicStatistics(any())).thenReturn(Lists.newArrayList(stats));
    state.onEnter();
    final NakadiCursor anyCursor = NakadiCursor.of(timeline, "0", "0");
    when(cursorConverter.convert(any(SubscriptionCursorWithoutToken.class))).thenReturn(anyCursor);
    state.refreshTopologyUnlocked(new Partition[] { new Partition(pk.getEventType(), pk.getPartition(), SESSION_ID, null, Partition.State.ASSIGNED) });
    Mockito.verify(zkMock, Mockito.times(1)).subscribeForOffsetChanges(Mockito.eq(pk), Mockito.any());
    Mockito.verify(offsetSubscription, Mockito.times(0)).close();
    Mockito.verify(offsetSubscription, Mockito.times(0)).getData();
    state.offsetChanged(pk);
    Mockito.verify(zkMock, Mockito.times(1)).subscribeForOffsetChanges(Mockito.eq(pk), Mockito.any());
    Mockito.verify(offsetSubscription, Mockito.times(0)).close();
    Mockito.verify(offsetSubscription, Mockito.times(1)).getData();
    // Verify that offset change listener is removed
    state.refreshTopologyUnlocked(new Partition[0]);
    Mockito.verify(zkMock, Mockito.times(1)).subscribeForOffsetChanges(Mockito.eq(pk), Mockito.any());
    Mockito.verify(offsetSubscription, Mockito.times(1)).close();
    Mockito.verify(offsetSubscription, Mockito.times(1)).getData();
}

Example 2 with NakadiCursor

use of org.zalando.nakadi.domain.NakadiCursor in project nakadi by zalando.

the class StartingStateTest method testGetSubscriptionOffsetsBegin.

@Test
public void testGetSubscriptionOffsetsBegin() throws Exception {
    when(subscription.getReadFrom()).thenReturn(SubscriptionBase.InitialPosition.BEGIN);
    final NakadiCursor beforeBegin0 = mock(NakadiCursor.class);
    final SubscriptionCursorWithoutToken beforeBegin0Converted = mock(SubscriptionCursorWithoutToken.class);
    when(cursorConverter.convertToNoToken(eq(beforeBegin0))).thenReturn(beforeBegin0Converted);
    final NakadiCursor beforeBegin1 = mock(NakadiCursor.class);
    final SubscriptionCursorWithoutToken beforeBegin1Converted = mock(SubscriptionCursorWithoutToken.class);
    when(cursorConverter.convertToNoToken(eq(beforeBegin1))).thenReturn(beforeBegin1Converted);
    final TopicRepository topicRepository = mock(TopicRepository.class);
    final PartitionStatistics resultForTopic0 = mock(PartitionStatistics.class);
    when(resultForTopic0.getBeforeFirst()).thenReturn(beforeBegin0);
    final PartitionStatistics resultForTopic1 = mock(PartitionStatistics.class);
    when(resultForTopic1.getBeforeFirst()).thenReturn(beforeBegin1);
    when(topicRepository.loadTopicStatistics(any())).thenReturn(Lists.newArrayList(resultForTopic0, resultForTopic1));
    when(timelineService.getTopicRepository(eq(timelineEt00))).thenReturn(topicRepository);
    final Storage storage = mock(Storage.class);
    when(timelineEt00.getStorage()).thenReturn(storage);
    when(timelineEt10.getStorage()).thenReturn(storage);
    final List<SubscriptionCursorWithoutToken> cursors = StartingState.calculateStartPosition(subscription, timelineService, cursorConverter);
    Assert.assertEquals(cursors.size(), 2);
    Assert.assertEquals(beforeBegin0Converted, cursors.get(0));
    Assert.assertEquals(beforeBegin1Converted, cursors.get(1));
}

Example 3 with NakadiCursor

use of org.zalando.nakadi.domain.NakadiCursor in project nakadi by zalando.

the class StartingStateTest method testGetSubscriptionOffsetsEnd.

@Test
public void testGetSubscriptionOffsetsEnd() throws Exception {
    when(subscription.getReadFrom()).thenReturn(SubscriptionBase.InitialPosition.END);
    final NakadiCursor end0 = mock(NakadiCursor.class);
    final SubscriptionCursorWithoutToken end0Converted = mock(SubscriptionCursorWithoutToken.class);
    when(cursorConverter.convertToNoToken(eq(end0))).thenReturn(end0Converted);
    final NakadiCursor end1 = mock(NakadiCursor.class);
    final SubscriptionCursorWithoutToken end1Converted = mock(SubscriptionCursorWithoutToken.class);
    when(cursorConverter.convertToNoToken(eq(end1))).thenReturn(end1Converted);
    final TopicRepository topicRepository = mock(TopicRepository.class);
    final PartitionStatistics statsForEt0 = mock(PartitionStatistics.class);
    when(statsForEt0.getLast()).thenReturn(end0);
    final PartitionStatistics statsForTopic1 = mock(PartitionStatistics.class);
    when(statsForTopic1.getLast()).thenReturn(end1);
    when(topicRepository.loadTopicEndStatistics(any())).thenReturn(Lists.newArrayList(statsForEt0, statsForTopic1));
    when(timelineService.getTopicRepository(eq(timelineEt01))).thenReturn(topicRepository);
    final Storage storage = mock(Storage.class);
    when(timelineEt01.getStorage()).thenReturn(storage);
    when(timelineEt11.getStorage()).thenReturn(storage);
    final List<SubscriptionCursorWithoutToken> cursors = StartingState.calculateStartPosition(subscription, timelineService, cursorConverter);
    Assert.assertEquals(cursors.size(), 2);
    Assert.assertEquals(end0Converted, cursors.get(0));
    Assert.assertEquals(end1Converted, cursors.get(1));
}

Example 4 with NakadiCursor

use of org.zalando.nakadi.domain.NakadiCursor in project nakadi by zalando.

the class SubscriptionValidationServiceTest method whenInvalidCursorThenException.

@Test(expected = WrongInitialCursorsException.class)
public void whenInvalidCursorThenException() throws Exception {
    subscriptionBase.setInitialCursors(ImmutableList.of(new SubscriptionCursorWithoutToken(ET1, P0, ""), new SubscriptionCursorWithoutToken(ET2, P0, ""), new SubscriptionCursorWithoutToken(ET3, P0, "")));
    final NakadiCursor cursor = mockCursorWithTimeline();
    when(cursorConverter.convert((SubscriptionCursorWithoutToken) any())).thenReturn(cursor);
    doThrow(new InvalidCursorException(CursorError.INVALID_FORMAT)).when(topicRepository).validateReadCursors(any());
    subscriptionValidationService.validateSubscription(subscriptionBase);
}

Example 5 with NakadiCursor

use of org.zalando.nakadi.domain.NakadiCursor in project nakadi by zalando.

the class SubscriptionValidationServiceTest method whenServiceUnavailableThenException.

@Test(expected = RepositoryProblemException.class)
public void whenServiceUnavailableThenException() throws Exception {
    subscriptionBase.setInitialCursors(ImmutableList.of(new SubscriptionCursorWithoutToken(ET1, P0, ""), new SubscriptionCursorWithoutToken(ET2, P0, ""), new SubscriptionCursorWithoutToken(ET3, P0, "")));
    final NakadiCursor cursor = mockCursorWithTimeline();
    when(cursorConverter.convert((SubscriptionCursorWithoutToken) any())).thenReturn(cursor);
    doThrow(new ServiceUnavailableException("")).when(topicRepository).validateReadCursors(any());
    subscriptionValidationService.validateSubscription(subscriptionBase);
}