From 1384e55d636f385b246b621ebddbc5a3d7674df1 Mon Sep 17 00:00:00 2001 From: p1c2u Date: Tue, 26 Sep 2023 13:33:29 +0000 Subject: [PATCH] Starlette binary request test fix --- .../starlette/data/v3.0/starletteproject/pets/endpoints.py | 6 ++---- .../integration/contrib/starlette/test_starlette_project.py | 5 +---- 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/tests/integration/contrib/starlette/data/v3.0/starletteproject/pets/endpoints.py b/tests/integration/contrib/starlette/data/v3.0/starletteproject/pets/endpoints.py index 535da4e5..eb0b2c76 100644 --- a/tests/integration/contrib/starlette/data/v3.0/starletteproject/pets/endpoints.py +++ b/tests/integration/contrib/starlette/data/v3.0/starletteproject/pets/endpoints.py @@ -26,10 +26,8 @@ def pet_photo_endpoint(request): if request.method == "GET": response = StreamingResponse([OPENID_LOGO], media_type="image/gif") elif request.method == "POST": - with request.form() as form: - filename = form["file"].filename - contents = form["file"].read() - response = Response(status_code=201) + contents = request.body() + response = Response(status_code=201) openapi_response = StarletteOpenAPIResponse(response) response_unmarshalled = unmarshal_response( openapi_request, openapi_response, spec=spec diff --git a/tests/integration/contrib/starlette/test_starlette_project.py b/tests/integration/contrib/starlette/test_starlette_project.py index ba406b53..c4783208 100644 --- a/tests/integration/contrib/starlette/test_starlette_project.py +++ b/tests/integration/contrib/starlette/test_starlette_project.py @@ -69,15 +69,12 @@ def test_post_valid(self, client, data_gif): "Api-Key": self.api_key_encoded, "Content-Type": content_type, } - data = { - "file": data_gif, - } cookies = {"user": "1"} response = client.post( "/v1/pets/1/photo", headers=headers, - data=data, + data=data_gif, cookies=cookies, )