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

#	count	content
1	n/a	"""TestCases for using the DB.join and DBCursor.join_item methods.
2	n/a	"""
3	n/a
4	n/a	import os
5	n/a
6	n/a	import unittest
7	n/a
8	n/a	from test_all import db, dbshelve, test_support, verbose, \
9	n/a	get_new_environment_path, get_new_database_path
10	n/a
11	n/a	#----------------------------------------------------------------------
12	n/a
13	n/a	ProductIndex = [
14	n/a	('apple', "Convenience Store"),
15	n/a	('blueberry', "Farmer's Market"),
16	n/a	('shotgun', "S-Mart"), # Aisle 12
17	n/a	('pear', "Farmer's Market"),
18	n/a	('chainsaw', "S-Mart"), # "Shop smart. Shop S-Mart!"
19	n/a	('strawberry', "Farmer's Market"),
20	n/a	]
21	n/a
22	n/a	ColorIndex = [
23	n/a	('blue', "blueberry"),
24	n/a	('red', "apple"),
25	n/a	('red', "chainsaw"),
26	n/a	('red', "strawberry"),
27	n/a	('yellow', "peach"),
28	n/a	('yellow', "pear"),
29	n/a	('black', "shotgun"),
30	n/a	]
31	n/a
32	n/a	class JoinTestCase(unittest.TestCase):
33	n/a	keytype = ''
34	n/a
35	n/a	def setUp(self):
36	n/a	self.filename = self.__class__.__name__ + '.db'
37	n/a	self.homeDir = get_new_environment_path()
38	n/a	self.env = db.DBEnv()
39	n/a	self.env.open(self.homeDir, db.DB_CREATE \| db.DB_INIT_MPOOL \| db.DB_INIT_LOCK )
40	n/a
41	n/a	def tearDown(self):
42	n/a	self.env.close()
43	n/a	test_support.rmtree(self.homeDir)
44	n/a
45	n/a	def test01_join(self):
46	n/a	if verbose:
47	n/a	print '\n', '-=' * 30
48	n/a	print "Running %s.test01_join..." % \
49	n/a	self.__class__.__name__
50	n/a
51	n/a	# create and populate primary index
52	n/a	priDB = db.DB(self.env)
53	n/a	priDB.open(self.filename, "primary", db.DB_BTREE, db.DB_CREATE)
54	n/a	map(lambda t, priDB=priDB: priDB.put(*t), ProductIndex)
55	n/a
56	n/a	# create and populate secondary index
57	n/a	secDB = db.DB(self.env)
58	n/a	secDB.set_flags(db.DB_DUP \| db.DB_DUPSORT)
59	n/a	secDB.open(self.filename, "secondary", db.DB_BTREE, db.DB_CREATE)
60	n/a	map(lambda t, secDB=secDB: secDB.put(*t), ColorIndex)
61	n/a
62	n/a	sCursor = None
63	n/a	jCursor = None
64	n/a	try:
65	n/a	# lets look up all of the red Products
66	n/a	sCursor = secDB.cursor()
67	n/a	# Don't do the .set() in an assert, or you can get a bogus failure
68	n/a	# when running python -O
69	n/a	tmp = sCursor.set('red')
70	n/a	self.assert_(tmp)
71	n/a
72	n/a	# FIXME: jCursor doesn't properly hold a reference to its
73	n/a	# cursors, if they are closed before jcursor is used it
74	n/a	# can cause a crash.
75	n/a	jCursor = priDB.join([sCursor])
76	n/a
77	n/a	if jCursor.get(0) != ('apple', "Convenience Store"):
78	n/a	self.fail("join cursor positioned wrong")
79	n/a	if jCursor.join_item() != 'chainsaw':
80	n/a	self.fail("DBCursor.join_item returned wrong item")
81	n/a	if jCursor.get(0)[0] != 'strawberry':
82	n/a	self.fail("join cursor returned wrong thing")
83	n/a	if jCursor.get(0): # there were only three red items to return
84	n/a	self.fail("join cursor returned too many items")
85	n/a	finally:
86	n/a	if jCursor:
87	n/a	jCursor.close()
88	n/a	if sCursor:
89	n/a	sCursor.close()
90	n/a	priDB.close()
91	n/a	secDB.close()
92	n/a
93	n/a
94	n/a	def test_suite():
95	n/a	suite = unittest.TestSuite()
96	n/a
97	n/a	suite.addTest(unittest.makeSuite(JoinTestCase))
98	n/a
99	n/a	return suite