Yessss....test from rfk. TYVM Fixes #76.

This commit is contained in:
Ryan Williams
2011-01-28 23:15:13 -08:00
parent f443bf514b
commit 8c808e9184
2 changed files with 11 additions and 4 deletions

View File

@@ -99,7 +99,7 @@ class Socket(__zmq__.Socket):
def _recv_message(self, flags=0, track=False):
if flags & __zmq__.NOBLOCK:
return super(Socket,self)._recv_message(flags)
return super(Socket, self)._recv_message(flags, track)
flags |= __zmq__.NOBLOCK
while True:
try:

View File

@@ -228,6 +228,13 @@ got '%s'" % (zmq.ZMQError(errno), zmq.ZMQError(e.errno)))
# but it's private __str__ appears to be the way to go
self.assertEqual([str(m) for m in recieved_msg], msg2)
@skip_unless_zmq
def test_recv_noblock_bug76(self):
req, rep, port = self.create_bound_pair(zmq.REQ, zmq.REP)
self.assertRaisesErrno(zmq.EAGAIN, rep.recv, zmq.NOBLOCK)
self.assertRaisesErrno(zmq.EAGAIN, rep.recv, zmq.NOBLOCK, True)
class TestThreadedContextAccess(TestCase):
"""zmq's Context must be unique within a hub