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]
152 lines
6.1 KiB
Python
152 lines
6.1 KiB
Python
"""
|
|
Tests for user enrollment.
|
|
"""
|
|
import ddt
|
|
import json
|
|
import unittest
|
|
|
|
from django.test.utils import override_settings
|
|
from django.core.urlresolvers import reverse
|
|
from rest_framework.test import APITestCase
|
|
from rest_framework import status
|
|
from django.conf import settings
|
|
from datetime import datetime
|
|
from mock import patch
|
|
from xmodule.modulestore.tests.django_utils import (
|
|
ModuleStoreTestCase, mixed_store_config
|
|
)
|
|
from xmodule.modulestore.tests.factories import CourseFactory, CourseAboutFactory
|
|
from student.tests.factories import UserFactory
|
|
from course_about.serializers import course_image_url
|
|
from course_about import api
|
|
from course_about.errors import CourseNotFoundError, CourseAboutError
|
|
from xmodule.modulestore.django import modulestore
|
|
|
|
|
|
@ddt.ddt
|
|
@unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
|
|
class CourseInfoTest(ModuleStoreTestCase, APITestCase):
|
|
"""
|
|
Test course information.
|
|
"""
|
|
USERNAME = "Bob"
|
|
EMAIL = "bob@example.com"
|
|
PASSWORD = "edx"
|
|
|
|
def setUp(self):
|
|
""" Create a course"""
|
|
super(CourseInfoTest, self).setUp()
|
|
|
|
self.course = CourseFactory.create()
|
|
self.user = UserFactory.create(username=self.USERNAME, email=self.EMAIL, password=self.PASSWORD)
|
|
self.client.login(username=self.USERNAME, password=self.PASSWORD)
|
|
|
|
def test_user_not_authenticated(self):
|
|
# Log out, so we're no longer authenticated
|
|
self.client.logout()
|
|
resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_200_OK)
|
|
self.assertIsNotNone(resp_data)
|
|
|
|
def test_with_valid_course_id(self):
|
|
_resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_200_OK)
|
|
|
|
def test_with_invalid_course_id(self):
|
|
resp = self.client.get(
|
|
reverse('courseabout', kwargs={"course_id": 'not/a/validkey'})
|
|
)
|
|
self.assertEqual(resp.status_code, status.HTTP_404_NOT_FOUND)
|
|
|
|
def test_get_course_details_all_attributes(self):
|
|
kwargs = dict()
|
|
kwargs["course_id"] = self.course.id
|
|
kwargs["course_runtime"] = self.course.runtime
|
|
CourseAboutFactory.create(**kwargs)
|
|
|
|
resp_data, status_code = self._get_course_about(self.course.id)
|
|
|
|
all_attributes = ['display_name', 'start', 'end', 'announcement', 'advertised_start', 'is_new', 'course_number',
|
|
'course_id',
|
|
'effort', 'media', 'course_image']
|
|
for attr in all_attributes:
|
|
self.assertIn(attr, str(resp_data))
|
|
self.assertEqual(status_code, status.HTTP_200_OK)
|
|
|
|
def test_get_course_about_valid_date(self):
|
|
module_store = modulestore()
|
|
self.course.start = datetime.now()
|
|
self.course.end = datetime.now()
|
|
self.course.announcement = datetime.now()
|
|
module_store.update_item(self.course, self.user.id)
|
|
|
|
resp_data, _status_code = self._get_course_about(self.course.id)
|
|
|
|
self.assertIsNotNone(resp_data["start"])
|
|
self.assertIsNotNone(resp_data["end"])
|
|
self.assertIsNotNone(resp_data["announcement"])
|
|
|
|
def test_get_course_about_none_date(self):
|
|
module_store = modulestore()
|
|
self.course.start = None
|
|
self.course.end = None
|
|
self.course.announcement = None
|
|
module_store.update_item(self.course, self.user.id)
|
|
|
|
resp_data, _status_code = self._get_course_about(self.course.id)
|
|
self.assertIsNone(resp_data["start"])
|
|
self.assertIsNone(resp_data["end"])
|
|
self.assertIsNone(resp_data["announcement"])
|
|
|
|
def test_get_course_details(self):
|
|
kwargs = dict()
|
|
kwargs["course_id"] = self.course.id
|
|
kwargs["course_runtime"] = self.course.runtime
|
|
kwargs["user_id"] = self.user.id
|
|
CourseAboutFactory.create(**kwargs)
|
|
|
|
resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_200_OK)
|
|
self.assertEqual(unicode(self.course.id), resp_data['course_id'])
|
|
self.assertIn('Run', resp_data['display_name'])
|
|
|
|
url = course_image_url(self.course)
|
|
self.assertEquals(url, resp_data['media']['course_image'])
|
|
|
|
@patch.object(api, "get_course_about_details")
|
|
def test_get_enrollment_course_not_found_error(self, mock_get_course_about_details):
|
|
mock_get_course_about_details.side_effect = CourseNotFoundError("Something bad happened.")
|
|
_resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_404_NOT_FOUND)
|
|
|
|
@patch.object(api, "get_course_about_details")
|
|
def test_get_enrollment_invalid_key_error(self, mock_get_course_about_details):
|
|
mock_get_course_about_details.side_effect = CourseNotFoundError('a/a/a', "Something bad happened.")
|
|
resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_404_NOT_FOUND)
|
|
self.assertIn('An error occurred', resp_data["message"])
|
|
|
|
@patch.object(api, "get_course_about_details")
|
|
def test_get_enrollment_internal_error(self, mock_get_course_about_details):
|
|
mock_get_course_about_details.side_effect = CourseAboutError('error')
|
|
resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
|
|
self.assertIn('An error occurred', resp_data["message"])
|
|
|
|
@override_settings(COURSE_ABOUT_DATA_API='foo')
|
|
def test_data_api_config_error(self):
|
|
# Retrive the invalid course
|
|
resp_data, status_code = self._get_course_about(self.course.id)
|
|
self.assertEqual(status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
|
|
self.assertIn('An error occurred', resp_data["message"])
|
|
|
|
def _get_course_about(self, course_id):
|
|
"""
|
|
helper function to get retrieve course about information.
|
|
args course_id (str): course id
|
|
"""
|
|
resp = self.client.get(
|
|
reverse('courseabout', kwargs={"course_id": unicode(course_id)})
|
|
)
|
|
return json.loads(resp.content), resp.status_code
|