def test_err_callback_and_failure_raised(self):
class MyException(Exception):
pass
class MyRepackagedException(Exception):
pass
def raise_something_else(ex):
raise MyRepackagedException(ex.args[0] + ' repackaged')
base_f = ThenableFuture()
new_f = base_f.then(None, raise_something_else)
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception())
with self.assertRaises(MyRepackagedException) as catcher:
new_f.result()
self.assert_(catcher.exception.args[0] == 'sad repackaged')
def test_err_callback_convert_to_success(self):
class MyException(Exception):
pass
class NotMatched(Exception):
pass
def on_failure(ex):
if isinstance(ex, MyException):
return ex.args[0] + ' repackaged'
else:
return NotMatched('?')
base_f = ThenableFuture()
new_f = base_f.catch(on_failure)
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(not new_f.exception())
self.assert_(new_f.result() == 'sad repackaged')
def test_err_callback_and_failure_repackage(self):
class MyException(Exception):
pass
class MyRepackagedException(Exception):
pass
class NotMatched(Exception):
pass
def on_failure(ex):
if isinstance(ex, MyException):
return MyRepackagedException(ex.args[0] + ' repackaged')
else:
return NotMatched('?')
base_f = ThenableFuture()
new_f = base_f.then(None, on_failure)
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception())
with self.assertRaises(MyRepackagedException) as catcher:
new_f.result()
self.assert_(catcher.exception.args[0] == 'sad repackaged')
def test_err_callback_convert_to_success(self):
class MyException(Exception):
pass
class NotMatched(Exception):
pass
def on_failure(ex):
if isinstance(ex, MyException):
return ex.args[0] + ' repackaged'
else:
return NotMatched('?')
base_f = ThenableFuture()
new_f = base_f.catch(on_failure)
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(not new_f.exception())
self.assert_(new_f.result() == 'sad repackaged')
def test_err_callback_and_failure_raised(self):
class MyException(Exception):
pass
class MyRepackagedException(Exception):
pass
def raise_something_else(ex):
raise MyRepackagedException(ex.args[0] + ' repackaged')
base_f = ThenableFuture()
new_f = base_f.then(None, raise_something_else)
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception())
with self.assertRaises(MyRepackagedException) as catcher:
new_f.result()
self.assert_(catcher.exception.args[0] == 'sad repackaged')
def test_err_callback_and_failure_repackage(self):
class MyException(Exception):
pass
class MyRepackagedException(Exception):
pass
class NotMatched(Exception):
pass
def on_failure(ex):
if isinstance(ex, MyException):
return MyRepackagedException(ex.args[0] + ' repackaged')
else:
return NotMatched('?')
base_f = ThenableFuture()
new_f = base_f.then(None, on_failure)
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception())
with self.assertRaises(MyRepackagedException) as catcher:
new_f.result()
self.assert_(catcher.exception.args[0] == 'sad repackaged')
def test_err_catch_ignore(self):
base_f = ThenableFuture()
new_f = base_f.catch()
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(Exception('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception() is None)
self.assert_(new_f.result() is None)
def test_err_catch_ignore(self):
base_f = ThenableFuture()
new_f = base_f.catch()
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(Exception('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception() is None)
self.assert_(new_f.result() is None)
def test_no_callbacks_and_failure(self):
class MyException(Exception):
pass
base_f = ThenableFuture()
new_f = base_f.then()
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception())
with self.assertRaises(MyException) as catcher:
new_f.result()
self.assert_(catcher.exception.args[0] == 'sad')
def test_no_callbacks_and_failure(self):
class MyException(Exception):
pass
base_f = ThenableFuture()
new_f = base_f.then()
self.assert_(base_f is not new_f)
self.assert_(not base_f.done())
self.assert_(not new_f.done())
base_f.set_exception(MyException('sad'))
self.assert_(base_f.done())
self.assert_(new_f.done())
self.assert_(new_f.exception())
with self.assertRaises(MyException) as catcher:
new_f.result()
self.assert_(catcher.exception.args[0] == 'sad')
