100% test coverage on location/models.py

src/location/models.py

@@ -4,8 +4,10 @@ from django.db import models
 
 
 class Country(models.Model):
-    """
-    Classe représentant les pays (basée sur la liste ISO 3166 de 2015).
+    """Classe représentant les pays (basée sur la liste ISO 3166 de 2015).
+
+    References:
+        https://fr.wikipedia.org/wiki/ISO_3166
     """
 
     class Meta:
@@ -46,7 +48,7 @@ class Place(models.Model):
     active = models.BooleanField(default=1, verbose_name="Active")
 
     def __str__(self):
-        return "%s (%s)" % (self.name, self.city)
+        return "%s (%s)" % (self.name, self.city if self.city else "?")
 
 
 class Club(models.Model):
@@ -70,4 +72,4 @@ class Club(models.Model):
     active = models.BooleanField(default=1, verbose_name="Active")
 
     def __str__(self):
-        return "%s (à %s)" % (self.name, self.place.city)
+        return "%s (à %s)" % (self.name, self.place.city if self.place.city else "?")

src/location/tests/test_models.py

@@ -0,0 +1,52 @@
+"""Tests liés au modèle des localisations, places, etc."""
+
+from django.test import TestCase
+
+from ..models import Club, Country, Place
+
+
+class TestCountry(TestCase):
+    def test_str_should_contain_name_and_iso2(self):
+        country = Country.objects.create(
+            nameus="Belgium",
+            namefr="Belgique",
+            isonum=56,
+            iso2="BE"
+        )
+
+        self.assertEqual(str(country), "Belgique (BE)")
+
+
+class TestPlace(TestCase):
+    def test_str_should_contain_name_and_city(self):
+        place = Place.objects.create(
+            name="Heaven",
+            postal=1080,
+            country=Country.objects.create(
+                nameus="Belgium",
+                namefr="Belgique",
+                isonum=56,
+                iso2="BE"
+            )
+        )
+
+        self.assertEqual(str(place), "Heaven (?)")
+
+
+class TestClub(TestCase):
+    def test_str_should_contain_name_and_location(self):
+        club = Club.objects.create(
+            name="RSCA",
+            place=Place.objects.create(
+                name="Heaven",
+                postal=1080,
+                country=Country.objects.create(
+                    nameus="Belgium",
+                    namefr="Belgique",
+                    isonum=56,
+                    iso2="BE"
+                )
+            )
+        )
+
+        self.assertEqual(str(club), "RSCA (à ?)")