yetangitu/funkwhale
1
0
Fork 0
mirror of https://code.eliotberriot.com/funkwhale/funkwhale.git synced 2025-10-04 05:49:16 +02:00
Code Issues Releases Wiki Activity

Actor factory and fixture

Browse source
This commit is contained in:
Eliot Berriot 2018-04-01 10:17:06 +02:00
parent 3650c3699b
commit 6fbf8fa44c
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
3 changed files with 44 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

32
api/funkwhale_api/federation/factories.py
View file

@ -5,6 +5,7 @@ import requests_http_signature
from funkwhale_api.factories import registry
from . import keys
from . import models
registry.register(keys.get_key_pair, name='federation.KeyPair')
@ -48,14 +49,23 @@ class SignedRequestFactory(factory.Factory):
self.headers.update(default_headers)
# @registry.register
# class ActorFactory(factory.DjangoModelFactory):
# url = factory.Faker('url')
# inbox_url = factory.Faker('url')
# outbox_url = factory.Faker('url')
# public_key = factory.LazyFunction(lambda: keys.get_key_pair()[1])
# preferred_username = factory.Faker('username')
# summary = factory.Faker('paragraph')
#
# class Meta:
# model = models.Actor
@registry.register
class ActorFactory(factory.DjangoModelFactory):
url = factory.Faker('url')
inbox_url = factory.Faker('url')
outbox_url = factory.Faker('url')
public_key = None
private_key = None
preferred_username = factory.Faker('user_name')
summary = factory.Faker('paragraph')
class Meta:
model = models.Actor
@classmethod
def _generate(cls, create, attrs):
has_public = attrs.get('public_key') is None
has_private = attrs.get('private_key') is None
if not has_public and not has_private:
attrs['private_key'], attrs['public'] = keys.get_key_pair()
return super()._generate(create, attrs)