add a first automated unit test

2 files changed, 34 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..d5565cd
--- /dev/null
+++ b/tests/__init__.py
@@ -0,0 +1,7 @@
+import aiogreen
+import unittest
+
+class TestCase(unittest.TestCase):
+    def setUp(self):
+        self.loop = aiogreen.EventLoop()
+        self.addCleanup(self.loop.close)
diff --git a/tests/test_add_reader.py b/tests/test_add_reader.py
new file mode 100644
index 0000000..14262ba
--- /dev/null
+++ b/tests/test_add_reader.py
@@ -0,0 +1,27 @@
+from __future__ import absolute_import
+from aiogreen import socketpair
+import tests
+
+
+class AddReaderTests(tests.TestCase):
+    def test_add_reader(self):
+        result = {'received': None}
+        rsock, wsock = socketpair()
+        self.addCleanup(rsock.close)
+        self.addCleanup(wsock.close)
+
+        def reader():
+            data = rsock.recv(100)
+            result['received'] = data
+            self.loop.remove_reader(rsock)
+            self.loop.stop()
+
+        def writer():
+            self.loop.remove_writer(wsock)
+            self.loop.call_soon(wsock.send, b'abc')
+
+        self.loop.add_reader(rsock, reader)
+        self.loop.add_writer(wsock, writer)
+
+        self.loop.run_forever()
+        self.assertEqual(result['received'], b'abc')