[tez] branch master updated: TEZ-4036. TestMockDAGAppMaster#testInternalPreemption should assert for failed state (Kuhu Shukla via jeagles)

[jeagles]

This is an automated email from the ASF dual-hosted git repository.

jeagles pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/tez.git


The following commit(s) were added to refs/heads/master by this push:
     new 3162aab  TEZ-4036. TestMockDAGAppMaster#testInternalPreemption should 
assert for failed state (Kuhu Shukla via jeagles)
3162aab is described below

commit 3162aab3411f885db3e103f0c16c92de3e2ca46b
Author: Jonathan Eagles <jeag...@gmail.com>
AuthorDate: Tue Feb 5 11:33:43 2019 -0600

    TEZ-4036. TestMockDAGAppMaster#testInternalPreemption should assert for 
failed state (Kuhu Shukla via jeagles)
---
 tez-dag/src/test/java/org/apache/tez/dag/app/TestMockDAGAppMaster.java | 3 ++-
 1 file changed, 2 insertions(+), 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
index 859537b..1a46595 100644
--- 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 class TestMockDAGAppMaster {
     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();
   }