@Test public void testWindowAssignment() { WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class); TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.milliseconds(5000)); assertThat(assigner.assignWindows("String", 0L, mockContext), contains(timeWindow(0, 5000))); assertThat(assigner.assignWindows("String", 4999L, mockContext), contains(timeWindow(0, 5000))); assertThat(assigner.assignWindows("String", 5000L, mockContext), contains(timeWindow(5000, 10000))); }
@Test public void testWindowAssignmentWithNegativeOffset() { WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class); TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(-100)); assertThat(assigner.assignWindows("String", 0L, mockContext), contains(timeWindow(-100, 4900))); assertThat(assigner.assignWindows("String", 4899L, mockContext), contains(timeWindow(-100, 4900))); assertThat(assigner.assignWindows("String", 4900L, mockContext), contains(timeWindow(4900, 9900))); }
@Test public void testWindowAssignmentWithOffset() { WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class); TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(100)); assertThat(assigner.assignWindows("String", 100L, mockContext), contains(timeWindow(100, 5100))); assertThat(assigner.assignWindows("String", 5099L, mockContext), contains(timeWindow(100, 5100))); assertThat(assigner.assignWindows("String", 5100L, mockContext), contains(timeWindow(5100, 10100))); }
@Test public void testTimeUnits() { // sanity check with one other time unit WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class); TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.seconds(5), Time.seconds(1)); assertThat(assigner.assignWindows("String", 1000L, mockContext), contains(timeWindow(1000, 6000))); assertThat(assigner.assignWindows("String", 5999L, mockContext), contains(timeWindow(1000, 6000))); assertThat(assigner.assignWindows("String", 6000L, mockContext), contains(timeWindow(6000, 11000))); }
Collection<TimeWindow> windows = windowAssigner.assignWindows(new Tuple2<>("key2", 1), timestamp, new WindowAssigner.WindowAssignerContext() { @Override public long getCurrentProcessingTime() {