wehub-resource-sync
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
|
|
import pytest
|
|
|
|
import ray
|
|
from ray._common.test_utils import wait_for_condition
|
|
from ray.cross_language import java_actor_class
|
|
|
|
# Source code of MyClass.java
|
|
_MY_CLASS_JAVA = """
|
|
public class MyClass {
|
|
public int printToLog(String line) {
|
|
System.err.println(line);
|
|
return 0;
|
|
}
|
|
}
|
|
"""
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.platform == "win32" or sys.platform == "darwin",
|
|
reason="Does not work on Windows and OSX.",
|
|
)
|
|
def test_log_java_worker_logs(shutdown_only, capsys):
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
print("using tmp_dir", tmp_dir)
|
|
with open(os.path.join(tmp_dir, "MyClass.java"), "w") as f:
|
|
f.write(_MY_CLASS_JAVA)
|
|
subprocess.check_call(["javac", "MyClass.java"], cwd=tmp_dir)
|
|
subprocess.check_call(["jar", "-cf", "myJar.jar", "MyClass.class"], cwd=tmp_dir)
|
|
|
|
ray.init(
|
|
job_config=ray.job_config.JobConfig(code_search_path=[tmp_dir]),
|
|
)
|
|
|
|
handle = java_actor_class("MyClass").remote()
|
|
ray.get(handle.printToLog.remote("here's my random line!"))
|
|
|
|
def check():
|
|
out, err = capsys.readouterr()
|
|
out += err
|
|
with capsys.disabled():
|
|
print(out)
|
|
return "here's my random line!" in out
|
|
|
|
wait_for_condition(check)
|
|
|
|
ray.shutdown()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Make subprocess happy in bazel.
|
|
os.environ["LC_ALL"] = "en_US.UTF-8"
|
|
os.environ["LANG"] = "en_US.UTF-8"
|
|
sys.exit(pytest.main(["-sv", __file__]))
|