mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-08 22:22:24 -05:00
update e2e test user
This commit is contained in:
60
migrations/versions/0416_readd_e2e_test_user.py
Normal file
60
migrations/versions/0416_readd_e2e_test_user.py
Normal file
@@ -0,0 +1,60 @@
|
||||
"""
|
||||
|
||||
Revision ID: 0416_readd_e2e_test_user
|
||||
Revises: 0415_add_message_cost
|
||||
Create Date: 2025-03-17 11:35:22.873930
|
||||
|
||||
"""
|
||||
|
||||
import datetime
|
||||
import os
|
||||
import uuid
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic import op
|
||||
|
||||
from app import db
|
||||
from app.dao.users_dao import get_user_by_email
|
||||
from app.models import User
|
||||
from app.utils import utc_now
|
||||
|
||||
revision = "0416_readd_e2e_test_user"
|
||||
down_revision = "0415_add_message_cost"
|
||||
|
||||
|
||||
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",
|
||||
"created_at": utc_now(),
|
||||
"password_changed_at": utc_now(),
|
||||
"failed_login_count": 0,
|
||||
"platform_admin": "f",
|
||||
"email_access_validated_at": utc_now(),
|
||||
}
|
||||
conn = op.get_bind()
|
||||
insert_sql = """
|
||||
insert into users (id, name, email_address, _password, mobile_number, state, created_at, password_changed_at, failed_login_count, platform_admin, email_access_validated_at)
|
||||
values (:id, :name, :email_address, :password, :mobile_number, :state, :created_at, :password_changed_at, :failed_login_count, :platform_admin, :email_access_validated_at)
|
||||
"""
|
||||
conn.execute(sa.text(insert_sql), data)
|
||||
|
||||
delete_sql = """
|
||||
delete from users where email_address='e2e-test-notify-user@fake.gov'
|
||||
"""
|
||||
|
||||
|
||||
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()
|
||||
Reference in New Issue
Block a user