mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 00:55:31 +08:00
Edward Oakes
37 lines
766 B
Python
37 lines
766 B
Python
import os
|
|
import subprocess
|
|
import tempfile
|
|
|
|
import ray
|
|
from ray import serve
|
|
|
|
|
|
def test_new_driver(serve_instance):
|
|
script = """
|
|
import ray
|
|
ray.init(address="{}")
|
|
|
|
from ray import serve
|
|
serve.init()
|
|
|
|
def driver(flask_request):
|
|
return "OK!"
|
|
|
|
serve.create_endpoint("driver", "/driver")
|
|
serve.create_backend("driver", driver)
|
|
serve.set_traffic("driver", {{"driver": 1.0}})
|
|
""".format(ray.worker._global_node._redis_address)
|
|
|
|
with tempfile.NamedTemporaryFile(mode="w", delete=False) as f:
|
|
path = f.name
|
|
f.write(script)
|
|
|
|
proc = subprocess.Popen(["python", path])
|
|
return_code = proc.wait(timeout=10)
|
|
assert return_code == 0
|
|
|
|
handle = serve.get_handle("driver")
|
|
assert ray.get(handle.remote()) == "OK!"
|
|
|
|
os.remove(path)
|