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()