Examples with OffsetRangeTracker org.apache.beam.sdk.transforms.splittabledofn.OffsetRangeTracker used on opensource projects

Example 6 with OffsetRangeTracker

use of org.apache.beam.sdk.transforms.splittabledofn.OffsetRangeTracker in project beam by apache.

the class OutputAndTimeBoundedSplittableProcessElementInvokerTest method testInvokeProcessElementOutputBounded.

@Test
public void testInvokeProcessElementOutputBounded() throws Exception {
    SplittableProcessElementInvoker<Integer, String, OffsetRange, OffsetRangeTracker>.Result<Integer, String, OffsetRange, OffsetRangeTracker> res = runTest(10000, Duration.ZERO);
    OffsetRange residualRange = res.getResidualRestriction();
    // Should process the first 100 elements.
    assertEquals(1000, residualRange.getFrom());
    assertEquals(10000, residualRange.getTo());
}