summaryrefslogtreecommitdiff
path: root/src/third_party/wiredtiger/test/suite/test_truncate01.py
diff options
context:
space:
mode:
authorRamon Fernandez <ramon@mongodb.com>2016-01-27 17:17:17 -0500
committerRamon Fernandez <ramon@mongodb.com>2016-01-27 17:17:21 -0500
commit90118b147a6943b19dc929862a11071538db1438 (patch)
treed1e2c409595332174953e7dbe2db262935d89ae5 /src/third_party/wiredtiger/test/suite/test_truncate01.py
parentb8cad6a59cbce2831e69e6b94f9544d83d6e00b0 (diff)
downloadmongo-90118b147a6943b19dc929862a11071538db1438.tar.gz
Import wiredtiger-wiredtiger-2.7.0-505-g7fea169.tar.gz from wiredtiger branch mongodb-3.4
ref: 44463c5..7fea169 WT-2355 Fix minor scratch buffer usage in logging. WT-2348 xargs -P isn't portable WT-2347 Java: schema format edge cases WT-2344 OS X compiler warning WT-2342 Enhance wtperf to support background create and drop operations WT-2340 Add logging guarantee assertions, whitespace WT-2339 format post-rebalance verify failure (stress run #11586) WT-2338 Disable using pre-allocated log files when backup cursor is open WT-2335 NULL pointer crash in config_check_search with invalid configuration string WT-2333 Add a flag so drop doesn't block WT-2332 Bug in logging write-no-sync mode WT-2331 Checking of search() result for reference cursors before join() WT-2328 schema drop does direct unlink, it should use a block manager interface. WT-2326 Change WTPERF to use new memory allocation functions instead of the standard WT-2321 WT-2321: race between eviction and worker threads on the eviction queue WT-2320 Only check copyright when cutting releases WT-2316 stress test failure: WT_CURSOR.prev out-of-order returns WT-2314 page-swap error handling is inconsistent WT-2313 sweep-server: conn_dhandle.c, 610: dhandle != conn->cache->evict_file_next WT-2312 re-creating a deleted column-store page can corrupt the in-memory tree WT-2308 custom extractor for ref_cursors in join cursor WT-2305 Fix coverity scan issues on 23/12/2015 WT-2296 New log algorithm needs improving for sync/flush settings WT-2295 WT_SESSION.create does a full-scan of the main table WT-2287 WT_SESSION.rebalance WT-2275 broken DB after application crash WT-2267 Improve wtperf throttling implementation to provide steady load WT-2247 variable-length column-store in-memory page splits WT-2242 WiredTiger treats dead trees the same as other trees in eviction WT-2142 Connection cleanup in Python tests WT-2073 metadata cleanups WT-1801 Add a directory sync after rollback of a WT_SESSION::rename operation WT-1517 schema format edge cases SERVER-22064 Coverity analysis defect 77699: Unchecked return value SERVER-21619 sys-perf: WT crash during core_workloads_WT execution
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_truncate01.py')
-rw-r--r--src/third_party/wiredtiger/test/suite/test_truncate01.py453
1 files changed, 453 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_truncate01.py b/src/third_party/wiredtiger/test/suite/test_truncate01.py
new file mode 100644
index 00000000000..77a476e40c1
--- /dev/null
+++ b/src/third_party/wiredtiger/test/suite/test_truncate01.py
@@ -0,0 +1,453 @@
+#!/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.
+#
+# test_truncate01.py
+# session level operations on tables
+#
+
+import wiredtiger, wttest
+from helper import confirm_empty,\
+ key_populate, value_populate, simple_populate,\
+ complex_populate, complex_value_populate
+from wtscenario import check_scenarios, multiply_scenarios, number_scenarios
+
+# Test truncation arguments.
+class test_truncate_arguments(wttest.WiredTigerTestCase):
+ name = 'test_truncate'
+
+ scenarios = check_scenarios([
+ ('file', dict(type='file:')),
+ ('table', dict(type='table:'))
+ ])
+
+ # Test truncation without URI or cursors specified, or with a URI and
+ # either cursor specified, expect errors.
+ def test_truncate_bad_args(self):
+ uri = self.type + self.name
+ simple_populate(self, uri, 'key_format=S', 100)
+ msg = '/either a URI or start/stop cursors/'
+ self.assertRaisesWithMessage(wiredtiger.WiredTigerError,
+ lambda: self.session.truncate(None, None, None, None), msg)
+ cursor = self.session.open_cursor(uri, None)
+ self.assertRaisesWithMessage(wiredtiger.WiredTigerError,
+ lambda: self.session.truncate(uri, cursor, None, None), msg)
+ self.assertRaisesWithMessage(wiredtiger.WiredTigerError,
+ lambda: self.session.truncate(uri, None, cursor, None), msg)
+
+ # Test truncation of cursors where no key is set, expect errors.
+ def test_truncate_cursor_notset(self):
+ uri = self.type + self.name
+ msg = '/requires key be set/'
+
+ simple_populate(self, uri, 'key_format=S', 100)
+
+ c1 = self.session.open_cursor(uri, None)
+ c2 = self.session.open_cursor(uri, None)
+ c2.set_key(key_populate(c2, 10))
+ self.assertRaisesWithMessage(wiredtiger.WiredTigerError,
+ lambda: self.session.truncate(None, c1, c2, None), msg)
+ self.assertRaisesWithMessage(wiredtiger.WiredTigerError,
+ lambda: self.session.truncate(None, c2, c1, None), msg)
+ c1.close()
+ c2.close()
+
+
+# Test truncation of an object using its URI.
+class test_truncate_uri(wttest.WiredTigerTestCase):
+ name = 'test_truncate'
+ scenarios = check_scenarios([
+ ('file', dict(type='file:')),
+ ('table', dict(type='table:'))
+ ])
+
+ # Populate an object, truncate it by URI, and confirm it's empty.
+ def test_truncate_uri(self):
+ uri = self.type + self.name
+
+ # A simple, one-file file or table object.
+ simple_populate(self, uri, 'key_format=S', 100)
+ self.session.truncate(uri, None, None, None)
+ confirm_empty(self, uri)
+ self.session.drop(uri, None)
+
+ if self.type == "table:":
+ complex_populate(self, uri, 'key_format=S', 100)
+ self.session.truncate(uri, None, None, None)
+ confirm_empty(self, uri)
+ self.session.drop(uri, None)
+
+
+# Test truncation of cursors in an illegal order.
+class test_truncate_cursor_order(wttest.WiredTigerTestCase):
+ name = 'test_truncate'
+
+ types = [
+ ('file', dict(type='file:')),
+ ('table', dict(type='table:'))
+ ]
+ keyfmt = [
+ ('integer', dict(keyfmt='i')),
+ ('recno', dict(keyfmt='r')),
+ ('string', dict(keyfmt='S')),
+ ]
+ scenarios = number_scenarios(multiply_scenarios('.', types, keyfmt))
+
+ # Test an illegal order, then confirm that equal cursors works.
+ def test_truncate_cursor_order(self):
+ uri = self.type + self.name
+ simple_populate(self, uri, 'key_format=' + self.keyfmt, 100)
+ c1 = self.session.open_cursor(uri, None)
+ c2 = self.session.open_cursor(uri, None)
+
+ c1.set_key(key_populate(c1, 20))
+ c2.set_key(key_populate(c2, 10))
+ msg = '/the start cursor position is after the stop cursor position/'
+ self.assertRaisesWithMessage(wiredtiger.WiredTigerError,
+ lambda: self.session.truncate(None, c1, c2, None), msg)
+ c2.set_key(key_populate(c2, 20))
+ self.session.truncate(None, c1, c2, None)
+
+
+# Test truncation of cursors past the end of the object.
+class test_truncate_cursor_end(wttest.WiredTigerTestCase):
+ name = 'test_truncate'
+
+ types = [
+ ('file', dict(type='file:')),
+ ('table', dict(type='table:'))
+ ]
+ keyfmt = [
+ ('integer', dict(keyfmt='i')),
+ ('recno', dict(keyfmt='r')),
+ ('string', dict(keyfmt='S')),
+ ]
+ scenarios = number_scenarios(multiply_scenarios('.', types, keyfmt))
+
+ # Test truncation of cursors past the end of the object.
+ def test_truncate_cursor_order(self):
+ uri = self.type + self.name
+
+ # A simple, one-file file or table object.
+ simple_populate(self, uri, 'key_format=' + self.keyfmt, 100)
+ c1 = self.session.open_cursor(uri, None)
+ c1.set_key(key_populate(c1, 1000))
+ c2 = self.session.open_cursor(uri, None)
+ c2.set_key(key_populate(c2, 2000))
+ self.session.truncate(None, c1, c2, None)
+ self.assertEquals(c1.close(), 0)
+ self.assertEquals(c2.close(), 0)
+ self.session.drop(uri)
+
+ if self.type == "table:":
+ complex_populate(self, uri, 'key_format=' + self.keyfmt, 100)
+ c1 = self.session.open_cursor(uri, None)
+ c1.set_key(key_populate(c1, 1000))
+ c2 = self.session.open_cursor(uri, None)
+ c2.set_key(key_populate(c2, 2000))
+ self.session.truncate(None, c1, c2, None)
+ self.assertEquals(c1.close(), 0)
+ self.assertEquals(c2.close(), 0)
+ self.session.drop(uri)
+
+
+# Test session.truncate.
+class test_truncate_cursor(wttest.WiredTigerTestCase):
+ name = 'test_truncate'
+
+ # Use a small page size because we want to create lots of pages.
+ # The underlying table routines don't easily support 8t value types, limit
+ # those tests to file objects.
+ types = [
+ ('file', dict(type='file:',\
+ config='allocation_size=512,leaf_page_max=512,key_format=')),
+ ('file8t', dict(type='file:',\
+ config='allocation_size=512,\
+ leaf_page_max=512,value_format=8t,key_format=')),
+ ('table', dict(type='table:',\
+ config='allocation_size=512,leaf_page_max=512,key_format=')),
+ ]
+ keyfmt = [
+ ('integer', dict(keyfmt='i')),
+ ('recno', dict(keyfmt='r')),
+ ('string', dict(keyfmt='S')),
+ ]
+ reopen = [
+ ('in-memory', dict(reopen=False)),
+ ('on-disk', dict(reopen=True)),
+ ]
+ size = [
+ ('small', dict(nentries=100,skip=7)),
+ ('big', dict(nentries=1000,skip=37)),
+ ]
+
+ scenarios = number_scenarios(
+ multiply_scenarios('.', types, keyfmt, size, reopen))
+
+ # Set a cursor key.
+ def cursorKey(self, uri, key):
+ if key == -1:
+ return None
+ cursor = self.session.open_cursor(uri, None)
+ cursor.set_key(key_populate(cursor, key))
+ return cursor
+
+ # Truncate a range using cursors, and check the results.
+ def truncateRangeAndCheck(self, uri, begin, end, expected):
+ self.pr('truncateRangeAndCheck: ' + str(begin) + ',' + str(end))
+ cur1 = self.cursorKey(uri, begin)
+ cur2 = self.cursorKey(uri, end)
+ self.session.truncate(None, cur1, cur2, None)
+ if not cur1:
+ begin = 1
+ else:
+ cur1.close()
+ if not cur2:
+ end = self.nentries
+ else:
+ cur2.close()
+
+ # If the object should be empty, confirm that.
+ if begin == 1 and end == self.nentries:
+ confirm_empty(self, uri)
+ return
+
+ # Check the expected values against the object.
+ cursor = self.session.open_cursor(uri, None)
+ for i in range(begin, end + 1):
+ expected[key_populate(cursor, i)] = [0]
+ for k, v in expected.iteritems():
+ cursor.set_key(k)
+ if v == [0] and \
+ cursor.key_format == 'r' and cursor.value_format == '8t':
+ cursor.search()
+ self.assertEqual(cursor.get_values(), [0])
+ elif v == [0]:
+ self.assertEqual(cursor.search(), wiredtiger.WT_NOTFOUND)
+ else:
+ cursor.search()
+ self.assertEqual(cursor.get_values(), v)
+ cursor.close()
+
+ # Test truncation of files and simple tables using cursors.
+ def test_truncate_simple(self):
+ uri = self.type + self.name
+
+ # layout:
+ # the number of initial skipped records
+ # the number of initial inserted records
+ # the number of trailing skipped records
+ # the number of trailing inserted records
+ layout = [
+ # simple set of rows
+ (0, 0, 0, 0),
+
+ # trailing append list, no delete point overlap
+ (0, 0, 0, self.skip - 3),
+
+ # trailing append list, delete point overlap
+ (0, 0, 0, self.skip + 3),
+
+ # trailing skipped list, no delete point overlap
+ (0, 0, self.skip - 3, 1),
+
+ # trailing skipped list, delete point overlap
+ (0, 0, self.skip + 3, 1),
+
+ # leading insert list, no delete point overlap
+ (0, self.skip - 3, 0, 0),
+
+ # leading insert list, delete point overlap
+ (0, self.skip + 3, 0, 0),
+
+ # leading skipped list, no delete point overlap
+ (self.skip - 3, 1, 0, 0),
+
+ # leading skipped list, delete point overlap
+ (self.skip + 3, 1, 0, 0),
+ ]
+
+ # list: truncation patterns applied on top of the layout.
+ #
+ # begin and end: -1 means pass None for the cursor arg to truncate. An
+ # integer N, with 1 <= N < self.nentries, truncates from/to a cursor
+ # positioned at that row.
+ list = [
+ (-1, self.nentries), # begin to end, begin = None
+ (1, -1), # begin to end, end = None
+ (1, self.nentries), # begin to end
+ (-1, self.nentries - self.skip), # begin to middle, begin = None
+ (1, self.nentries - self.skip), # begin to middle
+ (self.skip, -1), # middle to end, end = None
+ (self.skip, self.nentries), # middle to end
+ (self.skip, # middle to different middle
+ self.nentries - self.skip),
+ (1, 1), # begin to begin
+ (self.nentries, self.nentries), # end to end
+ (self.skip, self.skip) # middle to same middle
+ ]
+
+ # Build the layout we're going to test
+ total = self.nentries
+ for begin_skipped,begin_insert,end_skipped,end_insert in layout:
+
+ # skipped records require insert/append records
+ if begin_skipped and not begin_insert or \
+ end_skipped and not end_insert:
+ raise AssertionError('test error: skipped set without insert')
+
+ for begin,end in list:
+ '''
+ print '===== run:'
+ print 'key:', self.keyfmt, 'begin:', begin, 'end:', end
+ print 'total: ', total, \
+ 'begin_skipped:', begin_skipped, \
+ 'begin_insert:', begin_insert, \
+ 'end_skipped:', end_skipped, \
+ 'end_insert:', end_insert
+ '''
+
+ # Build a dictionary of what the object should look like for
+ # later comparison
+ expected = {}
+
+ # Create the object.
+ self.session.create(
+ uri, 'value_format=S,' + self.config + self.keyfmt)
+
+ # Insert the records that aren't skipped or inserted.
+ start = begin_skipped + begin_insert
+ stop = self.nentries - (end_skipped + end_insert)
+ cursor = self.session.open_cursor(uri, None)
+ for i in range(start + 1, stop + 1):
+ k = key_populate(cursor, i)
+ v = value_populate(cursor, i)
+ cursor[k] = v
+ expected[k] = [v]
+ cursor.close()
+
+ # Optionally close and re-open the object to get a disk image
+ # instead of a big insert list.
+ if self.reopen:
+ self.reopen_conn()
+
+ # Optionally insert initial skipped records.
+ cursor = self.session.open_cursor(uri, None, "overwrite")
+ start = 0
+ for i in range(0, begin_skipped):
+ start += 1
+ k = key_populate(cursor, start)
+ expected[k] = [0]
+
+ # Optionally insert leading records.
+ for i in range(0, begin_insert):
+ start += 1
+ k = key_populate(cursor, start)
+ v = value_populate(cursor, start)
+ cursor[k] = v
+ expected[k] = [v]
+
+ # Optionally insert trailing skipped records.
+ for i in range(0, end_skipped):
+ stop += 1
+ k = key_populate(cursor, stop)
+ expected[k] = [0]
+
+ # Optionally insert trailing records.
+ for i in range(0, end_insert):
+ stop += 1
+ k = key_populate(cursor, stop)
+ v = value_populate(cursor, stop)
+ cursor[k] = v
+ expected[k] = [v]
+ cursor.close()
+
+ self.truncateRangeAndCheck(uri, begin, end, expected)
+ self.session.drop(uri, None)
+
+ # Test truncation of complex tables using cursors. We can't do the kind of
+ # layout and detailed testing as we can with files, but this will at least
+ # smoke-test the handling of indexes and column-groups.
+ def test_truncate_complex(self):
+
+ # We only care about tables.
+ if self.type == 'file:':
+ return
+
+ uri = self.type + self.name
+
+ # list: truncation patterns
+ #
+ # begin and end: -1 means pass None for the cursor arg to truncate. An
+ # integer N, with 1 <= N < self.nentries, truncates from/to a cursor
+ # positioned at that row.
+ list = [
+ (-1, self.nentries), # begin to end, begin = None
+ (1, -1), # begin to end, end = None
+ (1, self.nentries), # begin to end
+ (-1, self.nentries - self.skip), # begin to middle, begin = None
+ (1, self.nentries - self.skip), # begin to middle
+ (self.skip, -1), # middle to end, end = None
+ (self.skip, self.nentries), # middle to end
+ (self.skip, # middle to different middle
+ self.nentries - self.skip),
+ (1, 1), # begin to begin
+ (self.nentries, self.nentries), # end to end
+ (self.skip, self.skip) # middle to same middle
+ ]
+
+ # Build the layout we're going to test
+ for begin,end in list:
+ '''
+ print '===== run:', uri
+ print 'key:', self.keyfmt, 'begin:', begin, 'end:', end
+ '''
+
+ # Create the object.
+ complex_populate(
+ self, uri, self.config + self.keyfmt, self.nentries)
+
+ # Build a dictionary of what the object should look like for
+ # later comparison
+ cursor = self.session.open_cursor(uri, None)
+ expected = {}
+ for i in range(1, self.nentries + 1):
+ expected[key_populate(cursor, i)] = \
+ complex_value_populate(cursor, i)
+ cursor.close()
+
+ # Optionally close and re-open the object to get a disk image
+ # instead of a big insert list.
+ if self.reopen:
+ self.reopen_conn()
+
+ self.truncateRangeAndCheck(uri, begin, end, expected)
+ self.session.drop(uri, None)
+
+
+if __name__ == '__main__':
+ wttest.run()