notifications-admin/tests/__init__.py

from flask.testing import FlaskClient
from flask import url_for
from app.main.dao import (users_dao)


class TestClient(FlaskClient):
    def login(self, user):
        # Skipping authentication here and just log them in
        with self.session_transaction() as session:
            session['user_details'] = {
                "email": user.email_address,
                "id": user.id}
        # Include mock_login fixture in test for this to work.
        # TODO would be better for it to be mocked in this
        # function
        response = self.post(
            url_for('main.two_factor'), data={'sms_code': '12345'})
        assert response.status_code == 302

    def logout(self, user):
        self.get(url_for("main.logout"))


def service_json(id_, name, users, limit=1000, active=False, restricted=True):
    return {
        'id': id_,
        'name': name,
        'users': users,
        'limit': limit,
        'active': active,
        'restricted': restricted
    }


def template_json(id_, name, type_, content, service_id):
    return {
        'id': id_,
        'name': name,
        'template_type': type_,
        'content': content,
        'service': service_id
    }


def api_key_json(id_, name, expiry_date=None):
    return {'id': id_,
            'name': name,
            'expiry_date': expiry_date
            }

TEST_USER_EMAIL = 'test@user.gov.uk'


def create_test_user(state):
    user = None
    users_dao.insert_user(user)
    return user


def create_test_api_user(state):
    from app.notify_client.user_api_client import User
    user_data = {'id': 1,
                 'name': 'Test User',
                 'password': 'somepassword',
                 'email_address': TEST_USER_EMAIL,
                 'mobile_number': '+441234123412',
                 'state': state
                 }
    user = User(user_data)
    return user


def create_another_test_user(state):
    user = None
    users_dao.insert_user(user)
    return user


def get_test_user():
    return users_dao.get_user_by_email(TEST_USER_EMAIL)