56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from ..models import Workflow, WorkflowState, WorkflowTransition
|
|
|
|
from .literals import (
|
|
TEST_WORKFLOW_INITIAL_STATE_COMPLETION, TEST_WORKFLOW_INITIAL_STATE_LABEL,
|
|
TEST_WORKFLOW_INTERNAL_NAME, TEST_WORKFLOW_LABEL,
|
|
TEST_WORKFLOW_STATE_COMPLETION, TEST_WORKFLOW_STATE_LABEL,
|
|
TEST_WORKFLOW_TRANSITION_LABEL, TEST_WORKFLOW_TRANSITION_LABEL_2
|
|
)
|
|
|
|
|
|
class WorkflowTestMixin(object):
|
|
def _create_test_workflow(self):
|
|
self.test_workflow = Workflow.objects.create(
|
|
label=TEST_WORKFLOW_LABEL,
|
|
internal_name=TEST_WORKFLOW_INTERNAL_NAME
|
|
)
|
|
|
|
def _create_test_workflow_state(self):
|
|
self.test_workflow_state = WorkflowState.objects.create(
|
|
workflow=self.test_workflow, label=TEST_WORKFLOW_STATE_LABEL,
|
|
completion=TEST_WORKFLOW_STATE_COMPLETION
|
|
)
|
|
|
|
def _create_test_workflow_states(self):
|
|
self.test_workflow_initial_state = WorkflowState.objects.create(
|
|
workflow=self.test_workflow, label=TEST_WORKFLOW_INITIAL_STATE_LABEL,
|
|
completion=TEST_WORKFLOW_INITIAL_STATE_COMPLETION, initial=True
|
|
)
|
|
self.test_workflow_state = WorkflowState.objects.create(
|
|
workflow=self.test_workflow, label=TEST_WORKFLOW_STATE_LABEL,
|
|
completion=TEST_WORKFLOW_STATE_COMPLETION
|
|
)
|
|
|
|
def _create_test_workflow_transition(self):
|
|
self._create_test_workflow_states()
|
|
self.test_workflow_transition = WorkflowTransition.objects.create(
|
|
workflow=self.test_workflow, label=TEST_WORKFLOW_TRANSITION_LABEL,
|
|
origin_state=self.test_workflow_initial_state,
|
|
destination_state=self.test_workflow_state
|
|
)
|
|
|
|
def _create_test_workflow_transitions(self):
|
|
self.workflow_transition = WorkflowTransition.objects.create(
|
|
workflow=self.test_workflow, label=TEST_WORKFLOW_TRANSITION_LABEL,
|
|
origin_state=self.test_workflow_initial_state,
|
|
destination_state=self.test_workflow_state
|
|
)
|
|
|
|
self.workflow_transition_2 = WorkflowTransition.objects.create(
|
|
workflow=self.workflow, label=TEST_WORKFLOW_TRANSITION_LABEL_2,
|
|
origin_state=self.workflow_initial_state,
|
|
destination_state=self.workflow_state
|
|
)
|