mirror of
https://github.com/django/django.git
synced 2024-12-30 13:05:45 +00:00
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import os
|
|
import re
|
|
import shutil
|
|
import tempfile
|
|
|
|
source_code_dir = os.path.dirname(__file__)
|
|
|
|
|
|
def copytree(src, dst):
|
|
shutil.copytree(src, dst, ignore=shutil.ignore_patterns("__pycache__"))
|
|
|
|
|
|
class POFileAssertionMixin:
|
|
def _assertPoKeyword(self, keyword, expected_value, haystack, use_quotes=True):
|
|
q = '"'
|
|
if use_quotes:
|
|
expected_value = '"%s"' % expected_value
|
|
q = "'"
|
|
needle = "%s %s" % (keyword, expected_value)
|
|
expected_value = re.escape(expected_value)
|
|
return self.assertTrue(
|
|
re.search("^%s %s" % (keyword, expected_value), haystack, re.MULTILINE),
|
|
"Could not find %(q)s%(n)s%(q)s in generated PO file"
|
|
% {"n": needle, "q": q},
|
|
)
|
|
|
|
def assertMsgId(self, msgid, haystack, use_quotes=True):
|
|
return self._assertPoKeyword("msgid", msgid, haystack, use_quotes=use_quotes)
|
|
|
|
|
|
class RunInTmpDirMixin:
|
|
"""
|
|
Allow i18n tests that need to generate .po/.mo files to run in an isolated
|
|
temporary filesystem tree created by tempfile.mkdtemp() that contains a
|
|
clean copy of the relevant test code.
|
|
|
|
Test classes using this mixin need to define a `work_subdir` attribute
|
|
which designates the subdir under `tests/i18n/` that will be copied to the
|
|
temporary tree from which its test cases will run.
|
|
|
|
The setUp() method sets the current working dir to the temporary tree.
|
|
It'll be removed when cleaning up.
|
|
"""
|
|
|
|
def setUp(self):
|
|
self._cwd = os.getcwd()
|
|
self.work_dir = tempfile.mkdtemp(prefix="i18n_")
|
|
# Resolve symlinks, if any, in test directory paths.
|
|
self.test_dir = os.path.realpath(os.path.join(self.work_dir, self.work_subdir))
|
|
copytree(os.path.join(source_code_dir, self.work_subdir), self.test_dir)
|
|
# Step out of the temporary working tree before removing it to avoid
|
|
# deletion problems on Windows. Cleanup actions registered with
|
|
# addCleanup() are called in reverse so preserve this ordering.
|
|
self.addCleanup(self._rmrf, self.test_dir)
|
|
self.addCleanup(os.chdir, self._cwd)
|
|
os.chdir(self.test_dir)
|
|
|
|
def _rmrf(self, dname):
|
|
if (
|
|
os.path.commonprefix([self.test_dir, os.path.abspath(dname)])
|
|
!= self.test_dir
|
|
):
|
|
return
|
|
shutil.rmtree(dname)
|