use of org.apache.flink.streaming.api.windowing.assigners.SessionWindowTimeGapExtractor in project flink by apache.
the class DynamicEventTimeSessionWindowsTest method testInvalidParameters.
@Test
public void testInvalidParameters() {
WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class);
try {
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(-1L);
DynamicEventTimeSessionWindows assigner = DynamicEventTimeSessionWindows.withDynamicGap(extractor);
assigner.assignWindows(Lists.newArrayList(new Object()), 1, mockContext);
fail("should fail");
} catch (IllegalArgumentException e) {
assertThat(e.toString(), containsString("0 < gap"));
}
try {
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(0L);
DynamicEventTimeSessionWindows assigner = DynamicEventTimeSessionWindows.withDynamicGap(extractor);
assigner.assignWindows(Lists.newArrayList(new Object()), 1, mockContext);
fail("should fail");
} catch (IllegalArgumentException e) {
assertThat(e.toString(), containsString("0 < gap"));
}
}
use of org.apache.flink.streaming.api.windowing.assigners.SessionWindowTimeGapExtractor in project flink by apache.
the class DynamicEventTimeSessionWindowsTest method testMergeCoveringWindow.
@Test
public void testMergeCoveringWindow() {
MergingWindowAssigner.MergeCallback callback = mock(MergingWindowAssigner.MergeCallback.class);
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(5000L);
DynamicEventTimeSessionWindows assigner = DynamicEventTimeSessionWindows.withDynamicGap(extractor);
assigner.mergeWindows(Lists.newArrayList(new TimeWindow(1, 1), new TimeWindow(0, 2), new TimeWindow(4, 7), new TimeWindow(5, 6)), callback);
verify(callback, times(1)).merge((Collection<TimeWindow>) argThat(containsInAnyOrder(new TimeWindow(1, 1), new TimeWindow(0, 2))), eq(new TimeWindow(0, 2)));
verify(callback, times(1)).merge((Collection<TimeWindow>) argThat(containsInAnyOrder(new TimeWindow(5, 6), new TimeWindow(4, 7))), eq(new TimeWindow(4, 7)));
verify(callback, times(2)).merge(anyCollection(), Matchers.anyObject());
}
use of org.apache.flink.streaming.api.windowing.assigners.SessionWindowTimeGapExtractor in project flink by apache.
the class DynamicProcessingTimeSessionWindowsTest method testProperties.
@Test
public void testProperties() {
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(5000L);
DynamicProcessingTimeSessionWindows assigner = DynamicProcessingTimeSessionWindows.withDynamicGap(extractor);
assertFalse(assigner.isEventTime());
assertEquals(new TimeWindow.Serializer(), assigner.getWindowSerializer(new ExecutionConfig()));
assertThat(assigner.getDefaultTrigger(mock(StreamExecutionEnvironment.class)), instanceOf(ProcessingTimeTrigger.class));
}
use of org.apache.flink.streaming.api.windowing.assigners.SessionWindowTimeGapExtractor in project flink by apache.
the class DynamicProcessingTimeSessionWindowsTest method testMergeSinglePointWindow.
@Test
public void testMergeSinglePointWindow() {
MergingWindowAssigner.MergeCallback callback = mock(MergingWindowAssigner.MergeCallback.class);
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(5000L);
DynamicProcessingTimeSessionWindows assigner = DynamicProcessingTimeSessionWindows.withDynamicGap(extractor);
assigner.mergeWindows(Lists.newArrayList(new TimeWindow(0, 0)), callback);
verify(callback, never()).merge(anyCollection(), Matchers.anyObject());
}
use of org.apache.flink.streaming.api.windowing.assigners.SessionWindowTimeGapExtractor in project flink by apache.
the class DynamicProcessingTimeSessionWindowsTest method testInvalidParameters.
@Test
public void testInvalidParameters() {
WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class);
try {
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(-1L);
DynamicProcessingTimeSessionWindows assigner = DynamicProcessingTimeSessionWindows.withDynamicGap(extractor);
assigner.assignWindows(Lists.newArrayList(new Object()), 1, mockContext);
fail("should fail");
} catch (IllegalArgumentException e) {
assertThat(e.toString(), containsString("0 < gap"));
}
try {
SessionWindowTimeGapExtractor extractor = mock(SessionWindowTimeGapExtractor.class);
when(extractor.extract(any())).thenReturn(-1L);
DynamicProcessingTimeSessionWindows assigner = DynamicProcessingTimeSessionWindows.withDynamicGap(extractor);
assigner.assignWindows(Lists.newArrayList(new Object()), 1, mockContext);
fail("should fail");
} catch (IllegalArgumentException e) {
assertThat(e.toString(), containsString("0 < gap"));
}
}
Aggregations