Python code coverage for Lib/bsddb/test/test_recno.py

#	count	content
1	n/a	"""TestCases for exercising a Recno DB.
2	n/a	"""
3	n/a
4	n/a	import os, sys
5	n/a	import errno
6	n/a	from pprint import pprint
7	n/a	import unittest
8	n/a
9	n/a	from test_all import db, test_support, verbose, get_new_environment_path, get_new_database_path
10	n/a
11	n/a	letters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
12	n/a
13	n/a
14	n/a	#----------------------------------------------------------------------
15	n/a
16	n/a	class SimpleRecnoTestCase(unittest.TestCase):
17	n/a	if sys.version_info < (2, 4) :
18	n/a	def assertFalse(self, expr, msg=None) :
19	n/a	return self.failIf(expr,msg=msg)
20	n/a	def assertTrue(self, expr, msg=None) :
21	n/a	return self.assert_(expr, msg=msg)
22	n/a
23	n/a	if (sys.version_info < (2, 7)) or ((sys.version_info >= (3, 0)) and
24	n/a	(sys.version_info < (3, 2))) :
25	n/a	def assertIsInstance(self, obj, datatype, msg=None) :
26	n/a	return self.assertEqual(type(obj), datatype, msg=msg)
27	n/a	def assertGreaterEqual(self, a, b, msg=None) :
28	n/a	return self.assertTrue(a>=b, msg=msg)
29	n/a
30	n/a
31	n/a	def setUp(self):
32	n/a	self.filename = get_new_database_path()
33	n/a	self.homeDir = None
34	n/a
35	n/a	def tearDown(self):
36	n/a	test_support.unlink(self.filename)
37	n/a	if self.homeDir:
38	n/a	test_support.rmtree(self.homeDir)
39	n/a
40	n/a	def test01_basic(self):
41	n/a	d = db.DB()
42	n/a
43	n/a	get_returns_none = d.set_get_returns_none(2)
44	n/a	d.set_get_returns_none(get_returns_none)
45	n/a
46	n/a	d.open(self.filename, db.DB_RECNO, db.DB_CREATE)
47	n/a
48	n/a	for x in letters:
49	n/a	recno = d.append(x * 60)
50	n/a	self.assertIsInstance(recno, int)
51	n/a	self.assertGreaterEqual(recno, 1)
52	n/a	if verbose:
53	n/a	print recno,
54	n/a
55	n/a	if verbose: print
56	n/a
57	n/a	stat = d.stat()
58	n/a	if verbose:
59	n/a	pprint(stat)
60	n/a
61	n/a	for recno in range(1, len(d)+1):
62	n/a	data = d[recno]
63	n/a	if verbose:
64	n/a	print data
65	n/a
66	n/a	self.assertIsInstance(data, str)
67	n/a	self.assertEqual(data, d.get(recno))
68	n/a
69	n/a	try:
70	n/a	data = d[0] # This should raise a KeyError!?!?!
71	n/a	except db.DBInvalidArgError, val:
72	n/a	if sys.version_info < (2, 6) :
73	n/a	self.assertEqual(val[0], db.EINVAL)
74	n/a	else :
75	n/a	self.assertEqual(val.args[0], db.EINVAL)
76	n/a	if verbose: print val
77	n/a	else:
78	n/a	self.fail("expected exception")
79	n/a
80	n/a	# test that has_key raises DB exceptions (fixed in pybsddb 4.3.2)
81	n/a	try:
82	n/a	d.has_key(0)
83	n/a	except db.DBError, val:
84	n/a	pass
85	n/a	else:
86	n/a	self.fail("has_key did not raise a proper exception")
87	n/a
88	n/a	try:
89	n/a	data = d[100]
90	n/a	except KeyError:
91	n/a	pass
92	n/a	else:
93	n/a	self.fail("expected exception")
94	n/a
95	n/a	try:
96	n/a	data = d.get(100)
97	n/a	except db.DBNotFoundError, val:
98	n/a	if get_returns_none:
99	n/a	self.fail("unexpected exception")
100	n/a	else:
101	n/a	self.assertEqual(data, None)
102	n/a
103	n/a	keys = d.keys()
104	n/a	if verbose:
105	n/a	print keys
106	n/a	self.assertIsInstance(keys, list)
107	n/a	self.assertIsInstance(keys[0], int)
108	n/a	self.assertEqual(len(keys), len(d))
109	n/a
110	n/a	items = d.items()
111	n/a	if verbose:
112	n/a	pprint(items)
113	n/a	self.assertIsInstance(items, list)
114	n/a	self.assertIsInstance(items[0], tuple)
115	n/a	self.assertEqual(len(items[0]), 2)
116	n/a	self.assertIsInstance(items[0][0], int)
117	n/a	self.assertIsInstance(items[0][1], str)
118	n/a	self.assertEqual(len(items), len(d))
119	n/a
120	n/a	self.assertTrue(d.has_key(25))
121	n/a
122	n/a	del d[25]
123	n/a	self.assertFalse(d.has_key(25))
124	n/a
125	n/a	d.delete(13)
126	n/a	self.assertFalse(d.has_key(13))
127	n/a
128	n/a	data = d.get_both(26, "z" * 60)
129	n/a	self.assertEqual(data, "z" * 60, 'was %r' % data)
130	n/a	if verbose:
131	n/a	print data
132	n/a
133	n/a	fd = d.fd()
134	n/a	if verbose:
135	n/a	print fd
136	n/a
137	n/a	c = d.cursor()
138	n/a	rec = c.first()
139	n/a	while rec:
140	n/a	if verbose:
141	n/a	print rec
142	n/a	rec = c.next()
143	n/a
144	n/a	c.set(50)
145	n/a	rec = c.current()
146	n/a	if verbose:
147	n/a	print rec
148	n/a
149	n/a	c.put(-1, "a replacement record", db.DB_CURRENT)
150	n/a
151	n/a	c.set(50)
152	n/a	rec = c.current()
153	n/a	self.assertEqual(rec, (50, "a replacement record"))
154	n/a	if verbose:
155	n/a	print rec
156	n/a
157	n/a	rec = c.set_range(30)
158	n/a	if verbose:
159	n/a	print rec
160	n/a
161	n/a	# test that non-existent key lookups work (and that
162	n/a	# DBC_set_range doesn't have a memleak under valgrind)
163	n/a	rec = c.set_range(999999)
164	n/a	self.assertEqual(rec, None)
165	n/a	if verbose:
166	n/a	print rec
167	n/a
168	n/a	c.close()
169	n/a	d.close()
170	n/a
171	n/a	d = db.DB()
172	n/a	d.open(self.filename)
173	n/a	c = d.cursor()
174	n/a
175	n/a	# put a record beyond the consecutive end of the recno's
176	n/a	d[100] = "way out there"
177	n/a	self.assertEqual(d[100], "way out there")
178	n/a
179	n/a	try:
180	n/a	data = d[99]
181	n/a	except KeyError:
182	n/a	pass
183	n/a	else:
184	n/a	self.fail("expected exception")
185	n/a
186	n/a	try:
187	n/a	d.get(99)
188	n/a	except db.DBKeyEmptyError, val:
189	n/a	if get_returns_none:
190	n/a	self.fail("unexpected DBKeyEmptyError exception")
191	n/a	else:
192	n/a	if sys.version_info < (2, 6) :
193	n/a	self.assertEqual(val[0], db.DB_KEYEMPTY)
194	n/a	else :
195	n/a	self.assertEqual(val.args[0], db.DB_KEYEMPTY)
196	n/a	if verbose: print val
197	n/a	else:
198	n/a	if not get_returns_none:
199	n/a	self.fail("expected exception")
200	n/a
201	n/a	rec = c.set(40)
202	n/a	while rec:
203	n/a	if verbose:
204	n/a	print rec
205	n/a	rec = c.next()
206	n/a
207	n/a	c.close()
208	n/a	d.close()
209	n/a
210	n/a	def test02_WithSource(self):
211	n/a	"""
212	n/a	A Recno file that is given a "backing source file" is essentially a
213	n/a	simple ASCII file. Normally each record is delimited by \n and so is
214	n/a	just a line in the file, but you can set a different record delimiter
215	n/a	if needed.
216	n/a	"""
217	n/a	homeDir = get_new_environment_path()
218	n/a	self.homeDir = homeDir
219	n/a	source = os.path.join(homeDir, 'test_recno.txt')
220	n/a	if not os.path.isdir(homeDir):
221	n/a	os.mkdir(homeDir)
222	n/a	f = open(source, 'w') # create the file
223	n/a	f.close()
224	n/a
225	n/a	d = db.DB()
226	n/a	# This is the default value, just checking if both int
227	n/a	d.set_re_delim(0x0A)
228	n/a	d.set_re_delim('\n') # and char can be used...
229	n/a	d.set_re_source(source)
230	n/a	d.open(self.filename, db.DB_RECNO, db.DB_CREATE)
231	n/a
232	n/a	data = "The quick brown fox jumped over the lazy dog".split()
233	n/a	for datum in data:
234	n/a	d.append(datum)
235	n/a	d.sync()
236	n/a	d.close()
237	n/a
238	n/a	# get the text from the backing source
239	n/a	text = open(source, 'r').read()
240	n/a	text = text.strip()
241	n/a	if verbose:
242	n/a	print text
243	n/a	print data
244	n/a	print text.split('\n')
245	n/a
246	n/a	self.assertEqual(text.split('\n'), data)
247	n/a
248	n/a	# open as a DB again
249	n/a	d = db.DB()
250	n/a	d.set_re_source(source)
251	n/a	d.open(self.filename, db.DB_RECNO)
252	n/a
253	n/a	d[3] = 'reddish-brown'
254	n/a	d[8] = 'comatose'
255	n/a
256	n/a	d.sync()
257	n/a	d.close()
258	n/a
259	n/a	text = open(source, 'r').read()
260	n/a	text = text.strip()
261	n/a	if verbose:
262	n/a	print text
263	n/a	print text.split('\n')
264	n/a
265	n/a	self.assertEqual(text.split('\n'),
266	n/a	"The quick reddish-brown fox jumped over the comatose dog".split())
267	n/a
268	n/a	def test03_FixedLength(self):
269	n/a	d = db.DB()
270	n/a	d.set_re_len(40) # fixed length records, 40 bytes long
271	n/a	d.set_re_pad('-') # sets the pad character...
272	n/a	d.set_re_pad(45) # ...test both int and char
273	n/a	d.open(self.filename, db.DB_RECNO, db.DB_CREATE)
274	n/a
275	n/a	for x in letters:
276	n/a	d.append(x * 35) # These will be padded
277	n/a
278	n/a	d.append('.' * 40) # this one will be exact
279	n/a
280	n/a	try: # this one will fail
281	n/a	d.append('bad' * 20)
282	n/a	except db.DBInvalidArgError, val:
283	n/a	if sys.version_info < (2, 6) :
284	n/a	self.assertEqual(val[0], db.EINVAL)
285	n/a	else :
286	n/a	self.assertEqual(val.args[0], db.EINVAL)
287	n/a	if verbose: print val
288	n/a	else:
289	n/a	self.fail("expected exception")
290	n/a
291	n/a	c = d.cursor()
292	n/a	rec = c.first()
293	n/a	while rec:
294	n/a	if verbose:
295	n/a	print rec
296	n/a	rec = c.next()
297	n/a
298	n/a	c.close()
299	n/a	d.close()
300	n/a
301	n/a
302	n/a	#----------------------------------------------------------------------
303	n/a
304	n/a
305	n/a	def test_suite():
306	n/a	return unittest.makeSuite(SimpleRecnoTestCase)
307	n/a
308	n/a
309	n/a	if __name__ == '__main__':
310	n/a	unittest.main(defaultTest='test_suite')