Merge pull request #86 from akhoronko/master

Use passed to create_zmq_stream events_backlog parameter

aiozmq/stream.py

@@ -32,7 +32,8 @@ def create_zmq_stream(zmq_type, *, bind=None, connect=None,
     """
     if loop is None:
         loop = asyncio.get_event_loop()
-    stream = ZmqStream(loop=loop, high=high_read, low=low_read)
+    stream = ZmqStream(loop=loop, high=high_read, low=low_read,
+                       events_backlog=events_backlog)
     tr, _ = yield from create_zmq_connection(
         lambda: stream._protocol,
         zmq_type,

tests/zmq_stream_test.py

@@ -624,6 +624,31 @@ def f(s, events):
 
         self.loop.run_until_complete(go())
 
+    def test_default_events_backlog(self):
+        @asyncio.coroutine
+        def go():
+            s1 = yield from aiozmq.create_zmq_stream(
+                zmq.DEALER,
+                bind='tcp://127.0.0.1:*',
+                loop=self.loop)
+
+            self.assertEqual(100, s1._event_queue.maxlen)
+
+        self.loop.run_until_complete(go())
+
+    def test_custom_events_backlog(self):
+        @asyncio.coroutine
+        def go():
+            s1 = yield from aiozmq.create_zmq_stream(
+                zmq.DEALER,
+                bind='tcp://127.0.0.1:*',
+                loop=self.loop,
+                events_backlog=1)
+
+            self.assertEqual(1, s1._event_queue.maxlen)
+
+        self.loop.run_until_complete(go())
+
 
 if __name__ == '__main__':
     unittest.main()