"""Test youtube channel."""

from datetime import datetime

from video.connectors import mysql
from video.models.sql.classes import youtube_channel

from tests.unit import ar_db_utils


@ar_db_utils.test_schema
def test_get() -> None:
    """Test get."""
    vendor_id = 12
    date_initiated = datetime.now().date().strftime("%Y-%m-%d")
    with mysql.ar_db_session(turn_off_foreign_key_constraint=True) as session:
        channel = youtube_channel.YoutubeChannel(
            youtube_channel_name="Bobby",
            youtube_channel_id="12",
            vendor_id=vendor_id,
            artist_id=123,
            date_initiated=date_initiated,
        )
        session.add(channel)
        session.flush()

    result = youtube_channel.get(vendor_id)
    assert result == [
        {
            "id": 1,
            "vendor_id": vendor_id,
            "channel_id": "12",
            "channel_name": "Bobby",
            "artist_id": 123,
            "date_initiated": date_initiated,
        }
    ]


def test_get_none_vendor_id() -> None:
    """Test get returns empty list when vendor_id is None."""
    assert youtube_channel.get(None) == []