Rework tests to be based on pyatspi only

dogtail is not packaged for ubuntu at the moment and dogtail adds
another layer which can make it difficult to debug intermittent
failures. We might switch back at some point but for now I think
it's easier to stay lower level and try to figure out how to
improve reliability there.

1 files changed, 64 insertions, 0 deletions

diff --git a/tests/tree.py b/tests/tree.py
new file mode 100644
index 0000000..e8746e2
--- /dev/null
+++ b/tests/tree.py
@@ -0,0 +1,64 @@
+import pyatspi
+
+def get_root():
+    return Node(pyatspi.Registry.getDesktop(0))
+
+class Node:
+    def __init__(self, accessible):
+        self._accessible = accessible
+
+    def _predicate(self, accessible, name, role_name): 
+        if name is not None and name != accessible.name:
+            return False
+
+        if role_name is not None and role_name != accessible.getRoleName():
+            return False
+
+        return True
+
+    def find_child(self, name=None, role_name=None):
+        def predicate(accessible):
+            return self._predicate(accessible, name, role_name)
+
+        accessible = pyatspi.findDescendant(self._accessible, predicate)
+
+        return Node(accessible)
+
+    def find_children(self, name=None, role_name=None):
+        def predicate(accessible):
+            return self._predicate(accessible, name, role_name)
+
+        all_accessibles = pyatspi.findAllDescendants(self._accessible, predicate)
+
+        return [Node(accessible) for accessible in all_accessibles]
+
+    def _dump_accessible(self, accessible, depth):
+        print "" * depth + str(accessible)
+
+    def _crawl_accessible(self, accessible, depth):
+        self._dump_accessible(accessible, depth)
+
+        for child in self.find_children():
+            self._crawl_accessible(child, depth + 1)
+
+    def dump(self):
+        self._crawl_accessible(self._accessible, 0)
+
+    def do_action(self, name):
+        action = self._accessible.queryAction()
+        for i in range(action.nActions):
+            if action.getName(i) == name:
+                action.doAction(i)
+
+    def click(self, button=1):
+        component = self._accessible.queryComponent()
+        x, y = component.getPosition(pyatspi.DESKTOP_COORDS)
+        pyatspi.Registry.generateMouseEvent(x, y, "b%sc" % button)
+
+    @property
+    def name(self):
+        return self._accessible.name
+
+    @property
+    def text(self):
+        return self._accessible.queryText().getText(0, -1)