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

#	count	content
1	n/a	"""
2	n/a	TestCases for testing the locking sub-system.
3	n/a	"""
4	n/a
5	n/a	import time
6	n/a
7	n/a	import unittest
8	n/a	from test_all import db, test_support, verbose, have_threads, \
9	n/a	get_new_environment_path, get_new_database_path
10	n/a
11	n/a	if have_threads :
12	n/a	from threading import Thread
13	n/a	import sys
14	n/a	if sys.version_info[0] < 3 :
15	n/a	from threading import currentThread
16	n/a	else :
17	n/a	from threading import current_thread as currentThread
18	n/a
19	n/a	#----------------------------------------------------------------------
20	n/a
21	n/a	class LockingTestCase(unittest.TestCase):
22	n/a	import sys
23	n/a	if sys.version_info < (2, 4) :
24	n/a	def assertTrue(self, expr, msg=None):
25	n/a	self.failUnless(expr,msg=msg)
26	n/a
27	n/a
28	n/a	def setUp(self):
29	n/a	self.homeDir = get_new_environment_path()
30	n/a	self.env = db.DBEnv()
31	n/a	self.env.open(self.homeDir, db.DB_THREAD \| db.DB_INIT_MPOOL \|
32	n/a	db.DB_INIT_LOCK \| db.DB_CREATE)
33	n/a
34	n/a
35	n/a	def tearDown(self):
36	n/a	self.env.close()
37	n/a	test_support.rmtree(self.homeDir)
38	n/a
39	n/a
40	n/a	def test01_simple(self):
41	n/a	if verbose:
42	n/a	print '\n', '-=' * 30
43	n/a	print "Running %s.test01_simple..." % self.__class__.__name__
44	n/a
45	n/a	anID = self.env.lock_id()
46	n/a	if verbose:
47	n/a	print "locker ID: %s" % anID
48	n/a	lock = self.env.lock_get(anID, "some locked thing", db.DB_LOCK_WRITE)
49	n/a	if verbose:
50	n/a	print "Aquired lock: %s" % lock
51	n/a	self.env.lock_put(lock)
52	n/a	if verbose:
53	n/a	print "Released lock: %s" % lock
54	n/a	self.env.lock_id_free(anID)
55	n/a
56	n/a
57	n/a	def test02_threaded(self):
58	n/a	if verbose:
59	n/a	print '\n', '-=' * 30
60	n/a	print "Running %s.test02_threaded..." % self.__class__.__name__
61	n/a
62	n/a	threads = []
63	n/a	threads.append(Thread(target = self.theThread,
64	n/a	args=(db.DB_LOCK_WRITE,)))
65	n/a	threads.append(Thread(target = self.theThread,
66	n/a	args=(db.DB_LOCK_READ,)))
67	n/a	threads.append(Thread(target = self.theThread,
68	n/a	args=(db.DB_LOCK_READ,)))
69	n/a	threads.append(Thread(target = self.theThread,
70	n/a	args=(db.DB_LOCK_WRITE,)))
71	n/a	threads.append(Thread(target = self.theThread,
72	n/a	args=(db.DB_LOCK_READ,)))
73	n/a	threads.append(Thread(target = self.theThread,
74	n/a	args=(db.DB_LOCK_READ,)))
75	n/a	threads.append(Thread(target = self.theThread,
76	n/a	args=(db.DB_LOCK_WRITE,)))
77	n/a	threads.append(Thread(target = self.theThread,
78	n/a	args=(db.DB_LOCK_WRITE,)))
79	n/a	threads.append(Thread(target = self.theThread,
80	n/a	args=(db.DB_LOCK_WRITE,)))
81	n/a
82	n/a	for t in threads:
83	n/a	import sys
84	n/a	if sys.version_info[0] < 3 :
85	n/a	t.setDaemon(True)
86	n/a	else :
87	n/a	t.daemon = True
88	n/a	t.start()
89	n/a	for t in threads:
90	n/a	t.join()
91	n/a
92	n/a	if db.version() >= (4, 2) :
93	n/a	def test03_lock_timeout(self):
94	n/a	self.env.set_timeout(0, db.DB_SET_LOCK_TIMEOUT)
95	n/a	self.assertEqual(self.env.get_timeout(db.DB_SET_LOCK_TIMEOUT), 0)
96	n/a	self.env.set_timeout(0, db.DB_SET_TXN_TIMEOUT)
97	n/a	self.assertEqual(self.env.get_timeout(db.DB_SET_TXN_TIMEOUT), 0)
98	n/a	self.env.set_timeout(123456, db.DB_SET_LOCK_TIMEOUT)
99	n/a	self.assertEqual(self.env.get_timeout(db.DB_SET_LOCK_TIMEOUT), 123456)
100	n/a	self.env.set_timeout(7890123, db.DB_SET_TXN_TIMEOUT)
101	n/a	self.assertEqual(self.env.get_timeout(db.DB_SET_TXN_TIMEOUT), 7890123)
102	n/a
103	n/a	def test04_lock_timeout2(self):
104	n/a	self.env.set_timeout(0, db.DB_SET_LOCK_TIMEOUT)
105	n/a	self.env.set_timeout(0, db.DB_SET_TXN_TIMEOUT)
106	n/a	self.env.set_timeout(123456, db.DB_SET_LOCK_TIMEOUT)
107	n/a	self.env.set_timeout(7890123, db.DB_SET_TXN_TIMEOUT)
108	n/a
109	n/a	def deadlock_detection() :
110	n/a	while not deadlock_detection.end :
111	n/a	deadlock_detection.count = \
112	n/a	self.env.lock_detect(db.DB_LOCK_EXPIRE)
113	n/a	if deadlock_detection.count :
114	n/a	while not deadlock_detection.end :
115	n/a	pass
116	n/a	break
117	n/a	time.sleep(0.01)
118	n/a
119	n/a	deadlock_detection.end=False
120	n/a	deadlock_detection.count=0
121	n/a	t=Thread(target=deadlock_detection)
122	n/a	import sys
123	n/a	if sys.version_info[0] < 3 :
124	n/a	t.setDaemon(True)
125	n/a	else :
126	n/a	t.daemon = True
127	n/a	t.start()
128	n/a	self.env.set_timeout(100000, db.DB_SET_LOCK_TIMEOUT)
129	n/a	anID = self.env.lock_id()
130	n/a	anID2 = self.env.lock_id()
131	n/a	self.assertNotEqual(anID, anID2)
132	n/a	lock = self.env.lock_get(anID, "shared lock", db.DB_LOCK_WRITE)
133	n/a	start_time=time.time()
134	n/a	self.assertRaises(db.DBLockNotGrantedError,
135	n/a	self.env.lock_get,anID2, "shared lock", db.DB_LOCK_READ)
136	n/a	end_time=time.time()
137	n/a	deadlock_detection.end=True
138	n/a	# Floating point rounding
139	n/a	self.assertTrue((end_time-start_time) >= 0.0999)
140	n/a	self.env.lock_put(lock)
141	n/a	t.join()
142	n/a
143	n/a	self.env.lock_id_free(anID)
144	n/a	self.env.lock_id_free(anID2)
145	n/a
146	n/a	if db.version() >= (4,6):
147	n/a	self.assertTrue(deadlock_detection.count>0)
148	n/a
149	n/a	def theThread(self, lockType):
150	n/a	import sys
151	n/a	if sys.version_info[0] < 3 :
152	n/a	name = currentThread().getName()
153	n/a	else :
154	n/a	name = currentThread().name
155	n/a
156	n/a	if lockType == db.DB_LOCK_WRITE:
157	n/a	lt = "write"
158	n/a	else:
159	n/a	lt = "read"
160	n/a
161	n/a	anID = self.env.lock_id()
162	n/a	if verbose:
163	n/a	print "%s: locker ID: %s" % (name, anID)
164	n/a
165	n/a	for i in xrange(1000) :
166	n/a	lock = self.env.lock_get(anID, "some locked thing", lockType)
167	n/a	if verbose:
168	n/a	print "%s: Aquired %s lock: %s" % (name, lt, lock)
169	n/a
170	n/a	self.env.lock_put(lock)
171	n/a	if verbose:
172	n/a	print "%s: Released %s lock: %s" % (name, lt, lock)
173	n/a
174	n/a	self.env.lock_id_free(anID)
175	n/a
176	n/a
177	n/a	#----------------------------------------------------------------------
178	n/a
179	n/a	def test_suite():
180	n/a	suite = unittest.TestSuite()
181	n/a
182	n/a	if have_threads:
183	n/a	suite.addTest(unittest.makeSuite(LockingTestCase))
184	n/a	else:
185	n/a	suite.addTest(unittest.makeSuite(LockingTestCase, 'test01'))
186	n/a
187	n/a	return suite
188	n/a
189	n/a
190	n/a	if __name__ == '__main__':
191	n/a	unittest.main(defaultTest='test_suite')