X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=tests%2Ftest_process_tokens.py;h=0cc431ccb0614098ce5fd0c5ec72e99c508db2e6;hb=e066c93875b3b9edcf6526728d81db7fe594e618;hp=2d30dc9b86561806d0c4b38f25e82171e3fc5724;hpb=6f433a4531f0e301f3534de7f0278afbcf79d50e;p=sqlconvert.git diff --git a/tests/test_process_tokens.py b/tests/test_process_tokens.py index 2d30dc9..0cc431c 100644 --- a/tests/test_process_tokens.py +++ b/tests/test_process_tokens.py @@ -4,10 +4,12 @@ from pytest import raises 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 -from sqlconvert.process_tokens import escape_strings, StatementGrouper +from sqlconvert.process_tokens import is_newline_statement, StatementGrouper + + +def test_newline_statement(): + parsed = parse("\n")[0] + assert is_newline_statement(parsed) def test_encoding(): @@ -23,57 +25,6 @@ def test_unicode(): assert query, u"INSERT INTO test (1 == 'тест')" -def test_directive(): - parsed = parse("select /*! test */ * from /* test */ `T`")[0] - remove_directive_tokens(parsed) - query = tlist2str(parsed) - assert query == u'SELECT * FROM /* test */ `T`' - - -def test_directive_statement(): - parsed = parse("/*! test */ test ;")[0] - assert not is_directive_statement(parsed) - parsed = parse("/*! test */ ;")[0] - assert is_directive_statement(parsed) - - -def test_requote(): - parsed = parse("select * from `T`")[0] - requote_names(parsed) - query = tlist2str(parsed) - assert query == u'SELECT * FROM "T"' - - -def test_unescape_string(): - parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0] - unescape_strings(parsed) - query = tlist2str(parsed) - assert query == u"INSERT INTO test VALUES ('\"te'st\"\n')" - - -def test_escape_string_postgres(): - parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0] - unescape_strings(parsed) - escape_strings(parsed, 'postgres') - query = tlist2str(parsed) - assert query == u"INSERT INTO test VALUES (E'\"te''st\"\\n')" - - -def test_escape_string_sqlite(): - parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0] - unescape_strings(parsed) - escape_strings(parsed, 'sqlite') - query = tlist2str(parsed) - assert query == u"INSERT INTO test VALUES ('\"te''st\"\n')" - - -def test_process(): - parsed = parse("select /*! test */ * from /* test */ `T`")[0] - statement = next(process_statement(parsed)) - query = tlist2str(statement) - assert query == u'SELECT * FROM /* test */ "T"' - - def test_incomplete(): grouper = StatementGrouper() grouper.process_line("select * from `T`")