2 from eventlet import event
3 from tests import LimitedTestCase
6 class TestEvent(LimitedTestCase):
7 def test_waiting_for_event(self):
13 eventlet.spawn_n(send_to_event)
14 self.assertEqual(evt.wait(), value)
16 def test_multiple_waiters(self):
17 self._test_multiple_waiters(False)
19 def test_multiple_waiters_with_exception(self):
20 self._test_multiple_waiters(True)
22 def _test_multiple_waiters(self, exception):
26 def wait_on_event(i_am_done):
35 for i in range(count):
36 waiters.append(event.Event())
37 eventlet.spawn_n(wait_on_event, waiters[-1])
38 eventlet.sleep() # allow spawns to start executing
44 self.assertEqual(len(results), count)
49 # calling reset before send should throw
50 self.assertRaises(AssertionError, evt.reset)
56 eventlet.spawn_n(send_to_event)
57 self.assertEqual(evt.wait(), value)
59 # now try it again, and we should get the same exact value,
60 # and we shouldn't be allowed to resend without resetting
61 value2 = 'second stuff'
62 self.assertRaises(AssertionError, evt.send, value2)
63 self.assertEqual(evt.wait(), value)
65 # reset and everything should be happy
70 eventlet.spawn_n(send_to_event2)
71 self.assertEqual(evt.wait(), value2)
73 def test_double_exception(self):
75 # send an exception through the event
76 evt.send(exc=RuntimeError('from test_double_exception'))
77 self.assertRaises(RuntimeError, evt.wait)
79 # shouldn't see the RuntimeError again
80 eventlet.Timeout(0.001)
81 self.assertRaises(eventlet.Timeout, evt.wait)