1 files changed, 125 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_cursor08.py b/src/third_party/wiredtiger/test/suite/test_cursor08.py
new file mode 100644
index 00000000000..ce9d3d68664
--- /dev/null
+++ b/src/third_party/wiredtiger/test/suite/test_cursor08.py
@@ -0,0 +1,125 @@
+#!/usr/bin/env python
+#
+# Public Domain 2014-2015 MongoDB, Inc.
+# Public Domain 2008-2014 WiredTiger, Inc.
+#
+# This is free and unencumbered software released into the public domain.
+#
+# Anyone is free to copy, modify, publish, use, compile, sell, or
+# distribute this software, either in source code form or as a compiled
+# binary, for any purpose, commercial or non-commercial, and by any
+# means.
+#
+# In jurisdictions that recognize copyright laws, the author or authors
+# of this software dedicate any and all copyright interest in the
+# software to the public domain. We make this dedication for the benefit
+# of the public at large and to the detriment of our heirs and
+# successors. We intend this dedication to be an overt act of
+# relinquishment in perpetuity of all present and future rights to this
+# software under copyright law.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+# IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+# OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+# ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+# OTHER DEALINGS IN THE SOFTWARE.
+#
+# test_cursor08.py
+# Log cursors with compression
+#
+
+import fnmatch, os, shutil, run, time
+from suite_subprocess import suite_subprocess
+from wiredtiger import wiredtiger_open, stat, WiredTigerError
+from wtscenario import multiply_scenarios, number_scenarios, check_scenarios
+import wttest
+
+class test_cursor08(wttest.WiredTigerTestCase, suite_subprocess):
+    logmax = "100K"
+    tablename = 'test_cursor08'
+    uri = 'table:' + tablename
+    nkeys = 5
+
+    reopens = check_scenarios([
+        ('regular', dict(reopen=False)),
+        ('reopen', dict(reopen=True))
+    ])
+    compress = check_scenarios([
+        ('bzip2', dict(compress='bzip2')),
+        ('nop', dict(compress='nop')),
+        ('snappy', dict(compress='snappy')),
+        ('zlib', dict(compress='zlib')),
+        ('none', dict(compress='')),
+    ])
+    scenarios = number_scenarios(multiply_scenarios('.', reopens, compress))
+
+    # Return the wiredtiger_open extension argument for a shared library.
+    def extensionArg(self, name):
+        if name == None:
+            return ''
+
+        testdir = os.path.dirname(__file__)
+        extdir = os.path.join(run.wt_builddir, 'ext/compressors')
+        extfile = os.path.join(
+            extdir, name, '.libs', 'libwiredtiger_' + name + '.so')
+        if not os.path.exists(extfile):
+            self.skipTest('compression extension "' + extfile + '" not built')
+        return ',extensions=["' + extfile + '"]'
+
+    # Overrides WiredTigerTestCase - add logging
+    def setUpConnectionOpen(self, dir):
+        self.home = dir
+        self.txn_sync = '(method=dsync,enabled)'
+        conn_params = \
+                'log=(archive=false,enabled,file_max=%s,' % self.logmax + \
+                'compressor=%s)' % self.compress + \
+                ',create,error_prefix="%s: ",' % self.shortid() + \
+                'transaction_sync="%s",' % self.txn_sync + \
+                self.extensionArg(self.compress)
+        # print "Creating conn at '%s' with config '%s'" % (dir, conn_params)
+        try:
+            conn = wiredtiger_open(dir, conn_params)
+        except WiredTigerError as e:
+            print "Failed conn at '%s' with config '%s'" % (dir, conn_params)
+        self.pr(`conn`)
+        self.session2 = conn.open_session()
+        return conn
+
+    def test_log_cursor(self):
+        # print "Creating %s with config '%s'" % (self.uri, self.create_params)
+        create_params = 'key_format=i,value_format=S'
+        self.session.create(self.uri, create_params)
+        c = self.session.open_cursor(self.uri, None)
+
+        # A binary value.
+        value = u'\u0001\u0002abcd\u0003\u0004'
+
+        self.session.begin_transaction()
+        for k in range(self.nkeys):
+            c[k] = value
+        self.session.commit_transaction()
+        c.close()
+
+        if self.reopen:
+            self.reopen_conn()
+
+        # Check for these values via a log cursor
+        c = self.session.open_cursor("log:", None)
+        count = 0
+        while c.next() == 0:
+            # lsn.file, lsn.offset, opcount
+            keys = c.get_key()
+            # txnid, rectype, optype, fileid, logrec_key, logrec_value
+            values = c.get_value()
+            try:
+                if value in str(values[5]):   # logrec_value
+                    count += 1
+            except:
+                pass
+        c.close()
+        self.assertEqual(count, self.nkeys)
+
+if __name__ == '__main__':
+    wttest.run()