96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
|
from json import loads as json_loads, dumps as json_dumps
|
||
|
|
from sanic import Sanic
|
||
|
|
from sanic.response import json, text
|
||
|
|
from helpers import sanic_endpoint_test
|
||
|
|
|
||
|
|
# ------------------------------------------------------------ #
|
||
|
|
# UTF-8
|
||
|
|
# ------------------------------------------------------------ #
|
||
|
|
|
||
|
|
def test_dynamic_route():
|
||
|
|
app = Sanic('test_dynamic_route')
|
||
|
|
|
||
|
|
results = []
|
||
|
|
|
||
|
|
@app.route('/folder/<name>')
|
||
|
|
async def handler(request, name):
|
||
|
|
results.append(name)
|
||
|
|
return text('OK')
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/test123')
|
||
|
|
|
||
|
|
assert response.text == 'OK'
|
||
|
|
assert results[0] == 'test123'
|
||
|
|
|
||
|
|
def test_dynamic_route_string():
|
||
|
|
app = Sanic('test_dynamic_route_string')
|
||
|
|
|
||
|
|
results = []
|
||
|
|
|
||
|
|
@app.route('/folder/<name:string>')
|
||
|
|
async def handler(request, name):
|
||
|
|
results.append(name)
|
||
|
|
return text('OK')
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/test123')
|
||
|
|
|
||
|
|
assert response.text == 'OK'
|
||
|
|
assert results[0] == 'test123'
|
||
|
|
|
||
|
|
def test_dynamic_route_int():
|
||
|
|
app = Sanic('test_dynamic_route_int')
|
||
|
|
|
||
|
|
results = []
|
||
|
|
|
||
|
|
@app.route('/folder/<folder_id:int>')
|
||
|
|
async def handler(request, folder_id):
|
||
|
|
results.append(folder_id)
|
||
|
|
return text('OK')
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/12345')
|
||
|
|
assert response.text == 'OK'
|
||
|
|
assert type(results[0]) is int
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/asdf')
|
||
|
|
assert response.status == 404
|
||
|
|
|
||
|
|
|
||
|
|
def test_dynamic_route_number():
|
||
|
|
app = Sanic('test_dynamic_route_int')
|
||
|
|
|
||
|
|
results = []
|
||
|
|
|
||
|
|
@app.route('/weight/<weight:number>')
|
||
|
|
async def handler(request, weight):
|
||
|
|
results.append(weight)
|
||
|
|
return text('OK')
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/weight/12345')
|
||
|
|
assert response.text == 'OK'
|
||
|
|
assert type(results[0]) is float
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/weight/1234.56')
|
||
|
|
assert response.status == 200
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/weight/1234-56')
|
||
|
|
assert response.status == 404
|
||
|
|
|
||
|
|
def test_dynamic_route_regex():
|
||
|
|
app = Sanic('test_dynamic_route_int')
|
||
|
|
|
||
|
|
@app.route('/folder/<folder_id:[A-Za-z0-9]{0,4}>')
|
||
|
|
async def handler(request, folder_id):
|
||
|
|
return text('OK')
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/test')
|
||
|
|
assert response.status == 200
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/test1')
|
||
|
|
assert response.status == 404
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/test-123')
|
||
|
|
assert response.status == 404
|
||
|
|
|
||
|
|
request, response = sanic_endpoint_test(app, uri='/folder/')
|
||
|
|
assert response.status == 200
|