77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
import os
|
|
import string
|
|
import random
|
|
import re
|
|
from polib import pofile
|
|
from unittest import TestCase
|
|
from datetime import datetime, timedelta
|
|
from pytz import UTC
|
|
|
|
import generate
|
|
from config import CONFIGURATION
|
|
|
|
|
|
class TestGenerate(TestCase):
|
|
"""
|
|
Tests functionality of i18n/generate.py
|
|
"""
|
|
generated_files = ('django-partial.po', 'djangojs.po', 'mako.po')
|
|
|
|
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(CONFIGURATION.source_messages_dir, random_name())
|
|
generate.merge(CONFIGURATION.source_locale, target=filename)
|
|
self.assertTrue(os.path.exists(filename))
|
|
os.remove(filename)
|
|
|
|
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()
|
|
for locale in CONFIGURATION.locales:
|
|
for filename in ('django', 'djangojs'):
|
|
mofile = filename+'.mo'
|
|
path = os.path.join(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)
|
|
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(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))
|