"""Tests for Spotify Track Metrics flow."""

from unittest import mock
from unittest.mock import MagicMock

from feed_ingestion.flows.spotify_track_metrics.flow import Flow


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

    spotify_track_metrics_flow.decider(schedule)

    schedule.assert_has_calls([
        mock.call('bootstrap', mock.ANY),
        mock.call('fetch_from_athena', mock.ANY, requires=[mock.ANY]),
        mock.call(
            'load_staging_raw_table', mock.ANY, requires=[mock.ANY]),
        mock.call(
            'set_overall_status_ingested', mock.ANY, requires=[mock.ANY]),
    ])