80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
import logging
|
|
|
|
import pytest
|
|
|
|
from sanic_routing.exceptions import NotFound
|
|
|
|
from sanic.signals import RESERVED_NAMESPACES
|
|
from sanic.touchup import TouchUp
|
|
|
|
|
|
def test_touchup_methods(app):
|
|
assert len(TouchUp._registry) == 9
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"verbosity,result", ((0, False), (1, False), (2, True), (3, True))
|
|
)
|
|
async def test_ode_removes_dispatch_events(app, caplog, verbosity, result):
|
|
with caplog.at_level(logging.DEBUG, logger="sanic.root"):
|
|
app.state.verbosity = verbosity
|
|
await app._startup()
|
|
logs = caplog.record_tuples
|
|
|
|
for signal in RESERVED_NAMESPACES["http"]:
|
|
assert (
|
|
(
|
|
"sanic.root",
|
|
logging.DEBUG,
|
|
f"Disabling event: {signal}",
|
|
)
|
|
in logs
|
|
) is result
|
|
|
|
|
|
@pytest.mark.parametrize("skip_it,result", ((False, True), (True, False)))
|
|
async def test_skip_touchup(app, caplog, skip_it, result):
|
|
app.config.TOUCHUP = not skip_it
|
|
with caplog.at_level(logging.DEBUG, logger="sanic.root"):
|
|
app.state.verbosity = 2
|
|
await app._startup()
|
|
assert app.signal_router.allow_fail_builtin is (not skip_it)
|
|
logs = caplog.record_tuples
|
|
|
|
for signal in RESERVED_NAMESPACES["http"]:
|
|
assert (
|
|
(
|
|
"sanic.root",
|
|
logging.DEBUG,
|
|
f"Disabling event: {signal}",
|
|
)
|
|
in logs
|
|
) is result
|
|
not_found_exceptions = 0
|
|
# Skip-touchup disables NotFound exceptions on the dispatcher
|
|
for signal in RESERVED_NAMESPACES["http"]:
|
|
try:
|
|
await app.dispatch(event=signal, inline=True)
|
|
except NotFound:
|
|
not_found_exceptions += 1
|
|
assert (not_found_exceptions > 0) is result
|
|
|
|
|
|
@pytest.mark.parametrize("skip_it,result", ((False, True), (True, True)))
|
|
async def test_skip_touchup_non_reserved(app, caplog, skip_it, result):
|
|
app.config.TOUCHUP = not skip_it
|
|
|
|
@app.signal("foo.bar.one")
|
|
def sync_signal(*_):
|
|
...
|
|
|
|
await app._startup()
|
|
assert app.signal_router.allow_fail_builtin is (not skip_it)
|
|
not_found_exception = False
|
|
# Skip-touchup doesn't disable NotFound exceptions for user-defined signals
|
|
try:
|
|
await app.dispatch(event="foo.baz.two", inline=True)
|
|
except NotFound:
|
|
not_found_exception = True
|
|
assert not_found_exception is result
|