import from kontor-flask

flask/tests/__init__.py

@@ -0,0 +1,38 @@
+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()