"""Test Context."""

from sqlalchemy import select

from video import config
from video.connectors import mysql
from video.models.sql.classes import context as context_class

from tests.unit import db_utils


@db_utils.test_schema
def test_context() -> None:
    """Test Context."""
    context = context_class.Context(
        primary_key=123,
    )
    db_utils.seed_models([context])
    with mysql.db_session() as session:
        query_response = session.execute(
            select(context_class.Context).where(
                context_class.Context.primary_key == 123
            )
        ).scalar_one_or_none()
        assert query_response is not None
        insert_datetime = query_response.datetime.strftime(
            config.RFC3339_DATETIME_FORMAT
        )
        query_response_dict = query_response.to_dict()

    assert query_response_dict == {
        "id": context.primary_key,
        "datetime": insert_datetime,
    }