1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#!/usr/bin/env python
#
# Public Domain 2014-2019 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.
#
# test_pack.py
# Tests packing using public methods
#
import wiredtiger, wttest
import re, sys
class test_pack(wttest.WiredTigerTestCase):
name = 'test_pack'
def dump_cursor(self, cursor, name):
cursor.reset()
while cursor.next() == 0:
x = cursor.get_key()
y = cursor.get_value()
self.tty(' ' + name + ': ' + str(x) + ' => ' + str(y))
cursor.reset()
def check(self, fmt, *v):
v = list(v)
fmtname = re.sub('([A-Z])', r'_\1', fmt)
uri = 'table:' + test_pack.name + '-' + fmtname
idx_uri = 'index:' + test_pack.name + '-' + fmtname + ':inverse'
nargs = len(v)
colnames = ",".join("v" + str(x) for x in range(nargs))
self.session.create(uri, "columns=(k," + colnames + ")," +
"key_format=i,value_format=" + fmt)
self.session.create(idx_uri, "columns=(" + colnames + ")")
forw = self.session.open_cursor(uri, None, None)
forw_idx = self.session.open_cursor(idx_uri + "(k)", None, None)
forw.set_key(1234)
forw.set_value(*v)
forw.insert()
#self.dump_cursor(forw, 'forw')
#self.dump_cursor(forw_idx, 'index')
forw.set_key(1234)
self.assertEquals(forw.search(), 0)
got = forw.get_value()
if nargs == 1: # API does not return a list, we want one for comparing
got = [got]
self.assertEquals(got, v)
forw_idx.set_key(*v)
self.assertEquals(forw_idx.search(), 0)
self.assertEquals(forw_idx.get_value(), 1234)
forw.close()
forw_idx.close()
def test_packing(self):
self.check('iii', 0, 101, -99)
self.check('3i', 0, 101, -99)
self.check('iS', 42, "forty two")
self.check('S', 'abc')
self.check('9S', 'a' * 9)
self.check('9SS', "forty two", "spam egg")
self.check('42S', 'a' * 42)
self.check('42SS', 'a' * 42, 'something')
self.check('S42S', 'something', 'a' * 42)
# nul terminated string with padding
self.check('10SS', 'aaaaa\x00\x00\x00\x00\x00', 'something')
self.check('S10S', 'something', 'aaaaa\x00\x00\x00\x00\x00')
self.check('u', b"\x42" * 20)
self.check('uu', b"\x42" * 10, b"\x42" * 10)
self.check('3u', b"\x04\x03\x02")
self.check('3uu', b"\x04\x03\x02", b"\x42" * 10)
self.check('u3u', b"\x42" * 10, b"\x04\x03\x02")
self.check('u', b'\x00')
self.check('u', b'')
self.check('uu', b'', b'\x00')
self.check('uu', b'\x00', b'')
self.check('uu', b'', b'')
self.check('s', "4")
self.check("1s", "4")
self.check("2s", "42")
if __name__ == '__main__':
wttest.run()
|
