From d51583dbd6dc9c082764b9ec06349678aaa71078 Mon Sep 17 00:00:00 2001
From: Kai Yang <kfstorm@outlook.com>
Date: Tue, 3 Dec 2019 16:34:07 +0800
Subject: [PATCH] Add test listener to show the test progress of java UT
 (#6341)

---
 .../org/ray/api/TestProgressListener.java     | 64 +++++++++++++++++++
 .../main/java/org/ray/api/test/BaseTest.java  |  2 -
 java/testng.xml                               |  1 +
 3 files changed, 65 insertions(+), 2 deletions(-)
 create mode 100644 java/test/src/main/java/org/ray/api/TestProgressListener.java

diff --git a/java/test/src/main/java/org/ray/api/TestProgressListener.java b/java/test/src/main/java/org/ray/api/TestProgressListener.java
new file mode 100644
index 000000000..e65acf8df
--- /dev/null
+++ b/java/test/src/main/java/org/ray/api/TestProgressListener.java
@@ -0,0 +1,64 @@
+package org.ray.api;
+
+import java.time.LocalDateTime;
+import org.testng.IInvokedMethod;
+import org.testng.IInvokedMethodListener;
+import org.testng.ITestContext;
+import org.testng.ITestListener;
+import org.testng.ITestResult;
+
+public class TestProgressListener implements IInvokedMethodListener, ITestListener {
+
+  private String getFullTestName(ITestResult iTestResult) {
+    return iTestResult.getTestClass().getName() + "."
+        + iTestResult.getMethod().getMethodName();
+  }
+
+  private void printInfo(String tag, String content) {
+    System.out.println(
+        "============ [" + LocalDateTime.now().toString() + "] [" + tag + "] " + content
+            + " ============");
+  }
+
+  @Override
+  public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {
+    printInfo("INVOKE METHOD", getFullTestName(testResult));
+  }
+
+  @Override
+  public void afterInvocation(IInvokedMethod method, ITestResult testResult) {
+  }
+
+  @Override
+  public void onTestStart(ITestResult result) {
+    printInfo("TEST START", getFullTestName(result));
+  }
+
+  @Override
+  public void onTestSuccess(ITestResult result) {
+    printInfo("TEST SUCCESS", getFullTestName(result));
+  }
+
+  @Override
+  public void onTestFailure(ITestResult result) {
+    printInfo("TEST FAILURE", getFullTestName(result));
+  }
+
+  @Override
+  public void onTestSkipped(ITestResult result) {
+    printInfo("TEST SKIPPED", getFullTestName(result));
+  }
+
+  @Override
+  public void onTestFailedButWithinSuccessPercentage(ITestResult result) {
+    printInfo("TEST FAILED BUT WITHIN SUCCESS PERCENTAGE", getFullTestName(result));
+  }
+
+  @Override
+  public void onStart(ITestContext context) {
+  }
+
+  @Override
+  public void onFinish(ITestContext context) {
+  }
+}
diff --git a/java/test/src/main/java/org/ray/api/test/BaseTest.java b/java/test/src/main/java/org/ray/api/test/BaseTest.java
index d15e4d708..34f44217d 100644
--- a/java/test/src/main/java/org/ray/api/test/BaseTest.java
+++ b/java/test/src/main/java/org/ray/api/test/BaseTest.java
@@ -18,8 +18,6 @@ public class BaseTest {
 
   @BeforeMethod(alwaysRun = true)
   public void setUpBase(Method method) {
-    LOGGER.info("===== Running test: "
-        + method.getDeclaringClass().getName() + "." + method.getName());
     Ray.init();
     // These files need to be deleted after each test case.
     filesToDelete = ImmutableList.of(
diff --git a/java/testng.xml b/java/testng.xml
index 9448cb30f..9d788abc6 100644
--- a/java/testng.xml
+++ b/java/testng.xml
@@ -9,5 +9,6 @@
     </test>
     <listeners>
         <listener class-name="org.ray.api.RayAlterSuiteListener" />
+        <listener class-name="org.ray.api.TestProgressListener" />
     </listeners>
 </suite>