"""Test for Category logic tier."""

from unittest.mock import MagicMock

from podcast.logic import category
from podcast.models import category as category_model


def test_get_categories(monkeypatch):
    """Test get categories for a podcast."""
    category_data = [
        dict(
            id=1,
            name='Arts',
            parent_id=None,
            active=True,
        ),
        dict(
            id=2,
            name='Beauty',
            parent_id=1,
            active=True,
        ),
        dict(
            id=3,
            name='Technology',
            parent_id=None,
            active=True,
        ),
    ]

    monkeypatch.setattr(
        category_model, 'get_categories', MagicMock(
            return_value={'items': category_data}))

    data_response = category.get_categories()['items']

    assert data_response == [
        dict(
            id=1,
            name='Arts',
            parent_id=None,
            active=True,
            children=[
                dict(
                    id=2,
                    name='Beauty',
                    parent_id=1,
                    active=True,
                ),
            ],
        ),
        dict(
            id=3,
            name='Technology',
            parent_id=None,
            active=True,
            children=[]
        ),

    ]