"""Test ContextField.""" from sqlalchemy import select from video import config from video.connectors import mysql from video.constants import field_types from video.models.sql.classes import context_field as context_field_class from tests.unit import db_utils @db_utils.test_schema def test_context_field() -> None: """Test ContextField.""" context_field = context_field_class.ContextField( primary_key=123, field_type=field_types.STRING, name="applesauce", value="bananas", context_id=333, ) db_utils.seed_models([context_field]) with mysql.db_session() as session: query_response = session.execute( select(context_field_class.ContextField).where( context_field_class.ContextField.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_field.primary_key, "type": context_field.field_type, "name": context_field.name, "value": context_field.value, "context_id": context_field.context_id, "datetime": insert_datetime, }