"""Unit tests for Category model."""

from podcast.models import category
from tests.utils import db_operations


def test_get_categories():
    """Test category model returns data when seeded."""
    result_categories = category.get_categories()
    assert len(result_categories['items']) == 2

    result_categories = result_categories['items'][0]
    assert result_categories['id'] == db_operations.category_data[0]['id']
    assert result_categories['name'] == db_operations.category_data[0]['name']


def test_get_categories_by_ids():
    """Test category model returns data when seeded."""
    ids = [3, 1]
    result_categories = category.get_categories_by_ids(ids)
    assert len(result_categories['items']) == 2

    categories = result_categories['items']

    assert categories[0] == db_operations.category_data[2]['name']
    assert categories[1] == db_operations.category_data[0]['name']