72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
import json
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
dirname = os.path.dirname(__file__)
|
|
sys.path.insert(0, os.path.join(dirname, '..', '..', '..'))
|
|
|
|
from SpiffWorkflow.bpmn.PythonScriptEngine import PythonScriptEngine
|
|
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
|
|
from .BaseTestCase import BaseTestCase
|
|
|
|
class ServiceTaskDelegate:
|
|
@staticmethod
|
|
def call_connector(name, params, task_data):
|
|
assertEqual(name, 'bamboohr/GetPayRate')
|
|
assertEqual(len(params), 3)
|
|
assertEqual(params['api_key']['value'], 'secret:BAMBOOHR_API_KEY')
|
|
assertEqual(params['employee_id']['value'], '109')
|
|
assertEqual(params['subdomain']['value'], 'statusdemo')
|
|
|
|
sample_response = {
|
|
"amount": "65000.00",
|
|
"currency": "USD",
|
|
"id": "4",
|
|
"payRate": "65000.00 USD",
|
|
}
|
|
|
|
return json.dumps(sample_response)
|
|
|
|
class ExampleCustomScriptEngine(PythonScriptEngine):
|
|
def call_service(self, operation_name, operation_params, task_data):
|
|
return ServiceTaskDelegate.call_connector(operation_name, operation_params,
|
|
task_data)
|
|
|
|
class ServiceTaskVariableTest(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
global assertEqual
|
|
assertEqual = self.assertEqual
|
|
|
|
spec, subprocesses = self.load_workflow_spec('service_task_variable.bpmn',
|
|
'Process_bd2e724555')
|
|
self.script_engine = ExampleCustomScriptEngine()
|
|
self.workflow = BpmnWorkflow(spec, subprocesses, script_engine=self.script_engine)
|
|
|
|
def testRunThroughHappy(self):
|
|
self.workflow.do_engine_steps()
|
|
self._assert_service_task()
|
|
|
|
def testRunThroughSaveRestore(self):
|
|
self.save_restore()
|
|
# Engine isn't preserved through save/restore, so we have to reset it.
|
|
self.workflow.script_engine = self.script_engine
|
|
self.workflow.do_engine_steps()
|
|
self.save_restore()
|
|
self._assert_service_task()
|
|
|
|
def _assert_service_task(self):
|
|
result = self.workflow.data['spiff__Activity_0xhr131_result']
|
|
self.assertEqual(len(result), 4)
|
|
self.assertEqual(result['amount'], '65000.00')
|
|
self.assertEqual(result['currency'], 'USD')
|
|
self.assertEqual(result['id'], '4')
|
|
self.assertEqual(result['payRate'], '65000.00 USD')
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromTestCase(ServiceTaskVariableTest)
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|