test: correct import error on test

openassessment/xblock/config_mixin.py

@@ -167,5 +167,5 @@ def is_due_date_extension_enabled(self):
         # .. toggle_description: Set to True to enable the due date extension feature
         # .. toggle_use_cases: circuit_breaker
         # .. toggle_creation_date: 2023-08-11
-        # .. toggle_tickets: 
+        # .. toggle_tickets:
         return self.is_feature_enabled(DUE_DATE_EXTENSION)

openassessment/xblock/openassessmentblock.py

@@ -1065,27 +1065,30 @@ def render_error(self, error_msg):
         context = {'error_msg': error_msg}
         template = get_template('openassessmentblock/oa_error.html')
         return Response(template.render(context), content_type='application/html', charset='UTF-8')
-    
+
     def can_use_date_extensions(self):
+        """
+        Check if the due date extension feature is enabled.
+        """
         if not self.is_due_date_extension_enabled:
             return False
-        
+
         if not self.due:
             return False
-        
+
         if not self.submission_due:
             return False
-        
+
         parsed_due = parse_date_value(self.due, self._)
-        parsed_submission_due_= parse_date_value(self.submission_due, self._)
-        
-        has_due_date_extension = parsed_due > parsed_submission_due_
+        parsed_submission_due = parse_date_value(self.submission_due, self._)
+
+        has_due_date_extension = parsed_due > parsed_submission_due
 
         if not has_due_date_extension:
             return False
-        
+
         return True
-    
+
     def date_ranges(self):
         """
         Generate a list of dates ranges for the submission and assessment steps.
@@ -1100,7 +1103,7 @@ def date_ranges(self):
         if self.can_use_date_extensions():
 
             parsed_due = parse_date_value(self.due, self._)
-            
+
             submission_range = (self.submission_start, self.due)
             assessment_ranges = []
 
@@ -1114,7 +1117,7 @@ def date_ranges(self):
                 parsed_assesment_due = parse_date_value(assesment_due, self._)
 
                 is_extended = parsed_due < parsed_assesment_due
-               
+
                 if is_extended:
                     assessment_ranges.append((assesment.get('start'), assesment_due))
                 else:
@@ -1127,7 +1130,6 @@ def date_ranges(self):
             ]
         return submission_range, assessment_ranges
 
-
     def is_closed(self, step=None, course_staff=None):
         """
         Checks if the question is closed.

openassessment/xblock/test/test_grade.py

@@ -5,6 +5,7 @@
 
 import copy
 import json
+from unittest.mock import Mock
 
 import ddt
 
@@ -22,6 +23,7 @@ class TestGrade(XBlockHandlerTestCase, SubmitAssessmentsMixin):
     @scenario('data/grade_scenario.xml', user_id='Greggs')
     def test_render_grade(self, xblock):
         # Submit, assess, and render the grade view
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         self.create_submission_and_assessments(
             xblock, self.SUBMISSION, self.PEERS, PEER_ASSESSMENTS, SELF_ASSESSMENT
         )
