Deprecate WaffleSwitch.override* methods
This allows us to get rid of the custom WaffleSwitch and WaffleSwitchNamespace classes from waffle_utils in favour of edx_toggles.toggles classes.
This commit is contained in:
@@ -18,13 +18,14 @@ from django.utils.timezone import now
|
||||
from mock import Mock
|
||||
from pytz import UTC
|
||||
|
||||
from student.admin import AllowedAuthUserForm, COURSE_ENROLLMENT_ADMIN_SWITCH, UserAdmin, CourseEnrollmentForm
|
||||
from openedx.core.djangoapps.content.course_overviews.tests.factories import CourseOverviewFactory
|
||||
from openedx.core.djangoapps.site_configuration.tests.mixins import SiteMixin
|
||||
from openedx.core.djangoapps.waffle_utils.testutils import override_waffle_switch
|
||||
from student.admin import COURSE_ENROLLMENT_ADMIN_SWITCH, AllowedAuthUserForm, CourseEnrollmentForm, UserAdmin
|
||||
from student.models import AllowedAuthUser, CourseEnrollment, LoginFailures
|
||||
from student.tests.factories import CourseEnrollmentFactory, UserFactory
|
||||
from xmodule.modulestore.tests.django_utils import SharedModuleStoreTestCase
|
||||
from xmodule.modulestore.tests.factories import CourseFactory
|
||||
from openedx.core.djangoapps.content.course_overviews.tests.factories import CourseOverviewFactory
|
||||
from openedx.core.djangoapps.site_configuration.tests.mixins import SiteMixin
|
||||
|
||||
|
||||
class AdminCourseRolesPageTest(SharedModuleStoreTestCase):
|
||||
@@ -243,7 +244,7 @@ class CourseEnrollmentAdminTest(SharedModuleStoreTestCase):
|
||||
"""
|
||||
Ensure CourseEnrollmentAdmin views can be enabled with the waffle switch.
|
||||
"""
|
||||
with COURSE_ENROLLMENT_ADMIN_SWITCH.override(active=True):
|
||||
with override_waffle_switch(COURSE_ENROLLMENT_ADMIN_SWITCH, active=True):
|
||||
response = getattr(self.client, method)(url)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
|
||||
@@ -257,7 +258,7 @@ class CourseEnrollmentAdminTest(SharedModuleStoreTestCase):
|
||||
course_id=self.course.id, # pylint: disable=no-member
|
||||
)
|
||||
search_url = '{}?q={}'.format(reverse('admin:student_courseenrollment_changelist'), self.user.username)
|
||||
with COURSE_ENROLLMENT_ADMIN_SWITCH.override(active=True):
|
||||
with override_waffle_switch(COURSE_ENROLLMENT_ADMIN_SWITCH, active=True):
|
||||
response = self.client.get(search_url)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
|
||||
@@ -283,7 +284,7 @@ class CourseEnrollmentAdminTest(SharedModuleStoreTestCase):
|
||||
'mode': self.course_enrollment.mode,
|
||||
}
|
||||
|
||||
with COURSE_ENROLLMENT_ADMIN_SWITCH.override(active=True):
|
||||
with override_waffle_switch(COURSE_ENROLLMENT_ADMIN_SWITCH, active=True):
|
||||
response = self.client.post(
|
||||
reverse('admin:student_courseenrollment_change', args=(self.course_enrollment.id, )),
|
||||
data=data,
|
||||
@@ -304,7 +305,7 @@ class CourseEnrollmentAdminTest(SharedModuleStoreTestCase):
|
||||
'mode': self.course_enrollment.mode,
|
||||
}
|
||||
|
||||
with COURSE_ENROLLMENT_ADMIN_SWITCH.override(active=True):
|
||||
with override_waffle_switch(COURSE_ENROLLMENT_ADMIN_SWITCH, active=True):
|
||||
with self.assertRaises(ValidationError):
|
||||
self.client.post(
|
||||
reverse('admin:student_courseenrollment_change', args=(self.course_enrollment.id, )),
|
||||
|
||||
Reference in New Issue
Block a user