class CommentFactory(django.DjangoModelFactory):
class Meta:
model = "movies.Comment"
body = Faker("word")
movie = SubFactory(MovieFactory)
class PermissionFactory(DjangoModelFactory):
FACTORY_FOR = Permission
name = Sequence(lambda n: 'test%s' % n)
codename = Sequence(lambda n: 'test%s' % n)
content_type = SubFactory(ContentTypeFactory)
class ContainerDescriptorFactory(LeafDescriptorFactory):
"""
Factory to generate XModuleDescriptors that are containers.
"""
runtime = SubFactory(ContainerDescriptorRuntimeFactory)
children = list(range(3))
class CarImageFactory(DjangoModelFactory):
car = SubFactory(CarFactory)
image = ImageField(width=1024, height=768)
class Meta:
model = CarImage
class LotteryInfoFactory(DjangoModelFactory):
class Meta:
model = models.LotteryInfo
participant = SubFactory(ParticipantFactory)
class SubpageFactory(DjangoModelFactory):
project = SubFactory(ProjectFactory)
name = Sequence(lambda n: 'subpage%s' % n)
class Meta:
model = Subpage
class DropboxUserSettingsFactory(ModularOdmFactory):
FACTORY_FOR = DropboxUserSettings
owner = SubFactory(UserFactory)
access_token = Sequence(lambda n: 'abcdef{0}'.format(n))
class NodeLogFactory(ModularOdmFactory):
class Meta:
model = NodeLog
action = 'file_added'
user = SubFactory(UserFactory)
class WatchConfigFactory(ModularOdmFactory):
class Meta:
model = WatchConfig
node = SubFactory(NodeFactory)
class NodeFactory(AbstractNodeFactory):
category = 'hypothesis'
parent = SubFactory(ProjectFactory)
class PointerFactory(ModularOdmFactory):
class Meta:
model = Pointer
node = SubFactory(NodeFactory)
class RecommendationFactory(BaseFactory):
"""Recommendation Factory."""
text = FuzzyText()
accuracy = SubFactory(RecommendationAccuracy)
class Meta:
"""Factory Configuration."""
model = Recommendation
@post_generation
def incident_priorities(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for priority in extracted:
self.incident_priorities.append(priority)
@post_generation
def incident_types(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for incident_type in extracted:
self.incident_types.append(incident_type)
@post_generation
def matched_terms(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for term in extracted:
self.matched_terms.append(term)
@post_generation
def documents(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for document in extracted:
self.documents.append(document)
@post_generation
def team_contacts(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for team_contact in extracted:
self.team_contacts.append(team_contact)
@post_generation
def service_contacts(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for service_contact in extracted:
self.service_contacts.append(service_contact)
@post_generation
def individual_contacts(self, create, extracted, **kwargs):
if not create:
return
if extracted:
for individual_contact in extracted:
self.individual_contacts.append(individual_contact)
class DocumentTagsFactory(DjangoModelFactory):
document = SubFactory(DocumentFactory)
tag = SubFactory(TagFactory)
class Meta:
model = models.Document.tags.through
class FileFactory(DjangoModelFactory):
name = Faker("name")
document = SubFactory(DocumentFactory)
class Meta:
model = models.File
class IdenticalTranslationFactory(TranslationFactory):
entity = SubFactory(EntityFactory, string=SelfAttribute('..string'))
class RetractionFactory(SanctionFactory):
class Meta:
model = Retraction
user = SubFactory(UserFactory)
class TranslatedResourceFactory(DjangoModelFactory):
resource = SubFactory(ResourceFactory)
locale = SubFactory(LocaleFactory)
class Meta:
model = TranslatedResource
class EmbargoFactory(SanctionFactory):
class Meta:
model = Embargo
user = SubFactory(UserFactory)
class LinkFactory(DjangoModelFactory):
FACTORY_FOR = Link
title = Sequence(lambda n: 'link title {0}'.format(n))
dataset = SubFactory(DatasetFactory)
format = SubFactory(FormatFactory)
class RegistrationApprovalFactory(SanctionFactory):
class Meta:
model = RegistrationApproval
user = SubFactory(UserFactory)
class DropboxNodeSettingsFactory(ModularOdmFactory):
FACTORY_FOR = DropboxNodeSettings
owner = SubFactory(ProjectFactory)
user_settings = SubFactory(DropboxUserSettingsFactory)
folder = 'Camera Uploads'
class AuthFactory(base.Factory):
class Meta:
model = Auth
user = SubFactory(UserFactory)
class IdentifierFactory(ModularOdmFactory):
FACTORY_FOR = Identifier
referent = SubFactory(RegistrationFactory)
category = 'carpid'
value = 'carp:/24601'
class OneDriveUserSettingsFactory(DjangoModelFactory):
class Meta:
model = UserSettings
owner = SubFactory(UserFactory)
class CourseEnrollmentFactory(DjangoModelFactory):
FACTORY_FOR = CourseEnrollment
user = SubFactory(UserFactory)
course_id = u'edX/toy/2012_Fall'
class EntityFactory(DjangoModelFactory):
resource = SubFactory(ResourceFactory)
string = Sequence(lambda n: 'string {0}'.format(n))
class Meta:
model = Entity
class CollectionMemberFactory(DjangoModelFactory):
collection = SubFactory(CollectionFactory, parent=None)
class ChangedEntityLocaleFactory(DjangoModelFactory):
entity = SubFactory(EntityFactory)
locale = SubFactory(LocaleFactory)
class Meta:
model = ChangedEntityLocale
class StudyInvitationFactory(DjangoModelFactory):
class Meta:
model = StudyInvitation
doctor = SubFactory(DoctorFactory)
study = SubFactory(StudyFactory)
class UserSettingsFactory(DjangoModelFactory):
FACTORY_FOR = UserSettings
user = SubFactory(UserFactory)