@@ -56,6 +58,8 @@ def test_render_grade(self, xblock):
     @scenario('data/grade_scenario_self_only.xml', user_id='Greggs')
     def test_render_grade_self_only(self, xblock):
         # Submit, assess, and render the grade view
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         self.create_submission_and_assessments(
             xblock, self.SUBMISSION, [], [], SELF_ASSESSMENT,
             waiting_for_peer=True
@@ -402,6 +406,7 @@ def _test_incomplete_helper(self, xblock, peers, self_assessment):
         """
         Check assessment completition status is shown correctly on assessment page.
         """
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         self.create_submission_and_assessments(
             xblock, self.SUBMISSION, peers, [PEER_ASSESSMENTS[0]] if peers else [], self_assessment
         )

openassessment/xblock/test/test_openassessment.py

@@ -38,6 +38,8 @@ def test_load_student_view(self, xblock):
         xblock_fragment = self.runtime.render(xblock, "student_view")
         self.assertIn("OpenAssessmentBlock", xblock_fragment.body_html())
 
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         # Validate Submission Rendering.
         submission_response = xblock.render_submission({})
         self.assertIsNotNone(submission_response)
@@ -292,6 +294,7 @@ def test_load_student_view_with_dates(self, time_zone, expected_date):
             time_zone_fn.return_value['user_timezone'] = pytz.timezone(time_zone)
 
             xblock = self.load_scenario('data/dates_scenario.xml')
+            xblock.is_due_date_extension_enabled = Mock(return_value=True)
             xblock_fragment = self.runtime.render(xblock, "student_view")
             self.assertIn("OpenAssessmentBlock", xblock_fragment.body_html())
 

openassessment/xblock/test/test_save_files_descriptions.py

@@ -33,6 +33,8 @@ def test_save_files_descriptions(self, xblock):
         # We're not worried about looking up shared uploads in this test
         xblock.has_team = mock.Mock(return_value=False)
 
+        xblock.is_due_date_extension_enabled = mock.Mock(return_value=True)
+
         xblock.xmodule_runtime = mock.Mock(
             user_is_staff=False,
             user_is_beta_tester=False,
@@ -66,6 +68,8 @@ def test_append_files_descriptions(self, xblock):
         # We're not worried about looking up shared uploads in this test
         xblock.has_team = mock.Mock(return_value=False)
 
+        xblock.is_due_date_extension_enabled = mock.Mock(return_value=True)
+
         xblock.xmodule_runtime = mock.Mock(
             user_is_staff=False,
             user_is_beta_tester=False,

openassessment/xblock/test/test_save_response.py

@@ -19,6 +19,8 @@ class SaveResponseTest(XBlockHandlerTestCase):
     def test_default_saved_response_blank(self, xblock):
         xblock.get_team_info = mock.Mock(return_value={})
 
+        xblock.is_due_date_extension_enabled = mock.Mock(return_value=True)
+
         xblock.xmodule_runtime = mock.Mock(
             user_is_staff=False,
             user_is_beta_tester=False,
@@ -34,6 +36,8 @@ def test_default_saved_response_blank(self, xblock):
     def test_save_response(self, xblock, data):
         xblock.get_team_info = mock.Mock(return_value={})
 
+        xblock.is_due_date_extension_enabled = mock.Mock(return_value=True)
+
         xblock.xmodule_runtime = mock.Mock(
             user_is_staff=False,
             user_is_beta_tester=False,

openassessment/xblock/test/test_staff_area.py

@@ -127,6 +127,7 @@ def test_is_course_staff(self, xblock):
     @scenario('data/basic_scenario.xml', user_id='Bob')
     def test_course_staff_area(self, xblock):
         # If we're not course staff, we shouldn't see the staff area
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         xblock.xmodule_runtime = self._create_mock_runtime(
             xblock.scope_ids.usage_id, False, False, "Bob"
         )
@@ -140,6 +141,7 @@ def test_course_staff_area(self, xblock):
 
     @scenario('data/basic_scenario.xml', user_id='Bob')
     def test_view_in_studio_button(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         xblock.xmodule_runtime = self._create_mock_runtime(
             xblock.scope_ids.usage_id, False, False, "Bob"
         )
@@ -210,6 +212,8 @@ def test_hide_course_staff_area_in_studio_preview(self, xblock):
     @scenario('data/staff_dates_scenario.xml', user_id='Bob')
     def test_staff_area_dates_table(self, xblock):
         # Simulate that we are course staff
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         xblock.xmodule_runtime = self._create_mock_runtime(
             xblock.scope_ids.usage_id, True, False, "Bob"
         )
@@ -232,6 +236,8 @@ def test_staff_area_dates_table(self, xblock):
     @scenario('data/basic_scenario.xml', user_id='Bob')
     def test_staff_area_dates_distant_past_and_future(self, xblock):
         # Simulate that we are course staff
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         xblock.xmodule_runtime = self._create_mock_runtime(
             xblock.scope_ids.usage_id, True, False, "Bob"
         )
@@ -1116,6 +1122,7 @@ def test_staff_area_student_user_state_not_used(self, xblock, waffle_patch):
 
     @scenario('data/team_submission.xml', user_id='Bob')
     def test_staff_area_has_team_info(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         # Given that we are course staff, managing a team assignment
         self._setup_xblock_and_create_submission(xblock)
 
@@ -1140,6 +1147,7 @@ def test_staff_area_student_info_has_team_info(self, xblock):
     @scenario('data/basic_scenario.xml', user_id='Bob')
     def test_staff_area_has_team_info_individual(self, xblock):
         # Given that we are course staff, managing an individual assignment
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         self._setup_xblock_and_create_submission(xblock)
 
         # When I get the staff context

openassessment/xblock/test/test_student_training.py

@@ -199,6 +199,7 @@ def test_updates_workflow(self, xblock, data):
 
     @scenario('data/feedback_only_criterion_student_training.xml', user_id='Bob')
     def test_feedback_only_criterion(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         xblock.create_submission(xblock.get_student_item_dict(), self.SUBMISSION)
         self.request(xblock, 'render_student_training', json.dumps({}))
 
@@ -264,6 +265,7 @@ def test_no_submission(self, xblock):
                 'Grammar': 'Poor'
             }
         }
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         resp = self.request(xblock, 'training_assess', json.dumps(selected_data))
         self.assertIn("Your scores could not be checked", resp.decode('utf-8'))
 
@@ -315,11 +317,13 @@ def test_no_student_training_defined(self, xblock):
 
     @scenario('data/student_training.xml', user_id="Plato")
     def test_no_submission(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         resp = self.request(xblock, 'render_student_training', json.dumps({}))
         self.assertIn("Not Available", resp.decode('utf-8'))
 
     @scenario('data/student_training.xml')
     def test_studio_preview(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         resp = self.request(xblock, 'render_student_training', json.dumps({}))
         self.assertIn("Not Available", resp.decode('utf-8'))
 

openassessment/xblock/test/test_submission.py

@@ -150,19 +150,22 @@ def test_cannot_submit_in_preview_mode(self, xblock):
 
     @scenario('data/over_grade_scenario.xml', user_id='Alice')
     def test_closed_submissions(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         resp = self.request(xblock, 'render_submission', json.dumps({}))
         self.assertIn("Incomplete", resp.decode('utf-8'))
 
     @scenario('data/line_breaks.xml')
     def test_prompt_line_breaks(self, xblock):
         # Verify that prompts with multiple lines retain line breaks
         # (backward compatibility in case if prompt_type == 'text')
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         resp = self.request(xblock, 'render_submission', json.dumps({}))
         expected_prompt = "<p><br>Line 1</p><p>Line 2</p><p>Line 3<br></p>"
         self.assertIn(expected_prompt, resp.decode('utf-8'))
 
     @scenario('data/prompt_html.xml')
     def test_prompt_html_to_text(self, xblock):
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         resp = self.request(xblock, 'render_submission', json.dumps({}))
         expected_prompt = "<code><strong>Question 123</strong></code>"
         self.assertIn(expected_prompt, resp.decode('utf-8'))
@@ -948,6 +951,8 @@ def test_open_saved_response(self, xblock):
 
         xblock.file_manager.append_uploads(*file_uploads)
 
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         # Save a response
         payload = json.dumps({'submission': ('A man must have a code', 'A man must have an umbrella too.')})
         resp = self.request(xblock, 'save_submission', payload, response_format='json')
@@ -1031,6 +1036,8 @@ def test_open_saved_response_deleted_file_uploads(self, xblock):
             anonymous_student_id='Pmn'
         )
 
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         # delete file-2
         with patch('openassessment.fileupload.api.remove_file'):
             xblock.file_manager.delete_upload(1)
@@ -1197,6 +1204,7 @@ def test_render_shared_files(self, xblock, mock_get_download_url):
         Test that we render files owned by Valchek and
         their teammates when files are shared with a team.
         """
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
         xblock.file_manager.get_uploads = Mock(return_value=[
             api.FileUpload(
                 description='file 1 description',
@@ -1600,6 +1608,8 @@ def test_closed_graded(self, xblock):
     @scenario('data/submission_open.xml', user_id="Bob")
     def test_integration(self, xblock):
         # Expect that the response step is open and displays the deadline
+        xblock.is_due_date_extension_enabled = Mock(return_value=True)
+
         resp = self.request(xblock, 'render_submission', json.dumps({}))
         self.assertIn('Enter your response to the prompt', resp.decode('utf-8'))
         self.assertIn('2999-05-06T00:00:00+00:00', resp.decode('utf-8'))