108 lines
4.3 KiB
Python
108 lines
4.3 KiB
Python
"""
|
|
Test the Data Aggregation Layer for Course Entitlements.
|
|
"""
|
|
import unittest
|
|
import uuid
|
|
|
|
import ddt
|
|
from django.conf import settings
|
|
|
|
from entitlements.models import CourseEntitlement
|
|
from student.tests.factories import UserFactory
|
|
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
from student.tests.factories import CourseEnrollmentFactory
|
|
|
|
|
|
@ddt.ddt
|
|
@unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
|
|
class EntitlementDataTest(ModuleStoreTestCase):
|
|
"""
|
|
Test course entitlement data aggregation.
|
|
"""
|
|
USERNAME = "Bob"
|
|
EMAIL = "bob@example.com"
|
|
PASSWORD = "edx"
|
|
|
|
def setUp(self):
|
|
"""Create a course and user, then log in. """
|
|
super(EntitlementDataTest, self).setUp()
|
|
self.course = CourseFactory.create()
|
|
self.course_uuid = uuid.uuid4()
|
|
self.user = UserFactory.create(username=self.USERNAME, email=self.EMAIL, password=self.PASSWORD)
|
|
self.client.login(username=self.USERNAME, password=self.PASSWORD)
|
|
|
|
def _add_entitlement_for_user(self, course, user, parent_uuid):
|
|
entitlement_data = {
|
|
'user': user,
|
|
'course_uuid': parent_uuid,
|
|
'mode': 'verified',
|
|
}
|
|
stored_entitlement, is_created = CourseEntitlement.update_or_create_new_entitlement(
|
|
user,
|
|
parent_uuid,
|
|
entitlement_data
|
|
)
|
|
return stored_entitlement, is_created
|
|
|
|
def test_get_entitlement_info(self):
|
|
stored_entitlement, is_created = self._add_entitlement_for_user(self.course, self.user, self.course_uuid)
|
|
self.assertTrue(is_created)
|
|
|
|
# Get the Entitlement and verify the data
|
|
entitlement = CourseEntitlement.get_user_course_entitlement(self.user, self.course_uuid)
|
|
self.assertEqual(entitlement.course_uuid, self.course_uuid)
|
|
self.assertEqual(entitlement.mode, 'verified')
|
|
self.assertIsNone(entitlement.enrollment_course_run)
|
|
|
|
def test_get_course_entitlements(self):
|
|
course2 = CourseFactory.create()
|
|
|
|
stored_entitlement, is_created = self._add_entitlement_for_user(self.course, self.user, self.course_uuid)
|
|
self.assertTrue(is_created)
|
|
|
|
course2_uuid = uuid.uuid4()
|
|
stored_entitlement2, is_created2 = self._add_entitlement_for_user(course2, self.user, course2_uuid)
|
|
self.assertTrue(is_created2)
|
|
|
|
# Get the Entitlement and verify the data
|
|
entitlement_list = CourseEntitlement.entitlements_for_user(self.user)
|
|
|
|
self.assertEqual(2, len(entitlement_list))
|
|
self.assertEqual(self.course_uuid, entitlement_list[0].course_uuid)
|
|
self.assertEqual(course2_uuid, entitlement_list[1].course_uuid)
|
|
|
|
def test_set_enrollment(self):
|
|
stored_entitlement, is_created = self._add_entitlement_for_user(self.course, self.user, self.course_uuid)
|
|
self.assertTrue(is_created)
|
|
|
|
# Entitlement set not enroll the user in the Course run
|
|
enrollment = CourseEnrollmentFactory(
|
|
user=self.user,
|
|
course_id=self.course.id,
|
|
mode="verified",
|
|
)
|
|
CourseEntitlement.update_entitlement_enrollment(self.user, self.course_uuid, enrollment)
|
|
|
|
entitlement = CourseEntitlement.get_user_course_entitlement(self.user, self.course_uuid)
|
|
self.assertIsNotNone(entitlement.enrollment_course_run)
|
|
|
|
def test_remove_enrollment(self):
|
|
stored_entitlement, is_created = self._add_entitlement_for_user(self.course, self.user, self.course_uuid)
|
|
self.assertTrue(is_created)
|
|
|
|
# Entitlement set not enroll the user in the Course run
|
|
enrollment = CourseEnrollmentFactory(
|
|
user=self.user,
|
|
course_id=self.course.id,
|
|
mode="verified",
|
|
)
|
|
CourseEntitlement.update_entitlement_enrollment(self.user, self.course_uuid, enrollment)
|
|
|
|
entitlement = CourseEntitlement.get_user_course_entitlement(self.user, self.course_uuid)
|
|
self.assertIsNotNone(entitlement.enrollment_course_run)
|
|
|
|
CourseEntitlement.update_entitlement_enrollment(self.user, self.course_uuid, None)
|
|
entitlement = CourseEntitlement.get_user_course_entitlement(self.user, self.course_uuid)
|
|
self.assertIsNone(entitlement.enrollment_course_run)
|