add es_meta to Model

Author: Rizziepit

elasticgit/models.py

@@ -250,9 +250,10 @@ class Model(Config):
 
     uuid = UUIDField('Unique Identifier')
 
-    def __init__(self, config_data, static=False):
+    def __init__(self, config_data, static=False, es_meta=None):
         super(Model, self).__init__(config_data, static=static)
         self._read_only = False
+        self.es_meta = es_meta
 
     def __eq__(self, other):
         own_data = dict(self)

elasticgit/search.py

@@ -43,7 +43,7 @@ def get_object(self):
         raise NotImplementedError
 
     def to_object(self):
-        obj = self.model_class(self._results_dict)
+        obj = self.model_class(self._results_dict, es_meta=self.es_meta)
         obj.set_read_only()  # might not be in sync with Git
         return obj
 

elasticgit/tests/test_search.py

@@ -161,3 +161,4 @@ def test_mapping_type_metadata(self):
         self.assertTrue(hasattr(person.es_meta, 'index'))
         self.assertEqual(person.es_meta.index,
                          '%s-master' % self.workspace1.index_prefix)
+        self.assertEqual(person.to_object().es_meta, person.es_meta)