1 files changed, 51 insertions, 0 deletions
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 0d6908374..9d99ed67d 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -895,6 +895,21 @@ class OperationTest(TestCase, OperationInitTest):
         self.assertEqual(ope_id, 'OP2011-1')
         self.assertEqual(town, self.towns[0].name)
 
+    def test_search_vector_update(self):
+        operation = self.operations[0]
+        town = self.create_towns({'numero_insee': '12346', 'name': 'Daisy'})[-1]
+        operation.towns.add(town)
+        town = self.create_towns(
+            {'numero_insee': '12347', 'name': 'Dirty old'})[-1]
+        operation.towns.add(town)
+        operation = models.Operation.objects.get(pk=operation.pk)
+        operation.comment = u"Zardoz"
+        operation.code_patriarche = u"HUIAAA5"
+        operation.save()
+        for key in ('old', 'op2010', 'dirty', 'daisy', "'2010'", "zardoz",
+                    "huiaaa5"):
+            self.assertIn(key, operation.search_vector)
+
     def test_cache_bulk_update(self):
         if settings.USE_SPATIALITE_FOR_TESTS:
             # using views - can only be tested with postgresql
@@ -1104,6 +1119,42 @@ class OperationSearchTest(TestCase, OperationInitTest):
         self.assertEqual(response.status_code, 200)
         self.assertEqual(json.loads(response.content)['total'], 1)
 
+    def test_town_search(self):
+        c = Client()
+        c.login(username=self.username, password=self.password)
+
+        data = {'numero_insee': '98989', 'name': 'base_town'}
+        base_town = self.create_towns(datas=data)[-1]
+
+        data = {'numero_insee': '56789', 'name': 'parent_town'}
+        parent_town = self.create_towns(datas=data)[-1]
+        parent_town.children.add(base_town)
+
+        data = {'numero_insee': '01234', 'name': 'child_town'}
+        child_town = self.create_towns(datas=data)[-1]
+        base_town.children.add(child_town)
+
+        ope = self.operations[1]
+        ope.towns.add(base_town)
+
+        # simple search
+        search = {'towns': base_town.pk}
+        response = c.get(reverse('get-operation'), search)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(json.loads(response.content)['total'], 1)
+
+        # parent search
+        search = {'towns': parent_town.pk}
+        response = c.get(reverse('get-operation'), search)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(json.loads(response.content)['total'], 1)
+
+        # child search
+        search = {'towns': child_town.pk}
+        response = c.get(reverse('get-operation'), search)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(json.loads(response.content)['total'], 1)
+
     def testOwnSearch(self):
         c = Client()
         response = c.get(reverse('get-operation'), {'year': '2010'})