Examples with MockInputSplitProvider org.apache.flink.runtime.operators.testutils.MockInputSplitProvider used on opensource projects

Example 6 with MockInputSplitProvider

use of org.apache.flink.runtime.operators.testutils.MockInputSplitProvider in project flink by apache.

the class StreamOperatorChainingTest method createMockTask.

private <IN, OT extends StreamOperator<IN>> StreamTask<IN, OT> createMockTask(StreamConfig streamConfig, String taskName) {
    final Object checkpointLock = new Object();
    final Environment env = new MockEnvironment(taskName, 3 * 1024 * 1024, new MockInputSplitProvider(), 1024);
    @SuppressWarnings("unchecked") StreamTask<IN, OT> mockTask = mock(StreamTask.class);
    when(mockTask.getName()).thenReturn("Mock Task");
    when(mockTask.getCheckpointLock()).thenReturn(checkpointLock);
    when(mockTask.getConfiguration()).thenReturn(streamConfig);
    when(mockTask.getEnvironment()).thenReturn(env);
    when(mockTask.getExecutionConfig()).thenReturn(new ExecutionConfig().enableObjectReuse());
    return mockTask;
}