Changeset 521

Timestamp:

02/01/10 19:36:22 (7 months ago)

Author:

olivier

Message:

fix unit tests for new location models

Location:

trunk/telemeta

Files:

• models/crem.py (modified) (2 diffs)
• models/cremquery.py (modified) (2 diffs)
• tests/model_tests.py (modified) (3 diffs)

trunk/telemeta/models/crem.py

@@ -557 +557 @@
 
     name             = CharField(_('name'), unique=True, max_length=150, required=True)
-    type             = IntegerField(_('type'), choices=TYPE_CHOICES, required=True, db_index=True)
+    type             = IntegerField(_('type'), choices=TYPE_CHOICES, default=OTHER_TYPE, db_index=True)
     complete_type    = ForeignKey('LocationType', related_name="locations", verbose_name=_('complete type'))
     current_location = WeakForeignKey('self', related_name="past_names", 
@@ -577 +577 @@
         return Location.objects.filter(ancestor_relations__ancestor_location=self)
 
+    def add_child(self, other):
+        LocationRelation.objects.create(location=other, ancestor_location=self, is_direct=True)
+        for location in self.ancestors():
+            #FIXME: might raise Duplicate Entry
+            LocationRelation.objects.create(location=other, ancestor_location=location)
+            
+    def add_parent(self, other):
+        LocationRelation.objects.create(location=self, ancestor_location=other, is_direct=True)
+        for location in self.descendants():
+            #FIXME: might raise Duplicate Entry
+            LocationRelation.objects.create(location=location, ancestor_location=other)
+
     def countries(self):
         if self.type == self.COUNTRY:
             return Location.objects.filter(pk=self.id)
         return self.ancestors().filter(type=self.COUNTRY)
+
+    def continents(self):
+        if self.type == self.CONTINENT:
+            return Location.objects.filter(pk=self.id)
+        return self.ancestors().filter(type=self.CONTINENT)
 
     class Meta(MetaCore):

trunk/telemeta/models/cremquery.py

@@ -181 +181 @@
         for lid in MediaItem.objects.filter(location__isnull=False).values_list('location', flat=True).distinct():
             location = Location.objects.get(pk=lid)
-            if not only_continent or (only_continent in location.ancestors().filter(type=Location.CONTINENT)):
+            if not only_continent or (only_continent in location.continents()):
                 for l in location.countries():
                     if not l in countries:
@@ -190 +190 @@
         for country in countries:
             count = country.collections().count()
-            for continent in country.ancestors().filter(type=Location.CONTINENT):
+            for continent in country.continents():
                 if not stat.has_key(continent):
                     stat[continent] = {}

trunk/telemeta/tests/model_tests.py

@@ -46 +46 @@
         self.olivier = User.objects.create(username="olivier", level="admin")    
 
-        self.country = LocationType.objects.create(id="country", name="Country")
-        self.continent = LocationType.objects.create(id="continent", name="Continent")
-        self.city = LocationType.objects.create(id="city", name="City")
-
-        self.paris = Location.objects.create(name="Paris", type="other", complete_type=self.city)
-        self.france = Location.objects.create(name="France", type="country", complete_type=self.country)
-        self.europe = Location.objects.create(name="Europe", type="continent", complete_type=self.continent)
-        self.belgique = Location.objects.create(name="Belgique", type="country", complete_type=self.country)
-
-        LocationRelation.objects.create(location=self.paris, parent_location=self.france)
-        LocationRelation.objects.create(location=self.france, parent_location=self.europe)
+        self.country = LocationType.objects.create(code="country", name="Country")
+        self.continent = LocationType.objects.create(code="continent", name="Continent")
+        self.city = LocationType.objects.create(code="city", name="City")
+
+        self.paris = Location.objects.create(name="Paris", type=Location.OTHER_TYPE, complete_type=self.city)
+        self.france = Location.objects.create(name="France", type=Location.COUNTRY, complete_type=self.country)
+        self.europe = Location.objects.create(name="Europe", type=Location.CONTINENT, complete_type=self.continent)
+        self.belgique = Location.objects.create(name="Belgique", type=Location.COUNTRY, complete_type=self.country)
+
+        self.europe.add_child(self.france)
+        self.france.add_child(self.paris)
+        self.europe.add_child(self.belgique)
 
         self.a = EthnicGroup.objects.create(name="a")
@@ -166 +167 @@
     def testLocationSearch(self):
         "Test by_country and by_continent properties of MediaCollection class"
-        self.assertEquals(self.collections.by_country("France")[0], self.persepolis)
-        self.assertEquals(self.collections.by_continent("Europe")[0], self.persepolis)
-        self.assertEquals(self.collections.by_country("Belgique").order_by("title")[0], self.nicolas)
-        self.assertEquals(self.collections.by_country("Belgique").order_by("title")[1], self.volonte)
+        self.assertEquals(self.collections.by_location(self.france)[0], self.persepolis)
+        self.assertEquals(self.collections.by_location(self.europe)[0], self.persepolis)
+        self.assertEquals(self.collections.by_location(self.belgique).order_by("title")[0], self.nicolas)
+        self.assertEquals(self.collections.by_location(self.belgique).order_by("title")[1], self.volonte)
 
     def testRecordingYear(self): 
@@ -270 +271 @@
     def testLocationRelation(self):
         "Test location country and continent resolving"
-        self.assertEquals(self.france, self.item_1.location.country())
-        self.assertEquals(self.europe, self.item_1.location.continent())
-        self.assertEquals(self.france, self.item_2.location.country())
-        self.assertEquals(self.europe, self.item_2.location.continent())
+        self.assertEquals(self.france, self.item_1.location.countries()[0])
+        self.assertEquals(self.europe, self.item_1.location.continents()[0])
+        self.assertEquals(self.france, self.item_2.location.countries()[0])
+        self.assertEquals(self.europe, self.item_2.location.continents()[0])
 
     def testCollectionCountries(self):