"""add_test_labels

Revision ID: c10d0f3d7b9e
Revises: 40a744571c9f
Create Date: 2023-04-07 15:03:30.769250

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = 'c10d0f3d7b9e'
down_revision = '711fbe1fddaf'
branch_labels = None
depends_on = None


def upgrade():
    bind = op.get_bind()
    bind.execute(
        """ INSERT INTO "Label" (id, name, external_name, country, abbreviation) VALUES
        (800, 'Decibel test label UK', 'Decibel test label UK', 'UK', 'TEST'),
        (900, 'Decibel test label USA', 'Decibel test label USA', null, null) """
    )


def downgrade():
    bind = op.get_bind()
    bind.execute(
        """ DELETE FROM "Label" WHERE id in (800, 900) """
    )