wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-09 12:20:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Serve] Migrate from Flask.Request to Starlette Request (#12852)

Browse Source
This commit is contained in:
architkulkarni
2020-12-21 15:34:15 -06:00
committed by GitHub
parent 5b48480e29
commit 8b4b4bf0a2
26 changed files with 140 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serve/tests/test_handle.py
+8 -8
View File
@@ -8,7 +8,7 @@ def test_handle_in_endpoint(serve_instance):
client = serve_instance
class Endpoint1:
def __call__(self, flask_request):
def __call__(self, starlette_request):
return "hello"
class Endpoint2:
@@ -40,12 +40,12 @@ def test_handle_http_args(serve_instance):
client = serve_instance
class Endpoint:
def __call__(self, request):
async def __call__(self, request):
return {
"args": dict(request.args),
"args": dict(request.query_params),
"headers": dict(request.headers),
"method": request.method,
"json": request.json
"json": await request.json()
}
client.create_backend("backend", Endpoint)
@@ -58,7 +58,7 @@ def test_handle_http_args(serve_instance):
"arg2": "2"
},
"headers": {
"X-Custom-Header": "value"
"x-custom-header": "value"
},
"method": "POST",
"json": {
@@ -81,10 +81,10 @@ def test_handle_http_args(serve_instance):
for resp in [resp_web, resp_handle]:
for field in ["args", "method", "json"]:
assert resp[field] == ground_truth[field]
resp["headers"]["X-Custom-Header"] == "value"
resp["headers"]["x-custom-header"] == "value"
def test_handle_inject_flask_request(serve_instance):
def test_handle_inject_starlette_request(serve_instance):
client = serve_instance
def echo_request_type(request):
@@ -103,7 +103,7 @@ def test_handle_inject_flask_request(serve_instance):
for route in ["/echo", "/wrapper"]:
resp = requests.get(f"http://127.0.0.1:8000{route}")
request_type = resp.text
assert request_type == "<class 'flask.wrappers.Request'>"
assert request_type == "<class 'starlette.requests.Request'>"
if __name__ == "__main__":