1. import os
    
  2. from pathlib import Path
    
  3. from unittest import mock
    
  4. 
    
  5. from django.core.exceptions import ImproperlyConfigured
    
  6. from django.test import SimpleTestCase, override_settings
    
  7. from django.urls.resolvers import LocaleRegexDescriptor, RegexPattern
    
  8. from django.utils import translation
    
  9. 
    
  10. here = os.path.dirname(os.path.abspath(__file__))
    
  11. 
    
  12. 
    
  13. @override_settings(LOCALE_PATHS=[os.path.join(here, "translations", "locale")])
    
  14. class LocaleRegexDescriptorTests(SimpleTestCase):
    
  15.     def setUp(self):
    
  16.         translation.trans_real._translations = {}
    
  17. 
    
  18.     def tearDown(self):
    
  19.         translation.trans_real._translations = {}
    
  20. 
    
  21.     def test_translated_regex_compiled_per_language(self):
    
  22.         provider = RegexPattern(translation.gettext_lazy("^foo/$"))
    
  23.         with translation.override("de"):
    
  24.             de_compiled = provider.regex
    
  25.             # compiled only once per language
    
  26.             error = AssertionError(
    
  27.                 "tried to compile url regex twice for the same language"
    
  28.             )
    
  29.             with mock.patch("django.urls.resolvers.re.compile", side_effect=error):
    
  30.                 de_compiled_2 = provider.regex
    
  31.         with translation.override("fr"):
    
  32.             fr_compiled = provider.regex
    
  33.         self.assertEqual(fr_compiled.pattern, "^foo-fr/$")
    
  34.         self.assertEqual(de_compiled.pattern, "^foo-de/$")
    
  35.         self.assertEqual(de_compiled, de_compiled_2)
    
  36. 
    
  37.     def test_nontranslated_regex_compiled_once(self):
    
  38.         provider = RegexPattern("^foo/$")
    
  39.         with translation.override("de"):
    
  40.             de_compiled = provider.regex
    
  41.         with translation.override("fr"):
    
  42.             # compiled only once, regardless of language
    
  43.             error = AssertionError("tried to compile non-translated url regex twice")
    
  44.             with mock.patch("django.urls.resolvers.re.compile", side_effect=error):
    
  45.                 fr_compiled = provider.regex
    
  46.         self.assertEqual(de_compiled.pattern, "^foo/$")
    
  47.         self.assertEqual(fr_compiled.pattern, "^foo/$")
    
  48. 
    
  49.     def test_regex_compile_error(self):
    
  50.         """Regex errors are re-raised as ImproperlyConfigured."""
    
  51.         provider = RegexPattern("*")
    
  52.         msg = '"*" is not a valid regular expression: nothing to repeat'
    
  53.         with self.assertRaisesMessage(ImproperlyConfigured, msg):
    
  54.             provider.regex
    
  55. 
    
  56.     def test_access_locale_regex_descriptor(self):
    
  57.         self.assertIsInstance(RegexPattern.regex, LocaleRegexDescriptor)
    
  58. 
    
  59. 
    
  60. @override_settings(LOCALE_PATHS=[Path(here) / "translations" / "locale"])
    
  61. class LocaleRegexDescriptorPathLibTests(LocaleRegexDescriptorTests):
    
  62.     pass