fix mock

backend/api/tests/test_views.py

@@ -1,8 +1,9 @@
-from unittest.mock import Mock, patch
+from unittest.mock import patch
 
 import pyotp
 from codeforlife.tests import CronTestCase
 from codeforlife.user.models import AuthFactor, User
+from django.core import management
 from django.test import TestCase
 from django.urls import reverse
 from django.utils import timezone
@@ -48,8 +49,7 @@ def test_post__otp(self):
 
 
 class TestClearExpiredView(CronTestCase):
-    @patch("django.core.management.call_command")
-    def test_clear_expired_view(self, call_command: Mock):
-        self.client.get(reverse("clear-expired-sessions"))
-
-        call_command.assert_called_once_with("clearsessions")
+    def test_clear_expired_view(self):
+        with patch.object(management, "call_command") as call_command:
+            self.client.get(reverse("clear-expired-sessions"))
+            call_command.assert_called_once_with("clearsessions")

backend/api/views.py

@@ -6,7 +6,7 @@
 from django.contrib.auth import login
 from django.contrib.auth.views import LoginView as _LoginView
 from django.contrib.sessions.models import Session, SessionManager
-from django.core.management import call_command
+from django.core import management
 from django.http import JsonResponse
 from rest_framework import status
 from rest_framework.response import Response
@@ -82,7 +82,7 @@ def get(self, request):
 
         # Clears expired sessions.
         # https://docs.djangoproject.com/en/3.2/ref/django-admin/#clearsessions
-        call_command("clearsessions")
+        management.call_command("clearsessions")
 
         after_session_count = session_objects.count()
         logging.info(f"Session count after clearance: {after_session_count}")