mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-24 01:49:42 -04:00
46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
"""
|
|
|
|
Revision ID: 0400_add_total_message_limit
|
|
Revises: 0399_remove_research_mode
|
|
Create Date: 2023-04-24 11:35:22.873930
|
|
|
|
"""
|
|
import os
|
|
import uuid
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
from app import db
|
|
from app.dao.users_dao import get_user_by_email
|
|
from app.models import User
|
|
|
|
revision = "0401_add_e2e_test_user"
|
|
down_revision = "0400_add_total_message_limit"
|
|
|
|
|
|
def upgrade():
|
|
email_address = os.getenv("NOTIFY_E2E_TEST_EMAIL")
|
|
password = os.getenv("NOTIFY_E2E_TEST_PASSWORD")
|
|
name = f"e2e_test_user_{uuid.uuid4()}"
|
|
data = {
|
|
"id": uuid.uuid4(),
|
|
"name": name,
|
|
"email_address": email_address,
|
|
"password": password,
|
|
"mobile_number": "+12025555555",
|
|
"state": "active",
|
|
}
|
|
user = User(**data)
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
|
|
|
|
def downgrade():
|
|
email_address = os.getenv("NOTIFY_E2E_TEST_EMAIL")
|
|
user_to_delete = get_user_by_email(email_address)
|
|
if not user_to_delete:
|
|
return
|
|
db.session.remove(user_to_delete)
|
|
db.session.commit()
|