Update ToolBase: Load cascade of config files, add -c parameter.
#!/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()