Python RuleEnforcer.enforce方法代码示例

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
    def test_triggered_execution(self):
        docs = {
            'trigger_type': copy.deepcopy(fixture.ARTIFACTS['trigger_type']),
            'trigger': copy.deepcopy(fixture.ARTIFACTS['trigger']),
            'rule': copy.deepcopy(fixture.ARTIFACTS['rule']),
            'trigger_instance': copy.deepcopy(fixture.ARTIFACTS['trigger_instance'])}

        # Trigger an action execution.
        trigger_type = TriggerType.add_or_update(
            TriggerTypeAPI.to_model(TriggerTypeAPI(**docs['trigger_type'])))
        trigger = Trigger.add_or_update(TriggerAPI.to_model(TriggerAPI(**docs['trigger'])))
        rule = RuleAPI.to_model(RuleAPI(**docs['rule']))
        rule.trigger = reference.get_str_resource_ref_from_model(trigger)
        rule = Rule.add_or_update(rule)
        trigger_instance = TriggerInstance.add_or_update(
            TriggerInstanceAPI.to_model(TriggerInstanceAPI(**docs['trigger_instance'])))
        trace_service.add_or_update_given_trace_context(
            trace_context={'trace_tag': 'test_triggered_execution_trace'},
            trigger_instances=[str(trigger_instance.id)])
        enforcer = RuleEnforcer(trigger_instance, rule)
        enforcer.enforce()

        # Wait for the action execution to complete and then confirm outcome.
        liveaction = LiveAction.get(context__trigger_instance__id=str(trigger_instance.id))
        self.assertIsNotNone(liveaction)
        liveaction = self._wait_on_status(liveaction, action_constants.LIVEACTION_STATUS_FAILED)

        execution = self._get_action_execution(
            liveaction__id=str(liveaction.id),
            raise_exception=True
        )

        self.assertDictEqual(execution.trigger, vars(TriggerAPI.from_model(trigger)))
        self.assertDictEqual(execution.trigger_type, vars(TriggerTypeAPI.from_model(trigger_type)))
        self.assertDictEqual(execution.trigger_instance,
                             vars(TriggerInstanceAPI.from_model(trigger_instance)))
        self.assertDictEqual(execution.rule, vars(RuleAPI.from_model(rule)))
        action = action_utils.get_action_by_ref(liveaction.action)
        self.assertDictEqual(execution.action, vars(ActionAPI.from_model(action)))
        runner = RunnerType.get_by_name(action.runner_type['name'])
        self.assertDictEqual(execution.runner, vars(RunnerTypeAPI.from_model(runner)))
        liveaction = LiveAction.get_by_id(str(liveaction.id))
        self.assertEqual(execution.start_timestamp, liveaction.start_timestamp)
        self.assertEqual(execution.end_timestamp, liveaction.end_timestamp)
        self.assertEqual(execution.result, liveaction.result)
        self.assertEqual(execution.status, liveaction.status)
        self.assertEqual(execution.context, liveaction.context)
        self.assertEqual(execution.liveaction['callback'], liveaction.callback)
        self.assertEqual(execution.liveaction['action'], liveaction.action)

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
 def test_ruleenforcement_create_on_success(self):
     enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, self.models['rules']['rule2.yaml'])
     execution_db = enforcer.enforce()
     self.assertTrue(execution_db is not None)
     self.assertTrue(RuleEnforcement.add_or_update.called)
     self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref,
                      self.models['rules']['rule2.yaml'].ref)

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
 def test_ruleenforcement_casts(self):
     enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, self.models['rules']['rule2.yaml'])
     execution_db = enforcer.enforce()
     self.assertTrue(execution_db is not None)
     self.assertTrue(action_service.request.called)
     self.assertTrue(isinstance(action_service.request.call_args[0][0].parameters['objtype'],
                                dict))

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
 def test_ruleenforcement_create_on_fail(self):
     enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, self.models['rules']['rule1.yaml'])
     execution_db = enforcer.enforce()
     self.assertTrue(execution_db is None)
     self.assertTrue(RuleEnforcement.add_or_update.called)
     self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].failure_reason,
                      FAILURE_REASON)

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
    def test_action_default_jinja_parameter_value_render_fail(self):
        # Action parameter render failure should result in a failed execution
        rule = self.models['rules']['rule_action_default_value_render_fail.yaml']

        enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, rule)
        execution_db = enforcer.enforce()

        self.assertTrue(execution_db is None)
        self.assertTrue(RuleEnforcement.add_or_update.called)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref, rule.ref)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].status,
                         RULE_ENFORCEMENT_STATUS_FAILED)
        self.assertFalse(action_service.request.called)

        self.assertTrue(action_service.create_request.called)
        self.assertEqual(action_service.create_request.call_args[0][0].action,
                         'wolfpack.a2_default_value')

        self.assertTrue(action_service.update_status.called)
        self.assertEqual(action_service.update_status.call_args[1]['new_status'],
                         action_constants.LIVEACTION_STATUS_FAILED)

        expected_msg = ('Failed to render parameter "arrtype": \'dict object\' has no '
                        'attribute \'arrtype_value\'')

        result = action_service.update_status.call_args[1]['result']
        self.assertEqual(result['error'], expected_msg)

        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].failure_reason,
                         expected_msg)

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
    def test_triggered_execution(self):
        docs = {
            'trigger_type': copy.deepcopy(fixture.ARTIFACTS['trigger_type']),
            'trigger': copy.deepcopy(fixture.ARTIFACTS['trigger']),
            'rule': copy.deepcopy(fixture.ARTIFACTS['rule']),
            'trigger_instance': copy.deepcopy(fixture.ARTIFACTS['trigger_instance'])}

        # Trigger an action execution.
        trigger_type = TriggerType.add_or_update(
            TriggerTypeAPI.to_model(TriggerTypeAPI(**docs['trigger_type'])))
        trigger = Trigger.add_or_update(TriggerAPI.to_model(TriggerAPI(**docs['trigger'])))
        rule = RuleAPI.to_model(RuleAPI(**docs['rule']))
        rule.trigger = reference.get_str_resource_ref_from_model(trigger)
        rule = Rule.add_or_update(rule)
        trigger_instance = TriggerInstance.add_or_update(
            TriggerInstanceAPI.to_model(TriggerInstanceAPI(**docs['trigger_instance'])))
        enforcer = RuleEnforcer(trigger_instance, rule)
        enforcer.enforce()

        # Wait for the action execution to complete and then confirm outcome.
        execution = ActionExecution.get(context__trigger_instance__id=str(trigger_instance.id))
        self.assertIsNotNone(execution)
        execution = ActionExecution.get_by_id(str(execution.id))
        self.assertEqual(execution.status, ACTIONEXEC_STATUS_SUCCEEDED)
        history = ActionExecutionHistory.get(execution__id=str(execution.id), raise_exception=True)
        self.assertDictEqual(history.trigger, vars(TriggerAPI.from_model(trigger)))
        self.assertDictEqual(history.trigger_type, vars(TriggerTypeAPI.from_model(trigger_type)))
        self.assertDictEqual(history.trigger_instance,
                             vars(TriggerInstanceAPI.from_model(trigger_instance)))
        self.assertDictEqual(history.rule, vars(RuleAPI.from_model(rule)))
        action_ref = ResourceReference.from_string_reference(ref=execution.action)
        action, _ = action_utils.get_action_by_dict(
            {'name': action_ref.name, 'pack': action_ref.pack})
        self.assertDictEqual(history.action, vars(ActionAPI.from_model(action)))
        runner = RunnerType.get_by_name(action.runner_type['name'])
        self.assertDictEqual(history.runner, vars(RunnerTypeAPI.from_model(runner)))
        execution = ActionExecution.get_by_id(str(execution.id))
        self.assertDictEqual(history.execution, vars(ActionExecutionAPI.from_model(execution)))

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
    def test_action_default_jinja_parameter_value_overridden_in_rule(self):
        # Verify that it works correctly if default parameter value is overridden in rule
        rule = self.models['rules']['rule_action_default_value_overridden.yaml']

        enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, rule)
        execution_db = enforcer.enforce()

        self.assertTrue(execution_db is not None)
        self.assertTrue(RuleEnforcement.add_or_update.called)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref, rule.ref)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].status,
                         RULE_ENFORCEMENT_STATUS_SUCCEEDED)

        call_parameters = action_service.request.call_args[0][0].parameters

        self.assertEqual(call_parameters['objtype'], {'t1_p': 't1_p_v'})
        self.assertEqual(call_parameters['strtype'], 't1_p_v')
        self.assertEqual(call_parameters['arrtype'], ['override 1', 'override 2'])

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
    def test_action_default_jinja_parameter_value_is_rendered(self):
        # Verify that a default action parameter which is a Jinja variable is correctly rendered
        rule = self.models['rules']['rule_action_default_value.yaml']

        enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, rule)
        execution_db = enforcer.enforce()

        self.assertTrue(execution_db is not None)
        self.assertTrue(RuleEnforcement.add_or_update.called)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref, rule.ref)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].status,
                         RULE_ENFORCEMENT_STATUS_SUCCEEDED)

        call_parameters = action_service.request.call_args[0][0].parameters

        self.assertEqual(call_parameters['objtype'], {'t1_p': 't1_p_v'})
        self.assertEqual(call_parameters['strtype'], 't1_p_v')
        self.assertEqual(call_parameters['arrtype'], ['one 1', 'two 2', 'three 3'])

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
 def test_ruleenforcement_occurs(self):
     enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, MOCK_RULE_1)
     execution_id = enforcer.enforce()
     self.assertTrue(execution_id is not None)

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
 def test_ruleenforcement_occurs(self):
     enforcer = RuleEnforcer(MOCK_TRIGGER_INSTANCE, self.models['rules']['rule1.yaml'])
     execution_db = enforcer.enforce()
     self.assertTrue(execution_db is not None)

