From 78b173de06afb3b0ead455a634594816d4333c93 Mon Sep 17 00:00:00 2001 From: Rokibat Oyeranmi Date: Mon, 10 Aug 2026 06:22:40 -0700 Subject: [PATCH] Fix streaming context test cleanup --- tests/test_helpers.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tests/test_helpers.py b/tests/test_helpers.py index c195b0280c..ed564536ed 100644 --- a/tests/test_helpers.py +++ b/tests/test_helpers.py @@ -278,7 +278,9 @@ def generate(): yield flask.request.args["name"] yield "!" - return flask.Response(flask.stream_with_context(Wrapper(generate()))) + return flask.Response( + flask.stream_with_context(Wrapper(generate())) + ) rv = client.get("/?name=World") assert rv.data == b"Hello World!" @@ -310,20 +312,22 @@ def gen(): return flask.Response(gen()) # response is closed without reading stream - client.get().close() + client.get("/").close() # response stream is read - with client.get() as rv: + with client.get("/") as rv: assert rv.text == "flask" # same as above, but with client context preservation with client: - client.get().close() + client.get("/").close() - with client, client.get() as rv: + with client, client.get("/") as rv: assert rv.text == "flask" + + class TestHelpers: @pytest.mark.parametrize( ("debug", "expect"),