"""Unit tests for podcast seasons."""
from unittest.mock import patch

from dev.backfill_podcast_seasons import create_podcast_seasons
from dev.backfill_podcast_seasons import get_all_seasons
from dev.backfill_podcast_seasons import get_podcast_seasons

from podcast.models import podcast_season as season_model


def test_get_all_seasons():
    """Test get all podcasts seasons."""
    all_seasons = get_all_seasons(season_model)
    assert all_seasons == [(1, 1), (1, 2), (1, 3), (2, 1), (2, 4)]


def test_get_podcast_seasons():
    """Test get seasons by podcast id."""
    podcast_id = 2

    seasons = get_podcast_seasons(season_model, podcast_id)
    assert seasons == [(1,), (4,)]


@patch('podcast.models.podcast_season.create_seasons')
def test_create_podcast_seasons(mock_create_seasons):
    """Test create podcast seasons."""
    mock_podcast_id = 2
    mock_season_number = 2
    mock_data = [
        dict(
            number=2,
            name='Season 2'
        ),
        dict(
            number=3,
            name='Season 3'
        ),
    ]
    create_podcast_seasons(mock_podcast_id, mock_season_number, season_model)
    mock_create_seasons.assert_called_once_with(
        mock_podcast_id, mock_data, False)