diff options
Diffstat (limited to 'test/ext/test_babelplugin.py')
| -rw-r--r-- | test/ext/test_babelplugin.py | 37 |
1 files changed, 36 insertions, 1 deletions
diff --git a/test/ext/test_babelplugin.py b/test/ext/test_babelplugin.py index 1dacf33..4f17ad9 100644 --- a/test/ext/test_babelplugin.py +++ b/test/ext/test_babelplugin.py @@ -1,7 +1,8 @@ import io -import mock +import os import unittest from mako.ext.babelplugin import extract +from .. import TemplateTest, template_base class Test_extract(unittest.TestCase): @@ -25,3 +26,37 @@ class Test_extract(unittest.TestCase): self.assertEqual( messages, [(4, '_', u'Message', [u'TRANSLATORS: This is a comment.'])]) + + +class ExtractMakoTestCase(TemplateTest): + def test_extract(self): + mako_tmpl = open(os.path.join(template_base, 'gettext.mako')) + messages = list(extract(mako_tmpl, {'_': None, 'gettext': None, + 'ungettext': (1, 2)}, + ['TRANSLATOR:'], {})) + expected = \ + [(1, '_', 'Page arg 1', []), + (1, '_', 'Page arg 2', []), + (10, 'gettext', 'Begin', []), + (14, '_', 'Hi there!', ['TRANSLATOR: Hi there!']), + (19, '_', 'Hello', []), + (22, '_', 'Welcome', []), + (25, '_', 'Yo', []), + (36, '_', 'The', ['TRANSLATOR: Ensure so and', 'so, thanks']), + (36, 'ungettext', ('bunny', 'bunnies', None), []), + (41, '_', 'Goodbye', ['TRANSLATOR: Good bye']), + (44, '_', 'Babel', []), + (45, 'ungettext', ('hella', 'hellas', None), []), + (62, '_', 'The', ['TRANSLATOR: Ensure so and', 'so, thanks']), + (62, 'ungettext', ('bunny', 'bunnies', None), []), + (68, '_', 'Goodbye, really!', ['TRANSLATOR: HTML comment']), + (71, '_', 'P.S. byebye', []), + (77, '_', 'Top', []), + (83, '_', 'foo', []), + (83, '_', 'hoho', []), + (85, '_', 'bar', []), + (92, '_', 'Inside a p tag', ['TRANSLATOR: <p> tag is ok?']), + (95, '_', 'Later in a p tag', ['TRANSLATOR: also this']), + (99, '_', 'No action at a distance.', []), + ] + self.assertEqual(expected, messages) |