"""Add red_tech_assignee and blue_tech_assignee to tests.

Revision ID: b050
Revises: b049
Create Date: 2026-06-19
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import UUID

revision = "b050"
down_revision = "b049"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("tests", sa.Column("red_tech_assignee", UUID(as_uuid=True), sa.ForeignKey("users.id"), nullable=True))
    op.add_column("tests", sa.Column("blue_tech_assignee", UUID(as_uuid=True), sa.ForeignKey("users.id"), nullable=True))


def downgrade() -> None:
    op.drop_column("tests", "blue_tech_assignee")
    op.drop_column("tests", "red_tech_assignee")
