def test9_End(self):
for closeable in (False, True):
db = SteadyPgConnection(closeable=closeable)
db.begin()
db.query('select test')
db.end()
self.assertEqual(db.session, ['begin', 'end'])
db.db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:end'), 'sql:end')
self.assertEqual(db.num_queries, 2)
db.begin()
db.query('select test')
db.commit()
self.assertEqual(db.session, ['begin', 'commit'])
db.db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:commit'), 'sql:commit')
self.assertEqual(db.num_queries, 2)
db.begin()
db.query('select test')
db.rollback()
self.assertEqual(db.session, ['begin', 'rollback'])
db.db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:rollback'), 'sql:rollback')
self.assertEqual(db.num_queries, 2)
def test8_End(self):
for closeable in (False, True):
db = SteadyPgConnection(closeable=closeable)
db.begin()
db.query('select test')
db.end()
self.assertEqual(db.session, ['begin', 'end'])
db.db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:end'), 'sql:end')
self.assertEqual(db.num_queries, 2)
db.begin()
db.query('select test')
db.commit()
self.assertEqual(db.session, ['begin', 'commit'])
db.db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:commit'), 'sql:commit')
self.assertEqual(db.num_queries, 2)
db.begin()
db.query('select test')
db.rollback()
self.assertEqual(db.session, ['begin', 'rollback'])
db.db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:rollback'), 'sql:rollback')
self.assertEqual(db.num_queries, 2)
def test8_Begin(self):
for closeable in (False, True):
db = SteadyPgConnection(closeable=closeable)
db.begin()
self.assertEqual(db.session, ['begin'])
db.query('select test')
self.assertEqual(db.num_queries, 1)
db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
db.begin()
self.assertEqual(db.session, ['begin'])
db.db.close()
self.assertRaises(pg.InternalError, db.query, 'select test')
self.assertEqual(db.num_queries, 0)
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:begin'), 'sql:begin')
self.assertEqual(db.num_queries, 2)
def test7_Begin(self):
for closeable in (False, True):
db = SteadyPgConnection(closeable=closeable)
db.begin()
self.assertEqual(db.session, ['begin'])
db.query('select test')
self.assertEqual(db.num_queries, 1)
db.close()
db.query('select test')
self.assertEqual(db.num_queries, 1)
db.begin()
self.assertEqual(db.session, ['begin'])
db.db.close()
self.assertRaises(InternalError, db.query, 'select test')
self.assertEqual(db.num_queries, 0)
db.query('select test')
self.assertEqual(db.num_queries, 1)
self.assertEqual(db.begin('select sql:begin'), 'sql:begin')
self.assertEqual(db.num_queries, 2)
