Implement tests

src/tests/settings.py

@@ -0,0 +1,21 @@
+"""
+Minimal Django settings file for tests
+"""
+
+from core.settings import *  # noqa: F403
+
+
+# Override database to SQLite
+DATABASES = {
+    "default": {
+        "ENGINE": "django.db.backends.sqlite3",
+        "NAME": BASE_DIR / "test_db.sqlite3",  # noqa: F405
+    }
+}
+
+# Use dummy cache
+CACHES = {
+    "default": {
+        "BACKEND": "django.core.cache.backends.dummy.DummyCache",
+    }
+}

src/tests/users/__init__.py

@@ -12,36 +12,61 @@ from core.settings import config, TESTS_DIR
 logger = logging.getLogger(__name__)
 
 
+def get_users_json():
+    """
+    Function to read test user data from JSON file
+    """
+    with open(os.path.join(TESTS_DIR, "users", "users.json"), "r") as f:
+        # Load JSON data
+        data = json.loads(f.read())
+        return data
+
+
 def generate_test_users():
     """
-    Function to create test users in DEBUG mode.
+    Function to create test users.
+    """
+    data = get_users_json()
+
+    for user in data["users"]:
+        # Check if user already exists
+        USER = CustomUser.objects.filter(username=user["username"]).first()
+        if not USER:
+            # Create user
+            if user["is_superuser"]:
+                USER = CustomUser.objects.create_superuser(
+                    username=user["username"],
+                    email=user["email"],
+                    password=config.DEBUG_USER_PASSWORD,
+                )
+                logger.info("Created Superuser:", user["username"])
+            else:
+                USER = CustomUser.objects.create_user(
+                    username=user["username"],
+                    email=user["email"],
+                    password=config.DEBUG_USER_PASSWORD,
+                )
+                logger.info("Created User:", user["username"])
+
+            # Additional user fields not covered by create() methods
+            USER.first_name = user["first_name"]
+            USER.last_name = user["last_name"]
+            USER.is_active = True
+            USER.save()
+
+
+def remove_test_users():
+    """
+    Function to remove test users in DEBUG mode.
     """
     if config.DEBUG:
-        with open(os.path.join(TESTS_DIR, "users", "users.json"), "r") as f:
-            # Load JSON data
-            data = json.loads(f.read())
-            for user in data["users"]:
-                # Check if user already exists
-                USER = CustomUser.objects.filter(email=user["email"]).first()
-                if not USER:
-                    # Create user
-                    if user["is_superuser"]:
-                        USER = CustomUser.objects.create_superuser(
-                            username=user["username"],
-                            email=user["email"],
-                            password=config.DEBUG_USER_PASSWORD,
-                        )
-                        print("Created Superuser:", user["email"])
-                    else:
-                        USER = CustomUser.objects.create_user(
-                            username=user["email"],
-                            email=user["email"],
-                            password=config.DEBUG_USER_PASSWORD,
-                        )
-                        print("Created User:", user["email"])
-
-                    # Additional user fields not covered by create() methods
-                    USER.first_name = user["first_name"]
-                    USER.last_name = user["last_name"]
-                    USER.is_active = True
-                    USER.save()
+        data = get_users_json()
+        for user in data["users"]:
+            # Check if user already exists
+            USER = CustomUser.objects.filter(username=user["username"]).first()
+            if USER:
+                USER.delete()
+            else:
+                logger.warning(
+                    f"Skipping user deletion for {user['username']}: Does not exist"
+                )

src/tests/users/test_user_creation_deletion.py

@@ -0,0 +1,38 @@
+import pytest
+import users
+
+from accounts.models import CustomUser
+
+
+def assert_users_created():
+    data = users.get_users_json()
+
+    for user in data["users"]:
+        USER = CustomUser.objects.filter(username=user["username"]).first()
+
+        # Assert user exists
+        assert USER
+
+        if user["is_superuser"]:
+            # Assert is superuser
+            assert USER.is_superuser
+
+
+def assert_users_removed():
+    data = users.get_users_json()
+    for user in data["users"]:
+        USER = CustomUser.objects.filter(username=user["username"]).first()
+
+        # Assert user does not exist
+        assert not USER
+
+
+@pytest.mark.django_db(transaction=True)
+def test_user_creation_deletion():
+    """
+    Test user creation and deletion
+    """
+    users.generate_test_users()
+    assert_users_created()
+    users.remove_test_users()
+    assert_users_removed()

src/tests/users/test_user_login_query_self.py

@@ -0,0 +1,38 @@
+import pytest
+from core.settings import config
+import users
+
+
+from rest_framework.test import APIClient
+
+client = APIClient()
+
+
+@pytest.mark.django_db(transaction=True)
+def test_user_login():
+    """
+    Test login
+    """
+
+    data = users.get_users_json()
+
+    # Generate test users
+    users.generate_test_users()
+
+    for user in data["users"]:
+        login_response = client.post(
+            "/api/v1/accounts/jwt/create/",
+            {"username": user["username"], "password": config.DEBUG_USER_PASSWORD},
+            format="json",
+        ).json()
+
+        # Check if login contains JWT data
+        assert {"access", "refresh"}.issubset(login_response)
+
+        access_token = login_response["access"]
+
+        client.credentials(HTTP_AUTHORIZATION=f"Bearer {access_token}")
+        # GET user info
+        user_response = client.get("/api/v1/accounts/users/me/")
+
+        assert user_response.json()  # Ensure there's response data

src/tests/users/users.json

@@ -16,7 +16,7 @@
         },
         {
             "username": "testuser2",
-            "email": "user1@test.com",
+            "email": "user2@test.com",
             "is_superuser": false,
             "first_name": "Test",
             "last_name": "User 2"