Add tests for logging in

tests/helpers.py

@@ -40,3 +40,12 @@ def client():
 
     with calsocial.app.app_context():
         db.drop_all()
+
+
+def login(client, username, password, no_redirect=False):
+    """Login with the specified username and password
+    """
+
+    return client.post('/login',
+                       data={'email': username, 'password': password},
+                       follow_redirects=not no_redirect)

tests/test_login.py

@@ -0,0 +1,84 @@
+# Calendar.social
+# Copyright (C) 2018  Gergely Polonkai
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+"""General tests for Calendar.social
+"""
+
+import calsocial
+from calsocial.models import db, User
+
+from helpers import client, login
+
+
+def test_index_no_login(client):
+    """Test the main page without logging in
+    """
+
+    page = client.get('/')
+    assert b'Welcome to Calendar.social' in page.data
+
+
+def test_login_invalid_user(client):
+    """Test logging in with a non-existing user
+    """
+
+    page = login(client, 'username', 'password')
+    assert b'Specified user does not exist' in page.data
+
+
+def test_login_bad_password(client):
+    """Test logging in with a bad password
+    """
+
+    with calsocial.app.app_context():
+        user = User(username='test', email='test@example.com', password='password')
+        db.session.add(user)
+        db.session.commit()
+
+    page = login(client, 'test', password='else')
+    assert b'Invalid password' in page.data
+
+
+def test_login_email(client):
+    """Test logging in with the email address instead of the username
+    """
+
+    with calsocial.app.app_context():
+        user = User(username='test', email='test@example.com', password='password', active=True)
+        db.session.add(user)
+        db.session.commit()
+
+    page = login(client, 'test@example.com', password='password')
+    assert b'Logged in as ' in page.data
+
+
+def test_login_first_steps(client):
+    """Test logging in with a new user
+
+    They must be redirected to the first login page
+    """
+
+    with calsocial.app.app_context():
+        user = User(username='test', email='test@example.com', password='password', active=True)
+        db.session.add(user)
+        db.session.commit()
+
+    page = login(client, 'test', password='password', no_redirect=True)
+    # First, we must be redirected to the main page
+    assert page.location == 'http://localhost/'
+
+    page = client.get('/')
+    assert page.location == 'http://localhost/first-steps'