Fix to issue djc#278

rdbisme · djc · commit f0230e6cb4a0 · 2016-04-27T20:15:55.000+02:00
Added an if statement in the _wrap_row method of
couchdb.mapping.Document to ensure the correct handling of the revision
'_rev' key inside the _data dictionary the class uses to wrap data from
the database

Added the test_wrapped_view unittest related to this change
diff --git a/couchdb/mapping.py b/couchdb/mapping.py
@@ -406,6 +406,8 @@ def _wrap_row(cls, row):
             return cls.wrap(doc)
         data = row['value']
         data['_id'] = row['id']
+        if 'rev' in data:  # When data is client.Document
+            data['_rev'] = data['rev']
         return cls.wrap(data)
 
 
diff --git a/couchdb/tests/mapping.py b/couchdb/tests/mapping.py
@@ -242,6 +242,12 @@ def test_view(self):
                                  include_docs=True)
         self.assertEqual(type(results.rows[0]), self.Item)
 
+    def test_wrapped_view(self):
+        self.Item().store(self.db)
+        results = self.db.view('_all_docs', wrapper=self.Item._wrap_row)
+        doc = results.rows[0]
+        self.db.delete(doc)
+
     def test_query(self):
         self.Item().store(self.db)
         results = self.Item.query(self.db, all_map_func, None)
