@@ -142,29 +142,29 @@ def test_pacing(self, self_paced, expected_pacing):
142142 result = self ._get_result (course )
143143 assert result ['pacing' ] == expected_pacing
144144
145- @mock .patch (
146- "lms.djangoapps.course_api.serializers.can_show_certificate_available_date_field" ,
147- return_value = True ,
148- )
149- def test_certificate_available_date_included (self , _mock ):
145+ def test_certificate_available_date_included (self ):
150146 """Test that certificate_available_date is included when the field should be shown."""
151147 course = self .create_course ()
152- result = self ._get_result (course )
148+ with mock .patch (
149+ "lms.djangoapps.course_api.serializers.can_show_certificate_available_date_field" ,
150+ return_value = True ,
151+ ):
152+ result = self ._get_result (course )
153153 assert 'certificate_available_date' in result
154154 actual = result ['certificate_available_date' ]
155155 if isinstance (actual , datetime ):
156156 assert actual .strftime ('%Y-%m-%dT%H:%M:%SZ' ) == '2015-08-14T00:00:00Z'
157157 else :
158158 assert actual == '2015-08-14T00:00:00Z'
159159
160- @mock .patch (
161- "lms.djangoapps.course_api.serializers.can_show_certificate_available_date_field" ,
162- return_value = False ,
163- )
164- def test_certificate_available_date_excluded (self , _mock ):
160+ def test_certificate_available_date_excluded (self ):
165161 """Test that certificate_available_date is excluded when the field should not be shown."""
166162 course = self .create_course ()
167- result = self ._get_result (course )
163+ with mock .patch (
164+ "lms.djangoapps.course_api.serializers.can_show_certificate_available_date_field" ,
165+ return_value = False ,
166+ ):
167+ result = self ._get_result (course )
168168 assert 'certificate_available_date' not in result
169169
170170
0 commit comments