chore: import upstream snapshot with attribution

python/ray/serve/tests/test_replica_request_context.py

@@ -0,0 +1,74 @@
+import sys
+
+import httpx
+import pytest
+from fastapi import FastAPI
+from fastapi.responses import PlainTextResponse
+
+from ray import serve
+from ray.serve._private.test_utils import get_application_url
+from ray.serve.context import _get_serve_request_context
+
+
+def _get_request_context_route() -> str:
+    return _get_serve_request_context().route
+
+
+class TestHTTPRoute:
+    def test_basic_route_prefix(self):
+        @serve.deployment
+        class A:
+            def __call__(self) -> str:
+                return _get_request_context_route()
+
+        # No route prefix, should return "/" regardless of full route.
+        serve.run(A.bind())
+        r = httpx.get(f"{get_application_url()}/")
+        assert r.status_code == 200
+        assert r.text == "/"
+        assert httpx.get(f"{get_application_url()}/subpath").text == "/"
+
+        # Configured route prefix should be set.
+        serve.run(A.bind(), route_prefix="/prefix")
+        base_url = get_application_url(exclude_route_prefix=True)
+        assert httpx.get(f"{base_url}/prefix").text == "/prefix"
+        assert httpx.get(f"{base_url}/prefix/subpath").text == "/prefix"
+
+    def test_matching_fastapi_route(self):
+        fastapi_app = FastAPI()
+
+        @serve.deployment
+        @serve.ingress(fastapi_app)
+        class A:
+            @fastapi_app.get("/fastapi-path")
+            def root(self) -> str:
+                return PlainTextResponse(_get_request_context_route())
+
+            @fastapi_app.get("/dynamic/{user_id}")
+            def dynamic(self) -> str:
+                return PlainTextResponse(_get_request_context_route())
+
+        # No route prefix, should return matched fastapi route.
+        serve.run(A.bind())
+        assert (
+            httpx.get(f"{get_application_url()}/fastapi-path").text == "/fastapi-path"
+        )
+        assert (
+            httpx.get(f"{get_application_url()}/dynamic/abc123").text
+            == "/dynamic/{user_id}"
+        )
+
+        # Configured route prefix, should return matched route prefix + fastapi route.
+        serve.run(A.bind(), route_prefix="/prefix")
+        base_url = get_application_url(exclude_route_prefix=True)
+        assert (
+            httpx.get(f"{base_url}/prefix/fastapi-path").text == "/prefix/fastapi-path"
+        )
+        assert (
+            httpx.get(f"{base_url}/prefix/dynamic/abc123").text
+            == "/prefix/dynamic/{user_id}"
+        )
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-v", "-s", __file__]))