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, PORT as test_port 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_simple_url_for_getting_with_duplicate_params(simple_app): kw = dict(arg1=['value1', 'value2'], _anchor='anchor') for letter in string.ascii_letters: url = simple_app.url_for(letter, **kw) assert url == '/{}?arg1=value1&arg1=value2#anchor'.format(letter) request, response = sanic_endpoint_test(simple_app, uri=url) assert response.status == 200 assert response.text == letter def test_simple_url_for_getting_with_special_params(simple_app): kw = dict(arg1='value1', _anchor='anchor', _scheme='http', _server='localhost:{}'.format(test_port), _external=True) url_fmt = 'http://localhost:{}/{}?arg1=value1#anchor' for letter in string.ascii_letters: url = simple_app.url_for(letter, **kw) assert url == url_fmt.format(test_port, 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) def test_fails_url_build_if_params_not_passed(): app = Sanic('fail_url_build') @app.route('/fail') def fail(): return text('this should fail') with pytest.raises(ValueError) as e: app.url_for('fail', _scheme='http') assert str(e.value) == 'When specifying _scheme, _external must be True' COMPLEX_PARAM_URL = ( '///' '//') 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/') 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/') # 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'