refactor: Move OrgStaffFactory helper to common/
This commit is contained in:
@@ -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 OrgStaffRole
|
||||
|
||||
# Factories are self documenting
|
||||
|
||||
@@ -272,6 +273,20 @@ class InstructorFactory(UserFactory):
|
||||
CourseInstructorRole(extracted).add_users(self)
|
||||
|
||||
|
||||
class OrgStaffFactory(UserFactory):
|
||||
"""
|
||||
Given a course Location, returns a User object with org-staff
|
||||
permissions for `course`.
|
||||
"""
|
||||
last_name = 'Org-Staff'
|
||||
|
||||
@factory.post_generation
|
||||
def course_key(self, _create, extracted, **kwargs):
|
||||
if extracted is None:
|
||||
raise ValueError('Must specify a CourseKey for an org-staff user')
|
||||
OrgStaffRole(extracted.org).add_users(self)
|
||||
|
||||
|
||||
class StaffFactory(UserFactory):
|
||||
"""
|
||||
Given a course Location, returns a User object with staff
|
||||
|
||||
@@ -18,7 +18,6 @@ from lms.djangoapps.courseware.models import (
|
||||
)
|
||||
from common.djangoapps.student.roles import (
|
||||
OrgInstructorRole,
|
||||
OrgStaffRole
|
||||
)
|
||||
# Imported to re-export
|
||||
from common.djangoapps.student.tests.factories import UserFactory
|
||||
@@ -39,20 +38,6 @@ class UserProfileFactory(StudentUserProfileFactory):
|
||||
# course_key(self, create, extracted, **kwargs)
|
||||
# pylint: disable=unused-argument
|
||||
|
||||
class OrgStaffFactory(UserFactory):
|
||||
"""
|
||||
Given a course Location, returns a User object with org-staff
|
||||
permissions for `course`.
|
||||
"""
|
||||
last_name = "Org-Staff"
|
||||
|
||||
@factory.post_generation
|
||||
def course_key(self, create, extracted, **kwargs):
|
||||
if extracted is None:
|
||||
raise ValueError("Must specify a CourseKey for an org-staff user")
|
||||
OrgStaffRole(extracted.org).add_users(self)
|
||||
|
||||
|
||||
class OrgInstructorFactory(UserFactory):
|
||||
"""
|
||||
Given a course Location, returns a User object with org-instructor
|
||||
|
||||
@@ -12,11 +12,11 @@ from django.urls import reverse
|
||||
from common.djangoapps.student.tests.factories import BetaTesterFactory
|
||||
from common.djangoapps.student.tests.factories import GlobalStaffFactory
|
||||
from common.djangoapps.student.tests.factories import InstructorFactory
|
||||
from common.djangoapps.student.tests.factories import OrgStaffFactory
|
||||
from common.djangoapps.student.tests.factories import StaffFactory
|
||||
from lms.djangoapps.courseware.access import has_access
|
||||
from lms.djangoapps.courseware.tests.factories import (
|
||||
OrgInstructorFactory,
|
||||
OrgStaffFactory,
|
||||
)
|
||||
from lms.djangoapps.courseware.tests.helpers import CourseAccessTestMixin, LoginEnrollmentTestCase
|
||||
from openedx.features.enterprise_support.tests.mixins.enterprise import EnterpriseTestConsentRequired
|
||||
|
||||
@@ -19,11 +19,11 @@ from common.djangoapps.course_modes.tests.factories import CourseModeFactory
|
||||
from common.djangoapps.student.tests.factories import BetaTesterFactory
|
||||
from common.djangoapps.student.tests.factories import GlobalStaffFactory
|
||||
from common.djangoapps.student.tests.factories import InstructorFactory
|
||||
from common.djangoapps.student.tests.factories import OrgStaffFactory
|
||||
from common.djangoapps.student.tests.factories import StaffFactory
|
||||
from lms.djangoapps.courseware.module_render import load_single_xblock
|
||||
from lms.djangoapps.courseware.tests.factories import (
|
||||
OrgInstructorFactory,
|
||||
OrgStaffFactory,
|
||||
)
|
||||
from lms.djangoapps.courseware.tests.helpers import MasqueradeMixin
|
||||
from lms.djangoapps.discussion.django_comment_client.tests.factories import RoleFactory
|
||||
|
||||
@@ -17,10 +17,10 @@ from common.djangoapps.student.tests.factories import BetaTesterFactory
|
||||
from common.djangoapps.student.tests.factories import TEST_PASSWORD, CourseEnrollmentFactory, UserFactory
|
||||
from common.djangoapps.student.tests.factories import GlobalStaffFactory
|
||||
from common.djangoapps.student.tests.factories import InstructorFactory
|
||||
from common.djangoapps.student.tests.factories import OrgStaffFactory
|
||||
from common.djangoapps.student.tests.factories import StaffFactory
|
||||
from lms.djangoapps.courseware.tests.factories import (
|
||||
OrgInstructorFactory,
|
||||
OrgStaffFactory,
|
||||
)
|
||||
from lms.djangoapps.courseware.tests.helpers import MasqueradeMixin
|
||||
from lms.djangoapps.discussion.django_comment_client.tests.factories import RoleFactory
|
||||
|
||||
@@ -24,6 +24,7 @@ from common.djangoapps.course_modes.tests.factories import CourseModeFactory
|
||||
from common.djangoapps.student.tests.factories import BetaTesterFactory
|
||||
from common.djangoapps.student.tests.factories import GlobalStaffFactory
|
||||
from common.djangoapps.student.tests.factories import InstructorFactory
|
||||
from common.djangoapps.student.tests.factories import OrgStaffFactory
|
||||
from common.djangoapps.student.tests.factories import StaffFactory
|
||||
from common.djangoapps.util.date_utils import strftime_localized_html
|
||||
from lms.djangoapps.experiments.models import ExperimentData
|
||||
@@ -32,7 +33,6 @@ from lms.djangoapps.commerce.utils import EcommerceService
|
||||
from lms.djangoapps.course_goals.api import add_course_goal, remove_course_goal
|
||||
from lms.djangoapps.courseware.tests.factories import (
|
||||
OrgInstructorFactory,
|
||||
OrgStaffFactory,
|
||||
)
|
||||
from lms.djangoapps.courseware.tests.helpers import get_expiration_banner_text
|
||||
from lms.djangoapps.courseware.utils import verified_upgrade_deadline_link
|
||||
|
||||
Reference in New Issue
Block a user