@Test
public void shouldPollInSubmitted() throws FlyteRunner.PollingException {

  RunState runState = RunState.create(WORKFLOW_INSTANCE, State.SUBMITTED, StateData.newBuilder()
      .executionId(EXECUTION_ID)
      .executionDescription(FLYTE_EXECUTION_DESCRIPTION)
      .build());
  flyteRunnerHandler.transitionInto(runState, eventRouter);
  verify(flyteRunner).poll(getFlyteExecutionId(FLYTE_EXECUTION_DESCRIPTION,
      EXECUTION_NAME), runState);
}