flask-testing

Author: mjhea0

config.py

@@ -10,6 +10,13 @@ class BaseConfig(object):
     print SQLALCHEMY_DATABASE_URI
 
 
+class TestConfig(BaseConfig):
+    DEBUG = True
+    TESTING = True
+    WTF_CSRF_ENABLED = False
+    SQLALCHEMY_DATABASE_URI = 'sqlite:///:memory:'
+
+
 class DevelopmentConfig(BaseConfig):
     DEBUG = True
 

requirements.txt

@@ -3,6 +3,7 @@ Flask-Bcrypt==0.6.0
 Flask-Migrate==1.2.0
 Flask-SQLAlchemy==1.0
 Flask-Script==2.0.5
+Flask-Testing==0.4.2
 Flask-WTF==0.10.2
 Jinja2==2.7.3
 Mako==1.0.0

tests.py

@@ -1,25 +1,43 @@
-from project import app
 import unittest
+from flask.ext.testing import TestCase
+from project import app, db
+from project.models import User, BlogPost
 
 
-class FlaskTestCase(unittest.TestCase):
+class BaseTestCase(TestCase):
+    """A base test case."""
+
+    def create_app(self):
+        app.config.from_object('config.TestConfig')
+        return app
+
+    def setUp(self):
+        db.create_all()
+        db.session.add(BlogPost("Test post", "This is a test. Only a test."))
+        db.session.add(User("admin", "ad@min.com", "admin"))
+        db.session.commit()
+
+    def tearDown(self):
+        db.session.remove()
+        db.drop_all()
+
+
+class FlaskTestCase(BaseTestCase):
 
     # Ensure that Flask was set up correctly
     def test_index(self):
-        tester = app.test_client(self)
-        response = tester.get('/login', content_type='html/text')
+        response = self.client.get('/login', content_type='html/text')
         self.assertEqual(response.status_code, 200)
 
     # Ensure that the login page loads correctly
     def test_login_page_loads(self):
-        tester = app.test_client(self)
-        response = tester.get('/login')
+        response = self.client.get('/login')
         self.assertIn(b'Please login', response.data)
 
     # Ensure login behaves correctly with correct credentials
     def test_correct_login(self):
-        tester = app.test_client()
-        response = tester.post(
+
+        response = self.client.post(
             '/login',
             data=dict(username="admin", password="admin"),
             follow_redirects=True
@@ -28,46 +46,41 @@ def test_correct_login(self):
 
     # Ensure login behaves correctly with incorrect credentials
     def test_incorrect_login(self):
-        tester = app.test_client()
-        response = tester.post(
+        response = self.client.post(
             '/login',
             data=dict(username="wrong", password="wrong"),
             follow_redirects=True
         )
-        self.assertIn(b'Invalid Credentials. Please try again.', response.data)
+        self.assertIn(b'Invalid username or password.', response.data)
 
     # Ensure logout behaves correctly
     def test_logout(self):
-        tester = app.test_client()
-        tester.post(
+        self.client.post(
             '/login',
             data=dict(username="admin", password="admin"),
             follow_redirects=True
         )
-        response = tester.get('/logout', follow_redirects=True)
+        response = self.client.get('/logout', follow_redirects=True)
         self.assertIn(b'You were logged out', response.data)
 
     # Ensure that main page requires user login
     def test_main_route_requires_login(self):
-        tester = app.test_client()
-        response = tester.get('/', follow_redirects=True)
+        response = self.client.get('/', follow_redirects=True)
         self.assertIn(b'You need to login first.', response.data)
 
     # Ensure that logout page requires user login
     def test_logout_route_requires_login(self):
-        tester = app.test_client()
-        response = tester.get('/logout', follow_redirects=True)
+        response = self.client.get('/logout', follow_redirects=True)
         self.assertIn(b'You need to login first.', response.data)
 
     # Ensure that posts show up on the main page
     def test_posts_show_up_on_main_page(self):
-        tester = app.test_client()
-        response = tester.post(
+        response = self.client.post(
             '/login',
             data=dict(username="admin", password="admin"),
             follow_redirects=True
         )
-        self.assertIn(b'Hello from the shell', response.data)
+        self.assertIn(b'This is a test. Only a test.', response.data)
 
 
 if __name__ == '__main__':