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

#	count	content
1	n/a	"""
2	n/a	Test cases adapted from the test_bsddb.py module in Python's
3	n/a	regression test suite.
4	n/a	"""
5	n/a
6	n/a	import os, string
7	n/a	import unittest
8	n/a
9	n/a	from test_all import db, hashopen, btopen, rnopen, verbose, \
10	n/a	get_new_database_path
11	n/a
12	n/a
13	n/a	class CompatibilityTestCase(unittest.TestCase):
14	n/a	def setUp(self):
15	n/a	self.filename = get_new_database_path()
16	n/a
17	n/a	def tearDown(self):
18	n/a	try:
19	n/a	os.remove(self.filename)
20	n/a	except os.error:
21	n/a	pass
22	n/a
23	n/a
24	n/a	def test01_btopen(self):
25	n/a	self.do_bthash_test(btopen, 'btopen')
26	n/a
27	n/a	def test02_hashopen(self):
28	n/a	self.do_bthash_test(hashopen, 'hashopen')
29	n/a
30	n/a	def test03_rnopen(self):
31	n/a	data = "The quick brown fox jumped over the lazy dog.".split()
32	n/a	if verbose:
33	n/a	print "\nTesting: rnopen"
34	n/a
35	n/a	f = rnopen(self.filename, 'c')
36	n/a	for x in range(len(data)):
37	n/a	f[x+1] = data[x]
38	n/a
39	n/a	getTest = (f[1], f[2], f[3])
40	n/a	if verbose:
41	n/a	print '%s %s %s' % getTest
42	n/a
43	n/a	self.assertEqual(getTest[1], 'quick', 'data mismatch!')
44	n/a
45	n/a	rv = f.set_location(3)
46	n/a	if rv != (3, 'brown'):
47	n/a	self.fail('recno database set_location failed: '+repr(rv))
48	n/a
49	n/a	f[25] = 'twenty-five'
50	n/a	f.close()
51	n/a	del f
52	n/a
53	n/a	f = rnopen(self.filename, 'w')
54	n/a	f[20] = 'twenty'
55	n/a
56	n/a	def noRec(f):
57	n/a	rec = f[15]
58	n/a	self.assertRaises(KeyError, noRec, f)
59	n/a
60	n/a	def badKey(f):
61	n/a	rec = f['a string']
62	n/a	self.assertRaises(TypeError, badKey, f)
63	n/a
64	n/a	del f[3]
65	n/a
66	n/a	rec = f.first()
67	n/a	while rec:
68	n/a	if verbose:
69	n/a	print rec
70	n/a	try:
71	n/a	rec = f.next()
72	n/a	except KeyError:
73	n/a	break
74	n/a
75	n/a	f.close()
76	n/a
77	n/a
78	n/a	def test04_n_flag(self):
79	n/a	f = hashopen(self.filename, 'n')
80	n/a	f.close()
81	n/a
82	n/a
83	n/a	def do_bthash_test(self, factory, what):
84	n/a	if verbose:
85	n/a	print '\nTesting: ', what
86	n/a
87	n/a	f = factory(self.filename, 'c')
88	n/a	if verbose:
89	n/a	print 'creation...'
90	n/a
91	n/a	# truth test
92	n/a	if f:
93	n/a	if verbose: print "truth test: true"
94	n/a	else:
95	n/a	if verbose: print "truth test: false"
96	n/a
97	n/a	f['0'] = ''
98	n/a	f['a'] = 'Guido'
99	n/a	f['b'] = 'van'
100	n/a	f['c'] = 'Rossum'
101	n/a	f['d'] = 'invented'
102	n/a	# 'e' intentionally left out
103	n/a	f['f'] = 'Python'
104	n/a	if verbose:
105	n/a	print '%s %s %s' % (f['a'], f['b'], f['c'])
106	n/a
107	n/a	if verbose:
108	n/a	print 'key ordering...'
109	n/a	start = f.set_location(f.first()[0])
110	n/a	if start != ('0', ''):
111	n/a	self.fail("incorrect first() result: "+repr(start))
112	n/a	while 1:
113	n/a	try:
114	n/a	rec = f.next()
115	n/a	except KeyError:
116	n/a	self.assertEqual(rec, f.last(), 'Error, last <> last!')
117	n/a	f.previous()
118	n/a	break
119	n/a	if verbose:
120	n/a	print rec
121	n/a
122	n/a	self.assert_(f.has_key('f'), 'Error, missing key!')
123	n/a
124	n/a	# test that set_location() returns the next nearest key, value
125	n/a	# on btree databases and raises KeyError on others.
126	n/a	if factory == btopen:
127	n/a	e = f.set_location('e')
128	n/a	if e != ('f', 'Python'):
129	n/a	self.fail('wrong key,value returned: '+repr(e))
130	n/a	else:
131	n/a	try:
132	n/a	e = f.set_location('e')
133	n/a	except KeyError:
134	n/a	pass
135	n/a	else:
136	n/a	self.fail("set_location on non-existent key did not raise KeyError")
137	n/a
138	n/a	f.sync()
139	n/a	f.close()
140	n/a	# truth test
141	n/a	try:
142	n/a	if f:
143	n/a	if verbose: print "truth test: true"
144	n/a	else:
145	n/a	if verbose: print "truth test: false"
146	n/a	except db.DBError:
147	n/a	pass
148	n/a	else:
149	n/a	self.fail("Exception expected")
150	n/a
151	n/a	del f
152	n/a
153	n/a	if verbose:
154	n/a	print 'modification...'
155	n/a	f = factory(self.filename, 'w')
156	n/a	f['d'] = 'discovered'
157	n/a
158	n/a	if verbose:
159	n/a	print 'access...'
160	n/a	for key in f.keys():
161	n/a	word = f[key]
162	n/a	if verbose:
163	n/a	print word
164	n/a
165	n/a	def noRec(f):
166	n/a	rec = f['no such key']
167	n/a	self.assertRaises(KeyError, noRec, f)
168	n/a
169	n/a	def badKey(f):
170	n/a	rec = f[15]
171	n/a	self.assertRaises(TypeError, badKey, f)
172	n/a
173	n/a	f.close()
174	n/a
175	n/a
176	n/a	#----------------------------------------------------------------------
177	n/a
178	n/a
179	n/a	def test_suite():
180	n/a	return unittest.makeSuite(CompatibilityTestCase)
181	n/a
182	n/a
183	n/a	if __name__ == '__main__':
184	n/a	unittest.main(defaultTest='test_suite')