diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_schema02.py')
| -rw-r--r-- | src/third_party/wiredtiger/test/suite/test_schema02.py | 288 |
1 files changed, 288 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_schema02.py b/src/third_party/wiredtiger/test/suite/test_schema02.py new file mode 100644 index 00000000000..6895e947efe --- /dev/null +++ b/src/third_party/wiredtiger/test/suite/test_schema02.py @@ -0,0 +1,288 @@ +#!/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 wtscenario import check_scenarios + +# test_schema02.py +# Columns, column groups, indexes +class test_schema02(wttest.WiredTigerTestCase): + """ + Test basic operations + """ + nentries = 1000 + + scenarios = check_scenarios([ + ('normal', { 'idx_config' : '' }), + ('lsm', { 'idx_config' : ',type=lsm' }), + ]) + + def expect_failure_colgroup(self, name, configstr, match): + self.assertRaisesWithMessage(wiredtiger.WiredTigerError, + lambda:self.session.create("colgroup:" + name, configstr), match) + + def test_colgroup_after_failure(self): + # bogus formats + self.assertRaisesWithMessage(wiredtiger.WiredTigerError, + lambda:self.session.create("table:main", + "key_format=Z,value_format=S"), + "/Invalid type 'Z' found in format 'Z'/") + self.assertRaisesWithMessage(wiredtiger.WiredTigerError, + lambda:self.session.create("table:main", + "key_format=S,value_format=Z"), + "/Invalid type 'Z' found in format 'Z'/") + + # These should succeed + self.session.create("table:main", "key_format=iS,value_format=SiSi," + "columns=(ikey,Skey,S1,i2,S3,i4),colgroups=(c1,c2)") + self.session.create("colgroup:main:c1", "columns=(S1,i2)") + + def test_colgroup_failures(self): + # too many columns + self.assertRaisesWithMessage(wiredtiger.WiredTigerError, + lambda:self.session.create("table:main", "key_format=S," + "value_format=,columns=(a,b)"), + "/Number of columns in '\(a,b\)' does not match " + "key format 'S' plus value format ''/") + # Note: too few columns is allowed + + # expect this to work + self.session.create("table:main", "key_format=iS,value_format=SiSi," + "columns=(ikey,Skey,S1,i2,S3,i4)," + "colgroups=(c1,c2)") + + # bad table name + self.expect_failure_colgroup("nomatch:c", "columns=(S1,i2)", + "/Can't create 'colgroup:nomatch:c'" + " for non-existent table 'nomatch'/") + # colgroup not declared in initial create + self.expect_failure_colgroup("main:nomatch", "columns=(S1,i2)", + "/Column group 'nomatch' not found" + " in table 'main'/") + # bad column + self.expect_failure_colgroup("main:c1", "columns=(S1,i2,bad)", + "/Column 'bad' not found/") + + # TODO: no columns allowed, or not? + #self.session.create("colgroup:main:c0", "columns=()") + + # key in a column group + self.expect_failure_colgroup("main:c1", "columns=(ikey,S1,i2)", + "/A column group cannot store key column" + " 'ikey' in its value/") + + # expect this to work + self.session.create("colgroup:main:c1", "columns=(S1,i2)") + + # exclusive: no error message + self.expect_failure_colgroup("main:c1", "columns=(S1,i2),exclusive", + "") + + # exists with different config + self.expect_failure_colgroup("main:c1", "columns=(S1,i4)", + "/does not match existing configuration/") + + # colgroup not declared in initial create + self.expect_failure_colgroup("main:c3", "columns=(S3,i4)", + "/Column group 'c3' not found in" + " table 'main'/") + + # this is the last column group, but there are missing columns + self.expect_failure_colgroup("main:c2", "columns=(S1,i4)", + "/Column 'S3' in 'table:main' does not" + " appear in a column group/") + + # TODO: is repartitioning column groups allowed? + # this does not raise an error + # self.expect_failure_colgroup("main:c2", "columns=(S1,S3,i4)" + + # expect this to work + self.session.create("colgroup:main:c2", "columns=(S3,i4)") + + # expect these to work - each table name is a separate namespace + self.session.create("table:main2", "key_format=iS,value_format=SiSi," + "columns=(ikey,Skey,S1,i2,S3,i4),colgroups=(c1,c2)") + self.session.create("colgroup:main2:c1", "columns=(S1,i2)") + self.session.create("colgroup:main2:c2", "columns=(S3,i4)") + + def test_index(self): + self.session.create("table:main", "key_format=iS,value_format=SiSi," + "columns=(ikey,Skey,S1,i2,S3,i4),colgroups=(c1,c2)") + + # should be able to create colgroups before indices + self.session.create("colgroup:main:c2", "columns=(S3,i4)") + + # should be able to create indices on all key combinations + self.session.create( + "index:main:ikey", "columns=(ikey)" + self.idx_config) + self.session.create( + "index:main:Skey", "columns=(Skey)" + self.idx_config) + self.session.create( + "index:main:ikeySkey", "columns=(ikey,Skey)" + self.idx_config) + self.session.create( + "index:main:Skeyikey", "columns=(Skey,ikey)" + self.idx_config) + + # should be able to create indices on all value combinations + self.session.create( + "index:main:S1", "columns=(S1)" + self.idx_config) + self.session.create( + "index:main:i2", "columns=(i2)" + self.idx_config) + self.session.create( + "index:main:i2S1", "columns=(i2,S1)" + self.idx_config) + self.session.create( + "index:main:S1i4", "columns=(S1,i4)" + self.idx_config) + + # somewhat nonsensical to repeat columns within an index, but allowed + self.session.create( + "index:main:i4S3i4S1", "columns=(i4,S3,i4,S1)" + self.idx_config) + + # should be able to create colgroups after indices + self.session.create("colgroup:main:c1", "columns=(S1,i2)") + + self.populate() + + # should be able to create indices after populating + self.session.create( + "index:main:i2S1i4", "columns=(i2,S1,i4)" + self.idx_config) + + self.check_entries() + self.check_indices() + + def populate(self): + cursor = self.session.open_cursor('table:main', None, None) + for i in range(0, self.nentries): + square = i * i + cube = square * i + cursor[(i, 'key' + str(i))] = \ + ('val' + str(square), square, 'val' + str(cube), cube) + cursor.close() + + def check_entries(self): + cursor = self.session.open_cursor('table:main', None, None) + # spot check via search + n = self.nentries + for i in (n / 5, 0, n - 1, n - 2, 1): + cursor.set_key(i, 'key' + str(i)) + square = i * i + cube = square * i + cursor.search() + (s1, i2, s3, i4) = cursor.get_values() + self.assertEqual(s1, 'val' + str(square)) + self.assertEqual(i2, square) + self.assertEqual(s3, 'val' + str(cube)) + self.assertEqual(i4, cube) + + i = 0 + count = 0 + # then check all via cursor + cursor.reset() + for ikey, skey, s1, i2, s3, i4 in cursor: + i = ikey + square = i * i + cube = square * i + self.assertEqual(ikey, i) + self.assertEqual(skey, 'key' + str(i)) + self.assertEqual(s1, 'val' + str(square)) + self.assertEqual(i2, square) + self.assertEqual(s3, 'val' + str(cube)) + self.assertEqual(i4, cube) + count += 1 + cursor.close() + self.assertEqual(count, n) + + def check_indices(self): + # we check an index that was created before populating + cursor = self.session.open_cursor('index:main:S1i4', None, None) + count = 0 + n = self.nentries + for s1key, i4key, s1, i2, s3, i4 in cursor: + i = int(i4key ** (1 / 3.0) + 0.0001) # cuberoot + #self.tty('index:main:S1i4[' + str(i) + '] (' + + # str(s1key) + ',' + + # str(i4key) + ') -> (' + + # str(s1) + ',' + + # str(i2) + ',' + + # str(s3) + ',' + + # str(i4) + ')') + self.assertEqual(s1key, s1) + self.assertEqual(i4key, i4) + ikey = i + skey = 'key' + str(i) + square = i * i + cube = square * i + self.assertEqual(ikey, i) + self.assertEqual(skey, 'key' + str(i)) + self.assertEqual(s1, 'val' + str(square)) + self.assertEqual(i2, square) + self.assertEqual(s3, 'val' + str(cube)) + self.assertEqual(i4, cube) + count += 1 + cursor.close() + self.assertEqual(count, n) + + # we check an index that was created after populating + cursor = self.session.open_cursor('index:main:i2S1i4', None, None) + count = 0 + for i2key, s1key, i4key, s1, i2, s3, i4 in cursor: + i = int(i4key ** (1 / 3.0) + 0.0001) # cuberoot + #self.tty('index:main:i2S1i4[' + str(i) + '] (' + + # str(i2key) + ',' + + # str(s1key) + ',' + + # str(i4key) + ') -> (' + + # str(s1) + ',' + + # str(i2) + ',' + + # str(s3) + ',' + + # str(i4) + ')') + self.assertEqual(i2key, i2) + self.assertEqual(s1key, s1) + self.assertEqual(i4key, i4) + ikey = i + skey = 'key' + str(i) + square = i * i + cube = square * i + self.assertEqual(ikey, i) + self.assertEqual(skey, 'key' + str(i)) + self.assertEqual(s1, 'val' + str(square)) + self.assertEqual(i2, square) + self.assertEqual(s3, 'val' + str(cube)) + self.assertEqual(i4, cube) + count += 1 + cursor.close() + self.assertEqual(count, n) + + def test_colgroups(self): + self.session.create("table:main", "key_format=iS,value_format=SiSi," + "columns=(ikey,Skey,S1,i2,S3,i4),colgroups=(c1,c2)") + self.session.create("colgroup:main:c1", "columns=(S1,i2)") + self.session.create("colgroup:main:c2", "columns=(S3,i4)") + self.populate() + self.check_entries() + + +if __name__ == '__main__': + wttest.run() |