def test_set_databases_correct(self):
mysql = Mysql(config)
mysql.set_databases(config['databases']['mysql'])
self.assertEqual(mysql.databases, config['databases']['mysql'].split(','))
databases = 'db1, db2, db3,db4'
mysql.set_databases(databases)
self.assertEqual(mysql.databases, [x.strip() for x in databases.split(',')])
def test_set_databases_error(self):
mysql = Mysql(config)
with self.assertRaises(ValueError):
mysql.set_databases('')