X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=tests%2Ftest_process_mysql.py;fp=tests%2Ftest_process_mysql.py;h=45e5020de20fdc91f8b80f1fad67cd1e57f02939;hb=d33677c7a57a955f59e46f2abe03e9f5a192d883;hp=500a35db746f146e3eed543cc6c10b633a6e4700;hpb=206d77dac80ce387d4714f589916de4842e99cef;p=sqlconvert.git diff --git a/tests/test_process_mysql.py b/tests/test_process_mysql.py index 500a35d..45e5020 100644 --- a/tests/test_process_mysql.py +++ b/tests/test_process_mysql.py @@ -5,7 +5,7 @@ from sqlparse import parse from sqlconvert.print_tokens import tlist2str from sqlconvert.process_mysql import remove_directive_tokens, \ is_directive_statement, requote_names, unescape_strings, \ - process_statement + is_insert, process_statement from sqlconvert.process_tokens import escape_strings @@ -53,6 +53,15 @@ def test_escape_string_sqlite(): assert query == u"INSERT INTO test VALUES ('\"te''st\"\n')" +def test_is_insert(): + parsed = parse("select /*! test */ * from /* test */ `T`")[0] + statement = next(process_statement(parsed)) + assert not is_insert(statement) + parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0] + statement = next(process_statement(parsed)) + assert is_insert(statement) + + def test_process(): parsed = parse("select /*! test */ * from /* test */ `T`")[0] statement = next(process_statement(parsed))