def get_steps_done_activity(self):
return task.ActivityTask(
activity.Activity(GetStepsDoneTask,
**self._get_step_activity_params()),
self.get_step_bucket(),
self.get_step_path_prefix(),
)
def run(self, num, force_steps=[]):
self.add_forced_steps(force_steps, "workflow_init")
taskf = self.submit(
Step('my_step',
task.ActivityTask(MyTask, num),
force_steps_if_executed=['my_step_2']))
futures.wait(taskf)
def run(self, num, force_steps=None, skip_steps=None):
self.add_forced_steps(force_steps or [], "workflow_init")
self.add_skipped_steps(skip_steps or [], "workflow_init")
taskf = self.submit(
Step(
"my_step",
task.ActivityTask(MyTask, num),
force_steps_if_executed=["my_step_2"],
))
futures.wait(taskf)
def test_task_applies_class_correctly():
assert task.ActivityTask(Double, 4).execute() == 8
def test_task_applies_function_correctly():
assert task.ActivityTask(double, 2).execute() == 4
def test_context_is_empty_for_non_swf_tasks():
assert task.ActivityTask(Double, 3).context is None