179 lines
7.3 KiB
Python
179 lines
7.3 KiB
Python
'''
|
|
Unit tests for enrollment methods in views.py
|
|
|
|
'''
|
|
|
|
from django.test.utils import override_settings
|
|
from django.contrib.auth.models import Group, User
|
|
from django.core.urlresolvers import reverse
|
|
from courseware.access import _course_staff_group_name
|
|
from courseware.tests.tests import LoginEnrollmentTestCase, TEST_DATA_XML_MODULESTORE, get_user
|
|
from xmodule.modulestore.django import modulestore
|
|
import xmodule.modulestore.django
|
|
from student.models import CourseEnrollment, CourseEnrollmentAllowed
|
|
from instructor.views import get_and_clean_student_list
|
|
|
|
|
|
@override_settings(MODULESTORE=TEST_DATA_XML_MODULESTORE)
|
|
class TestInstructorEnrollsStudent(LoginEnrollmentTestCase):
|
|
'''
|
|
Check Enrollment/Unenrollment with/without auto-enrollment on activation
|
|
'''
|
|
|
|
def setUp(self):
|
|
|
|
self.full = modulestore().get_course("edX/full/6.002_Spring_2012")
|
|
self.toy = modulestore().get_course("edX/toy/2012_Fall")
|
|
|
|
#Create instructor and student accounts
|
|
self.instructor = 'instructor1@test.com'
|
|
self.student1 = 'student1@test.com'
|
|
self.student2 = 'student2@test.com'
|
|
self.password = 'foo'
|
|
self.create_account('it1', self.instructor, self.password)
|
|
self.create_account('st1', self.student1, self.password)
|
|
self.create_account('st2', self.student2, self.password)
|
|
self.activate_user(self.instructor)
|
|
self.activate_user(self.student1)
|
|
self.activate_user(self.student2)
|
|
|
|
def make_instructor(course):
|
|
group_name = _course_staff_group_name(course.location)
|
|
g = Group.objects.create(name=group_name)
|
|
g.user_set.add(get_user(self.instructor))
|
|
|
|
make_instructor(self.toy)
|
|
|
|
#Enroll Students
|
|
self.logout()
|
|
self.login(self.student1, self.password)
|
|
self.enroll(self.toy)
|
|
|
|
self.logout()
|
|
self.login(self.student2, self.password)
|
|
self.enroll(self.toy)
|
|
|
|
#Enroll Instructor
|
|
self.logout()
|
|
self.login(self.instructor, self.password)
|
|
self.enroll(self.toy)
|
|
|
|
def test_unenrollment(self):
|
|
'''
|
|
Do un-enrollment test
|
|
'''
|
|
|
|
course = self.toy
|
|
url = reverse('instructor_dashboard', kwargs={'course_id': course.id})
|
|
response = self.client.post(url, {'action': 'Unenroll multiple students', 'multiple_students': 'student1@test.com, student2@test.com'})
|
|
|
|
#Check the page output
|
|
self.assertContains(response, '<td>student1@test.com</td>')
|
|
self.assertContains(response, '<td>student2@test.com</td>')
|
|
self.assertContains(response, '<td>un-enrolled</td>')
|
|
|
|
#Check the enrollment table
|
|
user = User.objects.get(email='student1@test.com')
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id, user=user)
|
|
self.assertEqual(0, len(ce))
|
|
|
|
user = User.objects.get(email='student2@test.com')
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id, user=user)
|
|
self.assertEqual(0, len(ce))
|
|
|
|
def test_enrollment_new_student_autoenroll_on(self):
|
|
'''
|
|
Do auto-enroll on test
|
|
'''
|
|
|
|
#Run the Enroll students command
|
|
course = self.toy
|
|
url = reverse('instructor_dashboard', kwargs={'course_id': course.id})
|
|
response = self.client.post(url, {'action': 'Enroll multiple students', 'multiple_students': 'test1_1@student.com, test1_2@student.com', 'auto_enroll': 'on'})
|
|
|
|
#Check the page output
|
|
self.assertContains(response, '<td>test1_1@student.com</td>')
|
|
self.assertContains(response, '<td>test1_2@student.com</td>')
|
|
self.assertContains(response, '<td>user does not exist, enrollment allowed, pending with auto enrollment on</td>')
|
|
|
|
#Check the enrollmentallowed db entries
|
|
cea = CourseEnrollmentAllowed.objects.filter(email='test1_1@student.com', course_id=course.id)
|
|
self.assertEqual(1, cea[0].auto_enroll)
|
|
cea = CourseEnrollmentAllowed.objects.filter(email='test1_2@student.com', course_id=course.id)
|
|
self.assertEqual(1, cea[0].auto_enroll)
|
|
|
|
#Check there is no enrollment db entry other than for the setup instructor and students
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id)
|
|
self.assertEqual(3, len(ce))
|
|
|
|
#Create and activate student accounts with same email
|
|
self.student1 = 'test1_1@student.com'
|
|
self.password = 'bar'
|
|
self.create_account('s1_1', self.student1, self.password)
|
|
self.activate_user(self.student1)
|
|
|
|
self.student2 = 'test1_2@student.com'
|
|
self.create_account('s1_2', self.student2, self.password)
|
|
self.activate_user(self.student2)
|
|
|
|
#Check students are enrolled
|
|
user = User.objects.get(email='test1_1@student.com')
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id, user=user)
|
|
self.assertEqual(1, len(ce))
|
|
|
|
user = User.objects.get(email='test1_2@student.com')
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id, user=user)
|
|
self.assertEqual(1, len(ce))
|
|
|
|
def test_enrollmemt_new_student_autoenroll_off(self):
|
|
'''
|
|
Do auto-enroll off test
|
|
'''
|
|
|
|
#Run the Enroll students command
|
|
course = self.toy
|
|
url = reverse('instructor_dashboard', kwargs={'course_id': course.id})
|
|
response = self.client.post(url, {'action': 'Enroll multiple students', 'multiple_students': 'test2_1@student.com, test2_2@student.com'})
|
|
|
|
#Check the page output
|
|
self.assertContains(response, '<td>test2_1@student.com</td>')
|
|
self.assertContains(response, '<td>test2_2@student.com</td>')
|
|
self.assertContains(response, '<td>user does not exist, enrollment allowed, pending with auto enrollment off</td>')
|
|
|
|
#Check the enrollmentallowed db entries
|
|
cea = CourseEnrollmentAllowed.objects.filter(email='test2_1@student.com', course_id=course.id)
|
|
self.assertEqual(0, cea[0].auto_enroll)
|
|
cea = CourseEnrollmentAllowed.objects.filter(email='test2_2@student.com', course_id=course.id)
|
|
self.assertEqual(0, cea[0].auto_enroll)
|
|
|
|
#Check there is no enrollment db entry other than for the setup instructor and students
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id)
|
|
self.assertEqual(3, len(ce))
|
|
|
|
#Create and activate student accounts with same email
|
|
self.student = 'test2_1@student.com'
|
|
self.password = 'bar'
|
|
self.create_account('s2_1', self.student, self.password)
|
|
self.activate_user(self.student)
|
|
|
|
self.student = 'test2_2@student.com'
|
|
self.create_account('s2_2', self.student, self.password)
|
|
self.activate_user(self.student)
|
|
|
|
#Check students are not enrolled
|
|
user = User.objects.get(email='test2_1@student.com')
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id, user=user)
|
|
self.assertEqual(0, len(ce))
|
|
user = User.objects.get(email='test2_2@student.com')
|
|
ce = CourseEnrollment.objects.filter(course_id=course.id, user=user)
|
|
self.assertEqual(0, len(ce))
|
|
|
|
def test_get_and_clean_student_list(self):
|
|
'''
|
|
Clean user input test
|
|
'''
|
|
|
|
string = "abc@test.com, def@test.com ghi@test.com \n \n jkl@test.com "
|
|
cleaned_string, cleaned_string_lc = get_and_clean_student_list(string)
|
|
self.assertEqual(cleaned_string, ['abc@test.com', 'def@test.com', 'ghi@test.com', 'jkl@test.com'])
|