test: update user check

edx · Mar 28, 2024 · 37daf7a · 37daf7a
1 parent 6a20932
commit 37daf7a
Showing 1 changed file with 2 additions and 2 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
@@ -61,10 +61,10 @@ def test_add_course_staff_with_new_user(self):
         with NamedTemporaryFile() as csv:
             csv = self._write_test_csv(csv, lines)
             call_command(self.command, f'--csv_path={csv.name}')
-            user = User.objects.filter(username=username, email=email)[0]
+            user = User.objects.filter(username=username, email=email)
             assert user.exists()
             assert CourseStaffRole.objects.filter(
-                user=user.id,
+                user=user[0].id,
                 course_id=self.course_id,
                 role=self.course_role,
             ).exists()