wehub-resource-sync
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
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__]))
|