Fixed wrong email login

githubstevemas · githubstevemas · commit 5b708aa32cb6 · 2024-07-15T20:36:58.000+02:00
diff --git a/server.py b/server.py
@@ -34,10 +34,14 @@ def index():
 
 @app.route('/showSummary', methods=['POST'])
 def showSummary():
-    club = [club for club in clubs if club['email'] == request.form['email']][
-        0]
-    return render_template('welcome.html', club=club,
-                           competitions=competitions)
+    try:
+        club = [club for club in clubs if club['email'] == request.form['email']][
+            0]
+        return render_template('welcome.html', club=club,
+                               competitions=competitions)
+    except IndexError:
+        flash('Wrong email-please try again')
+        return render_template('index.html')
 
 
 @app.route('/book/<competition>/<club>')
diff --git a/templates/index.html b/templates/index.html
@@ -5,12 +5,22 @@
     <title>GUDLFT Registration</title>
 </head>
 <body>
+
     <h1>Welcome to the GUDLFT Registration Portal!</h1>
+
     Please enter your secretary email to continue:
     <form action="showSummary" method="post">
         <label for="email">Email:</label>
         <input type="email" name="email" id=""/>
         <button type="submit">Enter</button>
     </form>
+
+    {% with messages = get_flashed_messages()%}
+        {% if messages %}
+           {% for message in messages %}
+                {{message}}
+            {% endfor %}
+        {% endif %}
+    {% endwith %}
 </body>
 </html>
diff --git a/tests/test_server.py b/tests/test_server.py
@@ -66,3 +66,20 @@ def test_purchase_places(client, test_clubs, test_competitions, mocker):
 
     assert int(test_clubs[0]['points']) == 4
     assert b'Great-booking complete!' in response.data
+
+
+def test_login(client):
+
+    response = client.post('/showSummary', data={
+        'email': 'john@simplylift.co'
+    })
+    assert response.status_code == 200
+
+
+def test_wrong_login(client):
+
+    response = client.post('/showSummary', data={
+        'email': 'wrong-email@test.com'
+    })
+    assert response.status_code == 200
+    assert b'Wrong email-please try again' in response.data
