"""Test Job.""" from sqlalchemy import select from video import config from video.connectors import mysql from video.models.sql.classes import job as job_class from tests.unit import db_utils @db_utils.test_schema def test_job() -> None: """Test job.""" job = job_class.Job( primary_key=123, job_type="TRANSFER_FROM_BROWSER_TO_S3", parent_id=789, context_id=44, ) db_utils.seed_models([job]) with mysql.db_session() as session: query_response = session.execute( select(job_class.Job).where(job_class.Job.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.primary_key, "type": job.job_type, "parent_id": job.parent_id, "context_id": job.context_id, "datetime": insert_datetime, }