1 files changed, 60 insertions, 0 deletions
diff --git a/tests/autopilot/tests/test_indicator_keyboard.py b/tests/autopilot/tests/test_indicator_keyboard.py
new file mode 100644
index 00000000..0ef197db
--- /dev/null
+++ b/tests/autopilot/tests/test_indicator_keyboard.py
@@ -0,0 +1,60 @@
+import autopilot.introspection.gtk
+import os
+import pyatspi.registry
+import pyatspi.utils
+import time
+import unity.tests
+
+def print_accessible(root, level=0):
+    print level * ' ', root
+
+    for node in root:
+        print_accessible(node, level + 1)
+
+def get_accessible_with_name_and_role(root, name, role):
+    is_accessible = lambda a: a.name == name and a.get_role_name() == role
+    return pyatspi.utils.findDescendant(root, is_accessible, True);
+
+def get_panel_accessible(root):
+    return get_accessible_with_name_and_role(root, 'unity-panel-service', 'application')
+
+def is_indicator_accessible(root):
+    return root.get_role_name() == 'panel' and \
+           len(root) == 1 and \
+           root[0].get_role_name() == 'image' and \
+           len(root[0]) == 1 and \
+           root[0][0].get_role_name() == 'menu' and \
+           len(root[0][0]) > 3 and \
+           root[0][0][-3].name == 'Character Map' and \
+           root[0][0][-3].get_role_name() == 'check menu item' and \
+           root[0][0][-2].name == 'Keyboard Layout Chart' and \
+           root[0][0][-2].get_role_name() == 'check menu item' and \
+           root[0][0][-1].name == 'Text Entry Settings...' and \
+           root[0][0][-1].get_role_name() == 'check menu item'
+
+def get_indicator_accessible(root):
+    return pyatspi.utils.findDescendant(root, is_indicator_accessible, True)
+
+def get_accessible_index(root, node):
+    for i in xrange(len(root)):
+        if root[i] == node:
+            return i
+
+    return -1
+
+class IndicatorKeyboardTestCase(unity.tests.UnityTestCase):
+
+    def setUp(self):
+        super(IndicatorKeyboardTestCase, self).setUp()
+
+        registry = pyatspi.registry.Registry()
+        desktop = registry.getDesktop(0)
+        panel = get_panel_accessible(desktop)
+        self.indicator = get_indicator_accessible(panel)
+
+        # This is needed on systems other than the EN locale
+        os.putenv("LC_ALL", "C")
+        self.addCleanup(os.unsetenv, "LC_ALL")
+
+    def test_indicator(self):
+        print_accessible(self.indicator)