1 files changed, 66 insertions, 0 deletions
diff --git a/tests/test_source.py b/tests/test_source.py
new file mode 100644
index 0000000..5e0f01b
--- /dev/null
+++ b/tests/test_source.py
@@ -0,0 +1,66 @@
+#!/usr/bin/env python
+
+import exceptions
+import os
+import sys
+import select
+import unittest
+
+from common import gobject
+
+class Idle(gobject.Idle):
+    def __init__(self, loop):
+        gobject.Idle.__init__(self)
+        self.count = 0
+        self.set_callback(self.callback, loop)
+
+    def callback(self, loop):
+        self.count += 1
+        return True
+
+class MySource(gobject.Source):
+    def __init__(self):
+        gobject.Source.__init__(self)
+
+    def prepare(self):
+        return True, 0
+
+    def check(self):
+        return True
+
+    def dispatch(self, callback, args):
+        return callback(*args)
+
+class TestSource(unittest.TestCase):
+    def timeout_callback(self, loop):
+        loop.quit()
+
+    def my_callback(self, loop):
+        self.pos += 1
+        return True
+
+    def setup_timeout(self, loop):
+        timeout = gobject.Timeout(500)
+        timeout.set_callback(self.timeout_callback, loop)
+        timeout.attach()
+
+    def testSources(self):
+        loop = gobject.MainLoop()
+
+        self.setup_timeout(loop)
+
+        idle = Idle(loop)
+        idle.attach()
+
+        self.pos = 0
+
+        m = MySource()
+        m.set_callback(self.my_callback, loop)
+        m.attach()
+
+        loop.run()
+
+        assert self.pos >= 0 and idle.count >= 0
+
+if __name__ == '__main__':
+    unittest.main()