import pytest
from weppy import session

from <%= appName %> import app
from <%= appName %>.dev_utils import setup_admin, remove_admin, setup_user, remove_user


class UserMock(object):
    def __init__(self, email, first_name, last_name, password):
        self.email = email
        self.first_name = first_name
        self.last_name = last_name
        self.password = password

    def __str__(self):
        print("Email: {}\n"
              "First_Name: {}\n"
              "Last_Name: {}\n"
              "Password: {}\n".format(self.email, self.first_name, self.last_name, self.password))


TEST_ADMIN = UserMock('test_admin@example.com',
                      'TestAdminFirst',
                      'TestAdminLast',
                      'testadmin')

TEST_USER = UserMock('test_user@example.com',
                     'TestUserFirst',
                     'TestUserLast',
                     'testuser')


@pytest.fixture(scope='function')
def client():
    return app.test_client()


@pytest.fixture(scope='function')
def logged_client(request):
    try:
        setup_user()
    except Exception as ex:
        # If running in dev mode, user may already exist
        # So it's possibly okay if this fails.
        print(ex)

    user_client = app.test_client()
    user_client.get("/account/login")
    user_client.post('/account/login', data={
        'email': TEST_USER.email,
        'password': TEST_USER.password,
        '_csrf_token': list(session._csrf)[-1]
    }, follow_redirects=True)

    def user_teardown():
        remove_user()
    request.addfinalizer(user_teardown)

    return user_client


@pytest.fixture(scope='function')
def admin_client(request):
    try:
        setup_admin()
    except Exception as ex:
        # If running in dev mode, admin may already exist
        # So it's possibly okay if this fails.
        print(ex)

    administrator = app.test_client()
    administrator.get("/account/login")
    administrator.post('/account/login', data={
        'email': TEST_ADMIN.email,
        'password': TEST_ADMIN.password,
        '_csrf_token': list(session._csrf)[-1]
    }, follow_redirects=True)

    def admin_teardown():
        remove_admin()
    request.addfinalizer(admin_teardown)

    return administrator