def test_keyboard_interrupt_on_result_does_not_block(self):
# This should raise a KeyboardInterrupt when result is called on it.
self.coordinator = TransferCoordinatorWithInterrupt()
self.future = self._get_transfer_future()
# result() should not block and immediately raise the keyboard
# interrupt exception.
with self.assertRaises(KeyboardInterrupt):
self.future.result()
def test_keyboard_interrupt_on_result_does_not_block(self):
# This should raise a KeyboardInterrupt when result is called on it.
self.coordinator = TransferCoordinatorWithInterrupt()
self.future = self._get_transfer_future()
# result() should not block and immediately raise the keyboard
# interrupt exception.
with self.assertRaises(KeyboardInterrupt):
self.future.result()
class TestTransferFuture(unittest.TestCase):
def setUp(self):
self.meta = TransferMeta()
self.coordinator = TransferCoordinator()
self.future = self._get_transfer_future()
def _get_transfer_future(self, **kwargs):
components = {
'meta': self.meta,
'coordinator': self.coordinator,
}
for component_name, component in kwargs.items():
components[component_name] = component
return TransferFuture(**components)
def test_meta(self):
self.assertIs(self.future.meta, self.meta)
def test_done(self):
self.assertFalse(self.future.done())
self.coordinator.set_result(None)
self.assertTrue(self.future.done())
def test_result(self):
result = 'foo'
self.coordinator.set_result(result)
self.coordinator.announce_done()
self.assertEqual(self.future.result(), result)
def test_keyboard_interrupt_on_result_does_not_block(self):
# This should raise a KeyboardInterrupt when result is called on it.
self.coordinator = TransferCoordinatorWithInterrupt()
self.future = self._get_transfer_future()
# result() should not block and immediately raise the keyboard
# interrupt exception.
with self.assertRaises(KeyboardInterrupt):
self.future.result()
def test_cancel(self):
self.future.cancel()
self.assertTrue(self.future.done())
self.assertEqual(self.coordinator.status, 'cancelled')
def setUp(self):
self.meta = TransferMeta()
self.coordinator = TransferCoordinator()
self.future = self._get_transfer_future()
def setUp(self):
self.meta = TransferMeta()
self.coordinator = TransferCoordinator()
self.future = self._get_transfer_future()
class TestTransferFuture(unittest.TestCase):
def setUp(self):
self.meta = TransferMeta()
self.coordinator = TransferCoordinator()
self.future = self._get_transfer_future()
def _get_transfer_future(self, **kwargs):
components = {
'meta': self.meta,
'coordinator': self.coordinator,
}
for component_name, component in kwargs.items():
components[component_name] = component
return TransferFuture(**components)
def test_meta(self):
self.assertIs(self.future.meta, self.meta)
def test_done(self):
self.assertFalse(self.future.done())
self.coordinator.set_result(None)
self.assertTrue(self.future.done())
def test_result(self):
result = 'foo'
self.coordinator.set_result(result)
self.coordinator.announce_done()
self.assertEqual(self.future.result(), result)
def test_keyboard_interrupt_on_result_does_not_block(self):
# This should raise a KeyboardInterrupt when result is called on it.
self.coordinator = TransferCoordinatorWithInterrupt()
self.future = self._get_transfer_future()
# result() should not block and immediately raise the keyboard
# interrupt exception.
with self.assertRaises(KeyboardInterrupt):
self.future.result()
def test_cancel(self):
self.future.cancel()
self.assertTrue(self.future.done())
self.assertEqual(self.coordinator.status, 'cancelled')
def test_set_exception(self):
# Set the result such that there is no exception
self.coordinator.set_result('result')
self.coordinator.announce_done()
self.assertEqual(self.future.result(), 'result')
self.future.set_exception(ValueError())
with self.assertRaises(ValueError):
self.future.result()
def test_set_exception_only_after_done(self):
with self.assertRaises(TransferNotDoneError):
self.future.set_exception(ValueError())
self.coordinator.set_result('result')
self.coordinator.announce_done()
self.future.set_exception(ValueError())
with self.assertRaises(ValueError):
self.future.result()
def test_wait_can_be_interrupted(self):
inject_interrupt_coordinator = TransferCoordinatorWithInterrupt()
self.coordinator_controller.add_transfer_coordinator(
inject_interrupt_coordinator)
with self.assertRaises(KeyboardInterrupt):
self.coordinator_controller.wait()
