refactor: upgrading factory-boy.
This commit is contained in:
@@ -30,7 +30,7 @@ def generate_dummy_image(_unused):
|
||||
)
|
||||
|
||||
|
||||
class CourseCompleteImageConfigurationFactory(DjangoModelFactory):
|
||||
class CourseCompleteImageConfigurationFactory(factory.django.DjangoModelFactory):
|
||||
"""
|
||||
Factory for BadgeImageConfigurations
|
||||
"""
|
||||
@@ -41,7 +41,7 @@ class CourseCompleteImageConfigurationFactory(DjangoModelFactory):
|
||||
icon = factory.LazyAttribute(generate_dummy_image)
|
||||
|
||||
|
||||
class BadgeClassFactory(DjangoModelFactory):
|
||||
class BadgeClassFactory(factory.django.DjangoModelFactory):
|
||||
"""
|
||||
Factory for BadgeClass
|
||||
"""
|
||||
@@ -64,7 +64,7 @@ class RandomBadgeClassFactory(BadgeClassFactory):
|
||||
slug = factory.lazy_attribute(lambda _: 'test_slug_' + str(random()).replace('.', '_'))
|
||||
|
||||
|
||||
class BadgeAssertionFactory(DjangoModelFactory):
|
||||
class BadgeAssertionFactory(factory.django.DjangoModelFactory):
|
||||
"""
|
||||
Factory for BadgeAssertions
|
||||
"""
|
||||
@@ -78,7 +78,7 @@ class BadgeAssertionFactory(DjangoModelFactory):
|
||||
image_url = 'http://example.com/image.png'
|
||||
|
||||
|
||||
class CourseEventBadgesConfigurationFactory(DjangoModelFactory):
|
||||
class CourseEventBadgesConfigurationFactory(factory.django.DjangoModelFactory):
|
||||
"""
|
||||
Factory for CourseEventsBadgesConfiguration
|
||||
"""
|
||||
|
||||
@@ -40,6 +40,7 @@ from lms.djangoapps.verify_student.models import VerificationDeadline
|
||||
from lms.djangoapps.verify_student.services import IDVerificationService
|
||||
from lms.djangoapps.verify_student.tests.factories import SoftwareSecurePhotoVerificationFactory
|
||||
from openedx.core.djangoapps.content.course_overviews.models import CourseOverview
|
||||
from openedx.core.djangoapps.content.course_overviews.tests.factories import CourseOverviewFactory
|
||||
from openedx.core.djangoapps.self_paced.models import SelfPacedConfiguration
|
||||
from openedx.core.djangoapps.site_configuration.tests.factories import SiteFactory # pylint: disable=unused-import
|
||||
from openedx.core.djangoapps.user_api.preferences.api import set_user_preference
|
||||
@@ -1016,7 +1017,8 @@ class TestScheduleOverrides(SharedModuleStoreTestCase):
|
||||
course = create_self_paced_course_run(days_till_start=-1, org_id='TestOrg')
|
||||
DynamicUpgradeDeadlineConfiguration.objects.create(enabled=True)
|
||||
if enroll_first:
|
||||
enrollment = CourseEnrollmentFactory(course_id=course.id, mode=CourseMode.AUDIT, course__self_paced=True)
|
||||
course_overview = CourseOverviewFactory.create(self_paced=True)
|
||||
enrollment = CourseEnrollmentFactory(course_id=course.id, mode=CourseMode.AUDIT, course=course_overview)
|
||||
OrgDynamicUpgradeDeadlineConfiguration.objects.create(
|
||||
enabled=org_config_enabled, opt_out=org_config_opt_out, org_id=course.id.org
|
||||
)
|
||||
@@ -1024,7 +1026,8 @@ class TestScheduleOverrides(SharedModuleStoreTestCase):
|
||||
enabled=course_config_enabled, opt_out=course_config_opt_out, course_id=course.id
|
||||
)
|
||||
if not enroll_first:
|
||||
enrollment = CourseEnrollmentFactory(course_id=course.id, mode=CourseMode.AUDIT, course__self_paced=True)
|
||||
course_overview = CourseOverviewFactory.create(self_paced=True)
|
||||
enrollment = CourseEnrollmentFactory(course_id=course.id, mode=CourseMode.AUDIT, course=course_overview)
|
||||
|
||||
# The enrollment has a schedule, and the upgrade_deadline is set when expected_dynamic_deadline is True
|
||||
if not enroll_first:
|
||||
|
||||
@@ -10,7 +10,7 @@ from common.djangoapps.student.tests.factories import UserFactory
|
||||
from lms.djangoapps.experiments.models import ExperimentData, ExperimentKeyValue
|
||||
|
||||
|
||||
class ExperimentDataFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class ExperimentDataFactory(factory.django.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class Meta:
|
||||
model = ExperimentData
|
||||
|
||||
@@ -20,7 +20,7 @@ class ExperimentDataFactory(factory.DjangoModelFactory): # lint-amnesty, pylint
|
||||
value = factory.Faker('word')
|
||||
|
||||
|
||||
class ExperimentKeyValueFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class ExperimentKeyValueFactory(factory.django.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class Meta:
|
||||
model = ExperimentKeyValue
|
||||
|
||||
|
||||
@@ -5,7 +5,7 @@ from common.djangoapps.student.tests.factories import UserFactory
|
||||
from lms.djangoapps.survey.models import SurveyAnswer, SurveyForm
|
||||
|
||||
|
||||
class SurveyFormFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class SurveyFormFactory(factory.django.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class Meta:
|
||||
model = SurveyForm
|
||||
|
||||
@@ -13,7 +13,7 @@ class SurveyFormFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: di
|
||||
form = '<form>First name:<input type="text" name="firstname"/></form>'
|
||||
|
||||
|
||||
class SurveyAnswerFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class SurveyAnswerFactory(factory.django.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
||||
class Meta:
|
||||
model = SurveyAnswer
|
||||
|
||||
|
||||
Reference in New Issue
Block a user