diff --git a/tests/test_router.py b/tests/test_router.py index bf0c5df..ff31325 100644 --- a/tests/test_router.py +++ b/tests/test_router.py @@ -4,7 +4,7 @@ import pytest import requests import werkzeug -from werkzeug.exceptions import MethodNotAllowed, NotFound +from werkzeug.exceptions import MethodNotAllowed, NotFound, SecurityError from werkzeug.routing import RequestRedirect, Submount from rolo import Request, Response, Router @@ -166,7 +166,7 @@ def test_port_host_dispatcher(self): assert collector.requests.pop()[2] == {"port": None} # invalid port - with pytest.raises(NotFound): + with pytest.raises((NotFound, SecurityError)): router.dispatch( Request( method="GET",