# -*- coding: utf-8 -*-

import unittest

from SpiffWorkflow.task import TaskState
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow
from SpiffWorkflow.bpmn.specs.events.event_definitions import MessageEventDefinition
from tests.SpiffWorkflow.bpmn.BpmnWorkflowTestCase import BpmnWorkflowTestCase

__author__ = 'matth'


class MessagesTest(BpmnWorkflowTestCase):

    def setUp(self):
        self.spec, self.subprocesses = self.load_workflow_spec('Test-Workflows/*.bpmn20.xml', 'Test Workflows', False)

    def testRunThroughHappy(self):

        self.workflow = BpmnWorkflow(self.spec, self.subprocesses)
        self.do_next_exclusive_step('Select Test', choice='Messages')
        self.workflow.do_engine_steps()
        self.assertEqual([], self.workflow.get_tasks(TaskState.READY))
        self.assertEqual(2, len(self.workflow.get_tasks(TaskState.WAITING)))
        self.workflow.catch(MessageEventDefinition('Wrong Message'))
        self.assertEqual([], self.workflow.get_tasks(TaskState.READY))
        self.workflow.catch(MessageEventDefinition('Test Message'))
        self.assertEqual(1, len(self.workflow.get_tasks(TaskState.READY)))

        self.assertEqual(
            'Test Message', self.workflow.get_tasks(TaskState.READY)[0].task_spec.description)

        self.workflow.do_engine_steps()
        self.assertEqual(
            0, len(self.workflow.get_tasks(TaskState.READY | TaskState.WAITING)))

    def testRunThroughSaveAndRestore(self):

        self.workflow = BpmnWorkflow(self.spec, self.subprocesses)
        self.do_next_exclusive_step('Select Test', choice='Messages')
        self.workflow.do_engine_steps()

        self.save_restore()

        self.assertEqual([], self.workflow.get_tasks(TaskState.READY))
        self.assertEqual(2, len(self.workflow.get_tasks(TaskState.WAITING)))
        self.workflow.catch(MessageEventDefinition('Wrong Message'))
        self.assertEqual([], self.workflow.get_tasks(TaskState.READY))
        self.workflow.catch(MessageEventDefinition('Test Message'))
        self.assertEqual(1, len(self.workflow.get_tasks(TaskState.READY)))

        self.save_restore()

        self.workflow.do_engine_steps()
        self.assertEqual(
            0, len(self.workflow.get_tasks(TaskState.READY | TaskState.WAITING)))


def suite():
    return unittest.TestLoader().loadTestsFromTestCase(MessagesTest)
if __name__ == '__main__':
    unittest.TextTestRunner(verbosity=2).run(suite())