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

testing: Add host argument to SanicTestClient

Browse Source 
Adds the ability to specify a host argument when using the
SanicTestClient.

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
This commit is contained in:
Eli Uriegas 2019-12-18 16:02:33 -08:00
parent 028778ed56
commit c3aed01096
No known key found for this signature in database
GPG Key ID: EE2C3B4893010973
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
sanic/testing.py
View File

@ -22,10 +22,11 @@ PORT = 42101
class SanicTestClient: class SanicTestClient:
def __init__(self, app, port=PORT): def __init__(self, app, port=PORT, host=HOST):
"""Use port=None to bind to a random port""" """Use port=None to bind to a random port"""
self.app = app self.app = app
self.port = port self.port = port
self.host = host
def get_new_session(self): def get_new_session(self):
return requests.Session() return requests.Session()
@ -71,6 +72,7 @@ class SanicTestClient:
gather_request=True, gather_request=True,
debug=False, debug=False,
server_kwargs={"auto_reload": False}, server_kwargs={"auto_reload": False},
host=None,
*request_args, *request_args,
**request_kwargs, **request_kwargs,
): ):
@ -95,11 +97,13 @@ class SanicTestClient:
return self.app.error_handler.default(request, exception) return self.app.error_handler.default(request, exception)
if self.port: if self.port:
server_kwargs = dict(host=HOST, port=self.port, **server_kwargs) server_kwargs = dict(
host, port = HOST, self.port host=host or self.host, port=self.port, **server_kwargs
)
host, port = host or self.host, self.port
else: else:
sock = socket() sock = socket()
sock.bind((HOST, 0)) sock.bind((host or self.host, 0))
server_kwargs = dict(sock=sock, **server_kwargs) server_kwargs = dict(sock=sock, **server_kwargs)
host, port = sock.getsockname() host, port = sock.getsockname()