/** {@inheritDoc} */ @Override public GridJunit3SerializableTest execute() throws GridException { final GridJunit3SerializableTest testArg = argument(0); assert testArg != null; TestResult collector = new TestResult(); collector.addListener( new TestListener() { /** */ private GridTestPrintStream out; /** */ private GridTestPrintStream err; /** {@inheritDoc} */ @Override public void addError(Test test, Throwable e) { assert test instanceof TestCase : "Errors can be added only to TestCases: " + test; testArg.findTestCase((TestCase) test).setError(e); } /** {@inheritDoc} */ @Override public void addFailure(Test test, AssertionFailedError e) { assert test instanceof TestCase : "Failures can be added only to TestCases: " + test; testArg.findTestCase((TestCase) test).setFailure(e); } /** {@inheritDoc} */ @Override public void startTest(Test test) { GridTestPrintStreamFactory.getStdOut() .println("Distributed test started: " + getTestName(test)); out = GridTestPrintStreamFactory.acquireOut(); err = GridTestPrintStreamFactory.acquireErr(); } /** {@inheritDoc} */ @Override public void endTest(Test test) { GridJunit3SerializableTestCase testCase = testArg.findTestCase((TestCase) test); try { testCase.setStandardOut(getBytes(out)); testCase.setStandardError(getBytes(err)); } catch (IOException e) { U.error(log, "Error resetting output.", e); if (testCase.getError() == null) { testCase.setError(e); } else if (testCase.getFailure() == null) { testCase.setFailure(e); } else { // Override initial error. testCase.setError(e); } } GridTestPrintStreamFactory.releaseOut(); GridTestPrintStreamFactory.releaseErr(); out = null; err = null; GridTestPrintStreamFactory.getStdOut() .println("Distributed test finished: " + getTestName(test)); } /** * @param out Output stream to gen bytes. * @return Output bytes. * @throws IOException If error occur. */ private byte[] getBytes(GridTestPrintStream out) throws IOException { ByteArrayOutputStream byteOut = new ByteArrayOutputStream(); out.purge(byteOut); return byteOut.toByteArray(); } /** * @param test JUnit3 test. * @return Test name. */ private String getTestName(Test test) { if (test instanceof TestSuite) { return ((TestSuite) test).getName(); } return test.getClass().getName() + '.' + ((TestCase) test).getName(); } }); // Run Junits. testArg.getTest().run(collector); return testArg; }