Add some mapping type tests.

notesapi/v1/models.py

@@ -82,6 +82,10 @@ def delete_in_index(sender, instance, **kwargs):
 
 
 class NoteMappingType(MappingType, Indexable):
+    """
+    Mapping type for Note.
+    """
+
     @classmethod
     def get_model(cls):
         return Note
@@ -107,7 +111,9 @@ class NoteMappingType(MappingType, Indexable):
 
     @classmethod
     def extract_document(cls, obj_id, obj=None):
-        """Converts this instance into an Elasticsearch document"""
+        """
+        Converts this instance into an Elasticsearch document.
+        """
         if obj is None:
             obj = cls.get_model().objects.get(pk=obj_id)
 
@@ -115,6 +121,11 @@ class NoteMappingType(MappingType, Indexable):
 
     @staticmethod
     def process_result(data):
+        """
+        Prepares result for response.
+
+        Also prepares ranges field and highlighting.
+        """
         for i, item in enumerate(data):
             if isinstance(item, dict):
                 for k, v in item.items():

notesapi/v1/tests/test_models.py

@@ -58,3 +58,21 @@ class NoteTest(TestCase):
         note.clean(self.note)
         note.save()
         self.assertEqual(NoteMappingType.extract_document(note.id), note.as_dict())
+
+    def test_get_model(self):
+        self.assertIsInstance(NoteMappingType.get_model()(), Note)
+
+    def test_get_mapping(self):
+        expected_mapping = {
+            'properties': {
+                'id': {'type': 'string', 'index': 'not_analyzed', 'store': True},
+                'user': {'type': 'string', 'index': 'not_analyzed', 'store': True},
+                'course_id': {'type': 'string', 'index': 'not_analyzed', 'store': True},
+                'usage_id': {'type': 'string', 'index': 'not_analyzed', 'store': True},
+                'text': {'type': 'string', 'analyzer': 'snowball', 'store': True},
+                'quote': {'type': 'string', 'analyzer': 'snowball', 'store': True},
+                'created': {'type': 'date', 'store': True},
+                'updated': {'type': 'date', 'store': True},
+            }
+        }
+        self.assertEqual(NoteMappingType.get_mapping(), expected_mapping)