tests/test_mymanager.py
author Radek Brich <brich.radek@ifortuna.cz>
Fri, 16 Aug 2013 13:19:26 +0200
changeset 86 b61b54aa9f96
parent 75 39f777341db4
child 104 d8ff52a0390f
permissions -rw-r--r--
Fix: ConfigParser add_argument -> add_option.

#!/usr/bin/env python3

from tests.config import Config
from mytoolkit import mymanager

import unittest
import logging


class TestMyManager(unittest.TestCase):

    def setUp(self):
        cfg = Config('pgtoolkit.conf')
        test_db_conn_params = cfg['databases']['test_mysql']
        params = self.params_to_mapping(test_db_conn_params)
        self.m = mymanager.get_instance()
        self.m.create_conn(**params)
        #self.setup_logging()

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

    def setup_logging(self):
        #FIXME: write test handler, check SQL log
        log = logging.getLogger('mymanager_sql')
        handler = logging.StreamHandler()
        log.addHandler(handler)
        log.setLevel(logging.DEBUG)

    def params_to_mapping(self, params):
        return dict([param.split('=') for param in params.split(' ')])

    def test_mysql_query(self):
        with self.m.cursor() as curs:
            ajaj = 1
            curs.execute('SELECT %(ajaj)s AS ajaj', locals())
            row = curs.fetchone_dict()
            self.assertEqual(row.ajaj, ajaj)

    def test_mysql_error(self):
        with self.m.cursor() as curs:
            self.assertRaises(mymanager.OperationalError,
                curs.execute, 'SELECT give_me_some_error;')


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