diff --git a/lms/djangoapps/instructor_task/tests/test_tasks_helper.py b/lms/djangoapps/instructor_task/tests/test_tasks_helper.py index 428d6ddc2d..0a8c5e788b 100644 --- a/lms/djangoapps/instructor_task/tests/test_tasks_helper.py +++ b/lms/djangoapps/instructor_task/tests/test_tasks_helper.py @@ -25,6 +25,7 @@ from freezegun import freeze_time from pytz import UTC import openedx.core.djangoapps.user_api.course_tag.api as course_tag_api +import openedx.core.djangoapps.content.block_structure.api as bs_api from xmodule.capa.tests.response_xml_factory import MultipleChoiceResponseXMLFactory # lint-amnesty, pylint: disable=wrong-import-order from common.djangoapps.course_modes.models import CourseMode from common.djangoapps.student.models import CourseEnrollment, CourseEnrollmentAllowed @@ -396,6 +397,8 @@ class TestInstructorGradeReport(InstructorGradeReportTestCase): ) _ = CreditCourseFactory(course_key=course.id) + bs_api.update_course_in_cache(course.id) + num_users = 5 for _ in range(num_users): user = UserFactory.create() @@ -406,7 +409,7 @@ class TestInstructorGradeReport(InstructorGradeReportTestCase): with patch('lms.djangoapps.instructor_task.tasks_helper.runner._get_current_task'): with check_mongo_calls(2): - with self.assertNumQueries(63): + with self.assertNumQueries(46): CourseGradeReport.generate(None, None, course.id, {}, 'graded') def test_inactive_enrollments(self):