"""Unit tests for backfill episodes guids."""

from dev.backfill_episodes_guids import get_all_podcasts
from dev.backfill_episodes_guids import get_episodes_by_podcast_id

from tests.utils import db_operations


def test_get_all_podcasts():
    """Test get all podcasts."""
    all_podcasts = get_all_podcasts()
    assert all_podcasts == [
        (1, 1, '6bf5bea4-a1d4-11e6-8dea-b334e2aa4710'),
        (2, 2, '6bf5bea4-a1d4-11e6-8dea-b334e2aa4720'),
        (3, 1, '6bf5bea4-a1d4-11e6-8dea-b334e2aa4730'),
        (4, 2, '6bf5bea4-a1d4-11e6-8dea-b334e2aa4740'),
    ]


def test_get_episodes_by_podcast_id():
    """Test get episodes by podcast id."""
    podcast_id = 1

    episodes = get_episodes_by_podcast_id(podcast_id)
    assert len(episodes) == 3
    assert episodes[0]['id'] == db_operations.episode_data[0]['id']
    assert episodes[1]['id'] == db_operations.episode_data[1]['id']