1 files changed, 63 insertions, 0 deletions
diff --git a/test/ext/test_linguaplugin.py b/test/ext/test_linguaplugin.py
new file mode 100644
index 0000000..6e2faa8
--- /dev/null
+++ b/test/ext/test_linguaplugin.py
@@ -0,0 +1,63 @@
+import os
+
+import pytest
+
+from mako.testing.assertions import eq_
+from mako.testing.config import config
+from mako.testing.exclusions import requires_lingua
+from mako.testing.fixtures import TemplateTest
+
+
+class MockOptions:
+    keywords = []
+    domain = None
+    comment_tag = True
+
+
+@requires_lingua
+class MakoExtractTest(TemplateTest):
+    @pytest.fixture(autouse=True)
+    def register_lingua_extractors(self):
+        from lingua.extractors import register_extractors
+
+        register_extractors()
+
+    def test_extract(self):
+        from mako.ext.linguaplugin import LinguaMakoExtractor
+
+        plugin = LinguaMakoExtractor({"comment-tags": "TRANSLATOR"})
+        messages = list(
+            plugin(
+                os.path.join(config.template_base, "gettext.mako"),
+                MockOptions(),
+            )
+        )
+        msgids = [(m.msgid, m.msgid_plural) for m in messages]
+        eq_(
+            msgids,
+            [
+                ("Page arg 1", None),
+                ("Page arg 2", None),
+                ("Begin", None),
+                ("Hi there!", None),
+                ("Hello", None),
+                ("Welcome", None),
+                ("Yo", None),
+                ("The", None),
+                ("bunny", "bunnies"),
+                ("Goodbye", None),
+                ("Babel", None),
+                ("hella", "hellas"),
+                ("The", None),
+                ("bunny", "bunnies"),
+                ("Goodbye, really!", None),
+                ("P.S. byebye", None),
+                ("Top", None),
+                ("foo", None),
+                ("hoho", None),
+                ("bar", None),
+                ("Inside a p tag", None),
+                ("Later in a p tag", None),
+                ("No action at a distance.", None),
+            ],
+        )