diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/__init__.py | 7 | ||||
-rw-r--r-- | tests/test_add_reader.py | 27 |
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') |