Move location identifying strings into index

cms/djangoapps/contentstore/courseware_index.py

@@ -190,6 +190,7 @@ class SearchIndexerBase(object):
                 item_index['id'] = item_id
                 if item.start:
                     item_index['start_date'] = item.start
+                item_index.update(cls.supplemental_fields(item))
 
                 searcher.index(cls.DOCUMENT_TYPE, item_index)
                 indexed_count["count"] += 1
@@ -271,6 +272,14 @@ class SearchIndexerBase(object):
         """
         pass
 
+    @classmethod
+    def supplemental_fields(cls, item):  # pylint: disable=unused-argument
+        """
+        Any supplemental fields that get added to the index for the specified
+        item. Base implementation returns an empty dictionary
+        """
+        return {}
+
 
 class CoursewareSearchIndexer(SearchIndexerBase):
     """
@@ -285,6 +294,8 @@ class CoursewareSearchIndexer(SearchIndexerBase):
         'category': 'courseware_index'
     }
 
+    UNNAMED_MODULE_NAME = _("(Unnamed)")
+
     @classmethod
     def normalize_structure_key(cls, structure_key):
         """ Normalizes structure key for use in indexing """
@@ -314,6 +325,30 @@ class CoursewareSearchIndexer(SearchIndexerBase):
         """
         CourseAboutSearchIndexer.index_about_information(modulestore, structure)
 
+    @classmethod
+    def supplemental_fields(cls, item):
+        """
+        Add location path to the item object
+
+        Once we've established the path of names, the first name is the course
+        name, and the next 3 names are the navigable path within the edx
+        application. Notice that we stop at that level because a full path to
+        deep children would be confusing.
+        """
+        location_path = []
+        parent = item
+        while parent is not None:
+            path_component_name = parent.display_name
+            if not path_component_name:
+                path_component_name = cls.UNNAMED_MODULE_NAME
+            location_path.append(path_component_name)
+            parent = parent.get_parent()
+        location_path.reverse()
+        return {
+            "course_name": location_path[0],
+            "location": location_path[1:4]
+        }
+
 
 class LibrarySearchIndexer(SearchIndexerBase):
     """

cms/djangoapps/contentstore/tests/test_courseware_index.py

@@ -145,10 +145,10 @@ class MixedWithOptionsTestCase(MixedSplitTestCase):
         """ Returns field_dictionary for default search """
         return {}
 
-    def search(self, field_dictionary=None):
+    def search(self, field_dictionary=None, query_string=None):
         """ Performs index search according to passed parameters """
         fields = field_dictionary if field_dictionary else self._get_default_search()
-        return self.searcher.search(field_dictionary=fields, doc_type=self.DOCUMENT_TYPE)
+        return self.searcher.search(query_string=query_string, field_dictionary=fields, doc_type=self.DOCUMENT_TYPE)
 
     def _perform_test_using_store(self, store_type, test_to_perform):
         """ Helper method to run a test function that uses a specific store """
@@ -220,7 +220,11 @@ class TestCoursewareSearchIndexer(MixedWithOptionsTestCase):
         """
         Set up the for the course outline tests.
         """
-        self.course = CourseFactory.create(modulestore=store, start=datetime(2015, 3, 1, tzinfo=UTC))
+        self.course = CourseFactory.create(
+            modulestore=store,
+            start=datetime(2015, 3, 1, tzinfo=UTC),
+            display_name="Search Index Test Course"
+        )
 
         self.chapter = ItemFactory.create(
             parent_location=self.course.location,
@@ -485,6 +489,50 @@ class TestCoursewareSearchIndexer(MixedWithOptionsTestCase):
         self.assertIn(CourseMode.HONOR, response["results"][0]["data"]["modes"])
         self.assertIn(CourseMode.VERIFIED, response["results"][0]["data"]["modes"])
 
+    def _test_course_location_info(self, store):
+        """ Test that course location information is added to index """
+        self.publish_item(store, self.vertical.location)
+        self.reindex_course(store)
+        response = self.search(query_string="Html Content")
+        self.assertEqual(response["total"], 1)
+
+        result = response["results"][0]["data"]
+        self.assertEqual(result["course_name"], "Search Index Test Course")
+        self.assertEqual(result["location"], ["Week 1", "Lesson 1", "Subsection 1"])
+
+    def _test_course_location_null(self, store):
+        """ Test that course location information is added to index """
+        sequential2 = ItemFactory.create(
+            parent_location=self.chapter.location,
+            category='sequential',
+            display_name=None,
+            modulestore=store,
+            publish_item=True,
+            start=datetime(2015, 3, 1, tzinfo=UTC),
+        )
+        # add a new vertical
+        vertical2 = ItemFactory.create(
+            parent_location=sequential2.location,
+            category='vertical',
+            display_name='Subsection 2',
+            modulestore=store,
+            publish_item=True,
+        )
+        ItemFactory.create(
+            parent_location=vertical2.location,
+            category="html",
+            display_name="Find Me",
+            publish_item=True,
+            modulestore=store,
+        )
+        self.reindex_course(store)
+        response = self.search(query_string="Find Me")
+        self.assertEqual(response["total"], 1)
+
+        result = response["results"][0]["data"]
+        self.assertEqual(result["course_name"], "Search Index Test Course")
+        self.assertEqual(result["location"], ["Week 1", CoursewareSearchIndexer.UNNAMED_MODULE_NAME, "Subsection 2"])
+
     @patch('django.conf.settings.SEARCH_ENGINE', 'search.tests.utils.ErroringIndexEngine')
     def _test_exception(self, store):
         """ Test that exception within indexing yields a SearchIndexingError """
@@ -536,6 +584,14 @@ class TestCoursewareSearchIndexer(MixedWithOptionsTestCase):
     def test_course_about_mode_index(self, store_type):
         self._perform_test_using_store(store_type, self._test_course_about_mode_index)
 
+    @ddt.data(*WORKS_WITH_STORES)
+    def test_course_location_info(self, store_type):
+        self._perform_test_using_store(store_type, self._test_course_location_info)
+
+    @ddt.data(*WORKS_WITH_STORES)
+    def test_course_location_null(self, store_type):
+        self._perform_test_using_store(store_type, self._test_course_location_null)
+
 
 @patch('django.conf.settings.SEARCH_ENGINE', 'search.tests.utils.ForceRefreshElasticSearchEngine')
 @ddt.ddt