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
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#!/usr/bin/env python
#
# Public Domain 2014-present 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.
from suite_subprocess import suite_subprocess
import wiredtiger, wttest
from wtscenario import make_scenarios
# test_util03.py
# Utilities: wt create
class test_util03(wttest.WiredTigerTestCase, suite_subprocess):
tablename = 'test_util03.a'
nentries = 1000
scenarios = make_scenarios([
('none', dict(key_format=None,value_format=None)),
('SS', dict(key_format='S',value_format='S')),
('rS', dict(key_format='r',value_format='S')),
('ri', dict(key_format='r',value_format='i')),
])
def test_create_process(self):
"""
Test create in a 'wt' process
"""
args = ["create"]
if self.key_format != None or self.value_format != None:
args.append('-c')
config = ''
if self.key_format != None:
config += 'key_format=' + self.key_format + ','
if self.value_format != None:
config += 'value_format=' + self.value_format
args.append(config)
args.append('table:' + self.tablename)
self.runWt(args)
cursor = self.session.open_cursor('table:' + self.tablename, None, None)
if self.key_format != None:
self.assertEqual(cursor.key_format, self.key_format)
if self.value_format != None:
self.assertEqual(cursor.value_format, self.value_format)
for key,val in cursor:
self.fail('table should be empty')
cursor.close()
class test_util03_import(wttest.WiredTigerTestCase, suite_subprocess):
tablename = 'test_util03.a'
nentries = 1000
create_config = 'allocation_size=512,key_format=i,value_format=i'
scenarios = make_scenarios([
('file_metadata', dict(repair=False)),
('repair', dict(repair=True)),
])
def test_create_process_import(self):
"""
Test create in a 'wt' process with the import flags.
"""
# Create and populate the table.
uri = 'file:' + self.tablename
self.session.create(uri, self.create_config)
cursor = self.session.open_cursor(uri)
for i in range(1, 1000):
cursor[i] = i
cursor.close()
self.session.checkpoint()
# Export the metadata for the file.
c = self.session.open_cursor('metadata:', None, None)
original_db_file_config = c[uri]
c.close()
# Now drop it. Keep the file there since we're about to import it back.
self.session.drop(uri, 'remove_files=false')
# Can't open the cursor on the file anymore since we dropped it.
self.assertRaisesException(wiredtiger.WiredTigerError,
lambda: self.session.open_cursor(uri))
# Import with the WT tool.
# We want to test the case where we have the associated file metadata AND when we don't
# and wish to repair it.
if self.repair:
import_config = 'import=(enabled,repair=true)'
else:
import_config = 'import=(enabled,repair=false,file_metadata=({}))'.format(original_db_file_config)
self.runWt(['create', '-c', import_config, uri])
# Check that the data got imported correctly.
cursor = self.session.open_cursor(uri)
for i in range(1, 1000):
self.assertEqual(cursor[i], i)
cursor.close()
if __name__ == '__main__':
wttest.run()
|