cde0dee14e
Previously to run a flow client code had to put together the flow,
an engine, logbook, flowdetail, and storage backend. This commit
adds two helper functions, run() and load(), so that simplest usecase
now looks like
taskflow.engines.run(flow)
Client code may also provide configuration for storage and engine if
needed, but if not needed it just works with defaults.
Engines are loaded via stevedore, as drivers in 'taskflow.engines'
backend. Now three entry points are defined in that namespace:
- 'default', for SingleThreadedActionEngine, used by default;
- 'serial', as another synonym for SingleThreadedActionEngine;
- 'parallel', for MultiThreadedActionEngine.
Closes-bug: #1224726
Change-Id: I7f4cb5c8ff7f5f12831ddd0952c202d2fd8cd6ef
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright (C) 2012-2013 Yahoo! Inc. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import taskflow.engines
|
|
|
|
from taskflow.patterns import linear_flow
|
|
from taskflow import task as base
|
|
from taskflow import test
|
|
|
|
|
|
def add(a, b):
|
|
return a + b
|
|
|
|
|
|
class BunchOfFunctions(object):
|
|
|
|
def __init__(self, values):
|
|
self.values = values
|
|
|
|
def run_one(self, *args, **kwargs):
|
|
self.values.append('one')
|
|
|
|
def revert_one(self, *args, **kwargs):
|
|
self.values.append('revert one')
|
|
|
|
def run_fail(self, *args, **kwargs):
|
|
self.values.append('fail')
|
|
raise RuntimeError('Woot!')
|
|
|
|
|
|
class FunctorTaskTest(test.TestCase):
|
|
|
|
def test_simple(self):
|
|
task = base.FunctorTask(add)
|
|
self.assertEquals(task.name, __name__ + '.add')
|
|
|
|
def test_other_name(self):
|
|
task = base.FunctorTask(add, name='my task')
|
|
self.assertEquals(task.name, 'my task')
|
|
|
|
def test_it_runs(self):
|
|
values = []
|
|
bof = BunchOfFunctions(values)
|
|
t = base.FunctorTask
|
|
|
|
flow = linear_flow.Flow('test')
|
|
flow.add(
|
|
t(bof.run_one, revert=bof.revert_one),
|
|
t(bof.run_fail)
|
|
)
|
|
with self.assertRaisesRegexp(RuntimeError, '^Woot'):
|
|
taskflow.engines.run(flow)
|
|
self.assertEquals(values, ['one', 'fail', 'revert one'])
|