class SwitchTestFlow(Flow):
start = flow.StartFunction(create_test_flow).Next(this.switch)
switch = flow.Switch().Case(this.task1,
cond=lambda process: True).Default(this.task2)
task1 = flow.Handler(handler).Next(this.end)
task2 = flow.Handler(handler).Next(this.end)
end = flow.End()
class TestFlow(Flow):
process_class = SavableProcess
start = flow.Start(CreateProcessView, fields=[]).Next(this.savable_task)
savable_task = flow.View(SavableProcessView) \
.Assign(username='******') \
.Next(this.if_task)
if_task = flow.If(1 == 1).Then(this.switch_task).Else(this.end)
switch_task = flow.Switch().Case(this.end, cond=(1 == 1))
end = flow.End()
def test_switch_activation_activate(self):
flow_task_mock = mock.Mock(spec=flow.Switch())
flow_task_mock._outgoing = mock.Mock(return_value=[])
type(flow_task_mock).branches = mock.PropertyMock(
return_value=[(mock.Mock(), lambda p: True)])
prev_activation_mock = mock.Mock(spec=activation.StartActivation())
act = gates.SwitchActivation.activate(flow_task_mock,
prev_activation_mock,
Token('start'))
act.task.save.assert_has_calls(())
class AllTaskFlow(Flow):
lock_impl = lock.cache_lock
start = flow.Start().Next(this.view)
view = flow.View(perform_task).Next(this.job)
job = celery.Job(tasks.dummy_job).Next(this.iff)
iff = flow.If(lambda act: True).OnTrue(this.switch).OnFalse(this.switch)
switch = flow.Switch().Default(this.split)
split = flow.Split().Always(this.join)
join = flow.Join().Next(this.first)
first = flow.First().Of(this.end)
end = flow.End()