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