b353ed2ea2
The existing pattern of using `override_settings(MODULESTORE=...)` prevented
us from having more than one layer of subclassing in modulestore tests.
In a structure like:
@override_settings(MODULESTORE=store_a)
class BaseTestCase(ModuleStoreTestCase):
def setUp(self):
# use store
@override_settings(MODULESTORE=store_b)
class ChildTestCase(BaseTestCase):
def setUp(self):
# use store
In this case, the store actions performed in `BaseTestCase` on behalf of
`ChildTestCase` would still use `store_a`, even though the `ChildTestCase`
had specified to use `store_b`. This is because the `override_settings`
decorator would be the innermost wrapper around the `BaseTestCase.setUp` method,
no matter what `ChildTestCase` does.
To remedy this, we move the call to `override_settings` into the
`ModuleStoreTestCase.setUp` method, and use a cleanup to remove the override.
Subclasses can just defined the `MODULESTORE` class attribute to specify which
modulestore to use _for the entire `setUp` chain_.
[PLAT-419]
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
"""
|
|
Tests for courseware middleware
|
|
"""
|
|
|
|
from django.core.urlresolvers import reverse
|
|
from django.test.utils import override_settings
|
|
from django.test.client import RequestFactory
|
|
from django.http import Http404
|
|
from mock import patch
|
|
|
|
import courseware.courses as courses
|
|
from courseware.middleware import RedirectUnenrolledMiddleware
|
|
from xmodule.modulestore.tests.django_utils import TEST_DATA_MOCK_MODULESTORE
|
|
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
|
|
|
|
class CoursewareMiddlewareTestCase(ModuleStoreTestCase):
|
|
"""Tests that courseware middleware is correctly redirected"""
|
|
|
|
def setUp(self):
|
|
super(CoursewareMiddlewareTestCase, self).setUp()
|
|
|
|
self.course = CourseFactory.create()
|
|
|
|
def check_user_not_enrolled_redirect(self):
|
|
"""A UserNotEnrolled exception should trigger a redirect"""
|
|
request = RequestFactory().get("dummy_url")
|
|
response = RedirectUnenrolledMiddleware().process_exception(
|
|
request, courses.UserNotEnrolled(self.course.id)
|
|
)
|
|
self.assertEqual(response.status_code, 302)
|
|
# make sure we redirect to the course about page
|
|
expected_url = reverse(
|
|
"about_course", args=[self.course.id.to_deprecated_string()]
|
|
)
|
|
|
|
target_url = response._headers['location'][1]
|
|
self.assertTrue(target_url.endswith(expected_url))
|
|
|
|
def test_user_not_enrolled_redirect(self):
|
|
self.check_user_not_enrolled_redirect()
|
|
|
|
@patch.dict("django.conf.settings.FEATURES", {"ENABLE_MKTG_SITE": True})
|
|
def test_user_not_enrolled_redirect_mktg(self):
|
|
self.check_user_not_enrolled_redirect()
|
|
|
|
def test_process_404(self):
|
|
"""A 404 should not trigger anything"""
|
|
request = RequestFactory().get("dummy_url")
|
|
response = RedirectUnenrolledMiddleware().process_exception(
|
|
request, Http404()
|
|
)
|
|
self.assertIsNone(response)
|