From 09942c79871be0e61bc08abfb244d554c5f8c223 Mon Sep 17 00:00:00 2001 From: Raphael Deem Date: Sun, 19 Feb 2017 17:29:06 -0800 Subject: [PATCH] condense methods --- sanic/testing.py | 36 ++++++++++++++++-------------------- 1 file changed, 16 insertions(+), 20 deletions(-) diff --git a/sanic/testing.py b/sanic/testing.py index 52e70f5a..109b44e0 100644 --- a/sanic/testing.py +++ b/sanic/testing.py @@ -1,8 +1,19 @@ from sanic.log import log +from functools import partial HOST = '127.0.0.1' PORT = 42101 +SUPPORTED_METHODS = [ + 'get', + 'post', + 'put', + 'delete', + 'patch', + 'options', + 'head', +] + class TestClient: def __init__(self, app): @@ -69,23 +80,8 @@ class TestClient: raise ValueError( "Request object expected, got ({})".format(results)) - def get(self, *args, **kwargs): - return self._sanic_endpoint_test('get', *args, **kwargs) - - def post(self, *args, **kwargs): - return self._sanic_endpoint_test('post', *args, **kwargs) - - 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) + def __getattr__(self, attr, *args, **kwargs): + if attr in SUPPORTED_METHODS: + return partial(self._sanic_endpoint_test, attr, *args, **kwargs) + else: + raise AttributeError