Fully test logs app

apps/logs/signals.py

@@ -54,8 +54,7 @@ def save_object(sender, instance, **kwargs):
         # On récupère alors l'utilisateur·trice connecté·e à la VM, et on récupère la note associée
         ip = "127.0.0.1"
         username = getpass.getuser()
-        if User.objects.filter(username=username).exists():
-            user = User.objects.get(username=username)
+        user = User.objects.get(username=username) if User.objects.filter(username=username).exists() else None
 
     # On n'enregistre pas les connexions
     # noinspection PyProtectedMember
@@ -113,8 +112,7 @@ def delete_object(sender, instance, **kwargs):
         # On récupère alors l'utilisateur·trice connecté·e à la VM, et on récupère la note associée
         ip = "127.0.0.1"
         username = getpass.getuser()
-        if User.objects.filter(username=username).exists():
-            user = User.objects.get(username=username)
+        user = User.objects.get(username=username) if User.objects.filter(username=username).exists() else None
 
     # On crée notre propre sérialiseur JSON pour pouvoir sauvegarder les modèles
     class CustomSerializer(ModelSerializer):
@@ -131,4 +129,4 @@ def delete_object(sender, instance, **kwargs):
                              previous=instance_json,
                              data="",
                              action="delete"
-                             ).save()
+                             )

apps/logs/tests.py

@@ -0,0 +1,21 @@
+from django.contrib.auth.models import User
+from django.contrib.contenttypes.models import ContentType
+from django.core.exceptions import ValidationError
+from django.test import TestCase
+
+from .models import Changelog
+
+
+class TestChangelog(TestCase):
+    def test_logs(self):
+        user = User.objects.create(email="admin@example.com")
+        self.assertTrue(Changelog.objects.filter(action="create", instance_pk=user.pk,
+                                                 model=ContentType.objects.get_for_model(User)).exists())
+        old_user_pk = user.pk
+        user.delete()
+        self.assertTrue(Changelog.objects.filter(action="delete", instance_pk=old_user_pk,
+                                                 model=ContentType.objects.get_for_model(User)).exists())
+
+        changelog = Changelog.objects.first()
+        self.assertRaises(ValidationError, changelog.delete)
+        str(Changelog.objects.all())