57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
"""
|
|
Common mixins for Course API Tests
|
|
"""
|
|
|
|
|
|
from datetime import datetime
|
|
|
|
from common.djangoapps.student.tests.factories import UserFactory, CourseEnrollmentFactory, CourseAccessRoleFactory
|
|
from xmodule.modulestore.tests.factories import ToyCourseFactory # lint-amnesty, pylint: disable=wrong-import-order
|
|
|
|
TEST_PASSWORD = 'edx'
|
|
|
|
|
|
class CourseApiFactoryMixin:
|
|
"""
|
|
Mixin to allow creation of test courses and users.
|
|
"""
|
|
|
|
@staticmethod
|
|
def create_course(end=datetime(2015, 9, 19, 18, 0, 0), **kwargs):
|
|
"""
|
|
Create a course for use in test cases
|
|
"""
|
|
return ToyCourseFactory.create(
|
|
end=end,
|
|
enrollment_start=datetime(2015, 6, 15, 0, 0, 0),
|
|
enrollment_end=datetime(2015, 7, 15, 0, 0, 0),
|
|
emit_signals=True,
|
|
**kwargs
|
|
)
|
|
|
|
@staticmethod
|
|
def create_user(username, is_staff):
|
|
"""
|
|
Create a user as identified by username, email, password and is_staff.
|
|
"""
|
|
return UserFactory(
|
|
username=username,
|
|
email=f'{username}@example.com',
|
|
password=TEST_PASSWORD,
|
|
is_staff=is_staff
|
|
)
|
|
|
|
@staticmethod
|
|
def create_enrollment(**kwargs):
|
|
"""
|
|
Create a CourseEnrollment to use in tests.
|
|
"""
|
|
return CourseEnrollmentFactory(**kwargs)
|
|
|
|
@staticmethod
|
|
def create_courseaccessrole(**kwargs):
|
|
"""
|
|
Create a CourseAccessRole to use in tests.
|
|
"""
|
|
return CourseAccessRoleFactory(**kwargs)
|