Added adult/child flags to search index (bug 852567)

Author: Rob Hudson

apps/addons/search.py

@@ -9,7 +9,7 @@
 
 import amo
 from amo.utils import create_es_index_if_missing
-from .models import Addon
+from .models import Addon, Flag
 from bandwagon.models import Collection
 from compat.models import AppCompat
 from stats.models import ClientData
@@ -60,6 +60,12 @@ def extract(addon):
     except ObjectDoesNotExist:
         d['has_version'] = None
     d['app'] = [app.id for app in addon.compatible_apps.keys()]
+    try:
+        d['flag_adult'] = addon.flag.adult_content
+        d['flag_child'] = addon.flag.child_content
+    except Flag.DoesNotExist:
+        d['flag_adult'] = d['flag_child'] = False
+
     if addon.type == amo.ADDON_PERSONA:
         # This would otherwise get attached when by the transformer.
         d['weekly_downloads'] = addon.persona.popularity

apps/addons/tests/test_search.py

@@ -0,0 +1,43 @@
+from nose.tools import eq_
+
+import amo.tests
+from addons.models import Addon, Flag
+from addons.search import extract
+from addons.tasks import (attach_categories, attach_devices, attach_prices,
+                          attach_tags, attach_translations)
+
+
+class TestExtract(amo.tests.TestCase):
+    fixtures = ['base/apps', 'base/users', 'base/addon_3615']
+
+    def setUp(self):
+        self.attrs = ('id', 'slug', 'app_slug', 'created', 'last_updated',
+                      'weekly_downloads', 'average_daily_users', 'status',
+                      'type', 'hotness', 'is_disabled', 'premium_type',
+                      'uses_flash')
+        self.transforms = (attach_categories, attach_devices, attach_prices,
+                           attach_tags, attach_translations)
+
+    def _extract(self):
+        qs = Addon.objects.filter(id__in=[3615])
+        for t in self.transforms:
+            qs = qs.transform(t)
+        self.addon = list(qs)[0]
+        return extract(self.addon)
+
+    def test_extract_attributes(self):
+        extracted = self._extract()
+        for attr in self.attrs:
+            eq_(extracted[attr], getattr(self.addon, attr))
+
+    def test_flags(self):
+        Flag.objects.create(addon_id=3615, adult_content=True,
+                            child_content=False)
+        extracted = self._extract()
+        eq_(extracted['flag_adult'], True)
+        eq_(extracted['flag_child'], False)
+
+    def test_no_flags(self):
+        extracted = self._extract()
+        eq_(extracted['flag_adult'], False)
+        eq_(extracted['flag_child'], False)