285 lines
12 KiB
Python
285 lines
12 KiB
Python
#!/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",
|
|
"")
|
|
|
|
# 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()
|