253 lines
6.7 KiB
Python
253 lines
6.7 KiB
Python
import pytest as pytest
|
|
|
|
from sanic import Sanic
|
|
from sanic.exceptions import InvalidUsage
|
|
from sanic.response import text, HTTPResponse
|
|
from sanic.views import HTTPMethodView, CompositionView
|
|
from sanic.blueprints import Blueprint
|
|
from sanic.request import Request
|
|
from sanic.utils import sanic_endpoint_test
|
|
from sanic.constants import HTTP_METHODS
|
|
|
|
|
|
@pytest.mark.parametrize('method', HTTP_METHODS)
|
|
def test_methods(method):
|
|
app = Sanic('test_methods')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request):
|
|
return text('', headers={'method': 'GET'})
|
|
|
|
def post(self, request):
|
|
return text('', headers={'method': 'POST'})
|
|
|
|
def put(self, request):
|
|
return text('', headers={'method': 'PUT'})
|
|
|
|
def head(self, request):
|
|
return text('', headers={'method': 'HEAD'})
|
|
|
|
def options(self, request):
|
|
return text('', headers={'method': 'OPTIONS'})
|
|
|
|
def patch(self, request):
|
|
return text('', headers={'method': 'PATCH'})
|
|
|
|
def delete(self, request):
|
|
return text('', headers={'method': 'DELETE'})
|
|
|
|
app.add_route(DummyView.as_view(), '/')
|
|
|
|
request, response = sanic_endpoint_test(app, method=method)
|
|
assert response.headers['method'] == method
|
|
|
|
|
|
def test_unexisting_methods():
|
|
app = Sanic('test_unexisting_methods')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
app.add_route(DummyView.as_view(), '/')
|
|
request, response = sanic_endpoint_test(app, method="get")
|
|
assert response.text == 'I am get method'
|
|
request, response = sanic_endpoint_test(app, method="post")
|
|
assert response.text == 'Error: Method POST not allowed for URL /'
|
|
|
|
|
|
def test_argument_methods():
|
|
app = Sanic('test_argument_methods')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request, my_param_here):
|
|
return text('I am get method with %s' % my_param_here)
|
|
|
|
app.add_route(DummyView.as_view(), '/<my_param_here>')
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/test123')
|
|
|
|
assert response.text == 'I am get method with test123'
|
|
|
|
|
|
def test_with_bp():
|
|
app = Sanic('test_with_bp')
|
|
bp = Blueprint('test_text')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
bp.add_route(DummyView.as_view(), '/')
|
|
|
|
app.blueprint(bp)
|
|
request, response = sanic_endpoint_test(app)
|
|
|
|
assert response.text == 'I am get method'
|
|
|
|
|
|
def test_with_bp_with_url_prefix():
|
|
app = Sanic('test_with_bp_with_url_prefix')
|
|
bp = Blueprint('test_text', url_prefix='/test1')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
bp.add_route(DummyView.as_view(), '/')
|
|
|
|
app.blueprint(bp)
|
|
request, response = sanic_endpoint_test(app, uri='/test1/')
|
|
|
|
assert response.text == 'I am get method'
|
|
|
|
|
|
def test_with_middleware():
|
|
app = Sanic('test_with_middleware')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
app.add_route(DummyView.as_view(), '/')
|
|
|
|
results = []
|
|
|
|
@app.middleware
|
|
async def handler(request):
|
|
results.append(request)
|
|
|
|
request, response = sanic_endpoint_test(app)
|
|
|
|
assert response.text == 'I am get method'
|
|
assert type(results[0]) is Request
|
|
|
|
|
|
def test_with_middleware_response():
|
|
app = Sanic('test_with_middleware_response')
|
|
|
|
results = []
|
|
|
|
@app.middleware('request')
|
|
async def process_response(request):
|
|
results.append(request)
|
|
|
|
@app.middleware('response')
|
|
async def process_response(request, response):
|
|
results.append(request)
|
|
results.append(response)
|
|
|
|
class DummyView(HTTPMethodView):
|
|
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
app.add_route(DummyView.as_view(), '/')
|
|
|
|
request, response = sanic_endpoint_test(app)
|
|
|
|
assert response.text == 'I am get method'
|
|
assert type(results[0]) is Request
|
|
assert type(results[1]) is Request
|
|
assert isinstance(results[2], HTTPResponse)
|
|
|
|
|
|
def test_with_custom_class_methods():
|
|
app = Sanic('test_with_custom_class_methods')
|
|
|
|
class DummyView(HTTPMethodView):
|
|
global_var = 0
|
|
|
|
def _iternal_method(self):
|
|
self.global_var += 10
|
|
|
|
def get(self, request):
|
|
self._iternal_method()
|
|
return text('I am get method and global var is {}'.format(self.global_var))
|
|
|
|
app.add_route(DummyView.as_view(), '/')
|
|
request, response = sanic_endpoint_test(app, method="get")
|
|
assert response.text == 'I am get method and global var is 10'
|
|
|
|
|
|
def test_with_decorator():
|
|
app = Sanic('test_with_decorator')
|
|
|
|
results = []
|
|
|
|
def stupid_decorator(view):
|
|
def decorator(*args, **kwargs):
|
|
results.append(1)
|
|
return view(*args, **kwargs)
|
|
return decorator
|
|
|
|
class DummyView(HTTPMethodView):
|
|
decorators = [stupid_decorator]
|
|
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
app.add_route(DummyView.as_view(), '/')
|
|
request, response = sanic_endpoint_test(app, method="get")
|
|
assert response.text == 'I am get method'
|
|
assert results[0] == 1
|
|
|
|
|
|
def test_composition_view_rejects_incorrect_methods():
|
|
def foo(request):
|
|
return text('Foo')
|
|
|
|
view = CompositionView()
|
|
|
|
with pytest.raises(InvalidUsage) as e:
|
|
view.add(['GET', 'FOO'], foo)
|
|
|
|
assert str(e.value) == 'FOO is not a valid HTTP method.'
|
|
|
|
|
|
def test_composition_view_rejects_duplicate_methods():
|
|
def foo(request):
|
|
return text('Foo')
|
|
|
|
view = CompositionView()
|
|
|
|
with pytest.raises(InvalidUsage) as e:
|
|
view.add(['GET', 'POST', 'GET'], foo)
|
|
|
|
assert str(e.value) == 'Method GET is already registered.'
|
|
|
|
|
|
def test_composition_view_runs_methods_as_expected():
|
|
app = Sanic('test_composition_view')
|
|
|
|
view = CompositionView()
|
|
view.add(['GET', 'POST', 'PUT'], lambda x: text('first method'))
|
|
view.add(['DELETE', 'PATCH'], lambda x: text('second method'))
|
|
|
|
app.add_route(view, '/')
|
|
|
|
for method in ['GET', 'POST', 'PUT']:
|
|
request, response = sanic_endpoint_test(app, uri='/', method=method)
|
|
assert response.text == 'first method'
|
|
|
|
for method in ['DELETE', 'PATCH']:
|
|
request, response = sanic_endpoint_test(app, uri='/', method=method)
|
|
assert response.text == 'second method'
|
|
|
|
def test_composition_view_rejects_invalid_methods():
|
|
app = Sanic('test_composition_view')
|
|
|
|
view = CompositionView()
|
|
view.add(['GET', 'POST', 'PUT'], lambda x: text('first method'))
|
|
|
|
app.add_route(view, '/')
|
|
for method in ['DELETE', 'PATCH']:
|
|
request, response = sanic_endpoint_test(app, uri='/', method=method)
|
|
assert response.status == 405
|