Moved test_generate from i18n-tools

common/lib/test_generate.py

@@ -0,0 +1,98 @@
+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))