Add new methods and data for testing the files

backend/api/tests/data.py

@@ -1,3 +1,10 @@
+super_user = {
+    "name": "Super User",
+    "email": "superu@email.com",
+    "type": "super",
+    "password": "testing1234"
+}
+
 test_user_1 = {
     "name": "John Doe",
     "email": "johndoe@email.com",

backend/api/tests/setup.py

@@ -2,6 +2,7 @@
 from api.auth.models import User, UserSchema
 from api.reminder.models import Contact
 from api.tests.data import (
+    super_user,
     test_user_1,
     test_user_2
 )
@@ -16,6 +17,25 @@ def reset_db():
     Contact.query.delete()
     db.session.commit()
 
+def create_super_user(client):
+    # Create super user
+    client.post(
+        "/api/auth/users",
+        json=super_user
+    )
+
+    super = User.find_by_email(super_user['email'])
+
+    # Login the super user
+    response = client.post(
+        "/api/auth/login",
+        json={"email": super_user['email'], "password": super_user['password']}
+    )
+
+    return response.json
+
+
+
 def create_one_test_user(client):
     # Create first user
     client.post(

backend/api/tests/test_auth.py

@@ -4,6 +4,7 @@
 )
 from api.tests.setup import (
     reset_db,
+    create_super_user,
     create_one_test_user,
     create_two_test_users
 )
@@ -31,20 +32,41 @@ def test_auth_create_new_user(client):
     # Reset the database after running all tests
     reset_db()
 
+def test_auth_login_user(client):
+
+    # Reset the database before the test begins
+    reset_db()
+
+    # Create a single user
+    create_one_test_user(client)
+
+    response = client.post(
+        "/api/auth/login",
+        json={"email": test_user_1['email'], "password": test_user_1['password']}
+    )
+
+    assert response.status_code == 200
+    assert response.json['token']
+    assert response.json['user_type'] == 'normal'
+
 def test_auth_get_all_users(client):
     # Reset the database before the test begins
     reset_db()
 
+    # Create super user
+    super = create_super_user(client)
+
     create_two_test_users(client)
 
     response = client.get(
-        "/api/auth/users"
+        "/api/auth/users",
+        headers={"Authorization": f"Bearer {super['token']}"}
     )
 
     assert response.status_code == 200
-    assert len(response.json) == 2
-    assert response.json[0]['email'] == test_user_1['email']
-    assert response.json[1]['email'] == test_user_2['email']
+    assert len(response.json) == 3
+    assert response.json[1]['email'] == test_user_1['email']
+    assert response.json[2]['email'] == test_user_2['email']
 
     # Reset the database before the test begins
     reset_db()
@@ -54,10 +76,14 @@ def test_auth_get_user_by_id(client):
     # Reset the database before the test begins
     reset_db()
 
+    # Create super user
+    super = create_super_user(client)
+    print(super)
     user = create_one_test_user(client)
 
     response = client.get(
-        f"/api/auth/users/{user['id']}"
+        f"/api/auth/users/{user['id']}",
+        headers={"Authorization": f"Bearer {super['token']}"}
     )
 
     assert response.status_code == 200