From 9ac6157ba2b9f86598eede69df7b67fa320f95b2 Mon Sep 17 00:00:00 2001
From: Johannes Christ <jc@jchri.st>
Date: Sun, 18 Nov 2018 21:13:47 +0100
Subject: Add the `MessageDeletionContext` model.

---
 api/tests/test_models.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

(limited to 'api/tests')

diff --git a/api/tests/test_models.py b/api/tests/test_models.py
index 1e4af806..92a59f46 100644
--- a/api/tests/test_models.py
+++ b/api/tests/test_models.py
@@ -1,12 +1,14 @@
+from datetime import datetime
+
 from django.test import SimpleTestCase
 
 from ..models import (
     DocumentationLink, Member,
-    Message, ModelReprMixin,
-    OffTopicChannelName, Role,
-    SnakeFact, SnakeIdiom,
-    SnakeName, SpecialSnake,
-    Tag
+    Message, MessageDeletionContext,
+    ModelReprMixin, OffTopicChannelName,
+    Role, SnakeFact,
+    SnakeIdiom, SnakeName,
+    SpecialSnake, Tag
 )
 
 
@@ -51,6 +53,13 @@ class StringDunderMethodTests(SimpleTestCase):
                 content="wooey",
                 embeds=[]
             ),
+            MessageDeletionContext(
+                actor=Member(
+                    id=5555, name='shawn',
+                    discriminator=555, avatar_hash=None
+                ),
+                creation=datetime.utcnow()
+            ),
             Member(
                 id=5, name='bob',
                 discriminator=1, avatar_hash=None
-- 
cgit v1.2.3