Add a battery of tests.

Antoine Goutenoir
1 parent ffdaeb0e
Showing 2 changed files with 55 additions and 0 deletions Show diff stats
tests/conftest.py
tests/test_models.py
@@ -0,0 +1,26 @@
+import pytest
+
+from flaskr import create_app
+from flaskr.models import db, User
+
+
+@pytest.fixture()
+def testapp(request):
+    app = create_app('flaskr.settings.TestConfig')
+    client = app.test_client()
+
+    db.app = app
+    db.create_all()
+
+    if getattr(request.module, "create_user", True):
+        admin = User('admin', 'supersafepassword')
+        db.session.add(admin)
+        db.session.commit()
+
+    def teardown():
+        db.session.remove()
+        db.drop_all()
+
+    request.addfinalizer(teardown)
+
+    return client
@@ -0,0 +1,29 @@
+#! ../env/bin/python
+# -*- coding: utf-8 -*-
+
+import pytest
+
+from flaskr.models import db, User
+
+create_user = False
+
+
+@pytest.mark.usefixtures("testapp")
+class TestModels:
+    def test_user_save(self, testapp):
+        """ Test Saving the user model to the database """
+
+        admin = User('admin', 'supersafepassword')
+        db.session.add(admin)
+        db.session.commit()
+
+        user = User.query.filter_by(username="admin").first()
+        assert user is not None
+
+    def test_user_password(self, testapp):
+        """ Test password hashing and checking """
+
+        admin = User('admin', 'supersafepassword')
+
+        assert admin.username == 'admin'
+        assert admin.check_password('supersafepassword')