9811926d97
This commit updates lms/djangoapps/courseware. These keys are now objects with a limited interface, and the particular internal representation is managed by the data storage layer (the modulestore). For the LMS, there should be no outward-facing changes to the system. The keys are, for now, a change to internal representation only. For Studio, the new serialized form of the keys is used in urls, to allow for further migration in the future. Co-Author: Andy Armstrong <andya@edx.org> Co-Author: Christina Roberts <christina@edx.org> Co-Author: David Baumgold <db@edx.org> Co-Author: Diana Huang <dkh@edx.org> Co-Author: Don Mitchell <dmitchell@edx.org> Co-Author: Julia Hansbrough <julia@edx.org> Co-Author: Nimisha Asthagiri <nasthagiri@edx.org> Co-Author: Sarina Canelake <sarina@edx.org> [LMS-2370]
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
# pylint: disable=C0111
|
|
# pylint: disable=W0621
|
|
|
|
import urllib
|
|
from lettuce import world
|
|
from django.contrib.auth.models import User, Group
|
|
from student.models import CourseEnrollment
|
|
from xmodule.modulestore.django import editable_modulestore
|
|
from xmodule.contentstore.django import contentstore
|
|
|
|
|
|
@world.absorb
|
|
def create_user(uname, password):
|
|
|
|
# If the user already exists, don't try to create it again
|
|
if len(User.objects.filter(username=uname)) > 0:
|
|
return
|
|
|
|
portal_user = world.UserFactory.build(username=uname, email=uname + '@edx.org')
|
|
portal_user.set_password(password)
|
|
portal_user.save()
|
|
|
|
registration = world.RegistrationFactory(user=portal_user)
|
|
registration.register(portal_user)
|
|
registration.activate()
|
|
|
|
world.UserProfileFactory(user=portal_user)
|
|
|
|
|
|
@world.absorb
|
|
def log_in(username='robot', password='test', email='robot@edx.org', name="Robot"):
|
|
"""
|
|
Use the auto_auth feature to programmatically log the user in
|
|
"""
|
|
url = '/auto_auth'
|
|
params = { 'username': username, 'password': password, 'email': email, 'full_name': name }
|
|
url += "?" + urllib.urlencode(params)
|
|
world.visit(url)
|
|
|
|
# Save the user info in the world scenario_dict for use in the tests
|
|
user = User.objects.get(username=username)
|
|
world.scenario_dict['USER'] = user
|
|
|
|
|
|
@world.absorb
|
|
def register_by_course_key(course_key, username='robot', password='test', is_staff=False):
|
|
create_user(username, password)
|
|
user = User.objects.get(username=username)
|
|
# Note: this flag makes the user global staff - that is, an edX employee - not a course staff.
|
|
# See courseware.tests.factories for StaffFactory and InstructorFactory.
|
|
if is_staff:
|
|
user.is_staff = True
|
|
user.save()
|
|
CourseEnrollment.enroll(user, course_key)
|
|
|
|
|
|
@world.absorb
|
|
def enroll_user(user, course_key):
|
|
# Activate user
|
|
registration = world.RegistrationFactory(user=user)
|
|
registration.register(user)
|
|
registration.activate()
|
|
# Enroll them in the course
|
|
CourseEnrollment.enroll(user, course_key)
|
|
|
|
|
|
@world.absorb
|
|
def clear_courses():
|
|
# Flush and initialize the module store
|
|
# Note that if your test module gets in some weird state
|
|
# (though it shouldn't), do this manually
|
|
# from the bash shell to drop it:
|
|
# $ mongo test_xmodule --eval "db.dropDatabase()"
|
|
editable_modulestore().collection.drop()
|
|
contentstore().fs_files.drop()
|