"""Test AutoCarveotDms."""

from typing import Any

import pytest
from sqlalchemy import select

from video.connectors import mysql
from video.models.sql.classes import auto_carveout_dms as auto_carveout_dms_class

from tests.unit import db_utils


@db_utils.test_schema
@pytest.mark.parametrize(
    ("auto_carveout_dms_id", "video_type_id", "dms_id", "expected_response"),
    [
        (1, 1, 2, {"auto_carveout_dms_id": 1, "video_type_id": 1, "dms_id": 2}),
        (2, 1, 3, {"auto_carveout_dms_id": 2, "video_type_id": 1, "dms_id": 3}),
    ],
)
def test_auto_carveout_dms(
    auto_carveout_dms_id: Any, video_type_id: Any, dms_id: Any, expected_response: Any
) -> None:
    """Test AutoCarveoutDms."""
    auto_carveout_dms = auto_carveout_dms_class.AutoCarveoutDms(
        auto_carveout_dms_id=auto_carveout_dms_id,
        video_type_id=video_type_id,
        dms_id=dms_id,
    )

    db_utils.seed_models([auto_carveout_dms])
    with mysql.db_session() as session:
        db_result = session.execute(
            select(auto_carveout_dms_class.AutoCarveoutDms).where(
                auto_carveout_dms_class.AutoCarveoutDms.auto_carveout_dms_id
                == auto_carveout_dms_id
            )
        ).scalar_one_or_none()
        query_response = db_result.to_dict() if db_result else None

    assert query_response == expected_response