def test_show_tables(): mocker = Mocker() connect = mocker.replace('MySQLdb.connect') link = connect(ARGS, KWARGS) mocker.result(link) cursor = link.cursor() cursor.execute('SHOW TABLES FROM `mysql`') iter(cursor) mocker.generate(['user', 'db']) cursor.close() mocker.replay() try: client = MySQLClient() tables = client.show_tables('mysql') assert_equals(tables, ['user', 'db']) finally: mocker.restore()