commit 3e2d6192fc346fe63342021e5d19c7921d6c59dd
author Volodymyr Vysotskyi <vvovyk@gmail.com> Fri Dec 21 14:37:45 2018 +0200
committer Vitalii Diravka <vitalii.diravka@gmail.com> Mon Dec 24 17:54:45 2018 +0200
tree c04fe5a6de111cf93594dfe2bd8ef5c646ab1fea
parent 44f17eaf1b9489fc16fadee0178e975f6aff5185

DRILL-6920: Fix TestClient.testBasics() yarn test failure

closes #1582

drill-yarn/src/test/java/org/apache/drill/yarn/client/TestClient.java

diff --git a/drill-yarn/src/test/java/org/apache/drill/yarn/client/TestClient.java b/drill-yarn/src/test/java/org/apache/drill/yarn/client/TestClient.java
index 73f1b00..8be3148 100644
--- a/drill-yarn/src/test/java/org/apache/drill/yarn/client/TestClient.java
+++ b/drill-yarn/src/test/java/org/apache/drill/yarn/client/TestClient.java
@@ -105,6 +105,7 @@
     } catch (SimulatedExitException e) {
       assert (e.exitCode == -1);
       assertTrue(TestContext.getOut().contains("Usage: "));
+      TestContext.resetOutput();
     }
 
     // Bogus command
@@ -115,16 +116,7 @@
     } catch (SimulatedExitException e) {
       assert (e.exitCode == -1);
       assertTrue(TestContext.getOut().contains("Usage: "));
-    }
-
-    // Help command
-
-    try {
-      DrillOnYarn.run(new String[] { "help" });
-      fail();
-    } catch (SimulatedExitException e) {
-      assert (e.exitCode == -1);
-      assertTrue(TestContext.getOut().contains("Usage: "));
+      TestContext.resetOutput();
     }
   }