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

closes #1582

1 files changed, 2 insertions, 10 deletions

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 73f1b00f8..8be314884 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 @@ public class TestClient {
     } catch (SimulatedExitException e) {
       assert (e.exitCode == -1);
       assertTrue(TestContext.getOut().contains("Usage: "));
+      TestContext.resetOutput();
     }
 
     // Bogus command
@@ -115,16 +116,7 @@ public class TestClient {
     } 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();
     }
   }