diff --git a/tests/contrib/sanic/test_sanic.py b/tests/contrib/sanic/test_sanic.py index 6e9c484ba972..7fe3b6f6a1a5 100644 --- a/tests/contrib/sanic/test_sanic.py +++ b/tests/contrib/sanic/test_sanic.py @@ -9,7 +9,6 @@ from sanic.exceptions import InvalidUsage 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 @@ -28,6 +27,16 @@ 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))