4a5e5b1a54
This replaces the routing logic with a new, faster strategy based on on-the-fly generation of Python code. As part of replacing the logic, the architecture has been refactored to decouple routing from falcon.API, thereby making it possible for users to easily replace the default router with their own. Closes #114 Co-Authored-By: Kurt Griffiths <mail@kgriffs.com>
183 lines
5.8 KiB
Python
183 lines
5.8 KiB
Python
import falcon
|
|
import falcon.testing as testing
|
|
import six
|
|
|
|
|
|
class IDResource(object):
|
|
def __init__(self):
|
|
self.id = None
|
|
self.name = None
|
|
self.called = False
|
|
|
|
def on_get(self, req, resp, id):
|
|
self.id = id
|
|
self.called = True
|
|
self.req = req
|
|
|
|
|
|
class NameResource(object):
|
|
def __init__(self):
|
|
self.id = None
|
|
self.name = None
|
|
self.called = False
|
|
|
|
def on_get(self, req, resp, id, name):
|
|
self.id = id
|
|
self.name = name
|
|
self.called = True
|
|
|
|
|
|
class TestUriTemplates(testing.TestBase):
|
|
|
|
def before(self):
|
|
self.resource = testing.TestResource()
|
|
|
|
def test_root_path(self):
|
|
self.api.add_route('/', self.resource)
|
|
self.simulate_request('/')
|
|
|
|
self.assertTrue(self.resource.called)
|
|
req = self.resource.req
|
|
|
|
self.assertEqual(req.get_param('id'), None)
|
|
|
|
def test_not_str(self):
|
|
self.assertRaises(TypeError, self.api.add_route, {}, self.resource)
|
|
self.assertRaises(TypeError, self.api.add_route, [], self.resource)
|
|
self.assertRaises(TypeError, self.api.add_route, set(), self.resource)
|
|
self.assertRaises(TypeError, self.api.add_route, self, self.resource)
|
|
|
|
def test_field_name_cannot_start_with_digit(self):
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/hello/{1world}', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/{524hello}/world', self.resource)
|
|
|
|
def test_whitespace_not_allowed(self):
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/{thing }/world', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/{ thing}/world', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/{ thing }/world', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/{thing}/wo rld', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/{thing} /world', self.resource)
|
|
|
|
def test_no_vars(self):
|
|
self.api.add_route('/hello/world', self.resource)
|
|
self.simulate_request('/hello/world')
|
|
|
|
self.assertTrue(self.resource.called)
|
|
req = self.resource.req
|
|
|
|
self.assertEqual(req.get_param('world'), None)
|
|
|
|
def test_unicode_literal_routes(self):
|
|
if six.PY3:
|
|
self.skipTest('Test only applies to Python 2')
|
|
|
|
self.api.add_route(u'/hello/world', self.resource)
|
|
self.simulate_request('/hello/world')
|
|
|
|
self.assertTrue(self.resource.called)
|
|
req = self.resource.req
|
|
|
|
self.assertEqual(req.get_param('world'), None)
|
|
|
|
def test_special_chars(self):
|
|
self.api.add_route('/hello/world.json', self.resource)
|
|
self.api.add_route('/hello(world)', self.resource)
|
|
|
|
self.simulate_request('/hello/world_json')
|
|
self.assertFalse(self.resource.called)
|
|
|
|
self.simulate_request('/helloworld')
|
|
self.assertFalse(self.resource.called)
|
|
|
|
self.simulate_request('/hello/world.json')
|
|
self.assertTrue(self.resource.called)
|
|
|
|
def test_single(self):
|
|
self.api.add_route('/widgets/{id}', self.resource)
|
|
|
|
self.simulate_request('/widgets/123')
|
|
self.assertTrue(self.resource.called)
|
|
|
|
req = self.resource.req
|
|
kwargs = self.resource.kwargs
|
|
self.assertEqual(kwargs['id'], '123')
|
|
self.assertNotIn(kwargs, 'Id')
|
|
self.assertEqual(req.get_param('id'), None)
|
|
|
|
def test_single_trailing_slash(self):
|
|
resource1 = IDResource()
|
|
self.api.add_route('/1/{id}/', resource1)
|
|
|
|
self.simulate_request('/1/123')
|
|
self.assertEqual(self.srmock.status, falcon.HTTP_200)
|
|
self.assertTrue(resource1.called)
|
|
self.assertEqual(resource1.id, '123')
|
|
self.assertEqual(resource1.name, None)
|
|
self.assertEqual(resource1.req.path, '/1/123')
|
|
|
|
resource2 = IDResource()
|
|
self.api.add_route('/2/{id}/', resource2)
|
|
|
|
self.simulate_request('/2/123/')
|
|
self.assertTrue(resource2.called)
|
|
self.assertEqual(resource2.id, '123')
|
|
self.assertEqual(resource2.name, None)
|
|
self.assertEqual(resource2.req.path, '/2/123')
|
|
|
|
resource3 = IDResource()
|
|
self.api.add_route('/3/{id}', resource3)
|
|
|
|
self.simulate_request('/3/123/')
|
|
self.assertTrue(resource3.called)
|
|
self.assertEqual(resource3.id, '123')
|
|
self.assertEqual(resource3.name, None)
|
|
self.assertEqual(resource3.req.path, '/3/123')
|
|
|
|
def test_multiple(self):
|
|
resource = NameResource()
|
|
self.api.add_route('/messages/{id}/names/{name}', resource)
|
|
|
|
test_id = self.getUniqueString()
|
|
test_name = self.getUniqueString()
|
|
path = '/messages/' + test_id + '/names/' + test_name
|
|
self.simulate_request(path)
|
|
self.assertTrue(resource.called)
|
|
|
|
self.assertEqual(resource.id, test_id)
|
|
self.assertEqual(resource.name, test_name)
|
|
|
|
def test_empty_path_component(self):
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'//', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'//begin', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/end//', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'/in//side', self.resource)
|
|
|
|
def test_relative_path(self):
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'no', self.resource)
|
|
|
|
self.assertRaises(ValueError, self.api.add_route,
|
|
'no/leading_slash', self.resource)
|