diff --git a/tests/contrib/sanic/test_sanic.py b/tests/contrib/sanic/test_sanic.py index e0dfa3dbe5d..d1972a85191 100644 --- a/tests/contrib/sanic/test_sanic.py +++ b/tests/contrib/sanic/test_sanic.py @@ -8,7 +8,6 @@ from sanic.config import DEFAULT_CONFIG from sanic.exceptions import ServerError from sanic.response import json -from sanic.response import stream from sanic.response import text from sanic.server import HttpProtocol @@ -27,6 +26,18 @@ sanic_version = tuple(map(int, sanic_version.split("."))) +try: + from sanic.response import ResponseStream + + def stream(*args, **kwargs): + return ResponseStream(*args, **kwargs) + + +except ImportError: + # stream was removed in sanic v22.6.0 + from sanic.response import stream + + def _response_status(response): return getattr(response, "status_code", getattr(response, "status", None))