"""Tests for ProductVideoOutputResolutions model class."""

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

from tests.unit import db_utils


@db_utils.test_schema
def test_get_success() -> None:
    db_utils.seed_models(
        [
            product_video_output_resolutions.ProductVideoOutputResolution(
                output_outer_width=1920,
                output_outer_height=1080,
            )
        ]
    )
    with mysql.db_session() as session:
        query_response = product_video_output_resolutions.get(1, session=session)
        assert query_response is not None
        assert query_response.to_dict() == {
            "id": 1,
            "output_outer_width": 1920,
            "output_outer_height": 1080,
        }


@db_utils.test_schema
def test_get_not_found_deleted() -> None:
    db_utils.seed_models(
        [
            product_video_output_resolutions.ProductVideoOutputResolution(
                output_outer_width=1920,
                output_outer_height=1080,
                is_deleted=True,
            )
        ]
    )
    with mysql.db_session() as session:
        query_response = product_video_output_resolutions.get(1, session=session)
    assert query_response is None


@db_utils.test_schema
def test_get_all_success() -> None:
    db_utils.seed_models(
        [
            product_video_output_resolutions.ProductVideoOutputResolution(
                output_outer_width=1920,
                output_outer_height=1080,
            ),
            product_video_output_resolutions.ProductVideoOutputResolution(
                output_outer_width=2048,
                output_outer_height=1024,
                is_deleted=True,
            ),
        ]
    )
    with mysql.db_session() as session:
        query_response = product_video_output_resolutions.get_all(session=session)
    assert query_response == [(1, 1920, 1080)]