9cf2f9f298
This will remove imports from __future__ that are no longer needed. https://docs.python.org/3.5/library/2to3.html#2to3fixer-future
31 lines
933 B
Python
31 lines
933 B
Python
"""
|
|
Provides helper functions for tests that want
|
|
to configure flags related to persistent grading.
|
|
"""
|
|
|
|
|
|
from contextlib import contextmanager
|
|
|
|
from edx_django_utils.cache import RequestCache
|
|
|
|
from lms.djangoapps.grades.config.models import CoursePersistentGradesFlag, PersistentGradesEnabledFlag
|
|
|
|
|
|
@contextmanager
|
|
def persistent_grades_feature_flags(
|
|
global_flag,
|
|
enabled_for_all_courses=False,
|
|
course_id=None,
|
|
enabled_for_course=False
|
|
):
|
|
"""
|
|
Most test cases will use a single call to this manager,
|
|
as they need to set the global setting and the course-specific
|
|
setting for a single course.
|
|
"""
|
|
RequestCache.clear_all_namespaces()
|
|
PersistentGradesEnabledFlag.objects.create(enabled=global_flag, enabled_for_all_courses=enabled_for_all_courses)
|
|
if course_id:
|
|
CoursePersistentGradesFlag.objects.create(course_id=course_id, enabled=enabled_for_course)
|
|
yield
|