"""Unit tests for InsertionPoint model."""

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


def test_get_insertion_points():
    """Test insertion point model returns data."""
    episode_id = 1
    mock_data = list(
        filter(
            lambda x: x['episode_id'] == episode_id,
            db_operations.insertion_point_data
        )
    )
    result = insertion_point.get_insertion_points(episode_id)
    assert result
    assert mock_data
    assert len(result['items']) == len(mock_data)


def test_create_insertion_points(mock_current_admin_user):
    """Test insertion point model creates rows."""
    episode_id = 1
    new_points = [{
        'point_type': 'mid',
        'count': 4,
        'timecode': '11.15'
    }]
    result = insertion_point.create_insertion_points(episode_id, new_points)
    assert len(result['items']) == len(new_points)
    assert result['items'][0]['created_by'] == 2


def test_delete_all(mock_current_admin_user):
    """Test delete all insertion points."""
    episode_id = 1
    new_points = [{
        'point_type': 'mid',
        'count': 4,
        'timecode': '11.15'
    }]
    result = insertion_point.create_insertion_points(episode_id, new_points)
    assert len(result['items']) == len(new_points)

    insertion_point.delete_all(episode_id)

    result = insertion_point.get_insertion_points(episode_id)
    assert len(result['items']) == 0