58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
"""
|
|
Tests for courseware middleware
|
|
"""
|
|
|
|
|
|
from django.http import Http404
|
|
from django.test.client import RequestFactory
|
|
|
|
from lms.djangoapps.courseware.exceptions import Redirect
|
|
from lms.djangoapps.courseware.middleware import RedirectMiddleware
|
|
from xmodule.modulestore.tests.django_utils import SharedModuleStoreTestCase # lint-amnesty, pylint: disable=wrong-import-order
|
|
from xmodule.modulestore.tests.factories import CourseFactory # lint-amnesty, pylint: disable=wrong-import-order
|
|
|
|
|
|
class CoursewareMiddlewareTestCase(SharedModuleStoreTestCase):
|
|
"""Tests that courseware middleware is correctly redirected"""
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
cls.course = CourseFactory.create()
|
|
|
|
@staticmethod
|
|
def get_headers(cache_response):
|
|
"""
|
|
Django 3.2 has no ._headers
|
|
See https://docs.djangoproject.com/en/3.2/releases/3.2/#requests-and-responses
|
|
"""
|
|
if hasattr(cache_response, '_headers'):
|
|
headers = cache_response._headers.copy() # pylint: disable=protected-access
|
|
else:
|
|
headers = {k.lower(): (k, v) for k, v in cache_response.items()}
|
|
|
|
return headers
|
|
|
|
def test_process_404(self):
|
|
"""A 404 should not trigger anything"""
|
|
request = RequestFactory().get("dummy_url")
|
|
response = RedirectMiddleware(get_response=lambda request: None).process_exception(
|
|
request, Http404()
|
|
)
|
|
assert response is None
|
|
|
|
def test_redirect_exceptions(self):
|
|
"""
|
|
Unit tests for handling of Redirect exceptions.
|
|
"""
|
|
request = RequestFactory().get("dummy_url")
|
|
test_url = '/test_url'
|
|
exception = Redirect(test_url)
|
|
response = RedirectMiddleware(get_response=lambda request: None).process_exception(
|
|
request, exception
|
|
)
|
|
assert response.status_code == 302
|
|
headers = self.get_headers(response)
|
|
target_url = headers['location'][1]
|
|
assert target_url.endswith(test_url)
|