"""Test JobStatus."""

from sqlalchemy import select

from video import config
from video.connectors import mysql
from video.models.sql.classes import job_status as job_status_class

from tests.unit import db_utils


@db_utils.test_schema
def test_job_status() -> None:
    """Test JobInput."""
    job_status = job_status_class.JobStatus(
        primary_key=123,
        job_id=456,
        status="PROGRESSING",
        context_id=44,
    )
    db_utils.seed_models([job_status])
    with mysql.db_session() as session:
        query_response = session.execute(
            select(job_status_class.JobStatus).where(
                job_status_class.JobStatus.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": job_status.primary_key,
        "job_id": job_status.job_id,
        "status": job_status.status,
        "context_id": job_status.context_id,
        "datetime": insert_datetime,
    }