spiff-arena/tests/SpiffWorkflow/camunda/CamundaParserTest.py

from SpiffWorkflow.bpmn.parser.util import full_tag
from SpiffWorkflow.camunda.specs.UserTask import UserTask
from SpiffWorkflow.camunda.parser.CamundaParser import CamundaParser
from SpiffWorkflow.camunda.parser.task_spec import UserTaskParser, BusinessRuleTaskParser
from SpiffWorkflow.dmn.specs.BusinessRuleTask import BusinessRuleTask

from .BaseTestCase import BaseTestCase

class CamundaParserTest(BaseTestCase):

    def setUp(self):
        self.parser = CamundaParser()

    def test_overrides(self):

        overrides = [
            ('userTask', UserTaskParser, UserTask),
            ('businessRuleTask', BusinessRuleTaskParser, BusinessRuleTask),
        ]

        for key, parser, spec in overrides:
            self.assertIn(full_tag(key), self.parser.OVERRIDE_PARSER_CLASSES)
            self.assertEqual((parser, spec), self.parser.OVERRIDE_PARSER_CLASSES.get(full_tag(key)))