def test_get_from_channel(self):
chan = Channel()
prom = chan.get()
chan.put(52)
chan.process()
self.assertTrue(prom.delivered)
self.assertEqual(prom.value, 52)
def test_channel_with_limit(self):
chan = Channel(limit=5)
put_proms = [chan.put(x) for x in range(5)]
get_proms = [chan.get() for _ in range(5)]
for prom in put_proms:
process_all()
self.assertTrue(prom.delivered)
self.assertTrue(prom.value)
for n, prom in enumerate(get_proms):
self.assertTrue(prom.delivered)
self.assertEqual(prom.value, n)
