@Override public void invokeOnTimer( String timerId, DoFnInvoker.ArgumentProvider<InputT, OutputT> arguments) { @Nullable OnTimerInvoker onTimerInvoker = onTimerInvokers.get(timerId); if (onTimerInvoker != null) { onTimerInvoker.invokeOnTimer(arguments); } else { throw new IllegalArgumentException( String.format( "Attempted to invoke timer %s on %s, but that timer is not registered." + " This is the responsibility of the runner, which must only deliver" + " registered timers.", timerId, delegate.getClass().getName())); } }
private void invokeOnTimer(DoFn<String, String> fn, String timerId) { OnTimerInvokers.forTimer(fn, timerId).invokeOnTimer(mockArgumentProvider); }