Examples with MockStreamConfig org.apache.flink.streaming.util.MockStreamConfig used on opensource projects

Example 1 with MockStreamConfig

use of org.apache.flink.streaming.util.MockStreamConfig in project flink by apache.

the class SourceOperatorEventTimeTest method createTestOperator.

// ------------------------------------------------------------------------
// test setup helpers
// ------------------------------------------------------------------------
private static <T> SourceOperator<T, MockSourceSplit> createTestOperator(SourceReader<T, MockSourceSplit> reader, WatermarkStrategy<T> watermarkStrategy, ProcessingTimeService timeService, boolean emitProgressiveWatermarks) throws Exception {
    final OperatorStateStore operatorStateStore = new MemoryStateBackend().createOperatorStateBackend(new MockEnvironmentBuilder().build(), "test-operator", Collections.emptyList(), new CloseableRegistry());
    final StateInitializationContext stateContext = new StateInitializationContextImpl(null, operatorStateStore, null, null, null);
    final SourceOperator<T, MockSourceSplit> sourceOperator = new TestingSourceOperator<>(reader, watermarkStrategy, timeService, emitProgressiveWatermarks);
    sourceOperator.setup(new SourceOperatorStreamTask<Integer>(new StreamMockEnvironment(new Configuration(), new Configuration(), new ExecutionConfig(), 1L, new MockInputSplitProvider(), 1, new TestTaskStateManager())), new MockStreamConfig(new Configuration(), 1), new MockOutput<>(new ArrayList<>()));
    sourceOperator.initializeState(stateContext);
    sourceOperator.open();
    return sourceOperator;
}

Example 2 with MockStreamConfig

use of org.apache.flink.streaming.util.MockStreamConfig in project flink by apache.

the class StreamTaskTest method setupEnvironment.

private MockEnvironment setupEnvironment(boolean... outputAvailabilities) {
    final Configuration configuration = new Configuration();
    new MockStreamConfig(configuration, outputAvailabilities.length);
    final List<ResultPartitionWriter> writers = new ArrayList<>(outputAvailabilities.length);
    for (int i = 0; i < outputAvailabilities.length; i++) {
        writers.add(new AvailabilityTestResultPartitionWriter(outputAvailabilities[i]));
    }
    final MockEnvironment environment = new MockEnvironmentBuilder().setTaskConfiguration(configuration).build();
    environment.addOutputs(writers);
    return environment;
}