147 lines
4.6 KiB
Python
147 lines
4.6 KiB
Python
import falcon
|
|
import falcon.testing as testing
|
|
|
|
|
|
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_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_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)
|