"""
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)