diff options
| author | Ivan A. Melnikov <imelnikov@griddynamics.com> | 2014-01-23 16:08:55 +0400 |
|---|---|---|
| committer | Ivan A. Melnikov <imelnikov@griddynamics.com> | 2014-01-30 17:08:27 +0400 |
| commit | 39da9727670b0baa06982fe3ea977545ffa034b4 (patch) | |
| tree | 387cad58c1dd9f40113d86aa211bf428ddcede8d /taskflow/tests/utils.py | |
| parent | a8f341a6c881acabb0fdb93e7f47cdb71d5fbd25 (diff) | |
| download | taskflow-39da9727670b0baa06982fe3ea977545ffa034b4.tar.gz | |
Tests: don't pass 'values' to task constructor
This might make it easier for worker-based engine to run it.
Change-Id: I94d7b8e0a447708e448c8c5d3dd7232f67c7adda
Diffstat (limited to 'taskflow/tests/utils.py')
| -rw-r--r-- | taskflow/tests/utils.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/taskflow/tests/utils.py b/taskflow/tests/utils.py index 43daf39..a62beef 100644 --- a/taskflow/tests/utils.py +++ b/taskflow/tests/utils.py @@ -78,13 +78,10 @@ class ProvidesRequiresTask(task.Task): class SaveOrderTask(task.Task): - def __init__(self, values=None, name=None, sleep=None, + def __init__(self, name=None, sleep=None, *args, **kwargs): super(SaveOrderTask, self).__init__(name=name, *args, **kwargs) - if values is None: - self.values = [] - else: - self.values = values + self.values = EngineTestBase.values self._sleep = sleep def execute(self, **kwargs): @@ -105,7 +102,6 @@ class SaveOrderTask(task.Task): class FailingTask(SaveOrderTask): - def execute(self, **kwargs): self.update_progress(0) if self._sleep: @@ -212,16 +208,19 @@ class NeverRunningTask(task.Task): class EngineTestBase(object): + values = None + def setUp(self): super(EngineTestBase, self).setUp() - self.values = [] + EngineTestBase.values = [] self.backend = impl_memory.MemoryBackend(conf={}) def tearDown(self): - super(EngineTestBase, self).tearDown() + EngineTestBase.values = None with contextlib.closing(self.backend) as be: with contextlib.closing(be.get_connection()) as conn: conn.clear_all() + super(EngineTestBase, self).tearDown() def _make_engine(self, flow, flow_detail=None): raise NotImplementedError() |
