def test_register_adds_default_routes_from_modelname(self):
class ParentView(ModelView):
pass
class FooView(ParentView):
model = FooModel
class BarView(ParentView):
model = BarModel
r = Router()
r.register(ParentView)
urls_module.urlpatterns = [url(r'^', include(r.urls))]
self.assertTrue(is_valid_path('/foo_model/', urls_module))
self.assertTrue(is_valid_path('/foo_model/1/', urls_module))
self.assertTrue(is_valid_path('/bar_model/12345/', urls_module))
self.assertFalse(is_valid_path('/bar_model/lalala/', urls_module))
self.assertFalse(is_valid_path('/another_model/', urls_module))
def test_register_obeys_custom_route_config(self):
class ParentView(ModelView):
pass
class FooView(ParentView):
model = FooModel
route = Route('foo', list_endpoint=False)
class BarView(ParentView):
model = BarModel
route = Route('bar', detail_endpoint=False)
r = Router()
r.register(ParentView)
urls_module.urlpatterns = [url(r'^', include(r.urls))]
self.assertFalse(is_valid_path('/foo/', urls_module))
self.assertTrue(is_valid_path('/foo/1/', urls_module))
self.assertTrue(is_valid_path('/bar/', urls_module))
self.assertFalse(is_valid_path('/bar/1/', urls_module))
def test_register_adds_custom_route_names(self):
class ParentView(ModelView):
pass
class FooView(ParentView):
model = FooModel
route = 'foo'
class BarView(ParentView):
model = BarModel
# Explicit Route objects should also be accepted
route = Route('bar')
r = Router()
r.register(ParentView)
urls_module.urlpatterns = [url(r'^', include(r.urls))]
self.assertTrue(is_valid_path('/foo/', urls_module))
self.assertTrue(is_valid_path('/foo/1/', urls_module))
self.assertTrue(is_valid_path('/bar/12345/', urls_module))
# Default named routes should not be there
self.assertFalse(is_valid_path('/foo_model/1/', urls_module))
self.assertFalse(is_valid_path('/bar_model/1/', urls_module))