def test_accumulated_risk_with_error(self): workflow = Workflow(name='workflow') workflow._execution_uid = 'some_uid' step1 = Step(name="step_one", app='HelloWorld', action='Buggy', risk=1) step2 = Step(name="step_two", app='HelloWorld', action='Buggy', risk=2) step3 = Step(name="step_three", app='HelloWorld', action='Buggy', risk=3.5) workflow.steps = { 'step_one': step1, 'step_two': step2, 'step_three': step3 } workflow._total_risk = 6.5 instance = AppInstance.create(app_name='HelloWorld', device_name='test_device_name') workflow._Workflow__execute_step(workflow.steps["step_one"], instance) self.assertAlmostEqual(workflow.accumulated_risk, 1.0 / 6.5) workflow._Workflow__execute_step(workflow.steps["step_two"], instance) self.assertAlmostEqual(workflow.accumulated_risk, (1.0 / 6.5) + (2.0 / 6.5)) workflow._Workflow__execute_step(workflow.steps["step_three"], instance) self.assertAlmostEqual(workflow.accumulated_risk, 1.0)
def test_execute_multiple_triggers(self): triggers = [ Condition('HelloWorld', action_name='regMatch', arguments=[Argument('regex', value='aaa')]) ] action = Action(app_name='HelloWorld', action_name='helloWorld', triggers=triggers) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') action.send_data_to_trigger({"data_in": {"data": 'a'}}) trigger_taken = {'triggered': 0} trigger_not_taken = {'triggered': 0} def callback_is_sent(sender, **kwargs): if kwargs['event'] == WalkoffEvent.TriggerActionTaken: trigger_taken['triggered'] += 1 elif kwargs['event'] == WalkoffEvent.TriggerActionNotTaken: action.send_data_to_trigger({"data_in": {"data": 'aaa'}}) trigger_not_taken['triggered'] += 1 WalkoffEvent.CommonWorkflowSignal.connect(callback_is_sent) action.execute(instance.instance, {}) self.assertEqual(trigger_taken['triggered'], 1) self.assertEqual(trigger_not_taken['triggered'], 1)
def __setup_app_instance(self, instances, step): device_id = (step.app, step.device) if device_id not in instances: instances[device_id] = AppInstance.create(step.app, step.device) data_sent.send(self, callback_name="App Instance Created", object_type="Workflow") logger.debug('Created new app instance: App {0}, device {1}'.format(step.app, step.device)) return device_id
def test_execute_with_complex_inputs(self): step = Step(app='HelloWorld', action='Json Sample', inputs={ 'json_in': { 'a': '-5.6', 'b': { 'a': '4.3', 'b': 5.3 }, 'c': ['1', '2', '3'], 'd': [{ 'a': '', 'b': 3 }, { 'a': '', 'b': -1.5 }, { 'a': '', 'b': -0.5 }] } }) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = step.execute(instance.instance, {}) self.assertAlmostEqual(result.result, 11.0) self.assertEqual(result.status, 'Success') self.assertEqual(step._output, result)
def test_execute_generates_uid(self): step = Step(app='HelloWorld', action='helloWorld') original_execution_uid = step.get_execution_uid() instance = AppInstance.create(app_name='HelloWorld', device_name='device1') step.execute(instance.instance, {}) self.assertNotEqual(step.get_execution_uid(), original_execution_uid)
def test_execute_with_accumulator_missing_step_callbacks(self): step = Step(app='HelloWorld', action='Add Three', inputs={ 'num1': '@1', 'num2': '@step2', 'num3': '10.2' }) accumulator = {'1': '-5.6', 'missing': '4.3', '3': '45'} instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = {'started_triggered': False, 'result_triggered': False} @callbacks.data_sent.connect def callback_is_sent(sender, **kwargs): if isinstance(sender, Step): self.assertIs(sender, step) self.assertIn('callback_name', kwargs) self.assertIn(kwargs['callback_name'], ('Step Started', 'Step Input Invalid')) self.assertIn('object_type', kwargs) self.assertEqual(kwargs['object_type'], 'Step') if kwargs['callback_name'] == 'Step Started': result['started_triggered'] = True else: result['result_triggered'] = True with self.assertRaises(InvalidInput): step.execute(instance.instance, accumulator) self.assertTrue(result['started_triggered']) self.assertTrue(result['result_triggered'])
def test_execute_with_accumulator_missing_action_callbacks(self): action = Action(app_name='HelloWorld', action_name='Add Three', arguments=[ Argument('num1', reference='1'), Argument('num2', reference='action2'), Argument('num3', value='10.2') ]) accumulator = {'1': '-5.6', 'missing': '4.3', '3': '45'} instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = {'started_triggered': False, 'result_triggered': False} def callback_is_sent(sender, **kwargs): if isinstance(sender, Action): self.assertIn('event', kwargs) self.assertIn(kwargs['event'], (WalkoffEvent.ActionStarted, WalkoffEvent.ActionArgumentsInvalid)) if kwargs['event'] == WalkoffEvent.ActionStarted: result['started_triggered'] = True else: result['result_triggered'] = True WalkoffEvent.CommonWorkflowSignal.connect(callback_is_sent) action.execute(instance.instance, accumulator) self.assertTrue(result['started_triggered']) self.assertTrue(result['result_triggered'])
def test_execute_with_complex_args(self): action = Action(app_name='HelloWorld', action_name='Json Sample', arguments=[ Argument('json_in', value={ 'a': '-5.6', 'b': { 'a': '4.3', 'b': 5.3 }, 'c': ['1', '2', '3'], 'd': [{ 'a': '', 'b': 3 }, { 'a': '', 'b': -1.5 }, { 'a': '', 'b': -0.5 }] }) ]) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = action.execute(instance.instance, {}) self.assertAlmostEqual(result.result, 11.0) self.assertEqual(result.status, 'Success') self.assertEqual(action._output, result)
def test_execute_action_which_raises_exception(self): from tests.testapps.HelloWorld.exceptions import CustomException step = Step(app='HelloWorld', action='Buggy') instance = AppInstance.create(app_name='HelloWorld', device_name='device1') with self.assertRaises(CustomException): step.execute(instance.instance, {})
def test_execute_sends_callbacks(self): action = Action(app_name='HelloWorld', action_name='Add Three', arguments=[ Argument('num1', value='-5.6'), Argument('num2', value='4.3'), Argument('num3', value='10.2') ]) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = {'started_triggered': False, 'result_triggered': False} def callback_is_sent(sender, **kwargs): if isinstance(sender, Action): self.assertIn('event', kwargs) self.assertIn(kwargs['event'], (WalkoffEvent.ActionStarted, WalkoffEvent.ActionExecutionSuccess)) if kwargs['event'] == WalkoffEvent.ActionStarted: result['started_triggered'] = True else: self.assertIn('data', kwargs) data = kwargs['data'] self.assertEqual(data['status'], 'Success') self.assertAlmostEqual(data['result'], 8.9) result['result_triggered'] = True WalkoffEvent.CommonWorkflowSignal.connect(callback_is_sent) action.execute(instance.instance, {}) self.assertTrue(result['started_triggered']) self.assertTrue(result['result_triggered'])
def test_execute_generates_uid(self): action = Action(app_name='HelloWorld', action_name='helloWorld') original_execution_uid = action.get_execution_uid() instance = AppInstance.create(app_name='HelloWorld', device_name='device1') action.execute(instance.instance, {}) self.assertNotEqual(action.get_execution_uid(), original_execution_uid)
def test_execute_no_args(self): step = Step(app='HelloWorld', action='helloWorld') instance = AppInstance.create(app_name='HelloWorld', device_name='device1') self.assertEqual(step.execute(instance.instance, {}), ActionResult({'message': 'HELLO WORLD'}, 'Success')) self.assertEqual(step._output, ActionResult({'message': 'HELLO WORLD'}, 'Success'))
def test_execute_global_action(self): action = Action(app_name='HelloWorld', action_name='global2', arguments=[Argument('arg1', value='something')]) instance = AppInstance.create(app_name='HelloWorld', device_name='') result = action.execute(instance.instance, {}) self.assertAlmostEqual(result.result, 'something') self.assertEqual(result.status, 'Success') self.assertEqual(action._output, result)
def test_execute_global_action(self): step = Step(app='HelloWorld', action='global2', inputs={'arg1': 'something'}) instance = AppInstance.create(app_name='HelloWorld', device_name='') result = step.execute(instance.instance, {}) self.assertAlmostEqual(result.result, 'something') self.assertEqual(result.status, 'Success') self.assertEqual(step._output, result)
def __setup_app_instance(self, instances, action): device_id = (action.app_name, action.device_id) if device_id not in instances: instances[device_id] = AppInstance.create(action.app_name, action.device_id) WalkoffEvent.CommonWorkflowSignal.send( self, event=WalkoffEvent.AppInstanceCreated) logger.debug( 'Created new app instance: App {0}, device {1}'.format( action.app_name, action.device_id)) return device_id
def test_execute_with_accumulator_missing_action(self): action = Action(app_name='HelloWorld', action_name='Add Three', arguments=[ Argument('num1', reference='1'), Argument('num2', reference='action2'), Argument('num3', value='10.2') ]) accumulator = {'1': '-5.6', 'missing': '4.3', '3': '45'} instance = AppInstance.create(app_name='HelloWorld', device_name='device1') action.execute(instance.instance, accumulator)
def test_execute_with_accumulator_missing_step(self): step = Step(app='HelloWorld', action='Add Three', inputs={ 'num1': '@1', 'num2': '@step2', 'num3': '10.2' }) accumulator = {'1': '-5.6', 'missing': '4.3', '3': '45'} instance = AppInstance.create(app_name='HelloWorld', device_name='device1') with self.assertRaises(InvalidInput): step.execute(instance.instance, accumulator)
def test_execute_with_args(self): step = Step(app='HelloWorld', action='Add Three', inputs={ 'num1': '-5.6', 'num2': '4.3', 'num3': '10.2' }) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = step.execute(instance.instance, {}) self.assertAlmostEqual(result.result, 8.9) self.assertEqual(result.status, 'Success') self.assertEqual(step._output, result)
def test_execute_with_args(self): action = Action(app_name='HelloWorld', action_name='Add Three', arguments=[ Argument('num1', value='-5.6'), Argument('num2', value='4.3'), Argument('num3', value='10.2') ]) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = action.execute(instance.instance, {}) self.assertAlmostEqual(result.result, 8.9) self.assertEqual(result.status, 'Success') self.assertEqual(action._output, result)
def test_execute_with_accumulator_with_extra_actions(self): action = Action(app_name='HelloWorld', action_name='Add Three', arguments=[ Argument('num1', reference='1'), Argument('num2', reference='action2'), Argument('num3', value='10.2') ]) accumulator = {'1': '-5.6', 'action2': '4.3', '3': '45'} instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = action.execute(instance.instance, accumulator) self.assertAlmostEqual(result.result, 8.9) self.assertEqual(result.status, 'Success') self.assertEqual(action._output, result)
def test_execute_with_accumulator_with_conversion(self): step = Step(app='HelloWorld', action='Add Three', inputs={ 'num1': '@1', 'num2': '@step2', 'num3': '10.2' }) accumulator = {'1': '-5.6', 'step2': '4.3'} instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = step.execute(instance.instance, accumulator) self.assertAlmostEqual(result.result, 8.9) self.assertEqual(result.status, 'Success') self.assertEqual(step._output, result)
def test_execute_with_triggers(self): triggers = [Flag(action='regMatch', args={'regex': 'aaa'})] step = Step(app='HelloWorld', action='helloWorld', triggers=triggers) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') step.send_data_to_trigger({"data_in": {"data": 'aaa'}}) result = {'triggered': False} @callbacks.data_sent.connect def callback_is_sent(sender, **kwargs): if kwargs['callback_name'] == "Trigger Step Taken": result['triggered'] = True step.execute(instance.instance, {}) self.assertTrue(result['triggered'])
def test_execute_multiple_triggers(self): triggers = [Flag(action='regMatch', args={'regex': 'aaa'})] step = Step(app='HelloWorld', action='helloWorld', triggers=triggers) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') step.send_data_to_trigger({"data_in": {"data": 'a'}}) trigger_taken = {'triggered': 0} trigger_not_taken = {'triggered': 0} @callbacks.data_sent.connect def callback_is_sent(sender, **kwargs): if kwargs['callback_name'] == "Trigger Step Taken": trigger_taken['triggered'] += 1 elif kwargs['callback_name'] == "Trigger Step Not Taken": step.send_data_to_trigger({"data_in": {"data": 'aaa'}}) trigger_not_taken['triggered'] += 1 step.execute(instance.instance, {}) self.assertEqual(trigger_taken['triggered'], 1) self.assertEqual(trigger_not_taken['triggered'], 1)
def test_execute_action_which_raises_exception_sends_callbacks(self): from tests.testapps.HelloWorld.exceptions import CustomException step = Step(app='HelloWorld', action='Buggy') instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = {'started_triggered': False} @callbacks.data_sent.connect def callback_is_sent(sender, **kwargs): if isinstance(sender, Step): self.assertIs(sender, step) self.assertIn('callback_name', kwargs) self.assertEqual(kwargs['callback_name'], 'Step Started') self.assertIn('object_type', kwargs) self.assertEqual(kwargs['object_type'], 'Step') result['started_triggered'] = True with self.assertRaises(CustomException): step.execute(instance.instance, {}) self.assertTrue(result['started_triggered'])
def test_execute_with_triggers(self): triggers = [ Condition('HelloWorld', action_name='regMatch', arguments=[Argument('regex', value='aaa')]) ] action = Action(app_name='HelloWorld', action_name='helloWorld', triggers=triggers) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') action.send_data_to_trigger({"data_in": {"data": 'aaa'}}) result = {'triggered': False} def callback_is_sent(sender, **kwargs): if kwargs['event'] == WalkoffEvent.TriggerActionTaken: result['triggered'] = True WalkoffEvent.CommonWorkflowSignal.connect(callback_is_sent) action.execute(instance.instance, {}) self.assertTrue(result['triggered'])
def test_execute_event(self): step = Step(app='HelloWorld', action='Sample Event', inputs={'arg1': 1}) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') import time from tests.testapps.HelloWorld.events import event1 import threading def sender(): time.sleep(0.1) event1.trigger(3) thread = threading.Thread(target=sender) start = time.time() thread.start() result = step.execute(instance.instance, {}) end = time.time() thread.join() self.assertEqual(result, ActionResult(4, 'Success')) self.assertGreater((end - start), 0.1)
def test_execute_action_which_raises_exception_sends_callbacks(self): action = Action(app_name='HelloWorld', action_name='Buggy') instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = {'started_triggered': False, 'result_triggered': False} def callback_is_sent(sender, **kwargs): if isinstance(sender, Action): self.assertIn('event', kwargs) self.assertIn(kwargs['event'], (WalkoffEvent.ActionStarted, WalkoffEvent.ActionExecutionError)) if kwargs['event'] == WalkoffEvent.ActionStarted: result['started_triggered'] = True elif kwargs['event'] == WalkoffEvent.ActionExecutionError: result['result_triggered'] = True WalkoffEvent.CommonWorkflowSignal.connect(callback_is_sent) action.execute(instance.instance, {}) self.assertTrue(result['started_triggered']) self.assertTrue(result['result_triggered'])
def test_execute_sends_callbacks(self): step = Step(app='HelloWorld', action='Add Three', inputs={ 'num1': '-5.6', 'num2': '4.3', 'num3': '10.2' }) instance = AppInstance.create(app_name='HelloWorld', device_name='device1') result = {'started_triggered': False, 'result_triggered': False} @callbacks.data_sent.connect def callback_is_sent(sender, **kwargs): if isinstance(sender, Step): self.assertIs(sender, step) self.assertIn('callback_name', kwargs) self.assertIn(kwargs['callback_name'], ('Step Started', 'Function Execution Success')) self.assertIn('object_type', kwargs) self.assertEqual(kwargs['object_type'], 'Step') if kwargs['callback_name'] == 'Step Started': result['started_triggered'] = True else: self.assertIn('data', kwargs) data = json.loads(kwargs['data']) self.assertIn('result', data) data = data['result'] self.assertEqual(data['status'], 'Success') self.assertAlmostEqual(data['result'], 8.9) result['result_triggered'] = True step.execute(instance.instance, {}) self.assertTrue(result['started_triggered']) self.assertTrue(result['result_triggered'])
def test_execute_action_which_raises_exception(self): action = Action(app_name='HelloWorld', action_name='Buggy') instance = AppInstance.create(app_name='HelloWorld', device_name='device1') action.execute(instance.instance, {}) self.assertIsNotNone(action.get_output())