def get_test_connection():
db_config = get_db_config()
con = pymysql.connect(**db_config)
try:
with con.cursor() as cur:
try:
cur.execute('DROP DATABASE {}'.format(DB_NAME))
except:
pass
cur.execute('CREATE DATABASE {}'.format(DB_NAME))
finally:
con.close()
db_config['database'] = DB_NAME
db_config['autocommit'] = True
mysql_conn = MySQLConnection(db_config)
mysql_conn.autocommit_mode = True
return mysql_conn
def get_test_connection(extra_config=None):
db_config = get_db_config()
con = pymysql.connect(**db_config)
try:
with con.cursor() as cur:
try:
cur.execute("DROP DATABASE {}".format(DB_NAME))
except:
pass
cur.execute("CREATE DATABASE {}".format(DB_NAME))
finally:
con.close()
db_config["database"] = DB_NAME
db_config["autocommit"] = True
if not extra_config:
extra_config = {}
mysql_conn = MySQLConnection({**db_config, **extra_config})
mysql_conn.autocommit_mode = True
return mysql_conn
