[tez] branch branch-0.9 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 branch-0.9
in repository https://gitbox.apache.org/repos/asf/tez.git


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

commit 429270959eae9fbd75af07e951a10c64a3b7fb4e
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)
    
    (cherry picked from commit 3162aab3411f885db3e103f0c16c92de3e2ca46b)
---
 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 6268912..349809e 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
@@ -213,7 +213,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();
   }