flask testing

Author: mjhea0

manage.py

@@ -19,7 +19,7 @@
 @manager.command
 def test():
     """Runs the unit tests without coverage."""
-    tests = unittest.TestLoader().discover('.')
+    tests = unittest.TestLoader().discover('tests')
     unittest.TextTestRunner(verbosity=2).run(tests)
 
 
@@ -28,7 +28,7 @@ def cov():
     """Runs the unit tests with coverage."""
     cov = coverage.coverage(branch=True, include='project/*')
     cov.start()
-    tests = unittest.TestLoader().discover('.')
+    tests = unittest.TestLoader().discover('tests')
     unittest.TextTestRunner(verbosity=2).run(tests)
     cov.stop()
     cov.save()

project/models.py

@@ -51,4 +51,4 @@ def get_id(self):
         return unicode(self.id)
 
     def __repr__(self):
-        return '<name {}'.format(self.name)
+        return '<name - {}>'.format(self.name)

readme.md

@@ -48,6 +48,7 @@ Please help us keep this free, open source project going. Purchase the [Real Pyt
 | 25   | [User Registration (functionality and unit tests)](http://youtu.be/kt4PEa5tsVw) | [part25](https://github.com/realpython/discover-flask/tree/part25) |
 | 26   | [Finalize Messaging System](http://youtu.be/WnT188ePHg4) | [part26](https://github.com/realpython/discover-flask/tree/part26) |
 | 27   | [Test Coverage with coverage.py](http://youtu.be/7Aqcn0-uAr0) | [part27](https://github.com/realpython/discover-flask/tree/part27) |
+| 28   | Flask Testing | Coming Soon! |
 
 **You can view the entire video playlist [here](http://www.youtube.com/watch?v=WfpFUmV1d0w&list=PLLjmbh6XPGK4ISY747FUHXEl9lBxre4mM&feature=share).**
 

tests/__init__.py

@@ -0,0 +1,23 @@
+from flask.ext.testing import TestCase
+
+from project import app, db
+from project.models import User, BlogPost
+
+
+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(User("admin", "ad@min.com", "admin"))
+        db.session.add(
+            BlogPost("Test post", "This is a test. Only a test.", "admin"))
+        db.session.commit()
+
+    def tearDown(self):
+        db.session.remove()
+        db.drop_all()

tests/base.py

@@ -0,0 +1 @@
+# tests/test_forms.py

tests/test_forms.py

@@ -1,29 +1,10 @@
+# tests/test_functional.py
+
 import unittest
 
-from flask.ext.testing import TestCase
 from flask.ext.login import current_user
 
-from project import app, db
-from project.models import User, BlogPost
-
-
-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(User("admin", "ad@min.com", "admin"))
-        db.session.add(
-            BlogPost("Test post", "This is a test. Only a test.", "admin"))
-        db.session.commit()
-
-    def tearDown(self):
-        db.session.remove()
-        db.drop_all()
+from base import BaseTestCase
 
 
 class FlaskTestCase(BaseTestCase):
@@ -98,17 +79,6 @@ def test_logout_route_requires_login(self):
         response = self.client.get('/logout', follow_redirects=True)
         self.assertIn(b'Please log in to access this page', response.data)
 
-    # Ensure user can register
-    def test_user_registeration(self):
-        with self.client:
-            response = self.client.post('register/', data=dict(
-                username='Michael', email='michael@realpython.com',
-                password='python', confirm='python'
-            ), follow_redirects=True)
-            self.assertIn(b'Welcome to Flask!', response.data)
-            self.assertTrue(current_user.name == "Michael")
-            self.assertTrue(current_user.is_active())
-
 
 if __name__ == '__main__':
     unittest.main()

tests.py renamed to tests/test_functional.py

@@ -0,0 +1,45 @@
+# tests/test_models.py
+
+
+import unittest
+
+from flask.ext.login import current_user
+
+from base import BaseTestCase
+from project import bcrypt
+from project.models import User
+
+
+class TestUser(BaseTestCase):
+
+    # Ensure user can register
+    def test_user_registeration(self):
+        with self.client:
+            response = self.client.post('register/', data=dict(
+                username='Michael', email='michael@realpython.com',
+                password='python', confirm='python'
+            ), follow_redirects=True)
+            self.assertIn(b'Welcome to Flask!', response.data)
+            self.assertTrue(current_user.name == "Michael")
+            self.assertTrue(current_user.is_active())
+            user = User.query.filter_by(email='michael@realpython.com').first()
+            self.assertTrue(str(user) == '<name - Michael>')
+
+    def test_get_by_id(self):
+        # Ensure id is correct for the current/logged in user
+        with self.client:
+            self.client.post('/login', data=dict(
+                username="admin", password='admin'
+            ), follow_redirects=True)
+            self.assertTrue(current_user.id == 1)
+            self.assertFalse(current_user.id == 20)
+
+    def test_check_password(self):
+        # Ensure given password is correct after unhashing
+        user = User.query.filter_by(email='ad@min.com').first()
+        self.assertTrue(bcrypt.check_password_hash(user.password, 'admin'))
+        self.assertFalse(bcrypt.check_password_hash(user.password, 'foobar'))
+
+
+if __name__ == '__main__':
+    unittest.main()