diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_compact01.py')
-rw-r--r-- | src/third_party/wiredtiger/test/suite/test_compact01.py | 105 |
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() |