/** * Returns the {@link ProcessingTimeService} responsible for getting the current * processing time and registering timers. */ protected ProcessingTimeService getProcessingTimeService() { return container.getProcessingTimeService(); }
public ProcessingTimeService getProcessingTimeService() { return task.getProcessingTimeService(); }
private static <T> void setupSourceOperator( StreamSource<T, ?> operator, ExecutionConfig executionConfig, Environment env, ProcessingTimeService timeProvider) { StreamConfig cfg = new StreamConfig(new Configuration()); cfg.setStateBackend(new MemoryStateBackend()); cfg.setTimeCharacteristic(TimeCharacteristic.EventTime); cfg.setOperatorID(new OperatorID()); StreamStatusMaintainer streamStatusMaintainer = mock(StreamStatusMaintainer.class); when(streamStatusMaintainer.getStreamStatus()).thenReturn(StreamStatus.ACTIVE); StreamTask<?, ?> mockTask = mock(StreamTask.class); when(mockTask.getName()).thenReturn("Mock Task"); when(mockTask.getCheckpointLock()).thenReturn(new Object()); when(mockTask.getConfiguration()).thenReturn(cfg); when(mockTask.getEnvironment()).thenReturn(env); when(mockTask.getExecutionConfig()).thenReturn(executionConfig); when(mockTask.getAccumulatorMap()).thenReturn(Collections.<String, Accumulator<?, ?>>emptyMap()); when(mockTask.getStreamStatusMaintainer()).thenReturn(streamStatusMaintainer); doAnswer(new Answer<ProcessingTimeService>() { @Override public ProcessingTimeService answer(InvocationOnMock invocation) throws Throwable { if (timeProvider == null) { throw new RuntimeException("The time provider is null."); } return timeProvider; } }).when(mockTask).getProcessingTimeService(); operator.setup(mockTask, cfg, (Output<StreamRecord<T>>) mock(Output.class)); }
@Override public void close() throws Exception { // verify that the timer service is still running Assert.assertTrue( ((SystemProcessingTimeService) getContainingTask().getProcessingTimeService()) .isAlive()); super.close(); } }
}).when(mockTask).getProcessingTimeService();
when(containingTask.getEnvironment()).thenReturn(environment); when(containingTask.getCheckpointLock()).thenReturn(lock); when(containingTask.getProcessingTimeService()).thenReturn(processingTimeService);
return processingTimeService; }).when(mockTask).getProcessingTimeService();
when(containingTask.getEnvironment()).thenReturn(environment); when(containingTask.getCheckpointLock()).thenReturn(lock); when(containingTask.getProcessingTimeService()).thenReturn(new TestProcessingTimeService());
operator.getContainingTask().getProcessingTimeService(), operator.getContainingTask().getCheckpointLock(), operator.getContainingTask().getStreamStatusMaintainer(),
/** * Returns the {@link ProcessingTimeService} responsible for getting the current * processing time and registering timers. */ protected ProcessingTimeService getProcessingTimeService() { return container.getProcessingTimeService(); }
/** * Returns the {@link ProcessingTimeService} responsible for getting the current * processing time and registering timers. */ protected ProcessingTimeService getProcessingTimeService() { return container.getProcessingTimeService(); }
/** * Returns the {@link ProcessingTimeService} responsible for getting the current * processing time and registering timers. */ protected ProcessingTimeService getProcessingTimeService() { return container.getProcessingTimeService(); }