262 lines
6.7 KiB
Python
262 lines
6.7 KiB
Python
import pytest as pytest
|
|
from urllib.parse import urlsplit, parse_qsl
|
|
|
|
from sanic import Sanic
|
|
from sanic.response import text
|
|
from sanic.views import HTTPMethodView
|
|
from sanic.blueprints import Blueprint
|
|
from sanic.utils import sanic_endpoint_test
|
|
from sanic.exceptions import URLBuildError
|
|
|
|
import string
|
|
|
|
|
|
def _generate_handlers_from_names(app, l):
|
|
for name in l:
|
|
# this is the easiest way to generate functions with dynamic names
|
|
exec('@app.route(name)\ndef {}(request):\n\treturn text("{}")'.format(
|
|
name, name))
|
|
|
|
|
|
@pytest.fixture
|
|
def simple_app():
|
|
app = Sanic('simple_app')
|
|
handler_names = list(string.ascii_letters)
|
|
|
|
_generate_handlers_from_names(app, handler_names)
|
|
|
|
return app
|
|
|
|
|
|
def test_simple_url_for_getting(simple_app):
|
|
for letter in string.ascii_letters:
|
|
url = simple_app.url_for(letter)
|
|
|
|
assert url == '/{}'.format(letter)
|
|
request, response = sanic_endpoint_test(
|
|
simple_app, uri=url)
|
|
assert response.status == 200
|
|
assert response.text == letter
|
|
|
|
|
|
def test_fails_if_endpoint_not_found():
|
|
app = Sanic('fail_url_build')
|
|
|
|
@app.route('/fail')
|
|
def fail():
|
|
return text('this should fail')
|
|
|
|
with pytest.raises(URLBuildError) as e:
|
|
app.url_for('passes')
|
|
|
|
assert str(e.value) == 'Endpoint with name `passes` was not found'
|
|
|
|
|
|
def test_fails_url_build_if_param_not_passed():
|
|
url = '/'
|
|
|
|
for letter in string.ascii_letters:
|
|
url += '<{}>/'.format(letter)
|
|
|
|
app = Sanic('fail_url_build')
|
|
|
|
@app.route(url)
|
|
def fail():
|
|
return text('this should fail')
|
|
|
|
fail_args = list(string.ascii_letters)
|
|
fail_args.pop()
|
|
|
|
fail_kwargs = {l: l for l in fail_args}
|
|
|
|
with pytest.raises(URLBuildError) as e:
|
|
app.url_for('fail', **fail_kwargs)
|
|
|
|
assert 'Required parameter `Z` was not passed to url_for' in str(e.value)
|
|
|
|
|
|
COMPLEX_PARAM_URL = (
|
|
'/<foo:int>/<four_letter_string:[A-z]{4}>/'
|
|
'<two_letter_string:[A-z]{2}>/<normal_string>/<some_number:number>')
|
|
PASSING_KWARGS = {
|
|
'foo': 4, 'four_letter_string': 'woof',
|
|
'two_letter_string': 'ba', 'normal_string': 'normal',
|
|
'some_number': '1.001'}
|
|
EXPECTED_BUILT_URL = '/4/woof/ba/normal/1.001'
|
|
|
|
|
|
def test_fails_with_int_message():
|
|
app = Sanic('fail_url_build')
|
|
|
|
@app.route(COMPLEX_PARAM_URL)
|
|
def fail():
|
|
return text('this should fail')
|
|
|
|
failing_kwargs = dict(PASSING_KWARGS)
|
|
failing_kwargs['foo'] = 'not_int'
|
|
|
|
with pytest.raises(URLBuildError) as e:
|
|
app.url_for('fail', **failing_kwargs)
|
|
|
|
expected_error = (
|
|
'Value "not_int" for parameter `foo` '
|
|
'does not match pattern for type `int`: \d+')
|
|
assert str(e.value) == expected_error
|
|
|
|
|
|
def test_fails_with_two_letter_string_message():
|
|
app = Sanic('fail_url_build')
|
|
|
|
@app.route(COMPLEX_PARAM_URL)
|
|
def fail():
|
|
return text('this should fail')
|
|
|
|
failing_kwargs = dict(PASSING_KWARGS)
|
|
failing_kwargs['two_letter_string'] = 'foobar'
|
|
|
|
with pytest.raises(URLBuildError) as e:
|
|
app.url_for('fail', **failing_kwargs)
|
|
|
|
expected_error = (
|
|
'Value "foobar" for parameter `two_letter_string` '
|
|
'does not satisfy pattern [A-z]{2}')
|
|
|
|
assert str(e.value) == expected_error
|
|
|
|
|
|
def test_fails_with_number_message():
|
|
app = Sanic('fail_url_build')
|
|
|
|
@app.route(COMPLEX_PARAM_URL)
|
|
def fail():
|
|
return text('this should fail')
|
|
|
|
failing_kwargs = dict(PASSING_KWARGS)
|
|
failing_kwargs['some_number'] = 'foo'
|
|
|
|
with pytest.raises(URLBuildError) as e:
|
|
app.url_for('fail', **failing_kwargs)
|
|
|
|
expected_error = (
|
|
'Value "foo" for parameter `some_number` '
|
|
'does not match pattern for type `float`: [0-9\\\\.]+')
|
|
|
|
assert str(e.value) == expected_error
|
|
|
|
|
|
def test_adds_other_supplied_values_as_query_string():
|
|
app = Sanic('passes')
|
|
|
|
@app.route(COMPLEX_PARAM_URL)
|
|
def passes():
|
|
return text('this should pass')
|
|
|
|
new_kwargs = dict(PASSING_KWARGS)
|
|
new_kwargs['added_value_one'] = 'one'
|
|
new_kwargs['added_value_two'] = 'two'
|
|
|
|
url = app.url_for('passes', **new_kwargs)
|
|
|
|
query = dict(parse_qsl(urlsplit(url).query))
|
|
|
|
assert query['added_value_one'] == 'one'
|
|
assert query['added_value_two'] == 'two'
|
|
|
|
|
|
@pytest.fixture
|
|
def blueprint_app():
|
|
app = Sanic('blueprints')
|
|
|
|
first_print = Blueprint('first', url_prefix='/first')
|
|
second_print = Blueprint('second', url_prefix='/second')
|
|
|
|
@first_print.route('/foo')
|
|
def foo():
|
|
return text('foo from first')
|
|
|
|
@first_print.route('/foo/<param>')
|
|
def foo_with_param(request, param):
|
|
return text(
|
|
'foo from first : {}'.format(param))
|
|
|
|
@second_print.route('/foo') # noqa
|
|
def foo():
|
|
return text('foo from second')
|
|
|
|
@second_print.route('/foo/<param>') # noqa
|
|
def foo_with_param(request, param):
|
|
return text(
|
|
'foo from second : {}'.format(param))
|
|
|
|
app.blueprint(first_print)
|
|
app.blueprint(second_print)
|
|
|
|
return app
|
|
|
|
|
|
def test_blueprints_are_named_correctly(blueprint_app):
|
|
first_url = blueprint_app.url_for('first.foo')
|
|
assert first_url == '/first/foo'
|
|
|
|
second_url = blueprint_app.url_for('second.foo')
|
|
assert second_url == '/second/foo'
|
|
|
|
|
|
def test_blueprints_work_with_params(blueprint_app):
|
|
first_url = blueprint_app.url_for('first.foo_with_param', param='bar')
|
|
assert first_url == '/first/foo/bar'
|
|
|
|
second_url = blueprint_app.url_for('second.foo_with_param', param='bar')
|
|
assert second_url == '/second/foo/bar'
|
|
|
|
|
|
@pytest.fixture
|
|
def methodview_app():
|
|
app = Sanic('methodview')
|
|
|
|
class ViewOne(HTTPMethodView):
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
def post(self, request):
|
|
return text('I am post method')
|
|
|
|
def put(self, request):
|
|
return text('I am put method')
|
|
|
|
def patch(self, request):
|
|
return text('I am patch method')
|
|
|
|
def delete(self, request):
|
|
return text('I am delete method')
|
|
|
|
app.add_route(ViewOne.as_view('view_one'), '/view_one')
|
|
|
|
class ViewTwo(HTTPMethodView):
|
|
def get(self, request):
|
|
return text('I am get method')
|
|
|
|
def post(self, request):
|
|
return text('I am post method')
|
|
|
|
def put(self, request):
|
|
return text('I am put method')
|
|
|
|
def patch(self, request):
|
|
return text('I am patch method')
|
|
|
|
def delete(self, request):
|
|
return text('I am delete method')
|
|
|
|
app.add_route(ViewTwo.as_view(), '/view_two')
|
|
|
|
return app
|
|
|
|
|
|
def test_methodview_naming(methodview_app):
|
|
viewone_url = methodview_app.url_for('ViewOne')
|
|
viewtwo_url = methodview_app.url_for('ViewTwo')
|
|
|
|
assert viewone_url == '/view_one'
|
|
assert viewtwo_url == '/view_two'
|