mirror of
https://github.com/sartography/spiff-arena.git
synced 2025-01-18 13:31:08 +00:00
0892db6fa7
git-subtree-dir: SpiffWorkflow git-subtree-split: 63db3e45947ec66b8d0efc2c74064004f8ff482c
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
import os
|
|
import sys
|
|
import unittest
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..'))
|
|
|
|
from tests.SpiffWorkflow.util import run_workflow
|
|
from .TaskSpecTest import TaskSpecTest
|
|
from SpiffWorkflow.task import TaskState
|
|
from SpiffWorkflow.specs import Execute
|
|
|
|
|
|
class ExecuteTest(TaskSpecTest):
|
|
CORRELATE = Execute
|
|
|
|
def create_instance(self):
|
|
if 'testtask' in self.wf_spec.task_specs:
|
|
del self.wf_spec.task_specs['testtask']
|
|
return Execute(self.wf_spec,
|
|
'testtask',
|
|
description='foo',
|
|
args=self.cmd_args)
|
|
|
|
def setUp(self):
|
|
self.cmd_args = ["python", "ExecuteProcessMock.py"]
|
|
TaskSpecTest.setUp(self)
|
|
|
|
def testConstructor(self):
|
|
TaskSpecTest.testConstructor(self)
|
|
self.assertEqual(self.spec.args, self.cmd_args)
|
|
|
|
def testPattern(self):
|
|
"""
|
|
Tests that we can create a task that executes a shell command
|
|
and that the workflow can be called to complete such tasks.
|
|
"""
|
|
self.wf_spec.start.connect(self.spec)
|
|
expected = 'Start\n testtask\n'
|
|
workflow = run_workflow(self, self.wf_spec, expected, '')
|
|
task = workflow.get_tasks_from_spec_name('testtask')[0]
|
|
self.assertEqual(task.state_history, [TaskState.FUTURE,
|
|
TaskState.WAITING,
|
|
TaskState.READY,
|
|
TaskState.COMPLETED])
|
|
self.assertIn(b'127.0.0.1', task.results[0])
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromTestCase(ExecuteTest)
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|