1 files changed, 80 insertions, 0 deletions
diff --git a/tests/testutil.py b/tests/testutil.py
new file mode 100644
index 0000000..ac98090
--- /dev/null
+++ b/tests/testutil.py
@@ -0,0 +1,80 @@
+# -*- mode: python -*-
+
+from gi.repository import GLib, Gio
+
+from dogtail.utils import isA11yEnabled, enableA11y
+if not isA11yEnabled():
+    enableA11y(True)
+
+from dogtail import tree
+from dogtail import utils
+from dogtail.predicate import *
+from dogtail.procedural import *
+
+import os, sys
+import subprocess
+
+APPLICATION_ID = "org.gnome.ScreenRecorder.Application"
+
+_bus = None
+
+class IsTextEqual(Predicate):
+    """Predicate subclass that looks for top-level windows"""
+    def __init__(self, text):
+        self.text = text
+
+    def satisfiedByNode(self, node):
+        try:
+            textIface = node.queryText()
+            #print textIface.getText(0, -1)
+            return textIface.getText(0, -1) == self.text
+        except NotImplementedError:
+            return False
+
+    def describeSearchResult(self):
+        return '%s text node' % self.text
+
+def _do_bus_call(method, params):
+    global _bus
+
+    if _bus == None:
+        _bus = Gio.bus_get_sync(Gio.BusType.SESSION)
+    _bus.call_sync(APPLICATION_ID, '/' + APPLICATION_ID.replace('.', '/'),
+                   'org.freedesktop.Application',
+                   method, params, None,
+                   Gio.DBusCallFlags.NONE,
+                   -1, None)
+
+def start():
+    builddir = os.environ.get('G_TEST_BUILDDIR', None)
+    if builddir and not 'TESTUTIL_DONT_START' in os.environ:
+        subprocess.Popen([os.path.join(builddir, '..', 'src', APPLICATION_ID)],
+                         cwd=os.path.join(builddir, '..'))
+    else:
+        _do_bus_call("Activate", GLib.Variant('(a{sv})', ([],)))
+    utils.doDelay(3)
+
+    app = tree.root.application(APPLICATION_ID)
+    focus.application(APPLICATION_ID)
+
+    return app
+
+def reset_settings():
+    # need to go through the parser because pygobject does not handle maybe types
+    parsed = GLib.Variant.parse(GLib.VariantType.new('av'),
+                                "[<(uint32 1, <('Linate Airport', 'LIML', "
+                                "false, @m(dd) (0.79296125100499293, "
+                                "0.16202472640904275), @m(dd) (0.79354303905785273, "
+                                "0.16057029118347829))>)>]")
+    settings.set_value("locations", parsed)
+
+def init():
+    global settings, _previous_locations
+
+    settings = Gio.Settings("org.gnome.ScreenRecorder.Application")
+    _previous_locations = settings.get_value("locations")
+    reset_settings()
+
+def fini():
+    settings.set_value("locations", _previous_locations)
+    _do_bus_call("ActivateAction", GLib.Variant('(sava{sv})', ('quit', [], [])))