class EntityFactory(DjangoModelFactory):
resource = SubFactory(ResourceFactory)
string = Sequence(lambda n: 'string {0}'.format(n))
class Meta:
model = Entity
class CourseEnrollmentFactory(DjangoModelFactory):
FACTORY_FOR = CourseEnrollment
user = SubFactory(UserFactory)
course_id = u'edX/toy/2012_Fall'
class NodeFactory(AbstractNodeFactory):
category = 'hypothesis'
parent = SubFactory(ProjectFactory)
class RoleEligibilityConditionFactory(DjangoModelFactory):
class Meta:
model = tickets.RoleEligibilityCondition
checklistitem = SubFactory(CheckListItemFactory)
role = "Teacher"
class ContainerDescriptorFactory(LeafDescriptorFactory):
"""
Factory to generate XModuleDescriptors that are containers.
"""
runtime = SubFactory(ContainerDescriptorRuntimeFactory)
children = range(3)
class FeaturedFactory(DjangoModelFactory):
class Meta:
abstract = True
page = SubFactory(WagtailHomepageFactory)
class HomepageFeaturedHighlightsFactory(FeaturedFactory):
class Meta:
model = HomepageFeaturedHighlights
highlight = SubFactory(HighlightFactory)
class IdenticalTranslationFactory(TranslationFactory):
entity = SubFactory(EntityFactory, string=SelfAttribute("..string"))
class TranslatedResourceFactory(DjangoModelFactory):
resource = SubFactory(ResourceFactory)
locale = SubFactory(LocaleFactory)
class Meta:
model = TranslatedResource
class EventValueFactory(CustomFieldValueFactory):
custom_field = SubFactory(EventFactory)
value = 'true'
class ChangedEntityLocaleFactory(DjangoModelFactory):
entity = SubFactory(EntityFactory)
locale = SubFactory(LocaleFactory)
class Meta:
model = ChangedEntityLocale
class ParticipantUserFactory(ParticipantFactory):
user = SubFactory(UserFactory)
class BFactoryWithSub(DjangoModelFactory):
class Meta:
model = ModelB
a = SubFactory(AFactory)
class StatsFactory(DjangoModelFactory):
resource = SubFactory(ResourceFactory)
locale = SubFactory(LocaleFactory)
class Meta:
model = Stats
class FacebookBannerLocaleFactory(Factory):
FACTORY_FOR = models.FacebookBannerLocale
banner = SubFactory(FacebookBannerFactory)
class SubpageFactory(DjangoModelFactory):
project = SubFactory(ProjectFactory)
name = Sequence(lambda n: "subpage%s" % n)
class Meta:
model = Subpage
class FacebookBannerInstanceFactory(Factory):
FACTORY_FOR = models.FacebookBannerInstance
banner = SubFactory(FacebookBannerFactory)
user = SubFactory(FacebookUserFactory)
text = Sequence(lambda n: 'test%s' % n)
class StudentInfoFactory(DjangoModelFactory):
FACTORY_FOR = XModuleStudentInfoField
field_name = 'existing_field'
value = json.dumps('old_value')
student = SubFactory(UserFactory)
class HomepageFeaturedBlogsFactory(FeaturedFactory):
class Meta:
model = HomepageFeaturedBlogs
blog = SubFactory(BlogPageFactory)
class SchoolClassFactory(BaseFactory):
class Meta:
model = StudentSchoolClass
student = SubFactory('videostore.db.factories.StudentFactory')
school_class = SubFactory('videostore.db.factories.SchoolClassFactory')
class UserProfileFactory(DjangoModelFactory):
FACTORY_FOR = models.UserProfile
user = SubFactory(UserFactory)
name = Sequence(lambda n: 'test{0}'.format(n))
class CategoryFeatureFactory(DjangoModelFactory):
class Meta:
model = 'device.CategoryFeature'
name = faker.Faker('name')
category = SubFactory(CategoryFactory)
class NoEventRoleExclusionFactory(DjangoModelFactory):
class Meta:
model = tickets.RoleExclusion
condition = SubFactory(RoleEligibilityConditionFactory)
role = "Teacher"
class FacebookClickStatsFactory(Factory):
FACTORY_FOR = models.FacebookClickStats
banner_instance = SubFactory(FacebookBannerInstanceFactory)
class BankConfirmationServiceFactory(ConfirmationServiceFactory):
bank = SubFactory(BankFactory)
class Meta:
model = BankConfirmationService
class AppNotificationFactory(Factory):
FACTORY_FOR = models.AppNotification
user = SubFactory(FacebookUserFactory)
message = Sequence(lambda n: random.choice(MESSAGES.keys()))
class ConfigFactory(BaseFactory):
"""This is a dummy Config factory, Referencing PropertyGroups with
no a-priori property relationships and arbitrary Accounts.
"""
class Meta:
model = Config
id = 1
# `PropertyGroup`s
member_group = SubFactory(MemberPropertyGroupFactory)
network_access_group = SubFactory(PropertyGroupFactory)
violation_group = SubFactory(PropertyGroupFactory)
traffic_limit_exceeded_group = SubFactory(PropertyGroupFactory)
external_group = SubFactory(PropertyGroupFactory)
payment_in_default_group = SubFactory(PropertyGroupFactory,
name='Payment in default',
granted=frozenset(
("payment_in_default", )),
denied=frozenset(
("network_access", "userwww",
"userdb")))
blocked_group = SubFactory(PropertyGroupFactory)
caretaker_group = SubFactory(PropertyGroupFactory)
treasurer_group = SubFactory(PropertyGroupFactory)
pre_member_group = SubFactory(PropertyGroupFactory)
# `Account`s
membership_fee_account = SubFactory(AccountFactory)
membership_fee_bank_account = SubFactory(BankAccountFactory)
class FacebookAccountLinkFactory(Factory):
FACTORY_FOR = models.FacebookAccountLink
facebook_user = SubFactory(FacebookUserFactory)
affiliates_user = SubFactory(UserFactory)
class PointerFactory(ModularOdmFactory):
class Meta:
model = Pointer
node = SubFactory(NodeFactory)
class BoxUserSettingsFactory(ModularOdmFactory):
FACTORY_FOR = BoxUserSettings
owner = SubFactory(UserFactory)
