Manage relation types in searches - relation type search for operations

3 files changed, 33 insertions, 4 deletions

diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py
index ec66a2a7a..294e3163c 100644
--- a/archaeological_operations/forms.py
+++ b/archaeological_operations/forms.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (C) 2010-2015  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+# Copyright (C) 2010-2016  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as
@@ -464,6 +464,9 @@ class OperationSelect(TableSelect):
     report_processing = forms.ChoiceField(label=_(u"Report processing"),
                                           choices=[])
     virtual_operation = forms.NullBooleanField(label=_(u"Virtual operation"))
+    relation_types = forms.MultipleChoiceField(
+        label=_(u"Search within relations"), choices=[],
+        widget=forms.CheckboxSelectMultiple)
 
     def __init__(self, *args, **kwargs):
         super(OperationSelect, self).__init__(*args, **kwargs)
@@ -481,12 +484,17 @@ class OperationSelect(TableSelect):
         self.fields['periods'].help_text = models.Period.get_help()
         self.fields['record_quality'].choices = \
             [('', '--')] + list(models.QUALITY)
+        self.fields['relation_types'].choices = models.RelationType.get_types(
+            empty_first=False)
 
     def get_input_ids(self):
         ids = super(OperationSelect, self).get_input_ids()
         ids.pop(ids.index('parcel'))
         ids.append('parcel_0')
         ids.append('parcel_1')
+        ids.pop(ids.index('relation_types'))
+        for idx, c in enumerate(self.fields['relation_types'].choices):
+            ids.append('relation_types_{}'.format(idx))
         return ids
 
 
diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index 795c1a805..2703db245 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (C) 2012-2015 Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+# Copyright (C) 2012-2016 Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 91ec1deb0..578382ccd 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -514,9 +514,9 @@ class OperationInitTest(object):
 
 class OperationTest(TestCase, OperationInitTest):
     fixtures = [settings.ROOT_PATH +
-                '../fixtures/initial_data.json',
+                '../fixtures/initial_data-auth-fr.json',
                 settings.ROOT_PATH +
-                '../ishtar_common/fixtures/initial_data.json',
+                '../ishtar_common/fixtures/initial_data-fr.json',
                 settings.ROOT_PATH +
                 '../archaeological_files/fixtures/initial_data.json',
                 settings.ROOT_PATH +
@@ -544,6 +544,27 @@ class OperationTest(TestCase, OperationInitTest):
                          {'operator': self.orgas[0].pk})
         self.assertTrue(json.loads(response.content)['total'] == 2)
 
+    def testRelatedSearch(self):
+        c = Client()
+        rel1 = models.RelationType.objects.create(
+            symmetrical=True, label='Include', txt_idx='include')
+        rel2 = models.RelationType.objects.create(
+            symmetrical=False, label='Included', txt_idx='included',
+            inverse_relation=rel1)
+        models.RecordRelations.objects.create(
+            left_record=self.operations[0],
+            right_record=self.operations[1],
+            relation_type=rel1)
+        self.operations[1].year = 2011
+        self.operations[1].save()
+        search = {'year': '2010', 'relation_types_0': rel2.pk}
+        response = c.get(reverse('get-operation'), search)
+        # no result when no authentification
+        self.assertTrue(not json.loads(response.content))
+        c.login(username=self.username, password=self.password)
+        response = c.get(reverse('get-operation'), search)
+        self.assertTrue(json.loads(response.content)['total'] == 2)
+
     def testOwnSearch(self):
         c = Client()
         response = c.get(reverse('get-operation'), {'year': '2010'})