@Test
public void testPollingExceptionFlyteAdminClientExecutionNotFound() {

  doThrow(new StatusRuntimeException(Status.NOT_FOUND))
      .when(flyteAdminClient).getExecution(anyString(), anyString(), anyString());
  assertThrows(
      FlyteRunner.ExecutionNotFoundException.class,
      () -> flyteRunner.poll(FLYTE_EXECUTION_ID, RUN_STATE));
}