tests/testpgmanager.py
author Radek Brich <radek.brich@devl.cz>
Tue, 06 Dec 2011 17:11:21 +0100
changeset 22 680631f35d83
child 37 5b0eb4b11940
permissions -rwxr-xr-x
Add test for PgManager.

#!/usr/bin/env python

import unittest

from config import Config
from pgtoolkit import pgmanager


class TestSuite(unittest.TestCase):

    def setUp(self):
        cfg = Config('tests.conf')
        self.m = pgmanager.get_instance()
        self.m.create_conn(**cfg)
        self.m.create_conn('autocommit', isolation_level='autocommit', **cfg)

    def tearDown(self):
        self.m.destroy_conn()
        self.m.destroy_conn('autocommit')

    def test_connection_get_put(self):
        conn = self.m.get_conn()

        curs = conn.cursor()
        curs.execute('SELECT %s AS t', [True])
        d = curs.fetchone_dict()
        self.assertTrue(d.t)

        self.m.put_conn(conn)

    def test_connection_with(self):
        with self.m.cursor() as curs:
            curs.execute('SELECT %s AS t', [True])
            d = curs.fetchone_dict()
            self.assertTrue(d.t)

    def test_connection_autocommit(self):
        conn = self.m.get_conn('autocommit')
        curs = conn.cursor()
        name = 'test'
        curs.execute('DELETE FROM test WHERE name = %s', [name])
        curs.execute('INSERT INTO test (name) VALUES (%s)', [name])
        conn.close()

        conn = self.m.get_conn('autocommit')
        curs = conn.cursor()
        curs.execute('SELECT * FROM test WHERE name = %s', [name])
        self.assertTrue(curs.rowcount == 1)
        self.m.put_conn(conn)

if __name__ == '__main__':
    unittest.main()