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)
