Properly handle `actor` via `PrimaryKeyRelatedField`.

1 files changed, 36 insertions, 2 deletions

diff --git a/pydis_site/apps/api/tests/test_deleted_messages.py b/pydis_site/apps/api/tests/test_deleted_messages.py
index cd5acab0..d1e9f2f5 100644
--- a/pydis_site/apps/api/tests/test_deleted_messages.py
+++ b/pydis_site/apps/api/tests/test_deleted_messages.py
@@ -3,10 +3,10 @@ from datetime import datetime
 from django_hosts.resolvers import reverse
 
 from .base import APISubdomainTestCase
-from ..models import User
+from ..models import MessageDeletionContext, User
 
 
-class DeletedMessagesTests(APISubdomainTestCase):
+class DeletedMessagesWithoutActorTests(APISubdomainTestCase):
     @classmethod
     def setUpTestData(cls):  # noqa
         cls.author = User.objects.create(
@@ -41,3 +41,37 @@ class DeletedMessagesTests(APISubdomainTestCase):
         url = reverse('bot:messagedeletioncontext-list', host='api')
         response = self.client.post(url, data=self.data)
         self.assertEqual(response.status_code, 201)
+        [context] = MessageDeletionContext.objects.all()
+        self.assertIsNone(context.actor)
+
+
+class DeletedMessagesWithActorTests(APISubdomainTestCase):
+    @classmethod
+    def setUpTestData(cls):  # noqa
+        cls.author = cls.actor = User.objects.create(
+            id=12904,
+            name='Joe Armstrong',
+            discriminator=1245,
+            avatar_hash=None
+        )
+
+        cls.data = {
+            'actor': cls.actor.id,
+            'creation': datetime.utcnow().isoformat(),
+            'deletedmessage_set': [
+                {
+                    'author': cls.author.id,
+                    'id': 12903,
+                    'channel_id': 1824,
+                    'content': "I hate trailing commas",
+                    'embeds': []
+                },
+            ]
+        }
+
+    def test_accepts_valid_data_and_sets_actor(self):
+        url = reverse('bot:messagedeletioncontext-list', host='api')
+        response = self.client.post(url, data=self.data)
+        self.assertEqual(response.status_code, 201)
+        [context] = MessageDeletionContext.objects.all()
+        self.assertEqual(context.actor.id, self.actor.id)