From d58dea336c051d4a93e7ccf1a896a7fcbca874ee Mon Sep 17 00:00:00 2001 From: Feanil Patel Date: Mon, 23 Sep 2024 09:42:49 -0400 Subject: [PATCH] fixup! test: Update the course in the cache after it's got new content. --- .../blocks/transformers/tests/test_milestones.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/lms/djangoapps/course_api/blocks/transformers/tests/test_milestones.py b/lms/djangoapps/course_api/blocks/transformers/tests/test_milestones.py index 8c7d60e938..5efea79c19 100644 --- a/lms/djangoapps/course_api/blocks/transformers/tests/test_milestones.py +++ b/lms/djangoapps/course_api/blocks/transformers/tests/test_milestones.py @@ -13,6 +13,7 @@ from edx_toggles.toggles.testutils import override_waffle_flag from lms.djangoapps.course_blocks.api import get_course_blocks from lms.djangoapps.course_blocks.transformers.tests.helpers import CourseStructureTestCase from lms.djangoapps.gating import api as lms_gating_api +import openedx.core.djangoapps.content.block_structure.api as bs_api from openedx.core.djangoapps.content.block_structure.transformers import BlockStructureTransformers from openedx.core.djangoapps.course_apps.toggles import EXAMS_IDA from openedx.core.lib.gating import api as gating_api @@ -166,7 +167,11 @@ class MilestonesTransformerTestCase(CourseStructureTestCase, MilestonesTestCaseM self.course.enable_subsection_gating = True self.setup_gated_section(self.blocks[gated_block_ref], self.blocks[gating_block_ref]) - with self.assertNumQueries(14): + # Cache the course blocks so that they don't need to be generated when we're trying to + # get data back. This would happen as a part of publishing in a production system. + bs_api.update_course_in_cache(self.course.id) + + with self.assertNumQueries(4): self.get_blocks_and_check_against_expected(self.user, expected_blocks_before_completion) # clear the request cache to simulate a new request