DataOps-platform/migrations/versions/20260723_130_bound_plan_lifecycle.py

"""Add an explicit tested state to physical execution plans."""

from alembic import op

revision = "20260723_130"
down_revision = "20260723_120"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.execute(
        """
        ALTER TABLE public.rule_execution_plans
            DROP CONSTRAINT rule_execution_plans_status_check;
        ALTER TABLE public.rule_execution_plans
            ADD CONSTRAINT rule_execution_plans_status_check
            CHECK (status IN ('compiled','tested','published','revoked'));
        """
    )


def downgrade() -> None:
    # Immutable runtime evidence and tested plans are retained on rollback.
    pass