2 files changed, 68 insertions, 1 deletions
diff --git a/tests/Makefile.am b/tests/Makefile.am
index 25a1a6d..b6ced0e 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -30,7 +30,8 @@ tests =  \
 	test_subprocess.py \
 	test_subtype.py \
 	test_gdkevent.py \
-	test_unknown.py
+	test_unknown.py \
+	test_source.py
 
 # This is a hack to make sure a shared library is built
 testhelper.la: $(testhelper_la_OBJECTS) $(testhelper_la_DEPENDENCIES)
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()