use decorator for waffle override flag

openedx-unsupported · Apr 9, 2020 · 5002293 · 5002293
1 parent 4f14967
commit 5002293
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/registrar/apps/api/v1/tests/test_views.py b/registrar/apps/api/v1/tests/test_views.py
@@ -1759,6 +1759,7 @@ def test_successful_update_course_staff_with_program_group_waffle(self, use_exte
     @mock_oauth_login
     @responses.activate
     @ddt.data(False, True)
+    @override_flag('enable_course_role_management', active=False)
     def test_failed_program_course_enrollment_write_with_course_staff(self, use_external_course_key):
         course_id = self.external_course_key if use_external_course_key else self.course_id
 
@@ -1774,7 +1775,6 @@ def test_failed_program_course_enrollment_write_with_course_staff(self, use_exte
                 course_id=course_id,
                 status_code=403,
         ):
-            with override_flag('enable_course_role_management', active=False):
                 response = self.request(
                     self.method, self.get_url(course_id=course_id), self.stem_admin, req_data
                 )