class URLParserTestCase(unittest.TestCase):
def setUp(self):
super(URLParserTestCase, self).setUp()
self.parser = URLParser()
def test_parse_static_url(self):
route = self.parser.define_route('/api', FakeGetView, 'GET')
self.assertIsInstance(route, PlainEndpoint)
def test_parse_dynamic_url(self):
route = self.parser.define_route('/api/{users}', FakeGetView, 'GET')
self.assertIsInstance(route, DynamicEndpoint)
def test_parse_invalid_url_1(self):
self.assertRaises(EndpointValueError, self.parser.define_route,
'/api/{users', FakeGetView, 'GET')
def test_parse_invalid_url_2(self):
self.assertRaises(EndpointValueError, self.parser.define_route,
'/api/{users{}}', FakeGetView, 'GET')
def test_parse_invalid_url_3(self):
self.assertRaises(EndpointValueError, self.parser.define_route,
'/api/{users{}', FakeGetView, 'GET')
def test_parse_invalid_url_4(self):
self.assertRaises(EndpointValueError, self.parser.define_route,
'/api/{users"}', FakeGetView, 'GET')
def test_parse_invalid_url_5(self):
self.assertRaises(EndpointValueError, self.parser.define_route,
r"/api/{users+++}", FakeGetView, 'GET')
def setUpClass(cls):
super(TestHyperlinkedRelatedField, cls).setUpClass()
session = SESSION()
user = cls.TestHyperlinkedRelatedFieldUserModel(name='admin')
address = cls.TestHyperlinkedRelatedFieldAddressModel(
email_address='*****@*****.**')
user_composite_pk = cls.TestHyperlinkedRelatedFieldCompositePkModel(
id=1, name='test_user')
user.addresses.append(address)
session.add(user)
session.add(address)
session.add(user_composite_pk)
session.commit()
session.close()
url_parser = URLParser()
cls.data = {
'urls': [
url_parser.define_route('/user/{pk}/',
FakeView, [
'GET',
],
name='user-detail'),
]
}
set_urlconf(cls.data)
def setUp(self):
super(TestReverse, self).setUpClass()
url_parser = URLParser()
data = {
'path': 'wss://127.0.0.1:8000',
'routes': {
'user-detail': url_parser.define_route(
'/user/{pk}/', FakeView, ['GET', ], name='user-detail'
)
}
}
set_urlconf(data)
def setUp(self):
super(TestReverse, self).setUpClass()
url_parser = URLParser()
data = {
'path': 'wss://127.0.0.1:8000',
'routes': {
'user-detail':
url_parser.define_route('/user/{pk}/',
FakeView, [
'GET',
],
name='user-detail')
}
}
set_urlconf(data)
def setUpClass(cls):
super(TestHyperlinkedRelatedField, cls).setUpClass()
cls.publisher = cls.Publisher.objects.create(name='publisher')
cls.book = cls.Book.objects.create(name="test book")
cls.book.publisher = cls.publisher
cls.book.save()
url_parser = URLParser()
cls.data = {
'urls': [
url_parser.define_route(
'/book/{pk}/', FakeView, ['GET', ], name='book-detail'
),
]
}
set_urlconf(cls.data)
class URLParserTestCase(unittest.TestCase):
def setUp(self):
super(URLParserTestCase, self).setUp()
self.parser = URLParser()
def test_parse_static_url(self):
route = self.parser.define_route('/api', FakeGetView, 'GET')
self.assertIsInstance(route, PlainEndpoint)
def test_parse_dynamic_url(self):
route = self.parser.define_route('/api/{users}', FakeGetView, 'GET')
self.assertIsInstance(route, DynamicEndpoint)
def test_parse_invalid_url_1(self):
self.assertRaises(
EndpointValueError,
self.parser.define_route, '/api/{users', FakeGetView, 'GET'
)
def test_parse_invalid_url_2(self):
self.assertRaises(
EndpointValueError,
self.parser.define_route,
'/api/{users{}}', FakeGetView, 'GET'
)
def test_parse_invalid_url_3(self):
self.assertRaises(
EndpointValueError,
self.parser.define_route,
'/api/{users{}', FakeGetView, 'GET'
)
def test_parse_invalid_url_4(self):
self.assertRaises(
EndpointValueError,
self.parser.define_route,
'/api/{users"}', FakeGetView, 'GET'
)
def test_parse_invalid_url_5(self):
self.assertRaises(
EndpointValueError,
self.parser.define_route,
r"/api/{users+++}", FakeGetView, 'GET'
)
def setUpClass(cls):
super(TestResolve, cls).setUpClass()
url_parser = URLParser()
cls.data = {
'path': 'wss://127.0.0.1:8000',
'urls': [
url_parser.define_route(
'/user/list/', FakeView, ['GET', ], name='user-list'
),
url_parser.define_route(
'/user/{pk}/', FakeView, ['GET', ]
),
url_parser.define_route(
'/user/', FakeView, ['GET', ]
)
]
}
def setUpClass(cls):
super(TestResolve, cls).setUpClass()
url_parser = URLParser()
cls.data = {
'path':
'wss://127.0.0.1:8000',
'urls': [
url_parser.define_route('/user/list/',
FakeView, [
'GET',
],
name='user-list'),
url_parser.define_route('/user/{pk}/', FakeView, [
'GET',
]),
url_parser.define_route('/user/', FakeView, [
'GET',
])
]
}
def setUp(self):
super(URLParserTestCase, self).setUp()
self.parser = URLParser()
class SimpleRouter(AbstractRouter):
"""
Default router class, used for working with REST over WebSockets.
"""
args_validator = RouteArgumentsValidator()
url_parser = URLParser()
def _correct_path(self, path):
"""
Convert path to valid value.
:param path: URL, which used to get access to API.
"""
path = path.strip()
if not path.endswith('/'):
path += '/'
return path
def register(self, path, handler, methods, name=None):
"""
Add new endpoint to the router.
:param path: URL, which used to get access to API.
:param handler: inherited class from the MethodBasedView, which used
for processing request.
:param methods: list of available for user methods or string with
concrete method name.
:param name: short name for endpoint.
"""
path = self._correct_path(path)
self.args_validator.validate(path, handler, methods, name)
route = self.url_parser.define_route(path, handler, methods, name)
self._register_url(route)
def register_endpoint(self, endpoint):
"""
Add new endpoint to the router.
:param endpoint: function with @endpoint decorator, which used for
processing request.
"""
path, handler, methods, name = endpoint()
self.register(path, handler, methods, name)
def extract_url(self, request):
"""
Extracting URL parameter for request.
:param request: request from the user.
"""
if not request.url:
raise NotSpecifiedURL()
return self._correct_path(request.url)
def search_handler(self, request, url):
"""
Searching handler by URL.
:param request: request from user.
:param url: path to the registered endpoint.
"""
args = ()
kwargs = {}
handler = None
# Iterate over the all endpoints
for route in self._urls:
# Find match between endpoint and request URL
match = route.match(url)
if match is not None:
handler = route.handler()
args = match
parameters = request.args
if parameters:
kwargs.update(parameters)
break
return handler, args, kwargs
def process_request(self, request):
"""
Handle received request from user.
:param request: request from user.
"""
logger.info("\"{method} {url}\" args={args}".format(
method=request.method, url=request.url, args=request.args))
response = Response()
try:
url = self.extract_url(request)
handler, args, kwargs = self.search_handler(request, url)
# Invoke handler for request
if handler:
for middleware in self.middlewares:
middleware.process_request(request, handler)
# Search serializer for response
format = request.get_argument('format')
serializer = handler.get_renderer(format, *args, **kwargs)
response.content = handler.dispatch(request, *args, **kwargs)
else:
raise NotSpecifiedHandler()
except BaseAPIException as exc:
logger.exception(exc)
response.wrap_exception(exc)
serializer = JSONRenderer()
response.append_request(request)
return serializer.render(response.content)
def _register_url(self, route):
"""
Register new endpoint.
:param route: instance of class, inherited from AbstractEndpoint.
"""
if not issubclass(type(route), (AbstractEndpoint, )):
raise TypeError(u"Custom route must be inherited from the "
u"AbstractEndpoint class.")
if route.name:
if route.name in self._routes.keys():
raise EndpointValueError(
'Duplicate {}, already handled by {}'.format(
route.name, self._routes[route.name]))
self._routes[route.name] = route
self._urls.append(route)
def include(self, router):
"""
Appending endpoints from another router to self.
:param router: instance of subclass, derived from AbstractRouter.
"""
if not issubclass(type(router), (AbstractRouter, )):
raise TypeError(u"Passed router must be inherited from the "
u"AbstractRouter class.")
self._urls.extend(router._urls)
self._routes.update(router._routes)
def setUp(self):
super(URLParserTestCase, self).setUp()
self.parser = URLParser()