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

#	count	content
1	n/a	"""
2	n/a	TestCases for exercising a Queue DB.
3	n/a	"""
4	n/a
5	n/a	import os, string
6	n/a	from pprint import pprint
7	n/a	import unittest
8	n/a
9	n/a	from test_all import db, verbose, get_new_database_path
10	n/a
11	n/a	#----------------------------------------------------------------------
12	n/a
13	n/a	class SimpleQueueTestCase(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_basic(self):
25	n/a	# Basic Queue tests using the deprecated DBCursor.consume method.
26	n/a
27	n/a	if verbose:
28	n/a	print '\n', '-=' * 30
29	n/a	print "Running %s.test01_basic..." % self.__class__.__name__
30	n/a
31	n/a	d = db.DB()
32	n/a	d.set_re_len(40) # Queues must be fixed length
33	n/a	d.open(self.filename, db.DB_QUEUE, db.DB_CREATE)
34	n/a
35	n/a	if verbose:
36	n/a	print "before appends" + '-' * 30
37	n/a	pprint(d.stat())
38	n/a
39	n/a	for x in string.letters:
40	n/a	d.append(x * 40)
41	n/a
42	n/a	self.assertEqual(len(d), len(string.letters))
43	n/a
44	n/a	d.put(100, "some more data")
45	n/a	d.put(101, "and some more ")
46	n/a	d.put(75, "out of order")
47	n/a	d.put(1, "replacement data")
48	n/a
49	n/a	self.assertEqual(len(d), len(string.letters)+3)
50	n/a
51	n/a	if verbose:
52	n/a	print "before close" + '-' * 30
53	n/a	pprint(d.stat())
54	n/a
55	n/a	d.close()
56	n/a	del d
57	n/a	d = db.DB()
58	n/a	d.open(self.filename)
59	n/a
60	n/a	if verbose:
61	n/a	print "after open" + '-' * 30
62	n/a	pprint(d.stat())
63	n/a
64	n/a	# Test "txn" as a positional parameter
65	n/a	d.append("one more", None)
66	n/a	# Test "txn" as a keyword parameter
67	n/a	d.append("another one", txn=None)
68	n/a
69	n/a	c = d.cursor()
70	n/a
71	n/a	if verbose:
72	n/a	print "after append" + '-' * 30
73	n/a	pprint(d.stat())
74	n/a
75	n/a	rec = c.consume()
76	n/a	while rec:
77	n/a	if verbose:
78	n/a	print rec
79	n/a	rec = c.consume()
80	n/a	c.close()
81	n/a
82	n/a	if verbose:
83	n/a	print "after consume loop" + '-' * 30
84	n/a	pprint(d.stat())
85	n/a
86	n/a	self.assertEqual(len(d), 0, \
87	n/a	"if you see this message then you need to rebuild " \
88	n/a	"Berkeley DB 3.1.17 with the patch in patches/qam_stat.diff")
89	n/a
90	n/a	d.close()
91	n/a
92	n/a
93	n/a
94	n/a	def test02_basicPost32(self):
95	n/a	# Basic Queue tests using the new DB.consume method in DB 3.2+
96	n/a	# (No cursor needed)
97	n/a
98	n/a	if verbose:
99	n/a	print '\n', '-=' * 30
100	n/a	print "Running %s.test02_basicPost32..." % self.__class__.__name__
101	n/a
102	n/a	if db.version() < (3, 2, 0):
103	n/a	if verbose:
104	n/a	print "Test not run, DB not new enough..."
105	n/a	return
106	n/a
107	n/a	d = db.DB()
108	n/a	d.set_re_len(40) # Queues must be fixed length
109	n/a	d.open(self.filename, db.DB_QUEUE, db.DB_CREATE)
110	n/a
111	n/a	if verbose:
112	n/a	print "before appends" + '-' * 30
113	n/a	pprint(d.stat())
114	n/a
115	n/a	for x in string.letters:
116	n/a	d.append(x * 40)
117	n/a
118	n/a	self.assertEqual(len(d), len(string.letters))
119	n/a
120	n/a	d.put(100, "some more data")
121	n/a	d.put(101, "and some more ")
122	n/a	d.put(75, "out of order")
123	n/a	d.put(1, "replacement data")
124	n/a
125	n/a	self.assertEqual(len(d), len(string.letters)+3)
126	n/a
127	n/a	if verbose:
128	n/a	print "before close" + '-' * 30
129	n/a	pprint(d.stat())
130	n/a
131	n/a	d.close()
132	n/a	del d
133	n/a	d = db.DB()
134	n/a	d.open(self.filename)
135	n/a	#d.set_get_returns_none(true)
136	n/a
137	n/a	if verbose:
138	n/a	print "after open" + '-' * 30
139	n/a	pprint(d.stat())
140	n/a
141	n/a	d.append("one more")
142	n/a
143	n/a	if verbose:
144	n/a	print "after append" + '-' * 30
145	n/a	pprint(d.stat())
146	n/a
147	n/a	rec = d.consume()
148	n/a	while rec:
149	n/a	if verbose:
150	n/a	print rec
151	n/a	rec = d.consume()
152	n/a
153	n/a	if verbose:
154	n/a	print "after consume loop" + '-' * 30
155	n/a	pprint(d.stat())
156	n/a
157	n/a	d.close()
158	n/a
159	n/a
160	n/a
161	n/a	#----------------------------------------------------------------------
162	n/a
163	n/a	def test_suite():
164	n/a	return unittest.makeSuite(SimpleQueueTestCase)
165	n/a
166	n/a
167	n/a	if __name__ == '__main__':
168	n/a	unittest.main(defaultTest='test_suite')