use of org.apache.beam.sdk.nexmark.sources.generator.Generator in project beam by apache.
the class GeneratorTest method splitPreservesOverallEventCount.
@Test
public void splitPreservesOverallEventCount() {
long n = 51237L;
GeneratorConfig initialConfig = makeConfig(n);
long expected = initialConfig.getStopEventId() - initialConfig.getStartEventId();
List<Generator> generators = new ArrayList<>();
for (GeneratorConfig subConfig : initialConfig.split(20)) {
generators.add(new Generator(subConfig));
}
long actual = 0;
for (Generator generator : generators) {
actual += consume(generator);
}
assertEquals(expected, actual);
}
use of org.apache.beam.sdk.nexmark.sources.generator.Generator in project beam by apache.
the class UnboundedEventSourceTest method resumeFromCheckpoint.
/**
* Check aggressively checkpointing and resuming a reader gives us exactly the same event stream
* as reading directly.
*/
@Ignore("TODO(BEAM-5070) Test is flaky. Fix before reenabling.")
@Test
public void resumeFromCheckpoint() throws IOException {
Random random = new Random(297);
int n = 47293;
GeneratorConfig config = makeConfig(n);
Generator modelGenerator = new Generator(config);
EventIdChecker checker = new EventIdChecker();
PipelineOptions options = TestPipeline.testingPipelineOptions();
UnboundedEventSource source = new UnboundedEventSource(config, 1, 0, false);
UnboundedReader<Event> reader = source.createReader(options, null);
while (n > 0) {
int m = Math.min(459 + random.nextInt(455), n);
System.out.printf("reading %d...%n", m);
checker.add(m, reader, modelGenerator);
n -= m;
System.out.printf("splitting with %d remaining...%n", n);
CheckpointMark checkpointMark = reader.getCheckpointMark();
reader = source.createReader(options, (GeneratorCheckpoint) checkpointMark);
}
assertFalse(reader.advance());
}
use of org.apache.beam.sdk.nexmark.sources.generator.Generator in project beam by apache.
the class GeneratorTest method splitAtFractionPreservesOverallEventCount.
@Test
public void splitAtFractionPreservesOverallEventCount() {
long n = 55729L;
GeneratorConfig initialConfig = makeConfig(n);
long expected = initialConfig.getStopEventId() - initialConfig.getStartEventId();
long actual = 0;
Generator initialGenerator = new Generator(initialConfig);
// Consume some events.
actual += consume(5000, initialGenerator);
// Split once.
GeneratorConfig remainConfig1 = initialGenerator.splitAtEventId(9000L);
Generator remainGenerator1 = new Generator(remainConfig1);
// Consume some more events.
actual += consume(2000, initialGenerator);
actual += consume(3000, remainGenerator1);
// Split again.
GeneratorConfig remainConfig2 = remainGenerator1.splitAtEventId(30000L);
Generator remainGenerator2 = new Generator(remainConfig2);
// Run to completion.
actual += consume(initialGenerator);
actual += consume(remainGenerator1);
actual += consume(remainGenerator2);
assertEquals(expected, actual);
}
Aggregations