X-Git-Url: https://git.phdru.name/?p=sqlconvert.git;a=blobdiff_plain;f=tests%2Ftest_process_tokens.py;h=0cc431ccb0614098ce5fd0c5ec72e99c508db2e6;hp=e561667d3c36de8a3d2a6ded7718743f9cee3d9e;hb=HEAD;hpb=29970ce1d9d4874df8e7ce6d85a4a6fa62337934 diff --git a/tests/test_process_tokens.py b/tests/test_process_tokens.py index e561667..d74cb47 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") + assert not parsed or is_newline_statement(parsed[0]) 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] - process_statement(parsed) - query = tlist2str(parsed) - assert query == u'SELECT * FROM /* test */ "T"' - - def test_incomplete(): grouper = StatementGrouper() grouper.process_line("select * from `T`")