notifications-admin/tests/app/main/__init__.py

from datetime import datetime

from app.main.dao import users_dao
from app.models import User


def create_test_user(state):
    user = User(name='Test User',
                password='somepassword',
                email_address='test@user.gov.uk',
                mobile_number='+441234123412',
                created_at=datetime.now(),
                role_id=1,
                state=state)
    users_dao.insert_user(user)
    return user


def create_another_test_user(state):
    user = User(name='Another Test User',
                password='someOtherpassword',
                email_address='another_test@user.gov.uk',
                mobile_number='+442233123412',
                created_at=datetime.now(),
                role_id=1,
                state=state)
    users_dao.insert_user(user)
    return user