@@ -68,10 +68,14 @@ def runQueries(i):
6868 this_db = persist .connection ()
6969 while 1 :
7070 try :
71- q = queryQueue [i ].get (1 , 1 )
71+ try :
72+ q = queryQueue [i ].get (1 , 1 )
73+ except TypeError :
74+ q = queryQueue [i ].get (1 )
7275 except Empty :
7376 q = None
74- if not q : break
77+ if not q :
78+ break
7579 db = persist .connection ()
7680 if db != this_db :
7781 r = 'error - not persistent'
@@ -87,7 +91,10 @@ def runQueries(i):
8791 r = cursor .fetchone ()
8892 cursor .close ()
8993 r = '%d(%d): %s' % (i , db ._usage , r )
90- resultQueue [i ].put (r , 1 , 0.1 )
94+ try :
95+ resultQueue [i ].put (r , 1 , 0.1 )
96+ except TypeError :
97+ resultQueue [i ].put (r , 1 )
9198 db .close ()
9299 from threading import Thread
93100 threads = []
@@ -96,35 +103,68 @@ def runQueries(i):
96103 threads .append (thread )
97104 thread .start ()
98105 for i in range (numThreads ):
99- queryQueue [i ].put ('ping' , 1 , 0.1 )
106+ try :
107+ queryQueue [i ].put ('ping' , 1 , 0.1 )
108+ except TypeError :
109+ queryQueue [i ].put ('ping' , 1 )
100110 for i in range (numThreads ):
101- r = resultQueue [i ].get (1 , 0.1 )
111+ try :
112+ r = resultQueue [i ].get (1 , 0.1 )
113+ except TypeError :
114+ r = resultQueue [i ].get (1 )
102115 self .assertEqual (r , '%d(0): ok - thread alive' % i )
103116 self .assert_ (threads [i ].isAlive ())
104117 for i in range (numThreads ):
105118 for j in range (i + 1 ):
106- queryQueue [i ].put ('select test%d' % j , 1 , 0.1 )
107- r = resultQueue [i ].get (1 , 0.1 )
119+ try :
120+ queryQueue [i ].put ('select test%d' % j , 1 , 0.1 )
121+ r = resultQueue [i ].get (1 , 0.1 )
122+ except TypeError :
123+ queryQueue [i ].put ('select test%d' % j , 1 )
124+ r = resultQueue [i ].get (1 )
108125 self .assertEqual (r , '%d(%d): test%d' % (i , j + 1 , j ))
109- queryQueue [1 ].put ('select test4' , 1 , 0.1 )
110- r = resultQueue [1 ].get (1 , 0.1 )
126+ try :
127+ queryQueue [1 ].put ('select test4' , 1 , 0.1 )
128+ except TypeError :
129+ queryQueue [1 ].put ('select test4' , 1 )
130+ try :
131+ r = resultQueue [1 ].get (1 , 0.1 )
132+ except TypeError :
133+ r = resultQueue [1 ].get (1 )
111134 self .assertEqual (r , '1(3): test4' )
112- queryQueue [1 ].put ('close' , 1 , 0.1 )
113- r = resultQueue [1 ].get (1 , 0.1 )
135+ try :
136+ queryQueue [1 ].put ('close' , 1 , 0.1 )
137+ r = resultQueue [1 ].get (1 , 0.1 )
138+ except TypeError :
139+ queryQueue [1 ].put ('close' , 1 )
140+ r = resultQueue [1 ].get (1 )
114141 self .assertEqual (r , '1(0): ok - connection closed' )
115142 for j in range (2 ):
116- queryQueue [1 ].put ('select test%d' % j , 1 , 0.1 )
117- r = resultQueue [1 ].get (1 , 0.1 )
143+ try :
144+ queryQueue [1 ].put ('select test%d' % j , 1 , 0.1 )
145+ r = resultQueue [1 ].get (1 , 0.1 )
146+ except TypeError :
147+ queryQueue [1 ].put ('select test%d' % j , 1 )
148+ r = resultQueue [1 ].get (1 )
118149 self .assertEqual (r , '1(%d): test%d' % (j + 1 , j ))
119150 for i in range (numThreads ):
120151 self .assert_ (threads [i ].isAlive ())
121- queryQueue [i ].put ('ping' , 1 , 0.1 )
152+ try :
153+ queryQueue [i ].put ('ping' , 1 , 0.1 )
154+ except TypeError :
155+ queryQueue [i ].put ('ping' , 1 )
122156 for i in range (numThreads ):
123- r = resultQueue [i ].get (1 , 0.1 )
157+ try :
158+ r = resultQueue [i ].get (1 , 0.1 )
159+ except TypeError :
160+ r = resultQueue [i ].get (1 )
124161 self .assertEqual (r , '%d(%d): ok - thread alive' % (i , i + 1 ))
125162 self .assert_ (threads [i ].isAlive ())
126163 for i in range (numThreads ):
127- queryQueue [i ].put (None , 1 , 1 )
164+ try :
165+ queryQueue [i ].put (None , 1 , 1 )
166+ except TypeError :
167+ queryQueue [i ].put (None , 1 )
128168
129169 def test4_PersistentDBMaxUsage (self ):
130170 persist = PersistentDB (dbapi , 20 )
0 commit comments