diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_cursor_compare.py')
| -rw-r--r-- | src/third_party/wiredtiger/test/suite/test_cursor_compare.py | 246 |
1 files changed, 246 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_cursor_compare.py b/src/third_party/wiredtiger/test/suite/test_cursor_compare.py new file mode 100644 index 00000000000..130f4e8ca96 --- /dev/null +++ b/src/third_party/wiredtiger/test/suite/test_cursor_compare.py @@ -0,0 +1,246 @@ +#!/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, exceptions +from helper import complex_populate, simple_populate, key_populate +from helper import complex_populate_index_name +from wtscenario import multiply_scenarios, number_scenarios + +# Test cursor comparisons. +class test_cursor_comparison(wttest.WiredTigerTestCase): + name = 'test_compare' + + types = [ + ('file', dict(type='file:', config='')), + ('lsm', dict(type='table:', config=',type=lsm')), + ('table', dict(type='table:', config='')) + ] + keyfmt = [ + ('integer', dict(keyfmt='i')), + ('recno', dict(keyfmt='r')), + ('string', dict(keyfmt='S')) + ] + scenarios = number_scenarios(multiply_scenarios('.', types, keyfmt)) + + def test_cursor_comparison(self): + uri = self.type + 'compare' + uriX = self.type + 'compareX' + + # Build the object. + if self.type == 'file:': + simple_populate( + self, uri, 'key_format=' + self.keyfmt + self.config, 100) + simple_populate( + self, uriX, 'key_format=' + self.keyfmt + self.config, 100) + ix0_0 = None + ix0_1 = None + ix1_0 = None + ixX_0 = None + else: + complex_populate( + self, uri, 'key_format=' + self.keyfmt + self.config, 100) + complex_populate( + self, uriX, 'key_format=' + self.keyfmt + self.config, 100) + ix0_0 = self.session.open_cursor( + complex_populate_index_name(self, uri, 0), None) + ix0_1 = self.session.open_cursor( + complex_populate_index_name(self, uri, 0), None) + ix1_0 = self.session.open_cursor( + complex_populate_index_name(self, uri, 1), None) + ixX_0 = self.session.open_cursor( + complex_populate_index_name(self, uriX, 0), None) + ix0_0.next() + ix0_1.next() + ix1_0.next() + ixX_0.next() + + c1 = self.session.open_cursor(uri, None) + c2 = self.session.open_cursor(uri, None) + + # Confirm failure unless the keys are set. + msg = '/requires key be set/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: c1.compare(c2), msg) + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: c2.compare(c1), msg) + + # Test cursors before they're positioned. + c1.set_key(key_populate(c1, 10)) + c2.set_key(key_populate(c2, 20)) + self.assertGreater(c2.compare(c1), 0) + self.assertLess(c1.compare(c2), 0) + c2.set_key(key_populate(c2, 10)) + self.assertEqual(c1.compare(c2), 0) + self.assertEqual(c2.compare(c1), 0) + + # Confirm failure for different objects. + cX = self.session.open_cursor(uriX, None) + cX.set_key(key_populate(cX, 10)) + msg = '/must reference the same object/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: cX.compare(c1), msg) + msg = '/wt_cursor.* is None/' + self.assertRaisesHavingMessage( + exceptions.RuntimeError, lambda: cX.compare(None), msg) + if ix0_0 != None: + self.assertEqual(ix0_0.compare(ix0_1), 0) + ix0_1.reset() + ix0_1.prev() + self.assertLess(ix0_0.compare(ix0_1), 0) + self.assertGreater(ix0_1.compare(ix0_0), 0) + # Main table vs. index not allowed + msg = '/must reference the same object/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: c1.compare(ix0_0), msg) + # Two unrelated indices not allowed + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: ixX_0.compare(ix0_0), msg) + # Two different indices from same table not allowed + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: ix0_0.compare(ix1_0), msg) + + # Test cursors after they're positioned (shouldn't matter for compare). + c1.set_key(key_populate(c1, 10)) + self.assertEqual(c1.search(), 0) + c2.set_key(key_populate(c2, 20)) + self.assertEqual(c2.search(), 0) + self.assertGreater(c2.compare(c1), 0) + self.assertLess(c1.compare(c2), 0) + c2.set_key(key_populate(c2, 10)) + self.assertEqual(c2.search(), 0) + self.assertEqual(c1.compare(c2), 0) + self.assertEqual(c2.compare(c1), 0) + + # Confirm failure for different objects. + cX = self.session.open_cursor(uriX, None) + cX.set_key(key_populate(cX, 10)) + self.assertEqual(cX.search(), 0) + msg = '/must reference the same object/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: cX.compare(c1), msg) + + def test_cursor_equality(self): + uri = self.type + 'equality' + uriX = self.type + 'compareX' + + # Build the object. + if self.type == 'file:': + simple_populate( + self, uri, 'key_format=' + self.keyfmt + self.config, 100) + simple_populate( + self, uriX, 'key_format=' + self.keyfmt + self.config, 100) + ix0_0 = None + ix0_1 = None + ix1_0 = None + ixX_0 = None + else: + complex_populate( + self, uri, 'key_format=' + self.keyfmt + self.config, 100) + complex_populate( + self, uriX, 'key_format=' + self.keyfmt + self.config, 100) + ix0_0 = self.session.open_cursor( + complex_populate_index_name(self, uri, 0), None) + ix0_1 = self.session.open_cursor( + complex_populate_index_name(self, uri, 0), None) + ix1_0 = self.session.open_cursor( + complex_populate_index_name(self, uri, 1), None) + ixX_0 = self.session.open_cursor( + complex_populate_index_name(self, uriX, 0), None) + ix0_0.next() + ix0_1.next() + ix1_0.next() + ixX_0.next() + + c1 = self.session.open_cursor(uri, None) + c2 = self.session.open_cursor(uri, None) + + # Confirm failure unless the keys are set. + msg = '/requires key be set/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: c1.equals(c2), msg) + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: c2.equals(c1), msg) + + # Test cursors before they're positioned. + c1.set_key(key_populate(c1, 10)) + c2.set_key(key_populate(c2, 20)) + self.assertFalse(c1.equals(c2)) + self.assertFalse(c2.equals(c1)) + c2.set_key(key_populate(c2, 10)) + self.assertTrue(c1.equals(c2)) + self.assertTrue(c2.equals(c1)) + + # Confirm failure for different objects. + cX = self.session.open_cursor(uriX, None) + cX.set_key(key_populate(cX, 10)) + msg = '/must reference the same object/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: cX.equals(c1), msg) + msg = '/wt_cursor.* is None/' + self.assertRaisesHavingMessage( + exceptions.RuntimeError, lambda: cX.equals(None), msg) + if ix0_0 != None: + self.assertTrue(ix0_0.equals(ix0_1)) + ix0_1.reset() + ix0_1.prev() + self.assertFalse(ix0_0.equals(ix0_1)) + # Main table vs. index not allowed + msg = '/must reference the same object/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: c1.equals(ix0_0), msg) + # Two unrelated indices not allowed + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: ixX_0.equals(ix0_0), msg) + # Two different indices from same table not allowed + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: ix0_0.equals(ix1_0), msg) + + # Test cursors after they're positioned (internally, it's a different + # search path if keys are positioned in the tree). + c1.set_key(key_populate(c1, 10)) + self.assertEqual(c1.search(), 0) + c2.set_key(key_populate(c2, 20)) + self.assertEqual(c2.search(), 0) + self.assertFalse(c1.equals(c2)) + self.assertFalse(c2.equals(c1)) + c2.set_key(key_populate(c2, 10)) + self.assertEqual(c2.search(), 0) + self.assertTrue(c1.equals(c2)) + self.assertTrue(c2.equals(c1)) + + # Confirm failure for different objects. + cX = self.session.open_cursor(uriX, None) + cX.set_key(key_populate(cX, 10)) + self.assertEqual(cX.search(), 0) + msg = '/must reference the same object/' + self.assertRaisesWithMessage( + wiredtiger.WiredTigerError, lambda: cX.equals(c1), msg) + + +if __name__ == '__main__': + wttest.run() |