"""Unit tests for Store Specific Itunes Priority Workflow."""
from unittest import mock
from unittest.mock import MagicMock

from feed_ingestion.flows.itunes_priority.flow import Flow


def test_flow_decider():
    """Test normal decider execution..."""
    schedule = MagicMock()
    schedule_result_object = MagicMock()
    schedule_result_object.result = {
    }
    schedule.return_value = schedule_result_object
    flow = Flow()
    flow.decider(schedule)

    schedule.assert_has_calls([
        mock.call('bootstrap', mock.ANY),
        mock.call('process_files', mock.ANY, requires=[mock.ANY]),
        mock.call('send_emails', mock.ANY, requires=[mock.ANY]),
    ])


def test_flow_decider_if_process_files_stopped():
    """Test normal decider execution..."""
    schedule = MagicMock()
    schedule_result_object = MagicMock()
    schedule_result_object.result = {
         'process_files.stop': True,
    }
    schedule.return_value = schedule_result_object
    flow = Flow()
    flow.decider(schedule)

    schedule.assert_has_calls([
        mock.call('bootstrap', mock.ANY),
        mock.call('process_files', mock.ANY, requires=[mock.ANY]),
    ])