165 lines
4.1 KiB
Python
165 lines
4.1 KiB
Python
import inspect
|
|
|
|
from sanic import Sanic
|
|
from sanic.blueprints import Blueprint
|
|
from sanic.response import json, text
|
|
from sanic.utils import sanic_endpoint_test
|
|
from sanic.exceptions import NotFound, ServerError, InvalidUsage
|
|
|
|
|
|
# ------------------------------------------------------------ #
|
|
# GET
|
|
# ------------------------------------------------------------ #
|
|
|
|
def test_bp():
|
|
app = Sanic('test_text')
|
|
bp = Blueprint('test_text')
|
|
|
|
@bp.route('/')
|
|
def handler(request):
|
|
return text('Hello')
|
|
|
|
app.blueprint(bp)
|
|
request, response = sanic_endpoint_test(app)
|
|
|
|
assert response.text == 'Hello'
|
|
|
|
def test_bp_with_url_prefix():
|
|
app = Sanic('test_text')
|
|
bp = Blueprint('test_text', url_prefix='/test1')
|
|
|
|
@bp.route('/')
|
|
def handler(request):
|
|
return text('Hello')
|
|
|
|
app.blueprint(bp)
|
|
request, response = sanic_endpoint_test(app, uri='/test1/')
|
|
|
|
assert response.text == 'Hello'
|
|
|
|
|
|
def test_several_bp_with_url_prefix():
|
|
app = Sanic('test_text')
|
|
bp = Blueprint('test_text', url_prefix='/test1')
|
|
bp2 = Blueprint('test_text2', url_prefix='/test2')
|
|
|
|
@bp.route('/')
|
|
def handler(request):
|
|
return text('Hello')
|
|
|
|
@bp2.route('/')
|
|
def handler2(request):
|
|
return text('Hello2')
|
|
|
|
app.blueprint(bp)
|
|
app.blueprint(bp2)
|
|
request, response = sanic_endpoint_test(app, uri='/test1/')
|
|
assert response.text == 'Hello'
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/test2/')
|
|
assert response.text == 'Hello2'
|
|
|
|
|
|
def test_bp_middleware():
|
|
app = Sanic('test_middleware')
|
|
blueprint = Blueprint('test_middleware')
|
|
|
|
@blueprint.middleware('response')
|
|
async def process_response(request, response):
|
|
return text('OK')
|
|
|
|
@app.route('/')
|
|
async def handler(request):
|
|
return text('FAIL')
|
|
|
|
app.blueprint(blueprint)
|
|
|
|
request, response = sanic_endpoint_test(app)
|
|
|
|
assert response.status == 200
|
|
assert response.text == 'OK'
|
|
|
|
def test_bp_exception_handler():
|
|
app = Sanic('test_middleware')
|
|
blueprint = Blueprint('test_middleware')
|
|
|
|
@blueprint.route('/1')
|
|
def handler_1(request):
|
|
raise InvalidUsage("OK")
|
|
|
|
@blueprint.route('/2')
|
|
def handler_2(request):
|
|
raise ServerError("OK")
|
|
|
|
@blueprint.route('/3')
|
|
def handler_3(request):
|
|
raise NotFound("OK")
|
|
|
|
@blueprint.exception(NotFound, ServerError)
|
|
def handler_exception(request, exception):
|
|
return text("OK")
|
|
|
|
app.blueprint(blueprint)
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/1')
|
|
assert response.status == 400
|
|
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/2')
|
|
assert response.status == 200
|
|
assert response.text == 'OK'
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/3')
|
|
assert response.status == 200
|
|
|
|
def test_bp_listeners():
|
|
app = Sanic('test_middleware')
|
|
blueprint = Blueprint('test_middleware')
|
|
|
|
order = []
|
|
|
|
@blueprint.listener('before_server_start')
|
|
def handler_1(sanic, loop):
|
|
order.append(1)
|
|
|
|
@blueprint.listener('after_server_start')
|
|
def handler_2(sanic, loop):
|
|
order.append(2)
|
|
|
|
@blueprint.listener('after_server_start')
|
|
def handler_3(sanic, loop):
|
|
order.append(3)
|
|
|
|
@blueprint.listener('before_server_stop')
|
|
def handler_4(sanic, loop):
|
|
order.append(5)
|
|
|
|
@blueprint.listener('before_server_stop')
|
|
def handler_5(sanic, loop):
|
|
order.append(4)
|
|
|
|
@blueprint.listener('after_server_stop')
|
|
def handler_6(sanic, loop):
|
|
order.append(6)
|
|
|
|
app.blueprint(blueprint)
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/')
|
|
|
|
assert order == [1,2,3,4,5,6]
|
|
|
|
def test_bp_static():
|
|
current_file = inspect.getfile(inspect.currentframe())
|
|
with open(current_file, 'rb') as file:
|
|
current_file_contents = file.read()
|
|
|
|
app = Sanic('test_static')
|
|
blueprint = Blueprint('test_static')
|
|
|
|
blueprint.static(current_file, '/testing.file')
|
|
|
|
app.blueprint(blueprint)
|
|
|
|
request, response = sanic_endpoint_test(app, uri='/testing.file')
|
|
assert response.status == 200
|
|
assert response.body == current_file_contents |