Update mysql.py and add the test for that