"""persist idempotency keys for run creation

Revision ID: 0002
Revises: 0001
"""
from alembic import op
import sqlalchemy as sa


revision = "0002"
down_revision = "0001"
branch_labels = None
depends_on = None


def upgrade():
    with op.batch_alter_table("test_runs") as batch:
        batch.add_column(sa.Column("idempotency_key", sa.String(length=128), nullable=True))
        batch.add_column(sa.Column("request_fingerprint", sa.String(length=128), nullable=False, server_default=""))
        batch.create_unique_constraint("uq_test_runs_idempotency_key", ["idempotency_key"])


def downgrade():
    with op.batch_alter_table("test_runs") as batch:
        batch.drop_constraint("uq_test_runs_idempotency_key", type_="unique")
        batch.drop_column("request_fingerprint")
        batch.drop_column("idempotency_key")