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

#	count	content
1	n/a	"""
2	n/a	TestCases for checking set_get_returns_none.
3	n/a	"""
4	n/a
5	n/a	import os, string
6	n/a	import unittest
7	n/a
8	n/a	from test_all import db, verbose, get_new_database_path
9	n/a
10	n/a
11	n/a	#----------------------------------------------------------------------
12	n/a
13	n/a	class GetReturnsNoneTestCase(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_get_returns_none(self):
25	n/a	d = db.DB()
26	n/a	d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
27	n/a	d.set_get_returns_none(1)
28	n/a
29	n/a	for x in string.letters:
30	n/a	d.put(x, x * 40)
31	n/a
32	n/a	data = d.get('bad key')
33	n/a	self.assertEqual(data, None)
34	n/a
35	n/a	data = d.get(string.letters[0])
36	n/a	self.assertEqual(data, string.letters[0]*40)
37	n/a
38	n/a	count = 0
39	n/a	c = d.cursor()
40	n/a	rec = c.first()
41	n/a	while rec:
42	n/a	count = count + 1
43	n/a	rec = c.next()
44	n/a
45	n/a	self.assertEqual(rec, None)
46	n/a	self.assertEqual(count, len(string.letters))
47	n/a
48	n/a	c.close()
49	n/a	d.close()
50	n/a
51	n/a
52	n/a	def test02_get_raises_exception(self):
53	n/a	d = db.DB()
54	n/a	d.open(self.filename, db.DB_BTREE, db.DB_CREATE)
55	n/a	d.set_get_returns_none(0)
56	n/a
57	n/a	for x in string.letters:
58	n/a	d.put(x, x * 40)
59	n/a
60	n/a	self.assertRaises(db.DBNotFoundError, d.get, 'bad key')
61	n/a	self.assertRaises(KeyError, d.get, 'bad key')
62	n/a
63	n/a	data = d.get(string.letters[0])
64	n/a	self.assertEqual(data, string.letters[0]*40)
65	n/a
66	n/a	count = 0
67	n/a	exceptionHappened = 0
68	n/a	c = d.cursor()
69	n/a	rec = c.first()
70	n/a	while rec:
71	n/a	count = count + 1
72	n/a	try:
73	n/a	rec = c.next()
74	n/a	except db.DBNotFoundError: # end of the records
75	n/a	exceptionHappened = 1
76	n/a	break
77	n/a
78	n/a	self.assertNotEqual(rec, None)
79	n/a	self.assert_(exceptionHappened)
80	n/a	self.assertEqual(count, len(string.letters))
81	n/a
82	n/a	c.close()
83	n/a	d.close()
84	n/a
85	n/a	#----------------------------------------------------------------------
86	n/a
87	n/a	def test_suite():
88	n/a	return unittest.makeSuite(GetReturnsNoneTestCase)
89	n/a
90	n/a
91	n/a	if __name__ == '__main__':
92	n/a	unittest.main(defaultTest='test_suite')