test: add test case

edx_exams/apps/core/management/commands/test/test_bulk_add_course_staff.py

@@ -93,6 +93,15 @@ def test_add_course_staff_with_not_default_batch_size(self):
             with self.assertNumQueries(8):
                 call_command(self.command, f'--csv_path={csv.name}', '--batch_size=1')
 
+    def test_add_course_staff_with_batch_size_larger_than_list(self):
+        """ Assert that the number of queries is correct given batch size larger than lines """
+        lines = ['pam,[email protected],staff,course-v1:edx+test+f20\n',
+                 'sam,[email protected],staff,course-v1:edx+test+f20\n']
+        with NamedTemporaryFile() as csv:
+            csv = self._write_test_csv(csv, lines)
+            with self.assertNumQueries(6):
+                call_command(self.command, f'--csv_path={csv.name}', '--batch_size=3')
+
     def test_add_course_staff_with_not_default_batch_delay(self):
         username, email = 'pam', '[email protected]'
         username2, email2 = 'cam', '[email protected]'
@@ -106,8 +115,8 @@ def test_add_course_staff_with_not_default_batch_delay(self):
 
     def test_num_queries_correct(self):
         """
-        Assert the number of queries to be 5 + 1 * number of lines:
-        2 for savepoint/release savepoint, 1 to get existing usernames,
+        Assert the number of queries to be 4 + 1 * number of lines:
+        2 for savepoint/release savepoint
         1 to bulk create users, 1 to bulk create course role
         1 for each user (to get user)
         """