add migration for e2e tests and improve code coverage

This commit is contained in:
Kenneth Kehl
2023-09-12 10:15:07 -07:00
parent a3d8c4835f
commit c082546aa0
3 changed files with 140 additions and 2 deletions

View File

@@ -0,0 +1,45 @@
"""
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()