"""Tests for util funcs of the Spotify Source of Stream Workflow."""

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

from analytics_aggregation.flows.apple_music_sos import utils


@patch('analytics_aggregation.flows.apple_music_sos.utils.feed_status')
def test_is_apple_music_ingested(feed_status):
    """Test for is_apple_music_ingested util function."""
    feed_status.STATUS_INGESTED = 'INGESTED'
    get_overall_status_mock = Mock(return_value=feed_status.STATUS_INGESTED)
    feed_status.get_overall_status = get_overall_status_mock

    result = utils.is_apple_music_ingested('2017-01-01')

    assert result
    get_overall_status_mock.assert_called_once_with(
        'apple_music_streams', '2017-01-01')


@patch('analytics_aggregation.flows.apple_music_sos.utils.feed_status')
def test_is_apple_music_ingested_not_ingested(feed_status):
    """Test for is_apple_music_ingested util function."""
    feed_status.STATUS_INGESTED = 'INGESTED'
    get_overall_status_mock = Mock(return_value='ANY_OTHER_STATUS')
    feed_status.get_overall_status = get_overall_status_mock

    result = utils.is_apple_music_ingested('2017-01-01')

    assert not result
    get_overall_status_mock.assert_called_once_with(
        'apple_music_streams', '2017-01-01')