f1597d77e6
Rather than compiling the routing tree directly to Python code,
first generate an AST and then use it to produce the code. This
provides several benefits, including:
* It makes the compilation process easier to reason about.
* It makes it easier to keep track of indentation and whitespace.
* It sets us up for being able to make transformations that we
will need to do to support URI template filters, etc. in the
future.
117 lines
3.8 KiB
Python
117 lines
3.8 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)
|
|
|
|
def find(self, uri):
|
|
pass
|
|
|
|
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 = '{{"uri_template": "{0}"}}'.format(req.uri_template)
|
|
|
|
class CustomRouter(object):
|
|
def __init__(self):
|
|
self.reached_backwards_compat = False
|
|
|
|
def find(self, uri):
|
|
if uri == '/test/42':
|
|
return resource, {'GET': resource}, {}, '/test/{id}'
|
|
|
|
if uri == '/test/42/no-uri-template':
|
|
return resource, {'GET': resource}, {}, None
|
|
|
|
if uri == '/test/42/uri-template/backwards-compat':
|
|
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/42')
|
|
self.assertEqual(body, [b'{"uri_template": "/test/{id}"}'])
|
|
|
|
body = self.simulate_request('/test/42/no-uri-template')
|
|
self.assertEqual(body, [b'{"uri_template": "None"}'])
|
|
|
|
body = self.simulate_request('/test/42/uri-template/backwards-compat')
|
|
self.assertEqual(body, [b'{"uri_template": "None"}'])
|
|
|
|
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)
|
|
|
|
def find(self, uri):
|
|
pass
|
|
|
|
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)
|
|
|
|
def test_custom_router_takes_req_positional_argument(self):
|
|
def responder(req, resp):
|
|
resp.body = 'OK'
|
|
|
|
class CustomRouter(object):
|
|
def find(self, uri, req):
|
|
if uri == '/test' and isinstance(req, falcon.Request):
|
|
return responder, {'GET': responder}, {}, None
|
|
|
|
router = CustomRouter()
|
|
self.api = falcon.API(router=router)
|
|
body = self.simulate_request('/test')
|
|
self.assertEqual(body[0], b'OK')
|
|
|
|
def test_custom_router_takes_req_keyword_argument(self):
|
|
def responder(req, resp):
|
|
resp.body = 'OK'
|
|
|
|
class CustomRouter(object):
|
|
def find(self, uri, req=None):
|
|
if uri == '/test' and isinstance(req, falcon.Request):
|
|
return responder, {'GET': responder}, {}, None
|
|
|
|
router = CustomRouter()
|
|
self.api = falcon.API(router=router)
|
|
body = self.simulate_request('/test')
|
|
self.assertEqual(body[0], b'OK')
|