cleaned Maraudeur model, created MaraudeurManager, created some tests

utilisateurs/tests.py

@@ -1,5 +1,50 @@
 from django.test import TestCase
 
+from .models import Maraudeur, Professionnel
 # Create your tests here.
 
 #TODO: Un seul objet Maraudeur peut avoir la propriété vraie 'is_referent'
+
+class MaraudeurTestCase(TestCase):
+
+    names = [
+        ('Arthur', 'Gerbaud'),
+        ('Thibault', 'Huet'),
+        ('Jacqueline', 'Julien'),
+    ]
+
+
+    def setUp(self):
+        for name in self.names:
+            Maraudeur.objects.create(*name)
+
+    def test_get_or_create_from_first_and_last_name(self):
+        # Existing Maraudeur
+        get_maraudeur = Maraudeur.objects.get(first_name="Thibault", last_name="Huet")
+        maraudeur, created = Maraudeur.objects.get_or_create('Thibault','Huet')
+        self.assertEqual(created, False)
+        self.assertEqual(maraudeur, get_maraudeur)
+        # Non-existing Maraudeur
+        with self.assertRaises(Maraudeur.DoesNotExist):
+            Maraudeur.objects.get(first_name="Thierry", last_name="Lhermitte")
+        maraudeur, created = Maraudeur.objects.get_or_create('Thierry', 'Lhermitte')
+        self.assertEqual(created, True)
+        self.assertEqual(maraudeur, Maraudeur.objects.get(username="t.lhermitte"))
+
+    def test_set_referent_maraude(self):
+        # Set a first referent, non-existing Maraudeur
+        referent1 = Maraudeur.objects.set_referent("Claudine", "Henry")
+        self.assertEqual(referent1.is_superuser, True)
+        self.assertEqual(referent1, Maraudeur.objects.get_referent())
+        # Set a new referent, existing Maraudeur
+        referent2 = Maraudeur.objects.set_referent("Arthur", "Gerbaud")
+        self.assertEqual(referent2.is_superuser, True)
+        self.assertEqual(referent2, Maraudeur.objects.get_referent())
+        self.test_referent_is_unique()
+
+    def test_referent_is_unique(self):
+        superusers = []
+        for m in Maraudeur.objects.all():
+            if m.is_superuser:
+                superusers.append(m)
+        self.assertLess(len(superusers), 2)