test: add batch test case

edx · Jul 30, 2024 · ef6be73 · ef6be73
1 parent fa873c1
commit ef6be73
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py b/edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py
@@ -102,6 +102,16 @@ def test_add_course_staff_with_batch_size_larger_than_list(self):
             with self.assertNumQueries(6):
                 call_command(self.command, f'--csv_path={csv.name}', '--batch_size=3')
 
+    def test_add_course_staff_with_batch_size_smaller_than_list(self):
+        """ Assert that the number of queries is correct given batch size smaller than lines """
+        lines = ['pam,[email protected],staff,course-v1:edx+test+f20\n',
+                 'sam,[email protected],staff,course-v1:edx+test+f20\n'
+                 'tam,[email protected],staff,course-v1:edx+test+f20\n']
+        with NamedTemporaryFile() as csv:
+            csv = self._write_test_csv(csv, lines)
+            with self.assertNumQueries(9):
+                call_command(self.command, f'--csv_path={csv.name}', '--batch_size=2')
+
     def test_add_course_staff_with_not_default_batch_delay(self):
         username, email = 'pam', '[email protected]'
         username2, email2 = 'cam', '[email protected]'