Implement ReminderCreationTests

1 files changed, 35 insertions, 0 deletions

diff --git a/pydis_site/apps/api/tests/test_reminders.py b/pydis_site/apps/api/tests/test_reminders.py
index 01f6d8e4..cb5e5867 100644
--- a/pydis_site/apps/api/tests/test_reminders.py
+++ b/pydis_site/apps/api/tests/test_reminders.py
@@ -1,6 +1,9 @@
+from datetime import datetime
+
 from django_hosts.resolvers import reverse
 
 from .base import APISubdomainTestCase
+from ..models import User
 
 
 class UnauthedReminderAPITests(APISubdomainTestCase):
@@ -40,3 +43,35 @@ class EmptyDatabaseReminderAPITests(APISubdomainTestCase):
         response = self.client.delete(url)
 
         self.assertEqual(response.status_code, 404)
+
+
+class ReminderCreationTests(APISubdomainTestCase):
+    def setUp(self):
+        super().setUp()
+        self.author = User.objects.create(
+            id=1234,
+            name='Mermaid Man',
+            discriminator=1234,
+            avatar_hash=None,
+        )
+        self.data = {
+            'author': self.author.id,
+            'content': 'Remember to...wait what was it again?',
+            'expiration': datetime.utcnow().isoformat(),
+            'jump_url': "https://www.google.com",
+            'channel_id': 123,
+        }
+        url = reverse('bot:reminder-list', host='api')
+        response = self.client.post(url, data=self.data)
+        self.assertEqual(response.status_code, 201)
+
+    def test_reminder_in_full_list(self):
+        url = reverse('bot:reminder-list', host='api')
+        response = self.client.get(url)
+
+        self.assertEqual(response.status_code, 200)
+
+        self.data['expiration'] += 'Z'  # Massaging a quirk of the response time format
+        self.data['active'] = True
+        self.data['id'] = 1
+        self.assertEqual(response.json(), [self.data])