wehub-resource-sync
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
import pytest
|
|
|
|
import ray
|
|
from ray import serve
|
|
from ray._common.network_utils import build_address
|
|
from ray.serve._private.common import RequestProtocol
|
|
from ray.serve._private.test_utils import get_application_urls
|
|
|
|
|
|
def test_get_application_urls(serve_instance):
|
|
@serve.deployment
|
|
def f():
|
|
return "Hello, world!"
|
|
|
|
serve.run(f.bind())
|
|
controller_details = ray.get(serve_instance._controller.get_actor_details.remote())
|
|
node_ip = controller_details.node_ip
|
|
assert get_application_urls(use_localhost=False) == [
|
|
f"http://{build_address(node_ip, 8000)}"
|
|
]
|
|
assert get_application_urls("gRPC", use_localhost=False) == [
|
|
build_address(node_ip, 9000)
|
|
]
|
|
assert get_application_urls(RequestProtocol.HTTP, use_localhost=False) == [
|
|
f"http://{build_address(node_ip, 8000)}"
|
|
]
|
|
assert get_application_urls(RequestProtocol.GRPC, use_localhost=False) == [
|
|
build_address(node_ip, 9000)
|
|
]
|
|
|
|
|
|
def test_get_application_urls_with_app_name(serve_instance):
|
|
@serve.deployment
|
|
def f():
|
|
return "Hello, world!"
|
|
|
|
serve.run(f.bind(), name="app1", route_prefix="/")
|
|
controller_details = ray.get(serve_instance._controller.get_actor_details.remote())
|
|
node_ip = controller_details.node_ip
|
|
assert get_application_urls("HTTP", app_name="app1", use_localhost=False) == [
|
|
f"http://{node_ip}:8000"
|
|
]
|
|
assert get_application_urls("gRPC", app_name="app1", use_localhost=False) == [
|
|
f"{node_ip}:9000"
|
|
]
|
|
|
|
|
|
def test_get_application_urls_with_route_prefix(serve_instance):
|
|
@serve.deployment
|
|
def f():
|
|
return "Hello, world!"
|
|
|
|
serve.run(f.bind(), name="app1", route_prefix="/app1")
|
|
controller_details = ray.get(serve_instance._controller.get_actor_details.remote())
|
|
node_ip = controller_details.node_ip
|
|
assert get_application_urls("HTTP", app_name="app1", use_localhost=False) == [
|
|
f"http://{node_ip}:8000/app1"
|
|
]
|
|
assert get_application_urls("gRPC", app_name="app1", use_localhost=False) == [
|
|
f"{node_ip}:9000"
|
|
]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
sys.exit(pytest.main(["-v", "-s", __file__]))
|