DataflowJobStateMapper { public static JobStatus map(String jobState) { DataflowJobState dfJobState = DataflowJobState.valueOf(jobState); if (DATAFLOW_TO_FEAST_JOB_STATUS.containsKey(dfJobState)) { return DATAFLOW_TO_FEAST_JOB_STATUS.get(dfJobState); } throw new IllegalArgumentException("Unknown job state: " + jobState); }  static JobStatus map(String jobState);  }
@Test public void shouldThrowIllegalArgumentExceptionForInvalidString() { assertThrows(IllegalArgumentException.class, () -> mapper.map("INVALID_STATE")); }