wehub-resource-sync
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
import ray
|
|
from ray._private.utils import update_envs
|
|
|
|
|
|
@pytest.mark.skipif("sys.platform != 'linux'")
|
|
def test_environ_file_on_linux(ray_start_10_cpus):
|
|
@ray.remote
|
|
class Actor1:
|
|
def __init__(self):
|
|
pass
|
|
|
|
def get_env_from_proc(self):
|
|
pid = os.getpid()
|
|
env = {}
|
|
with open("/proc/%s/environ" % pid) as fd:
|
|
for envspec in fd.read().split("\000"):
|
|
if not envspec:
|
|
continue
|
|
varname, varval = envspec.split("=", 1)
|
|
env[varname] = varval
|
|
return env
|
|
|
|
def get_os_environ(self):
|
|
return os.environ
|
|
|
|
a = Actor1.remote()
|
|
actor_proc_environ = ray.get(a.get_env_from_proc.remote())
|
|
actor_os_environ = ray.get(a.get_os_environ.remote())
|
|
assert len(actor_proc_environ) > 0
|
|
assert len(actor_os_environ) > 0
|
|
|
|
|
|
def test_update_envs():
|
|
with unittest.mock.patch.dict(os.environ):
|
|
env_vars = {
|
|
"PATH": "/test/lib/path:${PATH}",
|
|
"LD_LIBRARY_PATH": "/test/path1:${LD_LIBRARY_PATH}:./test/path2",
|
|
"DYLD_LIBRARY_PATH": "${DYLD_LIBRARY_PATH}:/test/path",
|
|
"LD_PRELOAD": "",
|
|
}
|
|
old_path = os.environ["PATH"]
|
|
os.environ["LD_LIBRARY_PATH"] = "./"
|
|
os.environ["DYLD_LIBRARY_PATH"] = "/lib64"
|
|
os.environ["LD_PRELOAD"] = "/lib:/usr/local/lib"
|
|
update_envs(env_vars)
|
|
assert os.environ["PATH"] == "/test/lib/path:" + old_path
|
|
assert os.environ["LD_LIBRARY_PATH"] == "/test/path1:./:./test/path2"
|
|
assert os.environ["DYLD_LIBRARY_PATH"] == "/lib64:/test/path"
|
|
assert os.environ["LD_PRELOAD"] == env_vars["LD_PRELOAD"]
|
|
|
|
# Test the empty string scenario
|
|
os.environ["LD_LIBRARY_PATH"] = ""
|
|
del os.environ["DYLD_LIBRARY_PATH"]
|
|
del os.environ["LD_PRELOAD"]
|
|
update_envs(env_vars)
|
|
assert os.environ["LD_LIBRARY_PATH"] == "/test/path1::./test/path2"
|
|
assert os.environ["DYLD_LIBRARY_PATH"] == ":/test/path"
|
|
assert os.environ["LD_PRELOAD"] == env_vars["LD_PRELOAD"]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
os.environ["LC_ALL"] = "en_US.UTF-8"
|
|
os.environ["LANG"] = "en_US.UTF-8"
|
|
sys.exit(pytest.main(["-sv", __file__]))
|