Thomas Peetz
39 lines
876 B
Python
39 lines
876 B
Python
from flask_testing import TestCase
|
|
from kontor import create_app
|
|
from kontor.models import User
|
|
|
|
|
|
class TestBase(TestCase):
|
|
|
|
def create_app(self):
|
|
|
|
# pass in test configuration
|
|
config_name = 'testing'
|
|
app = create_app(config_name)
|
|
return app
|
|
|
|
def setUp(self):
|
|
"""
|
|
Will be called before every test
|
|
"""
|
|
# create test admin user
|
|
admin = User()
|
|
admin.username="admin"
|
|
admin.password="admin2016"
|
|
admin.is_admin=True
|
|
admin.save()
|
|
|
|
# create test non-admin user
|
|
employee = User()
|
|
employee.username="test_user"
|
|
employee.password="test2016"
|
|
employee.save()
|
|
|
|
def tearDown(self):
|
|
"""
|
|
Will be called after every test
|
|
"""
|
|
users = User.objects.all()
|
|
for user in users:
|
|
user.delete()
|