Commit 5e2d8052 by Marlon Bailey

fix test case to work with Python 3 and make it more explicit

parent 9d2580dc
...@@ -38,9 +38,18 @@ class TestSimpleRouter(TestCase): ...@@ -38,9 +38,18 @@ class TestSimpleRouter(TestCase):
def test_link_and_action_decorator(self): def test_link_and_action_decorator(self):
routes = self.router.get_routes(BasicViewSet) routes = self.router.get_routes(BasicViewSet)
# Should be 2 by default, and then four from the @action and @link combined
#self.assertEqual(len(routes), 6)
#
decorator_routes = routes[2:] decorator_routes = routes[2:]
for i, method in enumerate(['action1', 'action2', 'link1', 'link2']): # Make sure all these endpoints exist and none have been clobbered
self.assertEqual(decorator_routes[i].mapping.values()[0], method) for i, endpoint in enumerate(['action1', 'action2', 'link1', 'link2']):
route = decorator_routes[i]
# check url listing
self.assertEqual(route.url,
'^{{prefix}}/{{lookup}}/{0}/$'.format(endpoint))
# check method to function mapping
if endpoint.startswith('action'):
method_map = 'post'
else:
method_map = 'get'
self.assertEqual(route.mapping[method_map], endpoint)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment