Skip to content
Snippets Groups Projects
factories.py 840 B
Newer Older
  • Learn to ignore specific revisions
  • import factory
    
    from funkwhale_api.factories import registry, NoUpdateOnCreate
    
    from funkwhale_api.federation import factories as federation_factories
    
    
    @registry.register
    class MutationFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
        fid = factory.Faker("federation_url")
        uuid = factory.Faker("uuid4")
        created_by = factory.SubFactory(federation_factories.ActorFactory)
        summary = factory.Faker("paragraph")
        type = "update"
    
        class Meta:
            model = "common.Mutation"
    
    
    Eliot Berriot's avatar
    Eliot Berriot committed
    
    @registry.register
    class AttachmentFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
        url = factory.Faker("federation_url")
        uuid = factory.Faker("uuid4")
        actor = factory.SubFactory(federation_factories.ActorFactory)
        file = factory.django.ImageField()
    
        class Meta:
            model = "common.Attachment"