From 521ae7f60e393d2c8215399570ba220a03088087 Mon Sep 17 00:00:00 2001 From: Adam Hopkins Date: Tue, 14 Jul 2020 10:41:28 +0300 Subject: [PATCH] squash --- sanic/testing.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/sanic/testing.py b/sanic/testing.py index 910b5c59..faabdfd1 100644 --- a/sanic/testing.py +++ b/sanic/testing.py @@ -9,6 +9,7 @@ from sanic.exceptions import MethodNotSupported from sanic.log import logger from sanic.response import text + ASGI_HOST = "mockserver" HOST = "127.0.0.1" PORT = None @@ -94,7 +95,9 @@ class SanicTestClient: @self.app.exception(MethodNotSupported) async def error_handler(request, exception): if request.method in ["HEAD", "PATCH", "PUT", "DELETE"]: - return text("", exception.status_code, headers=exception.headers) + return text( + "", exception.status_code, headers=exception.headers + ) else: return self.app.error_handler.default(request, exception) @@ -110,7 +113,9 @@ class SanicTestClient: host, port = sock.getsockname() self.port = port - if uri.startswith(("http:", "https:", "ftp:", "ftps://", "//", "ws:", "wss:")): + if uri.startswith( + ("http:", "https:", "ftp:", "ftps://", "//", "ws:", "wss:") + ): url = uri else: uri = uri if uri.startswith("/") else f"/{uri}" @@ -242,7 +247,9 @@ class SanicASGITestClient(httpx.AsyncClient): headers.setdefault("sec-websocket-key", "testserver==") headers.setdefault("sec-websocket-version", "13") if subprotocols is not None: - headers.setdefault("sec-websocket-protocol", ", ".join(subprotocols)) + headers.setdefault( + "sec-websocket-protocol", ", ".join(subprotocols) + ) scope = { "type": "websocket",