diff --git a/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/MonitoringUtil.java b/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/MonitoringUtil.java index a7dd5155b663..0237923e4e47 100644 --- a/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/MonitoringUtil.java +++ b/runners/google-cloud-dataflow-java/src/main/java/org/apache/beam/runners/dataflow/util/MonitoringUtil.java @@ -212,6 +212,7 @@ public static State toState(@Nullable String stateName) { case "JOB_STATE_UNKNOWN": return State.UNKNOWN; case "JOB_STATE_STOPPED": + case "JOB_STATE_PAUSED": // Job is paused; the closest mapping is STOPPED return State.STOPPED; case "JOB_STATE_FAILED": return State.FAILED; @@ -224,6 +225,7 @@ public static State toState(@Nullable String stateName) { case "JOB_STATE_PENDING": // Job has not yet started; closest mapping is RUNNING case "JOB_STATE_DRAINING": // Job is still active; the closest mapping is RUNNING case "JOB_STATE_CANCELLING": // Job is still active; the closest mapping is RUNNING + case "JOB_STATE_PAUSING": // Job is still active; the closest mapping is RUNNING case "JOB_STATE_RESOURCE_CLEANING_UP": // Job is still active; the closest mapping is RUNNING return State.RUNNING; diff --git a/runners/google-cloud-dataflow-java/src/test/java/org/apache/beam/runners/dataflow/util/MonitoringUtilTest.java b/runners/google-cloud-dataflow-java/src/test/java/org/apache/beam/runners/dataflow/util/MonitoringUtilTest.java index df169b66a58a..5f76b6750ffa 100644 --- a/runners/google-cloud-dataflow-java/src/test/java/org/apache/beam/runners/dataflow/util/MonitoringUtilTest.java +++ b/runners/google-cloud-dataflow-java/src/test/java/org/apache/beam/runners/dataflow/util/MonitoringUtilTest.java @@ -99,7 +99,9 @@ public void testToStateNormal() { assertEquals(State.UPDATED, MonitoringUtil.toState("JOB_STATE_UPDATED")); // Non-trivially mapped cases + assertEquals(State.STOPPED, MonitoringUtil.toState("JOB_STATE_PAUSED")); assertEquals(State.RUNNING, MonitoringUtil.toState("JOB_STATE_DRAINING")); + assertEquals(State.RUNNING, MonitoringUtil.toState("JOB_STATE_PAUSING")); assertEquals(State.DONE, MonitoringUtil.toState("JOB_STATE_DRAINED")); }