More flexible auth signature factory

api/funkwhale_api/federation/factories.py

@@ -16,7 +16,12 @@ class SignatureAuthFactory(factory.Factory):
     key = factory.LazyFunction(lambda: keys.get_key_pair()[0])
     key_id = factory.Faker('url')
     use_auth_header = False
-
+    headers = [
+        '(request-target)',
+        'user-agent',
+        'host',
+        'date',
+        'content-type',]
     class Meta:
         model = requests_http_signature.HTTPSignatureAuth
 
@@ -29,3 +34,28 @@ class SignedRequestFactory(factory.Factory):
 
     class Meta:
         model = requests.Request
+
+    @factory.post_generation
+    def headers(self, create, extracted, **kwargs):
+        default_headers = {
+            'User-Agent': 'Test',
+            'Host': 'test.host',
+            'Date': 'Right now',
+            'Content-Type': 'application/activity+json'
+        }
+        if extracted:
+            default_headers.update(extracted)
+        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