diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_overwrite.py')
-rw-r--r-- | src/third_party/wiredtiger/test/suite/test_overwrite.py | 140 |
1 files changed, 140 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_overwrite.py b/src/third_party/wiredtiger/test/suite/test_overwrite.py new file mode 100644 index 00000000000..e22cdab4dea --- /dev/null +++ b/src/third_party/wiredtiger/test/suite/test_overwrite.py @@ -0,0 +1,140 @@ +#!/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 key_populate, simple_populate +from wtscenario import check_scenarios + +# test_overwrite.py +# cursor overwrite configuration method +class test_overwrite(wttest.WiredTigerTestCase): + name = 'overwrite' + scenarios = check_scenarios([ + ('file-r', dict(type='file:',keyfmt='r')), + ('file-S', dict(type='file:',keyfmt='S')), + ('lsm-S', dict(type='lsm:',keyfmt='S')), + ('table-r', dict(type='table:',keyfmt='r')), + ('table-S', dict(type='table:',keyfmt='S')), + ]) + + # Confirm a cursor configured with/without overwrite correctly handles + # non-existent records during insert, remove and update operations. + def test_overwrite_insert(self): + uri = self.type + self.name + simple_populate(self, uri, 'key_format=' + self.keyfmt, 100) + + # Insert of an existing record with overwrite off fails. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 5)) + cursor.set_value('XXXXXXXXXX') + self.assertRaises(wiredtiger.WiredTigerError, lambda: cursor.insert()) + + # One additional test for the insert method: duplicate the cursor with + # overwrite configured and then the insert should succeed. This test + # is only for the insert method because the remove and update method + # failure modes are for non-existent records, and you cannot duplicate + # cursor pointing to non-existent records. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 5)) + dupc = self.session.open_cursor(None, cursor, "overwrite=true") + dupc.set_value('XXXXXXXXXX') + self.assertEquals(dupc.insert(), 0) + + # Insert of an existing record with overwrite on succeeds. + cursor = self.session.open_cursor(uri, None) + cursor.set_key(key_populate(cursor, 6)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.insert(), 0) + + # Insert of a non-existent record with overwrite off succeeds. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 200)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.insert(), 0) + + # Insert of a non-existent record with overwrite on succeeds. + cursor = self.session.open_cursor(uri, None) + cursor.set_key(key_populate(cursor, 201)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.insert(), 0) + + def test_overwrite_remove(self): + uri = self.type + self.name + simple_populate(self, uri, 'key_format=' + self.keyfmt, 100) + + # Remove of an existing record with overwrite off succeeds. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 5)) + self.assertEquals(cursor.remove(), 0) + + # Remove of an existing record with overwrite on succeeds. + cursor = self.session.open_cursor(uri, None) + cursor.set_key(key_populate(cursor, 6)) + self.assertEquals(cursor.remove(), 0) + + # Remove of a non-existent record with overwrite off fails. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 200)) + self.assertEquals(cursor.remove(), wiredtiger.WT_NOTFOUND) + + # Remove of a non-existent record with overwrite on succeeds. + cursor = self.session.open_cursor(uri, None) + cursor.set_key(key_populate(cursor, 201)) + self.assertEquals(cursor.remove(), 0) + + def test_overwrite_update(self): + uri = self.type + self.name + simple_populate(self, uri, 'key_format=' + self.keyfmt, 100) + + # Update of an existing record with overwrite off succeeds. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 5)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.update(), 0) + + # Update of an existing record with overwrite on succeeds. + cursor = self.session.open_cursor(uri, None) + cursor.set_key(key_populate(cursor, 6)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.update(), 0) + + # Update of a non-existent record with overwrite off fails. + cursor = self.session.open_cursor(uri, None, "overwrite=false") + cursor.set_key(key_populate(cursor, 200)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.update(), wiredtiger.WT_NOTFOUND) + + # Update of a non-existent record with overwrite on succeeds. + cursor = self.session.open_cursor(uri, None) + cursor.set_key(key_populate(cursor, 201)) + cursor.set_value('XXXXXXXXXX') + self.assertEquals(cursor.update(), 0) + + +if __name__ == '__main__': + wttest.run() |