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

condense methods

Browse Source
This commit is contained in:
Raphael Deem 2017-02-19 17:29:06 -08:00
parent 9c306899ba
commit 09942c7987
1 changed files with 16 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
sanic/testing.py
View File

@ -1,8 +1,19 @@
from sanic.log import log from sanic.log import log
from functools import partial
HOST = '127.0.0.1' HOST = '127.0.0.1'
PORT = 42101 PORT = 42101
SUPPORTED_METHODS = [
'get',
'post',
'put',
'delete',
'patch',
'options',
'head',
]
class TestClient: class TestClient:
def __init__(self, app): def __init__(self, app):
@ -69,23 +80,8 @@ class TestClient:
raise ValueError( raise ValueError(
"Request object expected, got ({})".format(results)) "Request object expected, got ({})".format(results))
def get(self, *args, **kwargs): def __getattr__(self, attr, *args, **kwargs):
return self._sanic_endpoint_test('get', *args, **kwargs) if attr in SUPPORTED_METHODS:
return partial(self._sanic_endpoint_test, attr, *args, **kwargs)
def post(self, *args, **kwargs): else:
return self._sanic_endpoint_test('post', *args, **kwargs) raise AttributeError
def put(self, *args, **kwargs):
return self._sanic_endpoint_test('put', *args, **kwargs)
def delete(self, *args, **kwargs):
return self._sanic_endpoint_test('delete', *args, **kwargs)
def patch(self, *args, **kwargs):
return self._sanic_endpoint_test('patch', *args, **kwargs)
def options(self, *args, **kwargs):
return self._sanic_endpoint_test('options', *args, **kwargs)
def head(self, *args, **kwargs):
return self._sanic_endpoint_test('head', *args, **kwargs)