1 files changed, 105 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_compact01.py b/src/third_party/wiredtiger/test/suite/test_compact01.py
new file mode 100644
index 00000000000..3af550708ed
--- /dev/null
+++ b/src/third_party/wiredtiger/test/suite/test_compact01.py
@@ -0,0 +1,105 @@
+#!/usr/bin/env python
+#
+# Public Domain 2014-2016 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.
+
+import wiredtiger, wttest
+from helper import complex_populate, simple_populate, key_populate
+from suite_subprocess import suite_subprocess
+from wiredtiger import stat
+from wtscenario import multiply_scenarios, number_scenarios
+
+# test_compact.py
+#    session level compact operation
+class test_compact(wttest.WiredTigerTestCase, suite_subprocess):
+    name = 'test_compact'
+
+    # Use a small page size because we want to create lots of pages.
+    config = 'allocation_size=512,' +\
+        'leaf_page_max=512,value_format=S,key_format=S'
+    nentries = 50000
+
+    # The table is a complex object, give it roughly 5 pages per underlying
+    # file.
+    types = [
+        ('file', dict(type='file:', pop=simple_populate, maxpages=5)),
+        ('table', dict(type='table:', pop=complex_populate, maxpages=50))
+        ]
+    compact = [
+        ('method', dict(utility=0,reopen=0)),
+        ('method_reopen', dict(utility=0,reopen=1)),
+        ('utility', dict(utility=1,reopen=0)),
+    ]
+    scenarios = number_scenarios(multiply_scenarios('.', types, compact))
+    # We want a large cache so that eviction doesn't happen
+    # (which could skew our compaction results).
+    conn_config = 'cache_size=250MB,statistics=(all)'
+
+    # Test compaction.
+    def test_compact(self):
+        # Populate an object
+        uri = self.type + self.name
+        self.pop(self, uri, self.config, self.nentries - 1)
+
+        # Reopen the connection to force the object to disk.
+        self.reopen_conn()
+
+        # Confirm the tree starts big
+        stat_cursor = self.session.open_cursor('statistics:' + uri, None, None)
+        self.assertGreater(stat_cursor[stat.dsrc.btree_row_leaf][2], self.maxpages)
+        stat_cursor.close()
+
+        # Remove most of the object.
+        c1 = self.session.open_cursor(uri, None)
+        c1.set_key(key_populate(c1, 5))
+        c2 = self.session.open_cursor(uri, None)
+        c2.set_key(key_populate(c2, self.nentries - 5))
+        self.session.truncate(None, c1, c2, None)
+        c1.close()
+        c2.close()
+
+        # Compact it, using either the session method or the utility.
+        if self.utility == 1:
+            self.session.checkpoint(None)
+            self.close_conn()
+            self.runWt(["compact", uri])
+        else:
+            # Optionally reopen the connection so we do more on-disk tests.
+            if self.reopen == 1:
+                self.session.checkpoint(None)
+                self.reopen_conn()
+
+            self.session.compact(uri, None)
+
+        # Confirm compaction worked: check the number of on-disk pages
+        self.reopen_conn()
+        stat_cursor = self.session.open_cursor('statistics:' + uri, None, None)
+        self.assertLess(stat_cursor[stat.dsrc.btree_row_leaf][2], self.maxpages)
+        stat_cursor.close()
+
+
+if __name__ == '__main__':
+    wttest.run()