minor fix

Author: brandonistfan

tcf_website/management/commands/fetch_enrollment.py

@@ -30,7 +30,7 @@
 # Maximum number of retry attempts
 MAX_TRIES = 8
 # Kept larger than MAX_WORKERS to handle connection lifecycle issues
-MAX_POOL_SIZE = 20 * 5
+MAX_POOL_SIZE = 5 * 5
 
 # Thread-local storage for requests sessions to avoid contention
 _session_local = threading.local()

tcf_website/tests/test_enrollment.py

@@ -1,5 +1,6 @@
 """Tests for enrollment data updates."""
 
+import asyncio
 from unittest.mock import patch
 
 from django.test import TestCase
@@ -35,7 +36,7 @@ def test_mocked_update_enrollment_data(self, mock_fetch):
             "waitlist_taken": 5,
             "waitlist_limit": 10,
         }
-        update_enrollment_data(self.section.course.id)
+        asyncio.run(update_enrollment_data(self.section.course.id))
         self.enrollment.refresh_from_db()
         self.assertEqual(self.enrollment.enrollment_taken, 10)
         self.assertEqual(self.enrollment.enrollment_limit, 20)
@@ -44,7 +45,7 @@ def test_mocked_update_enrollment_data(self, mock_fetch):
 
     def test_update_enrollment_data(self):
         """Test enrollment data update with real data."""
-        update_enrollment_data(self.section.course.id)
+        asyncio.run(update_enrollment_data(self.section.course.id))
         self.enrollment.refresh_from_db()
         self.assertGreaterEqual(self.enrollment.enrollment_taken, 0)
         self.assertGreaterEqual(self.enrollment.enrollment_limit, 0)