"""Tests for Flow class."""

from unittest import TestCase
from unittest.mock import Mock
from unittest.mock import patch

from flows.digital.flow import Flow


class FlowTest(TestCase):
    """Tets class for Flow class."""

    def setUp(self):
        """Setup function."""
        self.patched_tasks = patch('flows.digital.flow.tasks')
        self.tasks = self.patched_tasks.start()
        self.addCleanup(self.patched_tasks.stop)

    def test_decider(self):
        """Test decider function."""
        flow = Flow('test', 'test_digital', '1.0')
        schedule = Mock()
        flow.decider(schedule)
        tasks = {
            'unload_from_datawarehouse', 'load_to_temp_table',
            'cleanup_unload_files', 'move_temp_table_to_digital_revenue',
            'send_notification', 'finalize_etl_status'}
        for call in schedule.call_args_list:
            tasks.remove(call[0][0])
        assert not tasks