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)