mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-09-20 19:09:03 +03:00
410575dace
* url() is deprecated in Django 3.1 * update given feedbacks on url() is deprecated in Django 3.1 * Fix test_urlpatterns.py to continue testing mixed re_path() and path() * Fix one missed reference Co-authored-by: sanjusci <sanju.sci9@gmail.com>
197 lines
7.9 KiB
Python
197 lines
7.9 KiB
Python
from collections import namedtuple
|
|
|
|
from django.test import TestCase
|
|
from django.urls import Resolver404, URLResolver, include, path, re_path
|
|
from django.urls.resolvers import RegexPattern
|
|
|
|
from rest_framework.test import APIRequestFactory
|
|
from rest_framework.urlpatterns import format_suffix_patterns
|
|
|
|
# A container class for test paths for the test case
|
|
URLTestPath = namedtuple('URLTestPath', ['path', 'args', 'kwargs'])
|
|
|
|
|
|
def dummy_view(request, *args, **kwargs):
|
|
pass
|
|
|
|
|
|
class FormatSuffixTests(TestCase):
|
|
"""
|
|
Tests `format_suffix_patterns` against different URLPatterns to ensure the
|
|
URLs still resolve properly, including any captured parameters.
|
|
"""
|
|
def _resolve_urlpatterns(self, urlpatterns, test_paths, allowed=None):
|
|
factory = APIRequestFactory()
|
|
try:
|
|
urlpatterns = format_suffix_patterns(urlpatterns, allowed=allowed)
|
|
except Exception:
|
|
self.fail("Failed to apply `format_suffix_patterns` on the supplied urlpatterns")
|
|
resolver = URLResolver(RegexPattern(r'^/'), urlpatterns)
|
|
for test_path in test_paths:
|
|
try:
|
|
test_path, expected_resolved = test_path
|
|
except (TypeError, ValueError):
|
|
expected_resolved = True
|
|
|
|
request = factory.get(test_path.path)
|
|
try:
|
|
callback, callback_args, callback_kwargs = resolver.resolve(request.path_info)
|
|
except Resolver404:
|
|
callback, callback_args, callback_kwargs = (None, None, None)
|
|
if expected_resolved:
|
|
raise
|
|
except Exception:
|
|
self.fail("Failed to resolve URL: %s" % request.path_info)
|
|
|
|
if not expected_resolved:
|
|
assert callback is None
|
|
continue
|
|
|
|
assert callback_args == test_path.args
|
|
assert callback_kwargs == test_path.kwargs
|
|
|
|
def _test_trailing_slash(self, urlpatterns):
|
|
test_paths = [
|
|
(URLTestPath('/test.api', (), {'format': 'api'}), True),
|
|
(URLTestPath('/test/.api', (), {'format': 'api'}), False),
|
|
(URLTestPath('/test.api/', (), {'format': 'api'}), True),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths)
|
|
|
|
def test_trailing_slash(self):
|
|
urlpatterns = [
|
|
path('test/', dummy_view),
|
|
]
|
|
self._test_trailing_slash(urlpatterns)
|
|
|
|
def test_trailing_slash_django2(self):
|
|
urlpatterns = [
|
|
path('test/', dummy_view),
|
|
]
|
|
self._test_trailing_slash(urlpatterns)
|
|
|
|
def _test_format_suffix(self, urlpatterns):
|
|
test_paths = [
|
|
URLTestPath('/test', (), {}),
|
|
URLTestPath('/test.api', (), {'format': 'api'}),
|
|
URLTestPath('/test.asdf', (), {'format': 'asdf'}),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths)
|
|
|
|
def test_format_suffix(self):
|
|
urlpatterns = [
|
|
path('test', dummy_view),
|
|
]
|
|
self._test_format_suffix(urlpatterns)
|
|
|
|
def test_format_suffix_django2(self):
|
|
urlpatterns = [
|
|
path('test', dummy_view),
|
|
]
|
|
self._test_format_suffix(urlpatterns)
|
|
|
|
def test_format_suffix_django2_args(self):
|
|
urlpatterns = [
|
|
path('convtest/<int:pk>', dummy_view),
|
|
re_path(r'^retest/(?P<pk>[0-9]+)$', dummy_view),
|
|
]
|
|
test_paths = [
|
|
URLTestPath('/convtest/42', (), {'pk': 42}),
|
|
URLTestPath('/convtest/42.api', (), {'pk': 42, 'format': 'api'}),
|
|
URLTestPath('/convtest/42.asdf', (), {'pk': 42, 'format': 'asdf'}),
|
|
URLTestPath('/retest/42', (), {'pk': '42'}),
|
|
URLTestPath('/retest/42.api', (), {'pk': '42', 'format': 'api'}),
|
|
URLTestPath('/retest/42.asdf', (), {'pk': '42', 'format': 'asdf'}),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths)
|
|
|
|
def _test_default_args(self, urlpatterns):
|
|
test_paths = [
|
|
URLTestPath('/test', (), {'foo': 'bar', }),
|
|
URLTestPath('/test.api', (), {'foo': 'bar', 'format': 'api'}),
|
|
URLTestPath('/test.asdf', (), {'foo': 'bar', 'format': 'asdf'}),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths)
|
|
|
|
def test_default_args(self):
|
|
urlpatterns = [
|
|
path('test', dummy_view, {'foo': 'bar'}),
|
|
]
|
|
self._test_default_args(urlpatterns)
|
|
|
|
def test_default_args_django2(self):
|
|
urlpatterns = [
|
|
path('test', dummy_view, {'foo': 'bar'}),
|
|
]
|
|
self._test_default_args(urlpatterns)
|
|
|
|
def _test_included_urls(self, urlpatterns):
|
|
test_paths = [
|
|
URLTestPath('/test/path', (), {'foo': 'bar', }),
|
|
URLTestPath('/test/path.api', (), {'foo': 'bar', 'format': 'api'}),
|
|
URLTestPath('/test/path.asdf', (), {'foo': 'bar', 'format': 'asdf'}),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths)
|
|
|
|
def test_included_urls(self):
|
|
nested_patterns = [
|
|
path('path', dummy_view)
|
|
]
|
|
urlpatterns = [
|
|
path('test/', include(nested_patterns), {'foo': 'bar'}),
|
|
]
|
|
self._test_included_urls(urlpatterns)
|
|
|
|
def test_included_urls_mixed(self):
|
|
nested_patterns = [
|
|
path('path/<int:child>', dummy_view),
|
|
re_path(r'^re_path/(?P<child>[0-9]+)$', dummy_view)
|
|
]
|
|
urlpatterns = [
|
|
re_path(r'^pre_path/(?P<parent>[0-9]+)/', include(nested_patterns), {'foo': 'bar'}),
|
|
path('ppath/<int:parent>/', include(nested_patterns), {'foo': 'bar'}),
|
|
]
|
|
test_paths = [
|
|
# parent re_path() nesting child path()
|
|
URLTestPath('/pre_path/87/path/42', (), {'parent': '87', 'child': 42, 'foo': 'bar', }),
|
|
URLTestPath('/pre_path/87/path/42.api', (), {'parent': '87', 'child': 42, 'foo': 'bar', 'format': 'api'}),
|
|
URLTestPath('/pre_path/87/path/42.asdf', (), {'parent': '87', 'child': 42, 'foo': 'bar', 'format': 'asdf'}),
|
|
|
|
# parent path() nesting child re_path()
|
|
URLTestPath('/ppath/87/re_path/42', (), {'parent': 87, 'child': '42', 'foo': 'bar', }),
|
|
URLTestPath('/ppath/87/re_path/42.api', (), {'parent': 87, 'child': '42', 'foo': 'bar', 'format': 'api'}),
|
|
URLTestPath('/ppath/87/re_path/42.asdf', (), {'parent': 87, 'child': '42', 'foo': 'bar', 'format': 'asdf'}),
|
|
|
|
# parent path() nesting child path()
|
|
URLTestPath('/ppath/87/path/42', (), {'parent': 87, 'child': 42, 'foo': 'bar', }),
|
|
URLTestPath('/ppath/87/path/42.api', (), {'parent': 87, 'child': 42, 'foo': 'bar', 'format': 'api'}),
|
|
URLTestPath('/ppath/87/path/42.asdf', (), {'parent': 87, 'child': 42, 'foo': 'bar', 'format': 'asdf'}),
|
|
|
|
# parent re_path() nesting child re_path()
|
|
URLTestPath('/pre_path/87/re_path/42', (), {'parent': '87', 'child': '42', 'foo': 'bar', }),
|
|
URLTestPath('/pre_path/87/re_path/42.api', (), {'parent': '87', 'child': '42', 'foo': 'bar', 'format': 'api'}),
|
|
URLTestPath('/pre_path/87/re_path/42.asdf', (), {'parent': '87', 'child': '42', 'foo': 'bar', 'format': 'asdf'}),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths)
|
|
|
|
def _test_allowed_formats(self, urlpatterns):
|
|
allowed_formats = ['good', 'ugly']
|
|
test_paths = [
|
|
(URLTestPath('/test.good/', (), {'format': 'good'}), True),
|
|
(URLTestPath('/test.bad', (), {}), False),
|
|
(URLTestPath('/test.ugly', (), {'format': 'ugly'}), True),
|
|
]
|
|
self._resolve_urlpatterns(urlpatterns, test_paths, allowed=allowed_formats)
|
|
|
|
def test_allowed_formats_re_path(self):
|
|
urlpatterns = [
|
|
re_path(r'^test$', dummy_view),
|
|
]
|
|
self._test_allowed_formats(urlpatterns)
|
|
|
|
def test_allowed_formats_path(self):
|
|
urlpatterns = [
|
|
path('test', dummy_view),
|
|
]
|
|
self._test_allowed_formats(urlpatterns)
|