cd04997bce
Add testcase that executes all examples from taskflow/examples directory. For examples with somewhat deterministic output the output is also checked. This commit also fixes couple of examples broken by recent decorators refactoring. Change-Id: I3e65eabde43e1dd26b442d8f7dd3f062bf80703d
52 lines
1.0 KiB
Python
52 lines
1.0 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
logging.basicConfig(level=logging.ERROR)
|
|
|
|
my_dir_path = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.insert(0, os.path.join(os.path.join(my_dir_path, os.pardir),
|
|
os.pardir))
|
|
|
|
from taskflow import decorators
|
|
from taskflow.patterns import linear_flow as lf
|
|
|
|
|
|
def undo_call(context, result, cause):
|
|
print("Calling %s and apologizing." % result)
|
|
|
|
|
|
@decorators.task(revert_with=undo_call)
|
|
def call_jim(context):
|
|
print("Calling jim.")
|
|
print("Context = %s" % (context))
|
|
return context['jim_number']
|
|
|
|
|
|
@decorators.task(revert_with=undo_call)
|
|
def call_joe(context):
|
|
print("Calling joe.")
|
|
print("Context = %s" % (context))
|
|
return context['joe_number']
|
|
|
|
|
|
@decorators.task
|
|
def call_suzzie(context):
|
|
raise IOError("Suzzie not home right now.")
|
|
|
|
|
|
flow = lf.Flow("call-them")
|
|
flow.add(call_jim)
|
|
flow.add(call_joe)
|
|
flow.add(call_suzzie)
|
|
|
|
context = {
|
|
"joe_number": 444,
|
|
"jim_number": 555,
|
|
}
|
|
|
|
try:
|
|
flow.run(context)
|
|
except Exception as e:
|
|
print "Flow failed: %r" % e
|