Vasanko/sanic
2
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Adds testing manager to testing client property

Browse Source
This commit is contained in:
Adam Hopkins 2021-01-28 09:33:49 +02:00
parent 5545264cea
commit 693f2a5014
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sanic/app.py
View File

@ -86,6 +86,7 @@ class Sanic:
self.websocket_tasks: Set[Future] = set() self.websocket_tasks: Set[Future] = set()
self.named_request_middleware: Dict[str, MiddlewareType] = {} self.named_request_middleware: Dict[str, MiddlewareType] = {}
self.named_response_middleware: Dict[str, MiddlewareType] = {} self.named_response_middleware: Dict[str, MiddlewareType] = {}
self._test_manager = None
self._test_client = None self._test_client = None
self._asgi_client = None self._asgi_client = None
# Register alternative method names # Register alternative method names
@ -1035,6 +1036,8 @@ class Sanic:
def test_client(self): def test_client(self):
if self._test_client: if self._test_client:
return self._test_client return self._test_client
elif self._test_manager:
return self._test_manager.test_client
from sanic_testing.testing import SanicTestClient # type: ignore from sanic_testing.testing import SanicTestClient # type: ignore
self._test_client = SanicTestClient(self) self._test_client = SanicTestClient(self)
@ -1044,6 +1047,8 @@ class Sanic:
def asgi_client(self): def asgi_client(self):
if self._asgi_client: if self._asgi_client:
return self._asgi_client return self._asgi_client
elif self._test_manager:
return self._test_manager.test_client
from sanic_testing.testing import SanicASGITestClient # type: ignore from sanic_testing.testing import SanicASGITestClient # type: ignore
self._asgi_client = SanicASGITestClient(self) self._asgi_client = SanicASGITestClient(self)