1 files changed, 56 insertions, 0 deletions
diff --git a/testing/marionette/harness/marionette_harness/tests/unit/test_localization.py b/testing/marionette/harness/marionette_harness/tests/unit/test_localization.py
new file mode 100644
index 000000000..1b89f6f34
--- /dev/null
+++ b/testing/marionette/harness/marionette_harness/tests/unit/test_localization.py
@@ -0,0 +1,56 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+from marionette_driver import By
+from marionette_driver.errors import (
+    InvalidArgumentException,
+    NoSuchElementException,
+    UnknownException
+)
+from marionette_driver.localization import L10n
+
+from marionette_harness import MarionetteTestCase
+
+
+class TestL10n(MarionetteTestCase):
+
+    def setUp(self):
+        super(TestL10n, self).setUp()
+
+        self.l10n = L10n(self.marionette)
+
+    def test_localize_entity(self):
+        dtds = ['chrome://marionette/content/test_dialog.dtd']
+        value = self.l10n.localize_entity(dtds, 'testDialog.title')
+
+        self.assertEqual(value, 'Test Dialog')
+
+    def test_localize_entity_invalid_arguments(self):
+        dtds = ['chrome://marionette/content/test_dialog.dtd']
+
+        self.assertRaises(NoSuchElementException,
+                          self.l10n.localize_entity, dtds, 'notExistent')
+        self.assertRaises(InvalidArgumentException,
+                          self.l10n.localize_entity, dtds[0], 'notExistent')
+        self.assertRaises(InvalidArgumentException,
+                          self.l10n.localize_entity, dtds, True)
+
+    def test_localize_property(self):
+        properties = ['chrome://marionette/content/test_dialog.properties']
+
+        value = self.l10n.localize_property(properties, 'testDialog.title')
+        self.assertEqual(value, 'Test Dialog')
+
+        self.assertRaises(NoSuchElementException,
+                          self.l10n.localize_property, properties, 'notExistent')
+
+    def test_localize_property_invalid_arguments(self):
+        properties = ['chrome://global/locale/filepicker.properties']
+
+        self.assertRaises(NoSuchElementException,
+                          self.l10n.localize_property, properties, 'notExistent')
+        self.assertRaises(InvalidArgumentException,
+                          self.l10n.localize_property, properties[0], 'notExistent')
+        self.assertRaises(InvalidArgumentException,
+                          self.l10n.localize_property, properties, True)