Add tests for custom mapping in DocType

elastic · Jan 23, 2015 · 94e4241 · 94e4241
1 parent 0d6de74
commit 94e4241
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/test_elasticsearch_dsl/test_document.py b/test_elasticsearch_dsl/test_document.py
@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from elasticsearch_dsl import document, field
+from elasticsearch_dsl import document, field, Mapping
 
 class MyDoc(document.DocType):
     title = field.String(index='not_analyzed')
@@ -59,6 +59,21 @@ def test_declarative_mapping_definition():
         }
     } == MyDoc._doc_type.mapping.to_dict()
 
+def test_you_can_supply_own_mapping_instance():
+    class MyD(document.DocType):
+        title = field.String()
+
+        class Meta:
+            mapping = Mapping('my_d')
+            mapping = mapping.meta('_all', enabled=False)
+
+    assert {
+        'my_d': {
+            '_all': {'enabled': False},
+            'properties': {'title': {'type': 'string'}}
+        }
+    } == MyD._doc_type.mapping.to_dict()
+
 def test_document_can_be_created_dynamicaly():
     n = datetime.now()
     md = MyDoc(title='hello')