def test4_SetSession(self):
persist = PersistentPg(3, ('set datestyle', ))
db = persist.connection()
self.assertEqual(db._maxusage, 3)
self.assertEqual(db._setsession_sql, ('set datestyle', ))
self.assertEqual(db.db.session, ['datestyle'])
db.query('set test')
for i in range(3):
self.assertEqual(db.db.session, ['datestyle', 'test'])
db.query('select test')
self.assertEqual(db.db.session, ['datestyle'])
def test3_MaxUsage(self):
persist = PersistentPg(20)
db = persist.connection()
self.assertEqual(db._maxusage, 20)
for i in range(100):
r = db.query('select test%d' % i)
self.assertEqual(r, 'test%d' % i)
self.assertTrue(db.db.status)
j = i % 20 + 1
self.assertEqual(db._usage, j)
self.assertEqual(db.num_queries, j)
def test4_SetSession(self):
persist = PersistentPg(3, ('set datestyle',))
db = persist.connection()
self.assertEqual(db._maxusage, 3)
self.assertEqual(db._setsession_sql, ('set datestyle',))
self.assertEqual(db.db.session, ['datestyle'])
db.query('set test')
for i in range(3):
self.assertEqual(db.db.session, ['datestyle', 'test'])
db.query('select test')
self.assertEqual(db.db.session, ['datestyle'])
def test3_MaxUsage(self):
persist = PersistentPg(20)
db = persist.connection()
self.assertEqual(db._maxusage, 20)
for i in range(100):
r = db.query('select test%d' % i)
self.assertEqual(r, 'test%d' % i)
self.assertTrue(db.db.status)
j = i % 20 + 1
self.assertEqual(db._usage, j)
self.assertEqual(db.num_queries, j)
def test5_FailedTransaction(self):
persist = PersistentPg()
db = persist.connection()
db._con.close()
self.assertEqual(db.query('select test'), 'test')
db.begin()
db._con.close()
self.assertRaises(pg.InternalError, db.query, 'select test')
self.assertEqual(db.query('select test'), 'test')
db.begin()
self.assertEqual(db.query('select test'), 'test')
db.rollback()
db._con.close()
self.assertEqual(db.query('select test'), 'test')
def test5_FailedTransaction(self):
persist = PersistentPg()
db = persist.connection()
db._con.close()
self.assertEqual(db.query('select test'), 'test')
db.begin()
db._con.close()
self.assertRaises(pg.InternalError, db.query, 'select test')
self.assertEqual(db.query('select test'), 'test')
db.begin()
self.assertEqual(db.query('select test'), 'test')
db.rollback()
db._con.close()
self.assertEqual(db.query('select test'), 'test')
def test1_Close(self):
for closeable in (False, True):
persist = PersistentPg(closeable=closeable)
db = persist.connection()
self.assertTrue(db._con.db and db._con.valid)
db.close()
self.assertTrue(closeable
^ (db._con.db is not None and db._con.valid))
db.close()
self.assertTrue(closeable
^ (db._con.db is not None and db._con.valid))
db._close()
self.assertTrue(not db._con.db or not db._con.valid)
db._close()
self.assertTrue(not db._con.db or not db._con.valid)
def test1_Close(self):
for closeable in (False, True):
persist = PersistentPg(closeable=closeable)
db = persist.connection()
self.assertTrue(db._con.db and db._con.valid)
db.close()
self.assertTrue(closeable ^
(db._con.db is not None and db._con.valid))
db.close()
self.assertTrue(closeable ^
(db._con.db is not None and db._con.valid))
db._close()
self.assertTrue(not db._con.db or not db._con.valid)
db._close()
self.assertTrue(not db._con.db or not db._con.valid)
def test2_Threads(self):
numThreads = 3
persist = PersistentPg()
try:
from queue import Queue, Empty
except ImportError: # Python 2
from Queue import Queue, Empty
queryQueue, resultQueue = [], []
for i in range(numThreads):
queryQueue.append(Queue(1))
resultQueue.append(Queue(1))
def runQueries(i):
this_db = persist.connection().db
while 1:
try:
try:
q = queryQueue[i].get(1, 1)
except TypeError:
q = queryQueue[i].get(1)
except Empty:
q = None
if not q:
break
db = persist.connection()
if db.db != this_db:
r = 'error - not persistent'
else:
if q == 'ping':
r = 'ok - thread alive'
elif q == 'close':
db.db.close()
r = 'ok - connection closed'
else:
r = db.query(q)
r = '%d(%d): %s' % (i, db._usage, r)
try:
resultQueue[i].put(r, 1, 1)
except TypeError:
resultQueue[i].put(r, 1)
db.close()
from threading import Thread
threads = []
for i in range(numThreads):
thread = Thread(target=runQueries, args=(i, ))
threads.append(thread)
thread.start()
for i in range(numThreads):
try:
queryQueue[i].put('ping', 1, 1)
except TypeError:
queryQueue[i].put('ping', 1)
for i in range(numThreads):
try:
r = resultQueue[i].get(1, 1)
except TypeError:
r = resultQueue[i].get(1)
self.assertEqual(r, '%d(0): ok - thread alive' % i)
self.assertTrue(threads[i].isAlive())
for i in range(numThreads):
for j in range(i + 1):
try:
queryQueue[i].put('select test%d' % j, 1, 1)
r = resultQueue[i].get(1, 1)
except TypeError:
queryQueue[i].put('select test%d' % j, 1)
r = resultQueue[i].get(1)
self.assertEqual(r, '%d(%d): test%d' % (i, j + 1, j))
try:
queryQueue[1].put('select test4', 1, 1)
r = resultQueue[1].get(1, 1)
except TypeError:
queryQueue[1].put('select test4', 1)
r = resultQueue[1].get(1)
self.assertEqual(r, '1(3): test4')
try:
queryQueue[1].put('close', 1, 1)
r = resultQueue[1].get(1, 1)
except TypeError:
queryQueue[1].put('close', 1)
r = resultQueue[1].get(1)
self.assertEqual(r, '1(3): ok - connection closed')
for j in range(2):
try:
queryQueue[1].put('select test%d' % j, 1, 1)
r = resultQueue[1].get(1, 1)
except TypeError:
queryQueue[1].put('select test%d' % j, 1)
r = resultQueue[1].get(1)
self.assertEqual(r, '1(%d): test%d' % (j + 1, j))
for i in range(numThreads):
self.assertTrue(threads[i].isAlive())
try:
queryQueue[i].put('ping', 1, 1)
except TypeError:
queryQueue[i].put('ping', 1)
for i in range(numThreads):
try:
r = resultQueue[i].get(1, 1)
except TypeError:
r = resultQueue[i].get(1)
self.assertEqual(r, '%d(%d): ok - thread alive' % (i, i + 1))
self.assertTrue(threads[i].isAlive())
for i in range(numThreads):
try:
queryQueue[i].put(None, 1, 1)
except TypeError:
queryQueue[i].put(None, 1)