99 lines
3.8 KiB
Python
99 lines
3.8 KiB
Python
from datetime import datetime, timedelta
|
|
import os
|
|
import sys
|
|
import string
|
|
import random
|
|
import re
|
|
|
|
from unittest import TestCase
|
|
from mock import patch
|
|
from polib import pofile
|
|
from pytz import UTC
|
|
|
|
from i18n import config
|
|
from i18n import extract
|
|
from i18n import generate
|
|
from i18n import dummy
|
|
|
|
|
|
class TestGenerate(TestCase):
|
|
"""
|
|
Tests functionality of i18n/generate.py
|
|
"""
|
|
generated_files = ('django-partial.po', 'djangojs-partial.po', 'mako.po')
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cfg = os.path.join('conf', 'locale', 'config.yaml')
|
|
config.CONFIGURATION = config.Configuration(cfg)
|
|
sys.stderr.write(
|
|
"\nExtracting i18n strings and generating dummy translations; "
|
|
"this may take a few minutes\n"
|
|
)
|
|
sys.stderr.flush()
|
|
extract.main(verbosity=0, config=cfg)
|
|
dummy.main(verbosity=0, config=cfg)
|
|
|
|
def setUp(self):
|
|
# Subtract 1 second to help comparisons with file-modify time succeed,
|
|
# since os.path.getmtime() is not millisecond-accurate
|
|
self.start_time = datetime.now(UTC) - timedelta(seconds=1)
|
|
|
|
def test_merge(self):
|
|
"""
|
|
Tests merge script on English source files.
|
|
"""
|
|
filename = os.path.join(config.CONFIGURATION.source_messages_dir, random_name())
|
|
generate.merge(config.CONFIGURATION.source_locale, target=filename)
|
|
self.assertTrue(os.path.exists(filename))
|
|
os.remove(filename)
|
|
|
|
# Patch dummy_locales to not have esperanto present
|
|
@patch.object(config.CONFIGURATION, 'dummy_locales', ['fake2'])
|
|
def test_main(self):
|
|
"""
|
|
Runs generate.main() which should merge source files,
|
|
then compile all sources in all configured languages.
|
|
Validates output by checking all .mo files in all configured languages.
|
|
.mo files should exist, and be recently created (modified
|
|
after start of test suite)
|
|
"""
|
|
generate.main(verbosity=0, strict=False)
|
|
for locale in config.CONFIGURATION.translated_locales:
|
|
for filename in ('django', 'djangojs'):
|
|
mofile = filename+'.mo'
|
|
path = os.path.join(config.CONFIGURATION.get_messages_dir(locale), mofile)
|
|
exists = os.path.exists(path)
|
|
self.assertTrue(exists, msg='Missing file in locale %s: %s' % (locale, mofile))
|
|
self.assertTrue(datetime.fromtimestamp(os.path.getmtime(path), UTC) >= self.start_time,
|
|
msg='File not recently modified: %s' % path)
|
|
# Segmenting means that the merge headers don't work they way they
|
|
# used to, so don't make this check for now. I'm not sure if we'll
|
|
# get the merge header back eventually, or delete this code eventually.
|
|
# self.assert_merge_headers(locale)
|
|
|
|
def assert_merge_headers(self, locale):
|
|
"""
|
|
This is invoked by test_main to ensure that it runs after
|
|
calling generate.main().
|
|
|
|
There should be exactly three merge comment headers
|
|
in our merged .po file. This counts them to be sure.
|
|
A merge comment looks like this:
|
|
# #-#-#-#-# django-partial.po (0.1a) #-#-#-#-#
|
|
|
|
"""
|
|
path = os.path.join(config.CONFIGURATION.get_messages_dir(locale), 'django.po')
|
|
po = pofile(path)
|
|
pattern = re.compile('^#-#-#-#-#', re.M)
|
|
match = pattern.findall(po.header)
|
|
self.assertEqual(len(match), 3,
|
|
msg="Found %s (should be 3) merge comments in the header for %s" % \
|
|
(len(match), path))
|
|
|
|
|
|
def random_name(size=6):
|
|
"""Returns random filename as string, like test-4BZ81W"""
|
|
chars = string.ascii_uppercase + string.digits
|
|
return 'test-' + ''.join(random.choice(chars) for x in range(size))
|