Some basic testing.

g2g

api/consts.py

@@ -36,7 +36,7 @@
 ALL_FIELDS = '__all__'
 ID_FIELD = 'id'
 USERNAME_FIELD = 'username'
-MESSAGES_FIELD = 'messages'
+MESSAGES_FIELD = 'message'
 PASSWORD_FIELD = 'password'
 SENDER_FIELD = 'sender'
 RECEIVER_FIELD = 'receiver'

api/models.py

@@ -22,6 +22,9 @@ def __str__(self):
 
 @receiver(post_save, sender=settings.AUTH_USER_MODEL)
 def create_auth_token(sender, instance=None, created=False, **kwargs):
+    """
+    Django signal to create a token each time a user model is created.
+    """
     if created:
         Token.objects.create(user=instance)
 

api/serializers/message_serializers.py

@@ -2,7 +2,6 @@
 from api.consts import *
 from api.models import Message
 from api.serializers.user_serializers import UserMessageDisplaySerializer
-from api.user import CustomUser
 
 
 class MessageCreateSerializer(serializers.ModelSerializer):
@@ -22,6 +21,9 @@ class Meta:
     creation_data = serializers.SerializerMethodField()
 
     def get_creation_data(self, obj):
+        """
+        Used in order to format the date when displaying!
+        """
         return obj.creation_data.strftime(DATE_FORMAT)
 
 
@@ -36,5 +38,8 @@ class Meta:
     creation_data = serializers.SerializerMethodField()
 
     def get_creation_data(self, obj):
+        """
+        Used in order to format the date when displaying!
+        """
         return obj.creation_data.strftime(DATE_FORMAT)
 

api/serializers/user_serializers.py

@@ -1,6 +1,5 @@
 from rest_framework import serializers
 from api.consts import *
-from api.models import Message
 from api.user import CustomUser
 
 

api/tests.py

@@ -1,3 +1,84 @@
 from django.test import TestCase
+from rest_framework.test import APITestCase
+from .models import Message, CustomUser
+from .consts import *
+from api.serializers.message_serializers import MessageCreateSerializer, MessageFullDisplaySerializer
 
-# Create your tests here.
+# region model testing
+
+
+class MessageModelTestCase(TestCase):
+
+    @classmethod
+    def setUpTestData(cls):
+        cls.sender = CustomUser.objects.create(username="SENDER")
+        cls.receiver = CustomUser.objects.create(username="RECEIVER")
+
+        cls.message = Message.objects.create(sender=cls.sender,
+                                             receiver=cls.receiver,
+                                             message='TEST MESSAGE',
+                                             subject='TEST SUBJECT')
+
+    def test_message_content(self):
+        self.assertEqual('TEST MESSAGE', self.message.message)
+
+    def test_subject(self):
+        self.assertEqual('TEST SUBJECT', self.message.subject)
+
+    def test_sender(self):
+        self.assertEqual(self.sender.pk, self.message.sender.pk)
+
+    def test_receiver(self):
+        self.assertEqual(self.receiver.pk, self.message.receiver.pk)
+
+    def test_is_read(self):
+        self.assertEqual(False, self.message.is_read)
+
+    def test_str(self):
+        self.assertEqual("SENDER to RECEIVER", str(self.message))
+
+
+# endregion
+
+
+# region serializer testing
+
+
+class MessageSerializersTests(APITestCase):
+
+    @classmethod
+    def setUpTestData(cls):
+        cls.sender = CustomUser.objects.create(username="SENDER")
+        cls.receiver = CustomUser.objects.create(username="RECEIVER")
+
+        cls.message = Message.objects.create(sender=cls.sender,
+                                             receiver=cls.receiver,
+                                             message='TEST MESSAGE',
+                                             subject='TEST SUBJECT')
+
+        cls.message_data = {
+            SENDER_FIELD: str(cls.sender.pk),
+            RECEIVER_FIELD: str(cls.receiver.pk),
+            MESSAGES_FIELD: 'TEST MESSAGE',
+            SUBJECT_FIELD: 'TEST SUBJECT',
+            IS_READ_FIELD: False
+        }
+
+    def test_create_serializer(self):
+        create_serializer = MessageCreateSerializer(data=self.message_data)
+
+        self.assertTrue(create_serializer.is_valid())
+
+    def test_display_serializer(self):
+        display_serializer = MessageFullDisplaySerializer(self.message)
+
+        message_data = display_serializer.data
+
+        self.assertEqual('TEST MESSAGE', message_data[MESSAGES_FIELD])
+        self.assertEqual('TEST SUBJECT', message_data[SUBJECT_FIELD])
+        self.assertEqual(False, message_data[IS_READ_FIELD])
+        self.assertEqual('SENDER', message_data[SENDER_FIELD][USERNAME_FIELD])
+        self.assertEqual('RECEIVER', message_data[RECEIVER_FIELD][USERNAME_FIELD])
+
+
+# endregion

api/views.py

@@ -10,7 +10,6 @@
 from rest_framework.permissions import AllowAny, IsAuthenticated
 from rest_framework.authtoken.models import Token
 from django.db.models import Q
-from rest_framework.request import Request
 
 
 # region utils
@@ -49,6 +48,9 @@ def get_serializer_class(self):
         return MessageDisplaySerializer
 
     def destroy(self, request, *args, **kwargs):
+        """
+        When destroying a record, verifying the logged user is either the sender or receiver in order to do so!
+        """
         request_token = request.auth.key
         sender = self.get_object().sender
         receiver = self.get_object().receiver
@@ -132,6 +134,9 @@ def create(self, request, *args, **kwargs):
 
     @action(detail=False, methods=[POST])
     def register(self, request, *args, **kwargs):
+        """
+        Creating a new user, overrides default create
+        """
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
         self.perform_create(serializer)