diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_stat04.py')
-rw-r--r-- | src/third_party/wiredtiger/test/suite/test_stat04.py | 109 |
1 files changed, 109 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_stat04.py b/src/third_party/wiredtiger/test/suite/test_stat04.py new file mode 100644 index 00000000000..33e7a925c6a --- /dev/null +++ b/src/third_party/wiredtiger/test/suite/test_stat04.py @@ -0,0 +1,109 @@ +#!/usr/bin/env python +# +# Public Domain 2014-2015 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 os, struct +from suite_subprocess import suite_subprocess +from wtscenario import number_scenarios, multiply_scenarios +import wiredtiger, wttest +from wiredtiger import stat + +# test_stat04.py +# Statistics key/value pair count +class test_stat04(wttest.WiredTigerTestCase, suite_subprocess): + uripfx = 'table:test_stat04.' + + # Note: stats for fixed length bit fields (valuefmt='8t') + # do not include accurate counts for kv pairs. + keyfmt = [ + ('col', dict(keyfmt='r', valuefmt='S', storekind='col')), + ('row', dict(keyfmt='S', valuefmt='S', storekind='row')), + ] + nentries = [ + ('small', dict(nentries=100, valuesize=50)), + ('medium', dict(nentries=10000, valuesize=20)), + ('large', dict(nentries=100000, valuesize=1)), + ('jumboval', dict(nentries=100, valuesize=4200000)), + ] + scenarios = number_scenarios(multiply_scenarios('.', keyfmt, nentries)) + + # Override WiredTigerTestCase + def setUpConnectionOpen(self, dir): + return wiredtiger.wiredtiger_open(dir, + 'create,' + + 'statistics=(all),' + + 'error_prefix="%s: "' % self.shortid()) + + def init_test(self): + self.valuepfx = self.valuesize * 'X' + + def genkey(self, n): + if self.keyfmt == 'S': + return 'SOMEKEY' + str(n) + else: + return n + 1 + + def genvalue(self, n): + if self.valuefmt == 'S': + return self.valuepfx + str(n) + else: + return n & 0xff + + def checkcount(self, uri, expectpairs): + statcursor = self.session.open_cursor( + 'statistics:' + uri, None, 'statistics=(all,clear)') + self.assertEqual(statcursor[stat.dsrc.btree_entries][2], expectpairs) + statcursor.close() + + def test_stat_nentries(self): + """ + Test to make sure the number of key/value pairs is accurate. + """ + + self.init_test() + uri = self.uripfx + self.storekind + '.' + str(self.nentries) + self.session.create(uri, 'key_format=' + self.keyfmt + + ',value_format=' + self.valuefmt) + cursor = self.session.open_cursor(uri, None, None) + + count = 0 + # Insert entries, periodically checking that stats match. + for i in range(0, self.nentries): + if count % 50 == 0: + self.checkcount(uri, count) + cursor[self.genkey(i)] = self.genvalue(i) + count += 1 + # Remove a number of entries, at each step checking that stats match. + for i in range(0, self.nentries / 37): + cursor.set_key(self.genkey(i*11 % self.nentries)) + if cursor.remove() == 0: + count -= 1 + self.checkcount(uri, count) + cursor.close() + +if __name__ == '__main__': + wttest.run() |