TEZ-4036. TestMockDAGAppMaster#testInternalPreemption should assert f…

…or failed state (Kuhu Shukla via jeagles)
dingmy002 · Feb 5, 2019 · 3162aab · 3162aab
1 parent c78d348
commit 3162aab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tez-dag/src/test/java/org/apache/tez/dag/app/TestMockDAGAppMaster.java b/tez-dag/src/test/java/org/apache/tez/dag/app/TestMockDAGAppMaster.java
@@ -215,7 +215,8 @@ public void testInternalPreemption() throws Exception {
     TezVertexID vertexId = TezVertexID.getInstance(dagImpl.getID(), 0);
     TezTaskAttemptID killedTaId = TezTaskAttemptID.getInstance(TezTaskID.getInstance(vertexId, 0), 0);
     TaskAttempt killedTa = dagImpl.getVertex(vA.getName()).getTask(0).getAttempt(killedTaId);
-    Assert.assertEquals(TaskAttemptState.KILLED, killedTa.getState());
+    //Refer to TEZ-3950
+    Assert.assertTrue(killedTa.getState().equals(TaskAttemptState.KILLED) || killedTa.getState().equals(TaskAttemptState.FAILED));
     tezClient.stop();
   }