1 files changed, 56 insertions, 0 deletions
diff --git a/tests/gtkdemotest.py b/tests/gtkdemotest.py
new file mode 100644
index 00000000000..2caa0b4824c
--- /dev/null
+++ b/tests/gtkdemotest.py
@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+import unittest
+
+
+class GtkDemoTest(unittest.TestCase):
+    """
+    TestCase subclass which handles bringing up and shutting down gtk-demo as a fixture.  Used for writing other test cases.
+    """
+
+    def setUp(self):
+        import dogtail.config
+        dogtail.config.config.logDebugToStdOut = True
+        dogtail.config.config.logDebugToFile = False
+        import dogtail.utils
+        self.pid = dogtail.utils.run('gtk3-demo')
+        self.app = dogtail.tree.root.application('gtk3-demo')
+
+    def tearDown(self):
+        import os
+        import signal
+        import time
+        os.kill(self.pid, signal.SIGKILL)
+        # Sleep just enough to let the app actually die.
+        # AT-SPI doesn't like being hammered too fast.
+        time.sleep(0.5)
+
+    def runDemo(self, demoName):
+        """
+        Click on the named demo within the gtk-demo app.
+        """
+        tree = self.app.child(roleName="tree table")
+        tree.child(demoName).doActionNamed('activate')
+
+
+def trap_stdout(function, args=None):
+    """
+    Grab stdout output during function execution
+    """
+
+    import sys
+    from StringIO import StringIO
+
+    saved_stdout = sys.stdout
+    try:
+        out = StringIO()
+        sys.stdout = out
+        if type(args) is dict:
+            function(**args)
+        elif args:
+            function(args)
+        else:
+            function()
+        output = out.getvalue().strip()
+    finally:
+        sys.stdout = saved_stdout
+    return output