|
|
@ -15,7 +15,7 @@ class BoolTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
|
|
|
|
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
self.assertTrue(select)
|
|
|
|
self.assertTrue(select)
|
|
|
|
self.assertEquals(123, select.get())
|
|
|
|
self.assertEqual(123, select.get())
|
|
|
|
self.assertFalse(select)
|
|
|
|
self.assertFalse(select)
|
|
|
|
|
|
|
|
|
|
|
|
def test_receiver(self):
|
|
|
|
def test_receiver(self):
|
|
|
@ -27,7 +27,7 @@ class BoolTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
|
|
|
|
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
self.assertTrue(select)
|
|
|
|
self.assertTrue(select)
|
|
|
|
self.assertEquals('123', select.get().unpickle())
|
|
|
|
self.assertEqual('123', select.get().unpickle())
|
|
|
|
self.assertFalse(select)
|
|
|
|
self.assertFalse(select)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -38,34 +38,34 @@ class AddTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
select.add(latch)
|
|
|
|
select.add(latch)
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(latch, select._receivers[0])
|
|
|
|
self.assertEqual(latch, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, latch.notify)
|
|
|
|
self.assertEqual(select._put, latch.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_receiver(self):
|
|
|
|
def test_receiver(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
select.add(recv)
|
|
|
|
select.add(recv)
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(recv, select._receivers[0])
|
|
|
|
self.assertEqual(recv, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, recv.notify)
|
|
|
|
self.assertEqual(select._put, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_channel(self):
|
|
|
|
def test_channel(self):
|
|
|
|
context = self.router.local()
|
|
|
|
context = self.router.local()
|
|
|
|
chan = mitogen.core.Channel(self.router, context, 1234)
|
|
|
|
chan = mitogen.core.Channel(self.router, context, 1234)
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
select.add(chan)
|
|
|
|
select.add(chan)
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(chan, select._receivers[0])
|
|
|
|
self.assertEqual(chan, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, chan.notify)
|
|
|
|
self.assertEqual(select._put, chan.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_subselect_empty(self):
|
|
|
|
def test_subselect_empty(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
subselect = self.klass()
|
|
|
|
subselect = self.klass()
|
|
|
|
select.add(subselect)
|
|
|
|
select.add(subselect)
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(subselect, select._receivers[0])
|
|
|
|
self.assertEqual(subselect, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, subselect.notify)
|
|
|
|
self.assertEqual(select._put, subselect.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_subselect_nonempty(self):
|
|
|
|
def test_subselect_nonempty(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
@ -74,15 +74,15 @@ class AddTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
subselect.add(recv)
|
|
|
|
subselect.add(recv)
|
|
|
|
|
|
|
|
|
|
|
|
select.add(subselect)
|
|
|
|
select.add(subselect)
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(subselect, select._receivers[0])
|
|
|
|
self.assertEqual(subselect, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, subselect.notify)
|
|
|
|
self.assertEqual(select._put, subselect.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_subselect_loop_direct(self):
|
|
|
|
def test_subselect_loop_direct(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.add(select))
|
|
|
|
lambda: select.add(select))
|
|
|
|
self.assertEquals(str(exc), self.klass.loop_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.loop_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_subselect_loop_indirect(self):
|
|
|
|
def test_subselect_loop_indirect(self):
|
|
|
|
s0 = self.klass()
|
|
|
|
s0 = self.klass()
|
|
|
@ -93,7 +93,7 @@ class AddTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
s1.add(s2)
|
|
|
|
s1.add(s2)
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: s2.add(s0))
|
|
|
|
lambda: s2.add(s0))
|
|
|
|
self.assertEquals(str(exc), self.klass.loop_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.loop_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_double_add_receiver(self):
|
|
|
|
def test_double_add_receiver(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
@ -101,7 +101,7 @@ class AddTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select.add(recv)
|
|
|
|
select.add(recv)
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.add(recv))
|
|
|
|
lambda: select.add(recv))
|
|
|
|
self.assertEquals(str(exc), self.klass.owned_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.owned_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_double_add_subselect(self):
|
|
|
|
def test_double_add_subselect(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
@ -109,7 +109,7 @@ class AddTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select.add(select2)
|
|
|
|
select.add(select2)
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.add(select2))
|
|
|
|
lambda: select.add(select2))
|
|
|
|
self.assertEquals(str(exc), self.klass.owned_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.owned_msg)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class RemoveTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
class RemoveTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
@ -120,7 +120,7 @@ class RemoveTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.remove(recv))
|
|
|
|
lambda: select.remove(recv))
|
|
|
|
self.assertEquals(str(exc), self.klass.not_present_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.not_present_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_receiver_absent(self):
|
|
|
|
def test_receiver_absent(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
@ -129,22 +129,22 @@ class RemoveTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select.add(recv2)
|
|
|
|
select.add(recv2)
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.remove(recv))
|
|
|
|
lambda: select.remove(recv))
|
|
|
|
self.assertEquals(str(exc), self.klass.not_present_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.not_present_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_receiver_present(self):
|
|
|
|
def test_receiver_present(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
select.add(recv)
|
|
|
|
select.add(recv)
|
|
|
|
select.remove(recv)
|
|
|
|
select.remove(recv)
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(None, recv.notify)
|
|
|
|
self.assertEqual(None, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_latch_empty(self):
|
|
|
|
def test_latch_empty(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.remove(latch))
|
|
|
|
lambda: select.remove(latch))
|
|
|
|
self.assertEquals(str(exc), self.klass.not_present_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.not_present_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_latch_absent(self):
|
|
|
|
def test_latch_absent(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
@ -153,15 +153,15 @@ class RemoveTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select.add(latch2)
|
|
|
|
select.add(latch2)
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.remove(latch))
|
|
|
|
lambda: select.remove(latch))
|
|
|
|
self.assertEquals(str(exc), self.klass.not_present_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.not_present_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_latch_present(self):
|
|
|
|
def test_latch_present(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select.add(latch)
|
|
|
|
select.add(latch)
|
|
|
|
select.remove(latch)
|
|
|
|
select.remove(latch)
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(None, latch.notify)
|
|
|
|
self.assertEqual(None, latch.notify)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class CloseTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
class CloseTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
@ -176,24 +176,24 @@ class CloseTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select.add(latch)
|
|
|
|
select.add(latch)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(select._put, latch.notify)
|
|
|
|
self.assertEqual(select._put, latch.notify)
|
|
|
|
|
|
|
|
|
|
|
|
select.close()
|
|
|
|
select.close()
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(None, latch.notify)
|
|
|
|
self.assertEqual(None, latch.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_one_receiver(self):
|
|
|
|
def test_one_receiver(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
select.add(recv)
|
|
|
|
select.add(recv)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(select._put, recv.notify)
|
|
|
|
self.assertEqual(select._put, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
select.close()
|
|
|
|
select.close()
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(None, recv.notify)
|
|
|
|
self.assertEqual(None, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_one_subselect(self):
|
|
|
|
def test_one_subselect(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
@ -203,16 +203,16 @@ class CloseTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
subselect.add(recv)
|
|
|
|
subselect.add(recv)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(subselect._put, recv.notify)
|
|
|
|
self.assertEqual(subselect._put, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
select.close()
|
|
|
|
select.close()
|
|
|
|
|
|
|
|
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(subselect._put, recv.notify)
|
|
|
|
self.assertEqual(subselect._put, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
subselect.close()
|
|
|
|
subselect.close()
|
|
|
|
self.assertEquals(None, recv.notify)
|
|
|
|
self.assertEqual(None, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class EmptyTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
class EmptyTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
@ -262,20 +262,20 @@ class IterTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
|
|
|
|
|
|
|
|
def test_empty(self):
|
|
|
|
def test_empty(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
self.assertEquals([], list(select))
|
|
|
|
self.assertEqual([], list(select))
|
|
|
|
|
|
|
|
|
|
|
|
def test_nonempty_receiver(self):
|
|
|
|
def test_nonempty_receiver(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
select = self.klass([recv])
|
|
|
|
select = self.klass([recv])
|
|
|
|
msg = mitogen.core.Message.pickled('123')
|
|
|
|
msg = mitogen.core.Message.pickled('123')
|
|
|
|
recv._on_receive(msg)
|
|
|
|
recv._on_receive(msg)
|
|
|
|
self.assertEquals([msg], list(select))
|
|
|
|
self.assertEqual([msg], list(select))
|
|
|
|
|
|
|
|
|
|
|
|
def test_nonempty_latch(self):
|
|
|
|
def test_nonempty_latch(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select = self.klass([latch])
|
|
|
|
select = self.klass([latch])
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
self.assertEquals([123], list(select))
|
|
|
|
self.assertEqual([123], list(select))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class OneShotTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
class OneShotTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
@ -287,9 +287,9 @@ class OneShotTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
msg = mitogen.core.Message.pickled('123')
|
|
|
|
msg = mitogen.core.Message.pickled('123')
|
|
|
|
recv._on_receive(msg)
|
|
|
|
recv._on_receive(msg)
|
|
|
|
msg_ = select.get()
|
|
|
|
msg_ = select.get()
|
|
|
|
self.assertEquals(msg, msg_)
|
|
|
|
self.assertEqual(msg, msg_)
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(None, recv.notify)
|
|
|
|
self.assertEqual(None, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_false_receiver_persists_after_get(self):
|
|
|
|
def test_false_receiver_persists_after_get(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
@ -297,28 +297,28 @@ class OneShotTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
msg = mitogen.core.Message.pickled('123')
|
|
|
|
msg = mitogen.core.Message.pickled('123')
|
|
|
|
recv._on_receive(msg)
|
|
|
|
recv._on_receive(msg)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertEquals(msg, select.get())
|
|
|
|
self.assertEqual(msg, select.get())
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(recv, select._receivers[0])
|
|
|
|
self.assertEqual(recv, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, recv.notify)
|
|
|
|
self.assertEqual(select._put, recv.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_true_latch_removed_after_get(self):
|
|
|
|
def test_true_latch_removed_after_get(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select = self.klass([latch])
|
|
|
|
select = self.klass([latch])
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
self.assertEquals(123, select.get())
|
|
|
|
self.assertEqual(123, select.get())
|
|
|
|
self.assertEquals(0, len(select._receivers))
|
|
|
|
self.assertEqual(0, len(select._receivers))
|
|
|
|
self.assertEquals(None, latch.notify)
|
|
|
|
self.assertEqual(None, latch.notify)
|
|
|
|
|
|
|
|
|
|
|
|
def test_false_latch_persists_after_get(self):
|
|
|
|
def test_false_latch_persists_after_get(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select = self.klass([latch], oneshot=False)
|
|
|
|
select = self.klass([latch], oneshot=False)
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
|
|
|
|
|
|
|
|
self.assertEquals(123, select.get())
|
|
|
|
self.assertEqual(123, select.get())
|
|
|
|
self.assertEquals(1, len(select._receivers))
|
|
|
|
self.assertEqual(1, len(select._receivers))
|
|
|
|
self.assertEquals(latch, select._receivers[0])
|
|
|
|
self.assertEqual(latch, select._receivers[0])
|
|
|
|
self.assertEquals(select._put, latch.notify)
|
|
|
|
self.assertEqual(select._put, latch.notify)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class GetReceiverTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
class GetReceiverTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
@ -328,13 +328,13 @@ class GetReceiverTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.get())
|
|
|
|
lambda: select.get())
|
|
|
|
self.assertEquals(str(exc), self.klass.empty_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.empty_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_timeout_no_receivers(self):
|
|
|
|
def test_timeout_no_receivers(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.get(timeout=1.0))
|
|
|
|
lambda: select.get(timeout=1.0))
|
|
|
|
self.assertEquals(str(exc), self.klass.empty_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.empty_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_zero_timeout(self):
|
|
|
|
def test_zero_timeout(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
@ -353,7 +353,7 @@ class GetReceiverTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
select = self.klass([recv])
|
|
|
|
select = self.klass([recv])
|
|
|
|
msg = select.get()
|
|
|
|
msg = select.get()
|
|
|
|
self.assertEquals('123', msg.unpickle())
|
|
|
|
self.assertEqual('123', msg.unpickle())
|
|
|
|
|
|
|
|
|
|
|
|
def test_nonempty_multiple_items_before_add(self):
|
|
|
|
def test_nonempty_multiple_items_before_add(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
@ -361,9 +361,9 @@ class GetReceiverTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('234'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('234'))
|
|
|
|
select = self.klass([recv], oneshot=False)
|
|
|
|
select = self.klass([recv], oneshot=False)
|
|
|
|
msg = select.get()
|
|
|
|
msg = select.get()
|
|
|
|
self.assertEquals('123', msg.unpickle())
|
|
|
|
self.assertEqual('123', msg.unpickle())
|
|
|
|
msg = select.get()
|
|
|
|
msg = select.get()
|
|
|
|
self.assertEquals('234', msg.unpickle())
|
|
|
|
self.assertEqual('234', msg.unpickle())
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
lambda: select.get(block=False))
|
|
|
|
lambda: select.get(block=False))
|
|
|
|
|
|
|
|
|
|
|
@ -372,21 +372,21 @@ class GetReceiverTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select = self.klass([recv])
|
|
|
|
select = self.klass([recv])
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
msg = select.get()
|
|
|
|
msg = select.get()
|
|
|
|
self.assertEquals('123', msg.unpickle())
|
|
|
|
self.assertEqual('123', msg.unpickle())
|
|
|
|
|
|
|
|
|
|
|
|
def test_nonempty_receiver_attr_set(self):
|
|
|
|
def test_nonempty_receiver_attr_set(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
select = self.klass([recv])
|
|
|
|
select = self.klass([recv])
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
msg = select.get()
|
|
|
|
msg = select.get()
|
|
|
|
self.assertEquals(msg.receiver, recv)
|
|
|
|
self.assertEqual(msg.receiver, recv)
|
|
|
|
|
|
|
|
|
|
|
|
def test_drained_by_other_thread(self):
|
|
|
|
def test_drained_by_other_thread(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
select = self.klass([recv])
|
|
|
|
select = self.klass([recv])
|
|
|
|
msg = recv.get()
|
|
|
|
msg = recv.get()
|
|
|
|
self.assertEquals('123', msg.unpickle())
|
|
|
|
self.assertEqual('123', msg.unpickle())
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
lambda: select.get(timeout=0.0))
|
|
|
|
lambda: select.get(timeout=0.0))
|
|
|
|
|
|
|
|
|
|
|
@ -398,13 +398,13 @@ class GetLatchTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.get())
|
|
|
|
lambda: select.get())
|
|
|
|
self.assertEquals(str(exc), self.klass.empty_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.empty_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_timeout_no_receivers(self):
|
|
|
|
def test_timeout_no_receivers(self):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.get(timeout=1.0))
|
|
|
|
lambda: select.get(timeout=1.0))
|
|
|
|
self.assertEquals(str(exc), self.klass.empty_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.empty_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_zero_timeout(self):
|
|
|
|
def test_zero_timeout(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
@ -422,15 +422,15 @@ class GetLatchTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
select = self.klass([latch])
|
|
|
|
select = self.klass([latch])
|
|
|
|
self.assertEquals(123, select.get())
|
|
|
|
self.assertEqual(123, select.get())
|
|
|
|
|
|
|
|
|
|
|
|
def test_nonempty_multiple_items_before_add(self):
|
|
|
|
def test_nonempty_multiple_items_before_add(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(234)
|
|
|
|
latch.put(234)
|
|
|
|
select = self.klass([latch], oneshot=False)
|
|
|
|
select = self.klass([latch], oneshot=False)
|
|
|
|
self.assertEquals(123, select.get())
|
|
|
|
self.assertEqual(123, select.get())
|
|
|
|
self.assertEquals(234, select.get())
|
|
|
|
self.assertEqual(234, select.get())
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
lambda: select.get(block=False))
|
|
|
|
lambda: select.get(block=False))
|
|
|
|
|
|
|
|
|
|
|
@ -438,13 +438,13 @@ class GetLatchTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
select = self.klass([latch])
|
|
|
|
select = self.klass([latch])
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
self.assertEquals(123, latch.get())
|
|
|
|
self.assertEqual(123, latch.get())
|
|
|
|
|
|
|
|
|
|
|
|
def test_drained_by_other_thread(self):
|
|
|
|
def test_drained_by_other_thread(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
select = self.klass([latch])
|
|
|
|
select = self.klass([latch])
|
|
|
|
self.assertEquals(123, latch.get())
|
|
|
|
self.assertEqual(123, latch.get())
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
self.assertRaises(mitogen.core.TimeoutError,
|
|
|
|
lambda: select.get(timeout=0.0))
|
|
|
|
lambda: select.get(timeout=0.0))
|
|
|
|
|
|
|
|
|
|
|
@ -456,20 +456,20 @@ class GetEventTest(testlib.RouterMixin, testlib.TestCase):
|
|
|
|
select = self.klass()
|
|
|
|
select = self.klass()
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
exc = self.assertRaises(mitogen.select.Error,
|
|
|
|
lambda: select.get())
|
|
|
|
lambda: select.get())
|
|
|
|
self.assertEquals(str(exc), self.klass.empty_msg)
|
|
|
|
self.assertEqual(str(exc), self.klass.empty_msg)
|
|
|
|
|
|
|
|
|
|
|
|
def test_latch(self):
|
|
|
|
def test_latch(self):
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch = mitogen.core.Latch()
|
|
|
|
latch.put(123)
|
|
|
|
latch.put(123)
|
|
|
|
select = self.klass([latch])
|
|
|
|
select = self.klass([latch])
|
|
|
|
event = select.get_event()
|
|
|
|
event = select.get_event()
|
|
|
|
self.assertEquals(latch, event.source)
|
|
|
|
self.assertEqual(latch, event.source)
|
|
|
|
self.assertEquals(123, event.data)
|
|
|
|
self.assertEqual(123, event.data)
|
|
|
|
|
|
|
|
|
|
|
|
def test_receiver(self):
|
|
|
|
def test_receiver(self):
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv = mitogen.core.Receiver(self.router)
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
recv._on_receive(mitogen.core.Message.pickled('123'))
|
|
|
|
select = self.klass([recv])
|
|
|
|
select = self.klass([recv])
|
|
|
|
event = select.get_event()
|
|
|
|
event = select.get_event()
|
|
|
|
self.assertEquals(recv, event.source)
|
|
|
|
self.assertEqual(recv, event.source)
|
|
|
|
self.assertEquals('123', event.data.unpickle())
|
|
|
|
self.assertEqual('123', event.data.unpickle())
|
|
|
|