From dfb0ad0d3ef7b4a4205b3ea4bb3e093a0f7e4675 Mon Sep 17 00:00:00 2001
From: Qing Wang <kingchin1218@126.com>
Date: Wed, 15 Apr 2020 15:28:52 +0800
Subject: [PATCH] [Java] Fix Java CI exit code issue (#8028)

---
 java/test.sh | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/java/test.sh b/java/test.sh
index d88645405..56a5f650a 100755
--- a/java/test.sh
+++ b/java/test.sh
@@ -8,8 +8,12 @@ set -x
 ROOT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
 
 run_testng() {
-    $@
-    local exit_code=$?
+    local exit_code
+    if "$@"; then
+        exit_code=0
+    else
+        exit_code=$?
+    fi
     # exit_code == 2 means there are skipped tests.
     if [ $exit_code -ne 2 ] && [ $exit_code -ne 0 ] ; then
         exit $exit_code