123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740 |
- import sys
- import unittest
- from django.conf import settings
- from django.contrib import admin
- from django.contrib.admindocs import utils, views
- from django.contrib.admindocs.views import get_return_data_type, simplify_regex
- from django.contrib.auth.models import Permission, User
- from django.contrib.contenttypes.models import ContentType
- from django.contrib.sites.models import Site
- from django.db import models
- from django.db.models import fields
- from django.test import SimpleTestCase, modify_settings, override_settings
- from django.test.utils import captured_stderr
- from django.urls import include, path, reverse
- from django.utils.functional import SimpleLazyObject
- from .models import Company, Person
- from .tests import AdminDocsTestCase, TestDataMixin
- @unittest.skipUnless(utils.docutils_is_available, "no docutils installed.")
- class AdminDocViewTests(TestDataMixin, AdminDocsTestCase):
- def setUp(self):
- self.client.force_login(self.superuser)
- def test_index(self):
- response = self.client.get(reverse("django-admindocs-docroot"))
- self.assertContains(response, "<h1>Documentation</h1>", html=True)
- self.assertContains(
- response,
- '<div id="site-name"><a href="/admin/">Django administration</a></div>',
- )
- self.client.logout()
- response = self.client.get(reverse("django-admindocs-docroot"), follow=True)
- # Should display the login screen
- self.assertContains(
- response, '<input type="hidden" name="next" value="/admindocs/">', html=True
- )
- def test_bookmarklets(self):
- response = self.client.get(reverse("django-admindocs-bookmarklets"))
- self.assertContains(response, "/admindocs/views/")
- def test_templatetag_index(self):
- response = self.client.get(reverse("django-admindocs-tags"))
- self.assertContains(
- response, '<h3 id="built_in-extends">extends</h3>', html=True
- )
- def test_templatefilter_index(self):
- response = self.client.get(reverse("django-admindocs-filters"))
- self.assertContains(response, '<h3 id="built_in-first">first</h3>', html=True)
- def test_view_index(self):
- response = self.client.get(reverse("django-admindocs-views-index"))
- self.assertContains(
- response,
- '<h3><a href="/admindocs/views/django.contrib.admindocs.views.'
- 'BaseAdminDocsView/">/admindocs/</a></h3>',
- html=True,
- )
- self.assertContains(response, "Views by namespace test")
- self.assertContains(response, "Name: <code>test:func</code>.")
- self.assertContains(
- response,
- '<h3><a href="/admindocs/views/admin_docs.views.XViewCallableObject/">'
- "/xview/callable_object_without_xview/</a></h3>",
- html=True,
- )
- def test_view_index_with_method(self):
- """
- Views that are methods are listed correctly.
- """
- response = self.client.get(reverse("django-admindocs-views-index"))
- self.assertContains(
- response,
- "<h3>"
- '<a href="/admindocs/views/django.contrib.admin.sites.AdminSite.index/">'
- "/admin/</a></h3>",
- html=True,
- )
- def test_view_detail(self):
- url = reverse(
- "django-admindocs-views-detail",
- args=["django.contrib.admindocs.views.BaseAdminDocsView"],
- )
- response = self.client.get(url)
- # View docstring
- self.assertContains(response, "Base view for admindocs views.")
- def testview_docstring_links(self):
- summary = (
- '<h2 class="subhead">This is a view for '
- '<a class="reference external" href="/admindocs/models/myapp.company/">'
- "myapp.Company</a></h2>"
- )
- url = reverse(
- "django-admindocs-views-detail", args=["admin_docs.views.CompanyView"]
- )
- response = self.client.get(url)
- self.assertContains(response, summary, html=True)
- @override_settings(ROOT_URLCONF="admin_docs.namespace_urls")
- def test_namespaced_view_detail(self):
- url = reverse(
- "django-admindocs-views-detail", args=["admin_docs.views.XViewClass"]
- )
- response = self.client.get(url)
- self.assertContains(response, "<h1>admin_docs.views.XViewClass</h1>")
- def test_view_detail_illegal_import(self):
- url = reverse(
- "django-admindocs-views-detail",
- args=["urlpatterns_reverse.nonimported_module.view"],
- )
- response = self.client.get(url)
- self.assertEqual(response.status_code, 404)
- self.assertNotIn("urlpatterns_reverse.nonimported_module", sys.modules)
- def test_view_detail_as_method(self):
- """
- Views that are methods can be displayed.
- """
- url = reverse(
- "django-admindocs-views-detail",
- args=["django.contrib.admin.sites.AdminSite.index"],
- )
- response = self.client.get(url)
- self.assertEqual(response.status_code, 200)
- def test_model_index(self):
- response = self.client.get(reverse("django-admindocs-models-index"))
- self.assertContains(
- response,
- '<h2 id="app-auth">Authentication and Authorization (django.contrib.auth)'
- "</h2>",
- html=True,
- )
- def test_template_detail(self):
- response = self.client.get(
- reverse(
- "django-admindocs-templates", args=["admin_doc/template_detail.html"]
- )
- )
- self.assertContains(
- response,
- "<h1>Template: <q>admin_doc/template_detail.html</q></h1>",
- html=True,
- )
- def test_template_detail_loader(self):
- response = self.client.get(
- reverse("django-admindocs-templates", args=["view_for_loader_test.html"])
- )
- self.assertContains(response, "view_for_loader_test.html</code></li>")
- def test_missing_docutils(self):
- utils.docutils_is_available = False
- try:
- response = self.client.get(reverse("django-admindocs-docroot"))
- self.assertContains(
- response,
- "<h3>The admin documentation system requires Python’s "
- '<a href="https://docutils.sourceforge.io/">docutils</a> '
- "library.</h3>"
- "<p>Please ask your administrators to install "
- '<a href="https://pypi.org/project/docutils/">docutils</a>.</p>',
- html=True,
- )
- self.assertContains(
- response,
- '<div id="site-name"><a href="/admin/">Django administration</a></div>',
- )
- finally:
- utils.docutils_is_available = True
- @modify_settings(INSTALLED_APPS={"remove": "django.contrib.sites"})
- @override_settings(SITE_ID=None) # will restore SITE_ID after the test
- def test_no_sites_framework(self):
- """
- Without the sites framework, should not access SITE_ID or Site
- objects. Deleting settings is fine here as UserSettingsHolder is used.
- """
- Site.objects.all().delete()
- del settings.SITE_ID
- response = self.client.get(reverse("django-admindocs-views-index"))
- self.assertContains(response, "View documentation")
- def test_callable_urlconf(self):
- """
- Index view should correctly resolve view patterns when ROOT_URLCONF is
- not a string.
- """
- def urlpatterns():
- return (
- path("admin/doc/", include("django.contrib.admindocs.urls")),
- path("admin/", admin.site.urls),
- )
- with self.settings(ROOT_URLCONF=SimpleLazyObject(urlpatterns)):
- response = self.client.get(reverse("django-admindocs-views-index"))
- self.assertEqual(response.status_code, 200)
- @unittest.skipUnless(utils.docutils_is_available, "no docutils installed.")
- class AdminDocViewDefaultEngineOnly(TestDataMixin, AdminDocsTestCase):
- def setUp(self):
- self.client.force_login(self.superuser)
- def test_template_detail_path_traversal(self):
- cases = ["/etc/passwd", "../passwd"]
- for fpath in cases:
- with self.subTest(path=fpath):
- response = self.client.get(
- reverse("django-admindocs-templates", args=[fpath]),
- )
- self.assertEqual(response.status_code, 400)
- @override_settings(
- TEMPLATES=[
- {
- "NAME": "ONE",
- "BACKEND": "django.template.backends.django.DjangoTemplates",
- "APP_DIRS": True,
- },
- {
- "NAME": "TWO",
- "BACKEND": "django.template.backends.django.DjangoTemplates",
- "APP_DIRS": True,
- },
- ]
- )
- @unittest.skipUnless(utils.docutils_is_available, "no docutils installed.")
- class AdminDocViewWithMultipleEngines(AdminDocViewTests):
- def test_templatefilter_index(self):
- # Overridden because non-trivial TEMPLATES settings aren't supported
- # but the page shouldn't crash (#24125).
- response = self.client.get(reverse("django-admindocs-filters"))
- self.assertContains(response, "<title>Template filters</title>", html=True)
- def test_templatetag_index(self):
- # Overridden because non-trivial TEMPLATES settings aren't supported
- # but the page shouldn't crash (#24125).
- response = self.client.get(reverse("django-admindocs-tags"))
- self.assertContains(response, "<title>Template tags</title>", html=True)
- @unittest.skipUnless(utils.docutils_is_available, "no docutils installed.")
- class TestModelDetailView(TestDataMixin, AdminDocsTestCase):
- def setUp(self):
- self.client.force_login(self.superuser)
- with captured_stderr() as self.docutils_stderr:
- self.response = self.client.get(
- reverse("django-admindocs-models-detail", args=["admin_docs", "Person"])
- )
- def test_table_headers(self):
- tests = [
- ("Method", 1),
- ("Arguments", 1),
- ("Description", 2),
- ("Field", 1),
- ("Type", 1),
- ("Method", 1),
- ]
- for table_header, count in tests:
- self.assertContains(
- self.response, f'<th scope="col">{table_header}</th>', count=count
- )
- def test_method_excludes(self):
- """
- Methods that begin with strings defined in
- ``django.contrib.admindocs.views.MODEL_METHODS_EXCLUDE``
- shouldn't be displayed in the admin docs.
- """
- self.assertContains(self.response, "<td>get_full_name</td>")
- self.assertNotContains(self.response, "<td>_get_full_name</td>")
- self.assertNotContains(self.response, "<td>add_image</td>")
- self.assertNotContains(self.response, "<td>delete_image</td>")
- self.assertNotContains(self.response, "<td>set_status</td>")
- self.assertNotContains(self.response, "<td>save_changes</td>")
- def test_methods_with_arguments(self):
- """
- Methods that take arguments should also displayed.
- """
- self.assertContains(self.response, "<h3>Methods with arguments</h3>")
- self.assertContains(self.response, "<td>rename_company</td>")
- self.assertContains(self.response, "<td>dummy_function</td>")
- self.assertContains(self.response, "<td>dummy_function_keyword_only_arg</td>")
- self.assertContains(self.response, "<td>all_kinds_arg_function</td>")
- self.assertContains(self.response, "<td>suffix_company_name</td>")
- def test_methods_with_arguments_display_arguments(self):
- """
- Methods with arguments should have their arguments displayed.
- """
- self.assertContains(self.response, "<td>new_name</td>")
- self.assertContains(self.response, "<td>keyword_only_arg</td>")
- def test_methods_with_arguments_display_arguments_default_value(self):
- """
- Methods with keyword arguments should have their arguments displayed.
- """
- self.assertContains(self.response, "<td>suffix='ltd'</td>")
- def test_methods_with_multiple_arguments_display_arguments(self):
- """
- Methods with multiple arguments should have all their arguments
- displayed, but omitting 'self'.
- """
- self.assertContains(
- self.response, "<td>baz, rox, *some_args, **some_kwargs</td>"
- )
- self.assertContains(self.response, "<td>position_only_arg, arg, kwarg</td>")
- def test_instance_of_property_methods_are_displayed(self):
- """Model properties are displayed as fields."""
- self.assertContains(self.response, "<td>a_property</td>")
- def test_instance_of_cached_property_methods_are_displayed(self):
- """Model cached properties are displayed as fields."""
- self.assertContains(self.response, "<td>a_cached_property</td>")
- def test_method_data_types(self):
- company = Company.objects.create(name="Django")
- person = Person.objects.create(
- first_name="Human", last_name="User", company=company
- )
- self.assertEqual(
- get_return_data_type(person.get_status_count.__name__), "Integer"
- )
- self.assertEqual(get_return_data_type(person.get_groups_list.__name__), "List")
- def test_descriptions_render_correctly(self):
- """
- The ``description`` field should render correctly for each field type.
- """
- # help text in fields
- self.assertContains(
- self.response, "<td>first name - The person's first name</td>"
- )
- self.assertContains(
- self.response, "<td>last name - The person's last name</td>"
- )
- # method docstrings
- self.assertContains(self.response, "<p>Get the full name of the person</p>")
- link = '<a class="reference external" href="/admindocs/models/%s/">%s</a>'
- markup = "<p>the related %s object</p>"
- company_markup = markup % (link % ("admin_docs.company", "admin_docs.Company"))
- # foreign keys
- self.assertContains(self.response, company_markup)
- # foreign keys with help text
- self.assertContains(self.response, "%s\n - place of work" % company_markup)
- # many to many fields
- self.assertContains(
- self.response,
- "number of related %s objects"
- % (link % ("admin_docs.group", "admin_docs.Group")),
- )
- self.assertContains(
- self.response,
- "all related %s objects"
- % (link % ("admin_docs.group", "admin_docs.Group")),
- )
- # "raw" and "include" directives are disabled
- self.assertContains(
- self.response,
- "<p>"raw" directive disabled.</p>",
- )
- self.assertContains(
- self.response, ".. raw:: html\n :file: admin_docs/evilfile.txt"
- )
- self.assertContains(
- self.response,
- "<p>"include" directive disabled.</p>",
- )
- self.assertContains(self.response, ".. include:: admin_docs/evilfile.txt")
- out = self.docutils_stderr.getvalue()
- self.assertIn('"raw" directive disabled', out)
- self.assertIn('"include" directive disabled', out)
- def test_model_with_many_to_one(self):
- link = '<a class="reference external" href="/admindocs/models/%s/">%s</a>'
- response = self.client.get(
- reverse("django-admindocs-models-detail", args=["admin_docs", "company"])
- )
- self.assertContains(
- response,
- "number of related %s objects"
- % (link % ("admin_docs.person", "admin_docs.Person")),
- )
- self.assertContains(
- response,
- "all related %s objects"
- % (link % ("admin_docs.person", "admin_docs.Person")),
- )
- def test_model_with_no_backward_relations_render_only_relevant_fields(self):
- """
- A model with ``related_name`` of `+` shouldn't show backward
- relationship links.
- """
- response = self.client.get(
- reverse("django-admindocs-models-detail", args=["admin_docs", "family"])
- )
- fields = response.context_data.get("fields")
- self.assertEqual(len(fields), 2)
- def test_model_docstring_renders_correctly(self):
- summary = (
- '<h2 class="subhead">Stores information about a person, related to '
- '<a class="reference external" href="/admindocs/models/myapp.company/">'
- "myapp.Company</a>.</h2>"
- )
- subheading = "<p><strong>Notes</strong></p>"
- body = (
- '<p>Use <tt class="docutils literal">save_changes()</tt> when saving this '
- "object.</p>"
- )
- model_body = (
- '<dl class="docutils"><dt><tt class="'
- 'docutils literal">company</tt></dt><dd>Field storing <a class="'
- 'reference external" href="/admindocs/models/myapp.company/">'
- "myapp.Company</a> where the person works.</dd></dl>"
- )
- self.assertContains(self.response, "DESCRIPTION")
- self.assertContains(self.response, summary, html=True)
- self.assertContains(self.response, subheading, html=True)
- self.assertContains(self.response, body, html=True)
- self.assertContains(self.response, model_body, html=True)
- def test_model_docstring_built_in_tag_links(self):
- summary = "Links with different link text."
- body = (
- '<p>This is a line with tag <a class="reference external" '
- 'href="/admindocs/tags/#built_in-extends">extends</a>\n'
- 'This is a line with model <a class="reference external" '
- 'href="/admindocs/models/myapp.family/">Family</a>\n'
- 'This is a line with view <a class="reference external" '
- 'href="/admindocs/views/myapp.views.Index/">Index</a>\n'
- 'This is a line with template <a class="reference external" '
- 'href="/admindocs/templates/Index.html/">index template</a>\n'
- 'This is a line with filter <a class="reference external" '
- 'href="/admindocs/filters/#filtername">example filter</a></p>'
- )
- url = reverse("django-admindocs-models-detail", args=["admin_docs", "family"])
- response = self.client.get(url)
- self.assertContains(response, summary, html=True)
- self.assertContains(response, body, html=True)
- def test_model_detail_title(self):
- self.assertContains(self.response, "<h1>admin_docs.Person</h1>", html=True)
- def test_app_not_found(self):
- response = self.client.get(
- reverse("django-admindocs-models-detail", args=["doesnotexist", "Person"])
- )
- self.assertEqual(response.context["exception"], "App 'doesnotexist' not found")
- self.assertEqual(response.status_code, 404)
- def test_model_not_found(self):
- response = self.client.get(
- reverse(
- "django-admindocs-models-detail", args=["admin_docs", "doesnotexist"]
- )
- )
- self.assertEqual(
- response.context["exception"],
- "Model 'doesnotexist' not found in app 'admin_docs'",
- )
- self.assertEqual(response.status_code, 404)
- def test_model_permission_denied(self):
- person_url = reverse(
- "django-admindocs-models-detail", args=["admin_docs", "person"]
- )
- company_url = reverse(
- "django-admindocs-models-detail", args=["admin_docs", "company"]
- )
- staff_user = User.objects.create_user(
- username="staff", password="secret", is_staff=True
- )
- self.client.force_login(staff_user)
- response_for_person = self.client.get(person_url)
- response_for_company = self.client.get(company_url)
- # No access without permissions.
- self.assertEqual(response_for_person.status_code, 403)
- self.assertEqual(response_for_company.status_code, 403)
- company_content_type = ContentType.objects.get_for_model(Company)
- person_content_type = ContentType.objects.get_for_model(Person)
- view_company = Permission.objects.get(
- codename="view_company", content_type=company_content_type
- )
- change_person = Permission.objects.get(
- codename="change_person", content_type=person_content_type
- )
- staff_user.user_permissions.add(view_company, change_person)
- with captured_stderr():
- response_for_person = self.client.get(person_url)
- response_for_company = self.client.get(company_url)
- # View or change permission grants access.
- self.assertEqual(response_for_person.status_code, 200)
- self.assertEqual(response_for_company.status_code, 200)
- @unittest.skipUnless(utils.docutils_is_available, "no docutils installed.")
- class TestModelIndexView(TestDataMixin, AdminDocsTestCase):
- def test_model_index_superuser(self):
- self.client.force_login(self.superuser)
- index_url = reverse("django-admindocs-models-index")
- response = self.client.get(index_url)
- self.assertContains(
- response,
- '<a href="/admindocs/models/admin_docs.family/">Family</a>',
- html=True,
- )
- self.assertContains(
- response,
- '<a href="/admindocs/models/admin_docs.person/">Person</a>',
- html=True,
- )
- self.assertContains(
- response,
- '<a href="/admindocs/models/admin_docs.company/">Company</a>',
- html=True,
- )
- def test_model_index_with_model_permission(self):
- staff_user = User.objects.create_user(
- username="staff", password="secret", is_staff=True
- )
- self.client.force_login(staff_user)
- index_url = reverse("django-admindocs-models-index")
- response = self.client.get(index_url)
- # Models are not listed without permissions.
- self.assertNotContains(
- response,
- '<a href="/admindocs/models/admin_docs.family/">Family</a>',
- html=True,
- )
- self.assertNotContains(
- response,
- '<a href="/admindocs/models/admin_docs.person/">Person</a>',
- html=True,
- )
- self.assertNotContains(
- response,
- '<a href="/admindocs/models/admin_docs.company/">Company</a>',
- html=True,
- )
- company_content_type = ContentType.objects.get_for_model(Company)
- person_content_type = ContentType.objects.get_for_model(Person)
- view_company = Permission.objects.get(
- codename="view_company", content_type=company_content_type
- )
- change_person = Permission.objects.get(
- codename="change_person", content_type=person_content_type
- )
- staff_user.user_permissions.add(view_company, change_person)
- response = self.client.get(index_url)
- # View or change permission grants access.
- self.assertNotContains(
- response,
- '<a href="/admindocs/models/admin_docs.family/">Family</a>',
- html=True,
- )
- self.assertContains(
- response,
- '<a href="/admindocs/models/admin_docs.person/">Person</a>',
- html=True,
- )
- self.assertContains(
- response,
- '<a href="/admindocs/models/admin_docs.company/">Company</a>',
- html=True,
- )
- class CustomField(models.Field):
- description = "A custom field type"
- class DescriptionLackingField(models.Field):
- pass
- class TestFieldType(unittest.TestCase):
- def test_field_name(self):
- with self.assertRaises(AttributeError):
- views.get_readable_field_data_type("NotAField")
- def test_builtin_fields(self):
- self.assertEqual(
- views.get_readable_field_data_type(fields.BooleanField()),
- "Boolean (Either True or False)",
- )
- def test_char_fields(self):
- self.assertEqual(
- views.get_readable_field_data_type(fields.CharField(max_length=255)),
- "String (up to 255)",
- )
- self.assertEqual(
- views.get_readable_field_data_type(fields.CharField()),
- "String (unlimited)",
- )
- def test_custom_fields(self):
- self.assertEqual(
- views.get_readable_field_data_type(CustomField()), "A custom field type"
- )
- self.assertEqual(
- views.get_readable_field_data_type(DescriptionLackingField()),
- "Field of type: DescriptionLackingField",
- )
- class AdminDocViewFunctionsTests(SimpleTestCase):
- def test_simplify_regex(self):
- tests = (
- # Named and unnamed groups.
- (r"^(?P<a>\w+)/b/(?P<c>\w+)/$", "/<a>/b/<c>/"),
- (r"^(?P<a>\w+)/b/(?P<c>\w+)$", "/<a>/b/<c>"),
- (r"^(?P<a>\w+)/b/(?P<c>\w+)", "/<a>/b/<c>"),
- (r"^(?P<a>\w+)/b/(\w+)$", "/<a>/b/<var>"),
- (r"^(?P<a>\w+)/b/(\w+)", "/<a>/b/<var>"),
- (r"^(?P<a>\w+)/b/((x|y)\w+)$", "/<a>/b/<var>"),
- (r"^(?P<a>\w+)/b/((x|y)\w+)", "/<a>/b/<var>"),
- (r"^(?P<a>(x|y))/b/(?P<c>\w+)$", "/<a>/b/<c>"),
- (r"^(?P<a>(x|y))/b/(?P<c>\w+)", "/<a>/b/<c>"),
- (r"^(?P<a>(x|y))/b/(?P<c>\w+)ab", "/<a>/b/<c>ab"),
- (r"^(?P<a>(x|y)(\(|\)))/b/(?P<c>\w+)ab", "/<a>/b/<c>ab"),
- # Non-capturing groups.
- (r"^a(?:\w+)b", "/ab"),
- (r"^a(?:(x|y))", "/a"),
- (r"^(?:\w+(?:\w+))a", "/a"),
- (r"^a(?:\w+)/b(?:\w+)", "/a/b"),
- (r"(?P<a>\w+)/b/(?:\w+)c(?:\w+)", "/<a>/b/c"),
- (r"(?P<a>\w+)/b/(\w+)/(?:\w+)c(?:\w+)", "/<a>/b/<var>/c"),
- # Single and repeated metacharacters.
- (r"^a", "/a"),
- (r"^^a", "/a"),
- (r"^^^a", "/a"),
- (r"a$", "/a"),
- (r"a$$", "/a"),
- (r"a$$$", "/a"),
- (r"a?", "/a"),
- (r"a??", "/a"),
- (r"a???", "/a"),
- (r"a*", "/a"),
- (r"a**", "/a"),
- (r"a***", "/a"),
- (r"a+", "/a"),
- (r"a++", "/a"),
- (r"a+++", "/a"),
- (r"\Aa", "/a"),
- (r"\A\Aa", "/a"),
- (r"\A\A\Aa", "/a"),
- (r"a\Z", "/a"),
- (r"a\Z\Z", "/a"),
- (r"a\Z\Z\Z", "/a"),
- (r"\ba", "/a"),
- (r"\b\ba", "/a"),
- (r"\b\b\ba", "/a"),
- (r"a\B", "/a"),
- (r"a\B\B", "/a"),
- (r"a\B\B\B", "/a"),
- # Multiple mixed metacharacters.
- (r"^a/?$", "/a/"),
- (r"\Aa\Z", "/a"),
- (r"\ba\B", "/a"),
- # Escaped single metacharacters.
- (r"\^a", r"/^a"),
- (r"\\^a", r"/\\a"),
- (r"\\\^a", r"/\\^a"),
- (r"\\\\^a", r"/\\\\a"),
- (r"\\\\\^a", r"/\\\\^a"),
- (r"a\$", r"/a$"),
- (r"a\\$", r"/a\\"),
- (r"a\\\$", r"/a\\$"),
- (r"a\\\\$", r"/a\\\\"),
- (r"a\\\\\$", r"/a\\\\$"),
- (r"a\?", r"/a?"),
- (r"a\\?", r"/a\\"),
- (r"a\\\?", r"/a\\?"),
- (r"a\\\\?", r"/a\\\\"),
- (r"a\\\\\?", r"/a\\\\?"),
- (r"a\*", r"/a*"),
- (r"a\\*", r"/a\\"),
- (r"a\\\*", r"/a\\*"),
- (r"a\\\\*", r"/a\\\\"),
- (r"a\\\\\*", r"/a\\\\*"),
- (r"a\+", r"/a+"),
- (r"a\\+", r"/a\\"),
- (r"a\\\+", r"/a\\+"),
- (r"a\\\\+", r"/a\\\\"),
- (r"a\\\\\+", r"/a\\\\+"),
- (r"\\Aa", r"/\Aa"),
- (r"\\\Aa", r"/\\a"),
- (r"\\\\Aa", r"/\\\Aa"),
- (r"\\\\\Aa", r"/\\\\a"),
- (r"\\\\\\Aa", r"/\\\\\Aa"),
- (r"a\\Z", r"/a\Z"),
- (r"a\\\Z", r"/a\\"),
- (r"a\\\\Z", r"/a\\\Z"),
- (r"a\\\\\Z", r"/a\\\\"),
- (r"a\\\\\\Z", r"/a\\\\\Z"),
- # Escaped mixed metacharacters.
- (r"^a\?$", r"/a?"),
- (r"^a\\?$", r"/a\\"),
- (r"^a\\\?$", r"/a\\?"),
- (r"^a\\\\?$", r"/a\\\\"),
- (r"^a\\\\\?$", r"/a\\\\?"),
- # Adjacent escaped metacharacters.
- (r"^a\?\$", r"/a?$"),
- (r"^a\\?\\$", r"/a\\\\"),
- (r"^a\\\?\\\$", r"/a\\?\\$"),
- (r"^a\\\\?\\\\$", r"/a\\\\\\\\"),
- (r"^a\\\\\?\\\\\$", r"/a\\\\?\\\\$"),
- # Complex examples with metacharacters and (un)named groups.
- (r"^\b(?P<slug>\w+)\B/(\w+)?", "/<slug>/<var>"),
- (r"^\A(?P<slug>\w+)\Z", "/<slug>"),
- )
- for pattern, output in tests:
- with self.subTest(pattern=pattern):
- self.assertEqual(simplify_regex(pattern), output)
|