# 需要导入模块: from st2reactor.rules.enforcer import RuleEnforcer [as 别名]
# 或者: from st2reactor.rules.enforcer.RuleEnforcer import enforce [as 别名]
    def test_rule_enforcement_create_rule_none_param_casting(self):
        mock_trigger_instance = MOCK_TRIGGER_INSTANCE_2

        # 1. Non None value, should be serialized as regular string
        mock_trigger_instance.payload = {'t1_p': 'somevalue'}

        def mock_cast_string(x):
            assert x == 'somevalue'
            return casts._cast_string(x)
        casts.CASTS['string'] = mock_cast_string

        enforcer = RuleEnforcer(mock_trigger_instance,
                                self.models['rules']['rule_use_none_filter.yaml'])
        execution_db = enforcer.enforce()

        # Verify value has been serialized correctly
        call_args = action_service.request.call_args[0]
        live_action_db = call_args[0]
        self.assertEqual(live_action_db.parameters['actionstr'], 'somevalue')
        self.assertTrue(execution_db is not None)
        self.assertTrue(RuleEnforcement.add_or_update.called)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref,
                         self.models['rules']['rule_use_none_filter.yaml'].ref)

        # 2. Verify that None type from trigger instance is correctly serialized to
        # None when using "use_none" Jinja filter when invoking an action
        mock_trigger_instance.payload = {'t1_p': None}

        def mock_cast_string(x):
            assert x == NONE_MAGIC_VALUE
            return casts._cast_string(x)
        casts.CASTS['string'] = mock_cast_string

        enforcer = RuleEnforcer(mock_trigger_instance,
                                self.models['rules']['rule_use_none_filter.yaml'])
        execution_db = enforcer.enforce()

        # Verify None has been correctly serialized to None
        call_args = action_service.request.call_args[0]
        live_action_db = call_args[0]
        self.assertEqual(live_action_db.parameters['actionstr'], None)
        self.assertTrue(execution_db is not None)
        self.assertTrue(RuleEnforcement.add_or_update.called)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref,
                         self.models['rules']['rule_use_none_filter.yaml'].ref)

        casts.CASTS['string'] = casts._cast_string

        # 3. Parameter value is a compound string one of which values is None, but "use_none"
        # filter is not used
        mock_trigger_instance = MOCK_TRIGGER_INSTANCE_3
        mock_trigger_instance.payload = {'t1_p': None, 't2_p': 'value2'}

        enforcer = RuleEnforcer(mock_trigger_instance,
                                self.models['rules']['rule_none_no_use_none_filter.yaml'])
        execution_db = enforcer.enforce()

        # Verify None has been correctly serialized to None
        call_args = action_service.request.call_args[0]
        live_action_db = call_args[0]
        self.assertEqual(live_action_db.parameters['actionstr'], 'None-value2')
        self.assertTrue(execution_db is not None)
        self.assertTrue(RuleEnforcement.add_or_update.called)
        self.assertEqual(RuleEnforcement.add_or_update.call_args[0][0].rule.ref,
                         self.models['rules']['rule_none_no_use_none_filter.yaml'].ref)

        casts.CASTS['string'] = casts._cast_string