chore: import upstream snapshot with attribution

python/ray/serve/tests/test_api_2.py

@@ -0,0 +1,67 @@
+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__]))