def test_both(self):
marker = object()
second_marker = object()
result_marker = object()
def _mapper(_):
return marker
def _exception_mapper(_):
return second_marker
first_future = Future()
folded_future = future_fold(first_future,
result_mapper=_mapper,
exception_mapper=_exception_mapper)
folded_future_probe = FutureProbe(folded_future)
second_future = Future()
second_folded_future = future_fold(second_future,
result_mapper=_mapper,
exception_mapper=_exception_mapper)
second_folded_future_probe = FutureProbe(second_folded_future,
stop_cb=self.stop)
first_future.set_result(result_marker)
second_future.set_exception(MyException())
self.wait()
folded_future_probe.assert_single_result_call(self, marker)
second_folded_future_probe.assert_single_result_call(
self, second_marker)
def test_value_to_exception(self):
result_marker = object()
future = Future()
future_probe = FutureProbe(future)
def _mapper(result):
raise MyException(result_was=result)
res_future = future_fold(future, result_mapper=_mapper)
res_future_probe = FutureProbe(res_future, stop_cb=self.stop)
future.set_result(result_marker)
self.wait()
future_probe.assert_single_result_call(self, result_marker)
res_future_probe.assert_single_exception_call(self, MyException,
result_marker)
def test_exception_to_exception(self):
future = Future()
future_probe = FutureProbe(future)
def _exception_mapper(exception):
if isinstance(exception, MyException):
raise MyOtherException()
else:
return None
res_future = future_fold(future, exception_mapper=_exception_mapper)
res_future_probe = FutureProbe(res_future, stop_cb=self.stop)
future.set_exception(MyException())
self.wait()
future_probe.assert_single_exception_call(self, MyException)
res_future_probe.assert_single_exception_call(self, MyOtherException)
def test_value_to_value(self):
marker = object()
result_marker = object()
future = Future()
future_probe = FutureProbe(future)
def _mapper(result):
return marker, result
res_future = future_fold(future, result_mapper=_mapper)
check_res_future = FutureProbe(res_future, stop_cb=self.stop)
future.set_result(result_marker)
self.wait()
future_probe.assert_single_result_call(self, result_marker)
check_res_future.assert_single_result_call(self,
(marker, result_marker))
def test_exception_to_value(self):
marker = object()
future = Future()
future_probe = FutureProbe(future)
def _exception_mapper(exception):
# We need to check exception type, but here we can't raise AssertionException.
# So it returns None for failing in assertions bellow.
if isinstance(exception, MyException):
return marker
else:
return None
res_future = future_fold(future, exception_mapper=_exception_mapper)
res_future_probe = FutureProbe(res_future, stop_cb=self.stop)
future.set_exception(MyException())
self.wait()
future_probe.assert_single_exception_call(self, MyException)
res_future_probe.assert_single_result_call(self, marker)