"""Test model util functions."""

from analytics.models import utils


def test_playlists_with_status_empty():
    """Test playlists_with_status function with empty list."""
    playlists = []
    status_code = 0
    result = utils.playlists_with_status(playlists, status_code)
    assert result == []


def test_playlists_with_status():
    """Test playlists_with_status successfully."""
    playlists = [
        {'playlist_link': 'link1', 'random': 'data'},
        {'playlist_link': 'link2', 'other': 'values'}]
    status_code = 500
    expected_result = [
        {'playlist_link': 'link1', 'status_code': 500},
        {'playlist_link': 'link2', 'status_code': 500}]
    result = utils.playlists_with_status(playlists, status_code)
    assert result == expected_result