tests.py

from app import create_app
from app.config import test_config
from app.database import db
from app.user.models import User
from sqlalchemy.sql.expression import func
from faker import Faker
import unittest

admin_username = 'cburmeister'
admin_email = 'cburmeister@discogs.com'
admin_password = 'test123'

fake = Faker()


class TestCase(unittest.TestCase):
    def setUp(self):
        app = create_app(test_config)
        db.app = app  # hack for using db.init_app(app) in app/__init__.py
        self.app = app.test_client()

    def tearDown(self):
        pass

    def login(self, username, password):
        return self.app.post('/login', data=dict(
            username=username,
            password=password
        ), follow_redirects=True)

    def register_user(self, username, email, password):
        return self.app.post('/register', data=dict(
            username=username,
            email=email,
            password=password,
            confirm=password,
            accept_tos=True
        ), follow_redirects=True)

    def edit_user(self, user, email):
        return self.app.post('/user/edit/%s' % user.id, data=dict(
            username=user.username,
            email=user.email,
        ), follow_redirects=True)

    def delete_user(self, uid):
        return self.app.get('/user/delete/%s' % uid, follow_redirects=True)

    def test_404(self):
        resp = self.app.get('/nope', follow_redirects=True)
        assert resp.data, 'Page Not Found'

    def test_index(self):
        resp = self.app.get('/index', follow_redirects=True)
        assert resp.data, 'Flask Bones'

    def test_login(self):
        resp = self.login(admin_username, admin_password)
        assert resp.data, 'You were logged in'

    def test_logout(self):
        resp = self.login(admin_username, admin_password)
        resp = self.app.get('/logout', follow_redirects=True)
        assert resp.data, 'You were logged out'

    def test_register_user(self):
        username = fake.user_name()
        email = fake.email()
        password = fake.word() + fake.word()
        resp = self.register_user(username, email, password)
        assert resp.data, 'Sent verification email to %s' % email

    def test_edit_user(self):
        user = User.query.order_by(func.random()).first()
        resp = self.login(admin_username, admin_password)
        resp = self.edit_user(user, email=fake.email())
        assert resp.data, 'User %s edited' % user.username

    def test_delete_user(self):
        user = User.query.order_by(func.random()).first()
        resp = self.login(admin_username, admin_password)
        resp = self.delete_user(user.id)
        assert resp.data, 'User %s deleted' % user.username

    def test_user_list(self):
        resp = self.login(admin_username, admin_password)
        resp = self.app.get('/user/list', follow_redirects=True)
        assert resp.data, 'Users'