wehub-resource-sync
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import sys
|
|
|
|
import pytest
|
|
|
|
import ray
|
|
from ray._private.test_utils import external_redis_test_enabled
|
|
|
|
|
|
@pytest.fixture
|
|
def setup_tls(monkeypatch):
|
|
from pathlib import Path
|
|
|
|
tls_path = Path(__file__).parent
|
|
monkeypatch.setenv("RAY_REDIS_CA_CERT", f"{str(tls_path)}/tls/ca.crt")
|
|
monkeypatch.setenv("RAY_REDIS_CLIENT_CERT", f"{str(tls_path)}/tls/redis.crt")
|
|
monkeypatch.setenv("RAY_REDIS_CLIENT_KEY", f"{str(tls_path)}/tls/redis.key")
|
|
ray._raylet.Config.initialize("")
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def setup_replicas(request, monkeypatch):
|
|
monkeypatch.setenv("TEST_EXTERNAL_REDIS_REPLICAS", str(request.param))
|
|
yield
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
not external_redis_test_enabled(), reason="Only work for redis mode"
|
|
)
|
|
@pytest.mark.skipif(sys.platform != "linux", reason="Only work in linux")
|
|
@pytest.mark.parametrize("setup_replicas", [1, 3], indirect=True)
|
|
def test_redis_tls(setup_tls, setup_replicas, ray_start_cluster_head):
|
|
@ray.remote
|
|
def hello():
|
|
return "world"
|
|
|
|
assert ray.get(hello.remote()) == "world"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|