def test_async_shortcuts(self):
"""
Test the 'all_completed'/'as_completed' shortcut methods
"""
pool = Pool(TEST_SECRET_KEY, TEST_ACCESS_KEY)
mock = flexmock(AsyncResponse)
mock.should_receive('as_completed').once()
pool.as_completed([])
mock.should_receive('all_completed').once()
pool.all_completed([])
def test_async_shortcuts(self):
"""
Test the 'all_completed'/'as_completed' shortcut methods
"""
pool = Pool(TEST_SECRET_KEY, TEST_ACCESS_KEY)
mock = flexmock(AsyncResponse)
mock.should_receive('as_completed').once()
pool.as_completed([])
mock.should_receive('all_completed').once()
pool.all_completed([])
def test_all_completed(self):
"""
Test the all completed
"""
# Create mock futures
futures = [Future(), Future(), Future()]
# Create a default pool
pool = Pool(TEST_ACCESS_KEY, TEST_SECRET_KEY)
# Resolve futures with a simple object
for i in futures:
i.set_result(DUMMY_OBJECT)
# Make sure all the results are dummy objects
for i in pool.all_completed(futures):
self.assertEquals(i, DUMMY_OBJECT)
def test_all_completed(self):
"""
Test the all completed
"""
# Create mock futures
futures = [Future(), Future(), Future()]
# Create a default pool
pool = Pool(TEST_ACCESS_KEY, TEST_SECRET_KEY)
# Resolve futures with a simple object
for i in futures:
i.set_result(DUMMY_OBJECT)
# Make sure all the results are dummy objects
for i in pool.all_completed(futures):
self.assertEquals(i, DUMMY_OBJECT)