feat: added enclosing brackets in context manager with multiple contexts

mitodl · Mar 26, 2024 · 673b028 · 673b028
1 parent 435fe3b
commit 673b028
Showing 1 changed file with 20 additions and 10 deletions.
diff --git a/edx_sga/tests/integration_tests.py b/edx_sga/tests/integration_tests.py
@@ -299,8 +299,10 @@ def test_student_view_with_score(self, fragment, render_template):
         user = student["module"].student
         student_id = anonymous_id_for_user(user,self.course_id)
 
-        with mock.patch.object(StaffGradedAssignmentXBlock.get_submission,"__defaults__",(student_id,)),\
-        mock.patch.object(StaffGradedAssignmentXBlock.get_score, "__defaults__",(student_id,)):
+        with(
+            mock.patch.object(StaffGradedAssignmentXBlock.get_submission,"__defaults__",(student_id,)),
+            mock.patch.object(StaffGradedAssignmentXBlock.get_score, "__defaults__",(student_id,))
+            ):
             fragment = block.student_view()
             render_template.assert_called_once()
             template_arg = render_template.call_args[0][0]
@@ -553,8 +555,10 @@ def test_staff_download(self):
             user = student["module"].student
             student_id = anonymous_id_for_user(user,self.course_id)
 
-            with mock.patch.object(StaffGradedAssignmentXBlock.get_student_item_dict,"__defaults__",(student_id,)),\
-                  self.dummy_upload(filename, text) as (upload, __):
+            with(
+                mock.patch.object(StaffGradedAssignmentXBlock.get_student_item_dict,"__defaults__",(student_id,)),
+                self.dummy_upload(filename, text) as (upload, __)
+                ):
                 block.upload_assignment(mock.Mock(params={"assignment": upload}))
             students.append(
                 (
@@ -646,8 +650,10 @@ def test_staff_download_filename_with_spaces(self):
         user = student["module"].student
         student_id = anonymous_id_for_user(user,self.course_id)
 
-        with mock.patch.object(StaffGradedAssignmentXBlock.get_student_item_dict,"__defaults__",(student_id,)),\
-              self.dummy_upload(file_name) as (upload, expected):
+        with(
+            mock.patch.object(StaffGradedAssignmentXBlock.get_student_item_dict,"__defaults__",(student_id,)),
+            self.dummy_upload(file_name) as (upload, expected)
+            ):
             block.upload_assignment(mock.Mock(params={"assignment": upload}))
         response = block.staff_download(
             mock.Mock(params={"student_id": student["item"].student_id})
@@ -669,8 +675,10 @@ def test_file_download_comma_in_name(self, file_name):
         user = student["module"].student
         student_id = anonymous_id_for_user(user,self.course_id)
 
-        with mock.patch.object(StaffGradedAssignmentXBlock.get_student_item_dict,"__defaults__",(student_id,)),\
-              self.dummy_upload(file_name) as (upload, expected):
+        with(
+            mock.patch.object(StaffGradedAssignmentXBlock.get_student_item_dict,"__defaults__",(student_id,)),
+            self.dummy_upload(file_name) as (upload, expected)
+            ):
             block.upload_assignment(mock.Mock(params={"assignment": upload}))
         response = block.staff_download(
             mock.Mock(params={"student_id": student["item"].student_id})
@@ -686,8 +694,10 @@ def test_get_staff_grading_data_not_staff(self):
         test staff grading data for non staff members.
         """
         block = self.make_one()
-        with mock.patch("edx_sga.sga.StaffGradedAssignmentXBlock.is_course_staff", return_value=False),\
-              self.assertRaises(PermissionDenied):
+        with(
+            mock.patch("edx_sga.sga.StaffGradedAssignmentXBlock.is_course_staff", return_value=False),
+            self.assertRaises(PermissionDenied)
+            ):
             block.get_staff_grading_data(None)
 
     def test_get_staff_grading_data(self):