def test_duplicate(self):
futures = [Future()] * 5
result = []
def waiter():
result.append(gruvi.wait(futures))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
self.assertEqual(result, [])
futures[0].set_result(None)
gruvi.sleep(0)
self.assertEqual(result, [(futures, [])])
def test_duplicate(self):
futures = [Future()] * 5
result = []
def waiter():
result.append(gruvi.wait(futures))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
self.assertEqual(result, [])
futures[0].set_result(None)
gruvi.sleep(0)
self.assertEqual(result, [(futures, [])])
def test_partial(self):
# Ensure that the "count" argument of wait() can be used to wait for a
# subset of the futures to complete.
futures = [Future() for i in range(10)]
result = []
def waiter():
result.append(gruvi.wait(futures, count=5))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
for i in range(4):
futures[i].set_result(None)
self.assertTrue(fib.alive)
self.assertEqual(result, [])
futures[4].set_result(None)
gruvi.sleep(0)
self.assertFalse(fib.alive)
self.assertEqual(result, [(futures[:5], futures[5:])])
def test_basic(self):
# Simple wait() test waiting for two futures.
futures = [Future() for i in range(2)]
result = []
def waiter():
result.append(gruvi.wait(futures))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
self.assertTrue(fib.alive)
self.assertEqual(result, [])
futures[0].set_result(None)
gruvi.sleep(0)
self.assertTrue(fib.alive)
futures[1].set_result(None)
gruvi.sleep(0)
self.assertFalse(fib.alive)
self.assertEqual(result, [(futures, [])])
def test_partial(self):
# Ensure that the "count" argument of wait() can be used to wait for a
# subset of the futures to complete.
futures = [Future() for i in range(10)]
result = []
def waiter():
result.append(gruvi.wait(futures, count=5))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
for i in range(4):
futures[i].set_result(None)
self.assertTrue(fib.alive)
self.assertEqual(result, [])
futures[4].set_result(None)
gruvi.sleep(0)
self.assertFalse(fib.alive)
self.assertEqual(result, [(futures[:5], futures[5:])])
def test_basic(self):
# Simple wait() test waiting for two futures.
futures = [Future() for i in range(2)]
result = []
def waiter():
result.append(gruvi.wait(futures))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
self.assertTrue(fib.alive)
self.assertEqual(result, [])
futures[0].set_result(None)
gruvi.sleep(0)
self.assertTrue(fib.alive)
futures[1].set_result(None)
gruvi.sleep(0)
self.assertFalse(fib.alive)
self.assertEqual(result, [(futures, [])])
def test_object_types(self):
def sleeper():
gruvi.sleep(0.1)
objects = []
for i in range(2):
fib = Fiber(sleeper)
fib.start()
objects.append(fib)
for i in range(2):
objects.append(Event())
for i in range(2):
objects.append(Future())
for i in range(2):
proc = Process()
proc.spawn('true')
objects.append(proc)
result = []
def waiter():
result.append(gruvi.wait(objects))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
self.assertTrue(fib.alive)
objects[2].set()
objects[3].set()
objects[4].set_result(True)
objects[5].set_exception(RuntimeError)
fib.join()
self.assertFalse(fib.alive)
self.assertEqual(set(result[0][0]), set(objects))
self.assertFalse(objects[0].alive)
self.assertFalse(objects[1].alive)
self.assertEqual(objects[6].returncode, 0)
self.assertEqual(objects[7].returncode, 0)
for obj in objects:
self.assertIsNone(obj._callbacks)
objects[6].close()
objects[7].close()
def test_timeout_fibers(self):
futures = [Future() for i in range(10)]
result = []
def waiter():
result.append(gruvi.wait(futures, timeout=0.01))
fibers = [Fiber(waiter) for i in range(10)]
for fib in fibers:
fib.start()
for fib in fibers:
fib.join()
self.assertEqual(len(result), 10)
for res in result:
self.assertEqual(res, ([], futures))
for fut in futures:
self.assertIsNone(fut._callbacks)
def test_timeout_fibers(self):
futures = [Future() for i in range(10)]
exceptions = []
def waiter():
try:
list(gruvi.as_completed(futures, timeout=0.01))
except Timeout as e:
exceptions.append(e)
fibers = [Fiber(waiter) for i in range(10)]
for fib in fibers:
fib.start()
for fib in fibers:
fib.join()
self.assertEqual(len(exceptions), 10)
for exc in exceptions:
self.assertIsInstance(exc, Timeout)
for fut in futures:
self.assertIsNone(fut._callbacks)
def test_object_types(self):
def sleeper():
gruvi.sleep(0.1)
objects = []
for i in range(2):
fib = Fiber(sleeper)
fib.start()
objects.append(fib)
for i in range(2):
objects.append(Event())
for i in range(2):
objects.append(Future())
for i in range(2):
proc = Process()
proc.spawn('true')
objects.append(proc)
result = []
def waiter():
result.append(gruvi.wait(objects))
fib = Fiber(waiter)
fib.start()
gruvi.sleep(0)
self.assertTrue(fib.alive)
objects[2].set()
objects[3].set()
objects[4].set_result(True)
objects[5].set_exception(RuntimeError)
fib.join()
self.assertFalse(fib.alive)
self.assertEqual(set(result[0][0]), set(objects))
self.assertFalse(objects[0].alive)
self.assertFalse(objects[1].alive)
self.assertEqual(objects[6].returncode, 0)
self.assertEqual(objects[7].returncode, 0)
for obj in objects:
self.assertFalse(obj._callbacks)
objects[6].close()
objects[7].close()
def test_partial_fibers(self):
futures = [Future() for i in range(100)]
result = []
def waiter():
for fut in gruvi.as_completed(futures, count=2):
result.append(fut)
fibers = [Fiber(waiter) for i in range(100)]
for fib in fibers:
fib.start()
gruvi.sleep(0)
futures[0].set_result(None)
for fib in fibers:
self.assertTrue(fib.alive)
futures[len(futures) // 2].set_result(None)
gruvi.sleep(0)
for fib in fibers:
self.assertFalse(fib.alive)
self.assertEqual(len(result), len(fibers) * 2)
for fut in futures:
self.assertIsNone(fut._callbacks)
def test_partial_fibers(self):
# Have a lot of fibers wait on a small subet of a lot of futures.
futures = [Future() for i in range(100)]
def waiter():
gruvi.wait(futures, count=2)
fibers = [Fiber(waiter) for i in range(100)]
for fib in fibers:
fib.start()
gruvi.sleep(0)
futures[0].set_result(None)
gruvi.sleep(0)
for fib in fibers:
self.assertTrue(fib.alive)
futures[len(futures) // 2].set_result(None)
gruvi.sleep(0)
for fib in fibers:
self.assertFalse(fib.alive)
# Test cleanup
for fut in futures:
self.assertIsNone(fut._callbacks)