Fix for Python 3.4 compatibility

Add test for multiple reading audit log for multiple symbols

Author: Richard Bounds

arctic/store/version_store.py

@@ -208,7 +208,7 @@ def read_audit_log(self, symbol=None, message=None):
         """
         query = {}
         if symbol:
-            if isinstance(symbol, basestring):
+            if isinstance(symbol, six.string_types):
                 query['symbol'] = {'$regex': symbol}
             else:
                 query['symbol'] = {'$in': list(symbol)}

tests/integration/store/test_version_store_audit.py

@@ -41,6 +41,7 @@
 
 symbol = 'TS1'
 symbol2 = 'TS2'
+symbol3 = 'TS3'
 
 
 def test_ArcticTransaction_can_do_first_writes(library):
@@ -121,6 +122,9 @@ def test_metadata_changes_writes(library):
 
 
 def test_audit_read(library):
+    with ArcticTransaction(library, symbol3, 'u3', 'foo') as mt:
+        mt.write(symbol3, ts1)
+
     with ArcticTransaction(library, symbol, 'u1', 'l1') as mt:
         mt.write(symbol, ts1)
 
@@ -142,6 +146,14 @@ def test_audit_read(library):
     assert symbol_audit_log == [{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}]
 
+
+    symbols_audit_log = library.read_audit_log(symbol=[symbol, symbol2])
+
+    assert symbols_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}]
+
+
     symbol_message_audit_log = library.read_audit_log(symbol=symbol, message='l2')
 
     assert symbol_message_audit_log == [{u'new_v': 2, u'symbol': u'TS1', u'message': u'l2', u'user': u'u2', u'orig_v': 1}, ]