Andal.LND/edx-platform
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: Move OrgInstructorFactory helper to common/

Browse Source
This commit is contained in:
stvn
2021-04-13 19:46:16 -07:00
parent 31bc741410
commit f646971c74
6 changed files with 19 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
common/djangoapps/student/tests/factories.py
View File

@@ -29,6 +29,7 @@ from common.djangoapps.student.roles import GlobalStaff
from common.djangoapps.student.roles import CourseBetaTesterRole
from common.djangoapps.student.roles import CourseInstructorRole
from common.djangoapps.student.roles import CourseStaffRole
from common.djangoapps.student.roles import OrgInstructorRole
from common.djangoapps.student.roles import OrgStaffRole
# Factories are self documenting
@@ -233,7 +234,6 @@ class AccountRecoveryFactory(DjangoModelFactory): # lint-amnesty, pylint: disab
is_active = True
# pylint: disable=unused-argument
class BetaTesterFactory(UserFactory):
"""
Given a course Location, returns a User object with beta-tester
@@ -273,6 +273,20 @@ class InstructorFactory(UserFactory):
CourseInstructorRole(extracted).add_users(self)
class OrgInstructorFactory(UserFactory):
"""
Given a course Location, returns a User object with org-instructor
permissions for `course`.
"""
last_name = 'Org-Instructor'
@factory.post_generation
def course_key(self, _create, extracted, **kwargs):
if extracted is None:
raise ValueError('Must specify a CourseKey for an org-instructor user')
OrgInstructorRole(extracted.org).add_users(self)
class OrgStaffFactory(UserFactory):
"""
Given a course Location, returns a User object with org-staff