Merge pull request pandas-dev#145 from manahl/read_audit_log_mongo2x

Make read_audit_log work with mongo 2.x

Author: richardbounds

arctic/store/version_store.py

@@ -215,8 +215,11 @@ def read_audit_log(self, symbol=None, message=None):
         if message is not None:
             query['message'] = message
 
-        return list(self._audit.find(query, sort=[('_id', -1)],
-                                     projection={'_id': False}))
+        def _pop_id(x):
+            x.pop('_id')
+            return x
+
+        return [_pop_id(x) for x in self._audit.find(query, sort=[('_id', -1)])]
 
     def list_versions(self, symbol=None, snapshot=None, latest_only=False):
         """

tests/integration/store/test_version_store_audit.py

@@ -133,7 +133,14 @@ def test_audit_read(library):
 
     with ArcticTransaction(library, symbol2, 'u2', 'l2') as mt:
         mt.write(symbol2, ts2)
+        
+    audit_log = library.read_audit_log()
 
+    assert audit_log == [{u'new_v': 1, u'symbol': u'TS2', u'message': u'l2', u'user': u'u2', u'orig_v': 0},
+                         {u'new_v': 2, u'symbol': u'TS1', u'message': u'l2', u'user': u'u2', u'orig_v': 1},
+                         {u'new_v': 1, u'symbol': u'TS1', u'message': u'l1', u'user': u'u1', u'orig_v': 0},
+                         {u'new_v': 1, u'symbol': u'TS3', u'message': u'foo', u'user': u'u3', u'orig_v': 0},
+                         ]
 
     l2_audit_log = library.read_audit_log(message='l2')