feat: improved test coverage

Author: corp-0

src/tests/accounts/endpoints/test_confirm_account.py

@@ -0,0 +1,36 @@
+from django.urls import reverse
+from rest_framework import status
+from rest_framework.test import APITestCase
+
+from accounts.models import Account, AccountConfirmation
+
+
+class ConfirmAccountTest(APITestCase):
+    def setUp(self) -> None:
+        self.account = Account.objects.create_user(
+            username="confirmUser",
+            email="confirm@example.com",
+            unique_identifier="confirmUser",
+        )
+        self.account.set_password("aValidPss963")
+        self.account.is_confirmed = False
+        self.account.save()
+
+        self.confirmation = AccountConfirmation.objects.create(account=self.account, token="confirm-token")  # noqa: S106 - test-only credential
+        self.url = reverse("account:confirm")
+
+    def test_confirm_account_with_valid_token(self) -> None:
+        response = self.client.post(self.url, {"token": self.confirmation.token}, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        self.account.refresh_from_db()
+        self.assertTrue(self.account.is_confirmed)
+        self.assertFalse(AccountConfirmation.objects.filter(pk=self.confirmation.pk).exists())
+
+    def test_confirm_account_with_invalid_token(self) -> None:
+        response = self.client.post(self.url, {"token": "invalid-token"}, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.account.refresh_from_db()
+        self.assertFalse(self.account.is_confirmed)

src/tests/accounts/endpoints/test_login_credentials.py

@@ -1,3 +1,4 @@
+from django.core.cache import cache
 from django.urls import reverse
 from rest_framework import status
 from rest_framework.test import APITestCase
@@ -7,6 +8,7 @@
 
 class LoginCredentialsTest(APITestCase):
     def setUp(self):
+        cache.clear()
         self.url = reverse("account:login-credentials")
         self.valid_account = Account.objects.create_user(
             username="validUser",

src/tests/accounts/endpoints/test_login_token.py

@@ -1,3 +1,4 @@
+from django.core.cache import cache
 from django.urls import reverse
 from rest_framework import status
 from rest_framework.test import APITestCase
@@ -7,6 +8,7 @@
 
 class LoginTokenTest(APITestCase):
     def setUp(self):
+        cache.clear()
         self.url = reverse("account:login-token")
         self.valid_account = Account.objects.create_user(
             username="validUser",

src/tests/accounts/endpoints/test_register.py

@@ -1,3 +1,6 @@
+from unittest.mock import patch
+
+from django.core.cache import cache
 from django.urls import reverse
 from rest_framework import status
 from rest_framework.test import APITestCase
@@ -7,6 +10,7 @@
 
 class RegisterTest(APITestCase):
     def setUp(self):
+        cache.clear()
         self.url = reverse("account:register")
         self.valid_data = {
             "email": "validUser@valid.com",
@@ -101,3 +105,15 @@ def test_confirmed_state_at_registering(self):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         created_account = Account.objects.get(email=self.valid_data["email"])
         self.assertFalse(created_account.is_confirmed)
+
+    def test_register_sends_confirmation_email(self):
+        data = self.valid_data.copy()
+        data["email"] = "newuser@example.com"
+        data["unique_identifier"] = "newuser"
+        data["username"] = "newuser"
+
+        with patch("accounts.models.Account.send_confirmation_mail") as mock_send:
+            response = self.client.post(self.url, data, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        mock_send.assert_called_once()

src/tests/accounts/endpoints/test_reset_password.py

@@ -1,3 +1,6 @@
+from unittest.mock import patch
+
+from django.core.cache import cache
 from django.urls import reverse
 from rest_framework import status
 from rest_framework.test import APITestCase
@@ -11,6 +14,7 @@ def get_reset_url(token):
 
 class PasswordResetTest(APITestCase):
     def setUp(self):
+        cache.clear()
         self.valid_account = Account.objects.create_user(
             username="validUser",
             email="validUser@valid.com",
@@ -37,6 +41,16 @@ def test_request_password_reset_with_valid_email(self):
         password_reset_request = PasswordResetRequestModel.objects.first()
         self.assertIsNotNone(password_reset_request)
 
+    def test_request_password_reset_sends_email(self):
+        data = {"email": "validUser@valid.com"}
+
+        with patch("accounts.api.views.send_email_with_template") as mock_send:
+            response = self.client.post(self.url_request, data, format="json")
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        mock_send.assert_called_once()
+        self.assertEqual(mock_send.call_args.kwargs["recipient"], data["email"])
+
     def test_request_password_reset_with_invalid_email(self):
         data = {"email": "invalid@mail.com"}
 

src/tests/accounts/endpoints/test_server_verification_token.py

@@ -1,3 +1,4 @@
+from django.core.cache import cache
 from django.urls import reverse
 from rest_framework import status
 from rest_framework.test import APITestCase
@@ -7,6 +8,7 @@
 
 class ServerVerificationTokenTest(APITestCase):
     def setUp(self):
+        cache.clear()
         self.valid_account = Account.objects.create_user(
             username="validUser",
             email="validUser@valid.com",

src/tests/baby_serverlist/test_api.py

@@ -1,4 +1,5 @@
-from datetime import UTC, datetime
+from datetime import UTC, datetime, timedelta
+from uuid import uuid4
 
 from django.core.cache import cache
 from django.test import override_settings
@@ -102,6 +103,17 @@ def test_regenerate_token_rejects_non_owner(self) -> None:
 
         self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
 
+    def test_regenerate_token_not_found_returns_404(self) -> None:
+        self.client.force_authenticate(self.user)
+
+        response = self.client.post(
+            reverse("baby_serverlist:regenerate-token"),
+            {"server_id": str(uuid4())},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
+
     def test_post_server_status_stores_payload_in_cache(self) -> None:
         baby_server = BabyServer.objects.create(owner=self.user, whitelisted=True)
         payload = _sample_status_payload(baby_server.serverlist_token)
@@ -151,6 +163,12 @@ def test_list_owned_baby_servers_live_flag(self) -> None:
         response = self.client.get(reverse("baby_serverlist:list-owned"))
         self.assertTrue(response.json()[0]["live"])
 
+        stale_time = datetime.now(tz=UTC) - timedelta(seconds=13)
+        set_baby_server_heartbeat(str(baby_server.id), stale_time.isoformat())
+
+        response = self.client.get(reverse("baby_serverlist:list-owned"))
+        self.assertFalse(response.json()[0]["live"])
+
     def test_list_baby_servers_returns_whitelisted_status(self) -> None:
         baby_server = BabyServer.objects.create(owner=self.user, whitelisted=True)
         status_data = _sample_status_payload(baby_server.serverlist_token)