50 lines
1.0 KiB
Python
50 lines
1.0 KiB
Python
import factory
|
|
from student.models import (User, UserProfile, Registration,
|
|
CourseEnrollmentAllowed)
|
|
from django.contrib.auth.models import Group
|
|
from datetime import datetime
|
|
import uuid
|
|
|
|
|
|
class UserProfileFactory(factory.Factory):
|
|
FACTORY_FOR = UserProfile
|
|
|
|
user = None
|
|
name = 'Robot Studio'
|
|
courseware = 'course.xml'
|
|
|
|
|
|
class RegistrationFactory(factory.Factory):
|
|
FACTORY_FOR = Registration
|
|
|
|
user = None
|
|
activation_key = uuid.uuid4().hex
|
|
|
|
|
|
class UserFactory(factory.Factory):
|
|
FACTORY_FOR = User
|
|
|
|
username = 'robot'
|
|
email = 'robot@edx.org'
|
|
password = 'test'
|
|
first_name = 'Robot'
|
|
last_name = 'Tester'
|
|
is_staff = False
|
|
is_active = True
|
|
is_superuser = False
|
|
last_login = datetime.now()
|
|
date_joined = datetime.now()
|
|
|
|
|
|
class GroupFactory(factory.Factory):
|
|
FACTORY_FOR = Group
|
|
|
|
name = 'test_group'
|
|
|
|
|
|
class CourseEnrollmentAllowedFactory(factory.Factory):
|
|
FACTORY_FOR = CourseEnrollmentAllowed
|
|
|
|
email = 'test@edx.org'
|
|
course_id = 'edX/test/2012_Fall'
|