diff --git a/docs/reqcontext.rst b/docs/reqcontext.rst index f109bd27..0dc62e93 100644 --- a/docs/reqcontext.rst +++ b/docs/reqcontext.rst @@ -170,8 +170,8 @@ will not fail. During testing, it can be useful to defer popping the contexts after the request ends, so that their data can be accessed in the test function. -Using the :meth:`~Flask.test_client` as a ``with`` block to preserve the -contexts until the with block exits. +Use the :meth:`~Flask.test_client` as a ``with`` block to preserve the +contexts until the ``with`` block exits. .. code-block:: python diff --git a/tests/test_helpers.py b/tests/test_helpers.py index 78a56221..21735af1 100644 --- a/tests/test_helpers.py +++ b/tests/test_helpers.py @@ -11,6 +11,7 @@ import datetime import io import os +import sys import uuid import pytest @@ -780,12 +781,17 @@ class TestSendfile(object): assert rv.data.strip() == b"Hello Subdomain" rv.close() - def test_send_from_directory_bad_request(self, app, req_ctx): + def test_send_from_directory_null_character(self, app, req_ctx): app.root_path = os.path.join( os.path.dirname(__file__), "test_apps", "subdomaintestmodule" ) - with pytest.raises(BadRequest): + if sys.version_info >= (3, 8): + exception = NotFound + else: + exception = BadRequest + + with pytest.raises(exception): flask.send_from_directory("static", "bad\x00")