7f9ce56515
The documentation for custom routers states that the find() method of the router should return None when no route is found. However, the API only allows for the (None, None, None) tuple. Add support for a simple None return value as well. Closes #750
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
import falcon
|
|
from falcon import testing
|
|
|
|
|
|
class TestCustomRouter(testing.TestBase):
|
|
|
|
def test_custom_router_add_route_should_be_used(self):
|
|
|
|
check = []
|
|
|
|
class CustomRouter(object):
|
|
def add_route(self, uri_template, *args, **kwargs):
|
|
check.append(uri_template)
|
|
|
|
api = falcon.API(router=CustomRouter())
|
|
api.add_route('/test', 'resource')
|
|
|
|
self.assertEqual(len(check), 1)
|
|
self.assertIn('/test', check)
|
|
|
|
def test_custom_router_find_should_be_used(self):
|
|
|
|
def resource(req, resp, **kwargs):
|
|
resp.body = '{"status": "ok"}'
|
|
|
|
class CustomRouter(object):
|
|
def __init__(self):
|
|
self.reached_backwards_compat = False
|
|
|
|
def find(self, uri):
|
|
if uri == '/test':
|
|
return resource, {'GET': resource}, {}
|
|
|
|
if uri == '/404/backwards-compat':
|
|
self.reached_backwards_compat = True
|
|
return (None, None, None)
|
|
|
|
return None
|
|
|
|
router = CustomRouter()
|
|
self.api = falcon.API(router=router)
|
|
body = self.simulate_request('/test')
|
|
self.assertEqual(body, [b'{"status": "ok"}'])
|
|
|
|
for uri in ('/404', '/404/backwards-compat'):
|
|
body = self.simulate_request(uri)
|
|
self.assertFalse(body)
|
|
self.assertEqual(self.srmock.status, falcon.HTTP_404)
|
|
|
|
self.assertTrue(router.reached_backwards_compat)
|
|
|
|
def test_can_pass_additional_params_to_add_route(self):
|
|
|
|
check = []
|
|
|
|
class CustomRouter(object):
|
|
def add_route(self, uri_template, method_map, resource, name):
|
|
self._index = {name: uri_template}
|
|
check.append(name)
|
|
|
|
api = falcon.API(router=CustomRouter())
|
|
api.add_route('/test', 'resource', name='my-url-name')
|
|
|
|
self.assertEqual(len(check), 1)
|
|
self.assertIn('my-url-name', check)
|
|
|
|
# Also as arg.
|
|
api.add_route('/test', 'resource', 'my-url-name-arg')
|
|
|
|
self.assertEqual(len(check), 2)
|
|
self.assertIn('my-url-name-arg', check)
|