def test_simple_run(self):
r = Runnable()
# async run with valid state
f = r.run(State())
self.assertIsInstance(f, Future)
self.assertNotIsInstance(f, Present)
self.assertRaises(NotImplementedError, f.result)
# sync run with valid state
f = r.run(State(), defer=False)
self.assertIsInstance(f, Present)
self.assertRaises(NotImplementedError, f.result)
# run with error state, check exc is propagated (default)
f = r.run(Present(exception=ZeroDivisionError()))
self.assertRaises(ZeroDivisionError, f.result)
class MyRunnable(Runnable):
def init(self, state):
self.first = state.problem
def next(self, state):
return state.updated(problem=state.problem + 1)
r = MyRunnable()
s1 = State(problem=1)
s2 = r.run(s1).result()
self.assertEqual(r.first, s1.problem)
self.assertEqual(s2.problem, s1.problem + 1)
def test_error_prop(self):
class MyRunnable(Runnable):
def next(self, state):
return state
def error(self, exc):
return State(error=True)
r = MyRunnable()
s1 = Present(exception=KeyError())
s2 = r.run(s1).result()
self.assertEqual(s2.error, True)
def test_error_prop(self):
class ErrorSilencer(Runnable):
def next(self, state):
return state
def error(self, exc):
return State(error=True)
class Identity(Runnable):
def next(self, state):
return state
branch = ErrorSilencer() | Identity()
s1 = Present(exception=KeyError())
s2 = branch.run(s1).result()
self.assertEqual(s2.error, True)
def test_error_and_init(self):
runnable = Lambda(
next=lambda self, state: state.updated(c=state.a * state.b),
error=lambda self, exc: State(error=exc),
init=lambda self, state: setattr(self, 'first', state.c))
# test init
state = State(a=2, b=3, c=0)
result = runnable.run(state).result()
self.assertEqual(runnable.first, 0)
self.assertEqual(result.c, state.a * state.b)
# test error prop
exc = ZeroDivisionError()
result = runnable.run(Present(exception=exc)).result()
self.assertEqual(result.error, exc)
def error(self, exc):
"""Pass on the exception from input to the error handler of the first
runnable in branch.
"""
return self.next(Present(exception=exc))
def test_exc(self):
for exc in ValueError, KeyError, ZeroDivisionError:
f = Present(exception=exc())
self.assertIsInstance(f, Future)
self.assertTrue(f.done())
self.assertRaises(exc, f.result)
def test_res(self):
for val in 1, 'x', True, False, State(problem=1), lambda: None:
f = Present(result=val)
self.assertIsInstance(f, Future)
self.assertTrue(f.done())
self.assertEqual(f.result(), val)