commit 5e099dc43717ee54751f7f572119f7f4eee7e3cb
author Jianyong Dai <daijy@apache.org> Sun May 24 20:51:37 2015 +0000
committer Jianyong Dai <daijy@apache.org> Sun May 24 20:51:37 2015 +0000
tree 0f4d4f7b0f346f335a8586f135a59e2e9d144073
parent 94abd5629273f2ea1a9a8a74eb9c1e32cb46db5d

PIG-3994: Implement getting backend exception for Tez (PIG-3994-2.patch)

git-svn-id: https://svn.apache.org/repos/asf/pig/branches/branch-0.15@1681527 13f79535-47bb-0310-9956-ffa450edef68

test/org/apache/pig/test/TestAssert.java

diff --git a/test/org/apache/pig/test/TestAssert.java b/test/org/apache/pig/test/TestAssert.java
index 5c434cb..6d4b5c6 100644
--- a/test/org/apache/pig/test/TestAssert.java
+++ b/test/org/apache/pig/test/TestAssert.java
@@ -116,8 +116,13 @@
       try {
           pigServer.openIterator("A");
       } catch (FrontendException fe) {
-          Assert.assertTrue(fe.getCause().getMessage().contains(
-                  "Job terminated with anomalous status FAILED"));
+          if (pigServer.getPigContext().getExecType().toString().startsWith("TEZ")) {
+              Assert.assertTrue(fe.getCause().getMessage().contains(
+                      "Assertion violated: i should be greater than 1"));
+          } else {
+              Assert.assertTrue(fe.getCause().getMessage().contains(
+                      "Job terminated with anomalous status FAILED"));
+          }
       }
   }
 
@@ -142,8 +147,13 @@
       try {
           pigServer.openIterator("A");
       } catch (FrontendException fe) {
-          Assert.assertTrue(fe.getCause().getMessage().contains(
-                  "Job terminated with anomalous status FAILED"));
+          if (pigServer.getPigContext().getExecType().toString().startsWith("TEZ")) {
+              Assert.assertTrue(fe.getCause().getMessage().contains(
+                      "Assertion violated: i should be greater than 1"));
+          } else {
+              Assert.assertTrue(fe.getCause().getMessage().contains(
+                      "Job terminated with anomalous status FAILED"));
+          }
       }
   }