From b5fafe1fce23b29e913b8fd51cfd1aff110427fe Mon Sep 17 00:00:00 2001
From: chaokunyang <shawn.ck.yang@gmail.com>
Date: Mon, 22 Jun 2020 16:56:47 +0800
Subject: [PATCH] [Java] fix  java test (#9079)

---
 .../java/io/ray/test/MultiThreadingTest.java   |  2 +-
 .../test_cross_language_invocation.py          | 18 +++++++++---------
 java/testng.xml                                |  2 +-
 3 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/java/test/src/main/java/io/ray/test/MultiThreadingTest.java b/java/test/src/main/java/io/ray/test/MultiThreadingTest.java
index 5a63cdda4..c03d9f941 100644
--- a/java/test/src/main/java/io/ray/test/MultiThreadingTest.java
+++ b/java/test/src/main/java/io/ray/test/MultiThreadingTest.java
@@ -149,7 +149,7 @@ public class MultiThreadingTest extends BaseTest {
         Ray::getRuntimeContext,
         () -> Ray.task(MultiThreadingTest::echo, 1).remote(),
         () -> Ray.actor(Echo::new).remote(),
-        () -> fooActor.task(Echo::echo, 1),
+        () -> fooActor.task(Echo::echo, 1).remote(),
     };
   }
 
diff --git a/java/test/src/main/resources/test_cross_language_invocation.py b/java/test/src/main/resources/test_cross_language_invocation.py
index 50841b46b..f14bf1da2 100644
--- a/java/test/src/main/resources/test_cross_language_invocation.py
+++ b/java/test/src/main/resources/test_cross_language_invocation.py
@@ -13,32 +13,32 @@ def py_return_input(v):
 def py_func_call_java_function():
     try:
         # None
-        r = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        r = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "returnInput").remote(None)
         assert ray.get(r) is None
         # bool
-        r = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        r = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "returnInputBoolean").remote(True)
         assert ray.get(r) is True
         # int
-        r = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        r = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "returnInputInt").remote(100)
         assert ray.get(r) == 100
         # double
-        r = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        r = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "returnInputDouble").remote(1.23)
         assert ray.get(r) == 1.23
         # string
-        r = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        r = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "returnInputString").remote("Hello World!")
         assert ray.get(r) == "Hello World!"
         # list (tuple will be packed by pickle,
         # so only list can be transferred across language)
-        r = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        r = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "returnInputIntArray").remote([1, 2, 3])
         assert ray.get(r) == [1, 2, 3]
         # pack
-        f = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+        f = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                               "pack")
         input = [100, "hello", 1.23, [1, "2", 3.0]]
         r = f.remote(*input)
@@ -52,7 +52,7 @@ def py_func_call_java_function():
 def py_func_call_java_actor(value):
     assert isinstance(value, bytes)
     c = ray.java_actor_class(
-        "io.ray.api.test.CrossLanguageInvocationTest$TestActor")
+        "io.ray.test.CrossLanguageInvocationTest$TestActor")
     java_actor = c.remote(b"Counter")
     r = java_actor.concat.remote(value)
     return ray.get(r)
@@ -77,7 +77,7 @@ def py_func_call_python_actor_from_handle(value):
 @ray.remote
 def py_func_pass_python_actor_handle():
     counter = Counter.remote(2)
-    f = ray.java_function("io.ray.api.test.CrossLanguageInvocationTest",
+    f = ray.java_function("io.ray.test.CrossLanguageInvocationTest",
                           "callPythonActorHandle")
     r = f.remote(counter._serialization_helper()[0])
     return ray.get(r)
diff --git a/java/testng.xml b/java/testng.xml
index 2c5756315..20ed7c3ee 100644
--- a/java/testng.xml
+++ b/java/testng.xml
@@ -3,8 +3,8 @@
 <suite name="RAY suite" verbose="2">
     <test name = "RAY test">
         <packages>
-            <package name = "io.ray.api.test.*" />
             <package name = "io.ray.runtime.*" />
+            <package name = "io.ray.test.*" />
         </packages>
     </test>
     <listeners>