From dd9a027c5a36bb5c7e9086c2d0977cd26759748b Mon Sep 17 00:00:00 2001 From: Kai Chappell Date: Wed, 21 May 2025 19:49:22 +0100 Subject: [PATCH] feat(backend): test cases migration --- .../alembic/versions/003_add_test_cases.py | 34 +++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 backend/alembic/versions/003_add_test_cases.py diff --git a/backend/alembic/versions/003_add_test_cases.py b/backend/alembic/versions/003_add_test_cases.py new file mode 100644 index 0000000..dcb9db4 --- /dev/null +++ b/backend/alembic/versions/003_add_test_cases.py @@ -0,0 +1,34 @@ +"""add test cases + +Revision ID: 003 +Revises: 11f3a598b82d +Create Date: 2025-04-12 + +""" + +from collections.abc import Sequence + +import sqlalchemy as sa +from alembic import op +from sqlalchemy.dialects import postgresql + +revision: str = "003" +down_revision: str | None = "11f3a598b82d" +branch_labels: str | Sequence[str] | None = None +depends_on: str | Sequence[str] | None = None + + +def upgrade() -> None: + op.add_column( + "questions", + sa.Column("function_signature", sa.Text(), nullable=True), + ) + op.add_column( + "questions", + sa.Column("test_cases", postgresql.JSONB(astext_type=sa.Text()), nullable=True), + ) + + +def downgrade() -> None: + op.drop_column("questions", "test_cases") + op.drop_column("questions", "function_signature")