"""Unit tests for backfill user show family table."""

from dev.backfill_user_show_family_table import get_all_users_podcasts
from dev.backfill_user_show_family_table import get_all_users_show_families
from dev.backfill_user_show_family_table import update_user_and_show_families


def test_get_all_users_podcasts():
    """Test get all user_podcast rows."""
    all_users_podcasts = get_all_users_podcasts()
    assert all_users_podcasts == [(11, 1), (6, 2)]


def test_get_all_users_show_families():
    """Test get all user_show_family rows."""
    all_users_show_families = get_all_users_show_families()
    assert all_users_show_families == [(11, 1)]


def test_update_user_and_show_families():
    """Test update_user_and_show_families."""
    updated_user = update_user_and_show_families(11, 2, 366)
    assert updated_user['id'] == 11
    assert updated_user['updated_by'] == 366
    assert len(updated_user['show_families']) == 2
    assert updated_user['show_families'][0]['id'] == 1
    assert updated_user['show_families'][1]['id'] == 2


def test_update_user_and_show_families_duplicate():
    """Test update_user_and_show_families doesn't add duplicate show family."""
    updated_user = update_user_and_show_families(11, 1, 366)
    assert updated_user['id'] == 11
    assert updated_user['updated_by'] == 366
    assert len(updated_user['show_families']) == 1
    assert updated_user['show_families'][0]['id'] == 1