tests/test_process_mysql.py

   1 # -*- coding: utf-8 -*-
   2
   3 from sqlparse import parse
   4
   5 from sqlconvert.print_tokens import tlist2str
   6 from sqlconvert.process_mysql import remove_directive_tokens, \
   7         is_directive_statement, requote_names, unescape_strings, \
   8         process_statement
   9 from sqlconvert.process_tokens import escape_strings
  10
  11
  12 def test_directive():
  13     parsed = parse("select /*! test */ * from /* test */ `T`")[0]
  14     remove_directive_tokens(parsed)
  15     query = tlist2str(parsed)
  16     assert query == u'SELECT * FROM /* test */ `T`'
  17
  18
  19 def test_directive_statement():
  20     parsed = parse("/*! test */ test ;")[0]
  21     assert not is_directive_statement(parsed)
  22     parsed = parse("/*! test */ ;")[0]
  23     assert is_directive_statement(parsed)
  24
  25
  26 def test_requote():
  27     parsed = parse("select * from `T`")[0]
  28     requote_names(parsed)
  29     query = tlist2str(parsed)
  30     assert query == u'SELECT * FROM "T"'
  31
  32
  33 def test_unescape_string():
  34     parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0]
  35     unescape_strings(parsed)
  36     query = tlist2str(parsed)
  37     assert query == u"INSERT INTO test VALUES ('\"te'st\"\n')"
  38
  39
  40 def test_escape_string_postgres():
  41     parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0]
  42     unescape_strings(parsed)
  43     escape_strings(parsed, 'postgres')
  44     query = tlist2str(parsed)
  45     assert query == u"INSERT INTO test VALUES (E'\"te''st\"\\n')"
  46
  47
  48 def test_escape_string_sqlite():
  49     parsed = parse("insert into test values ('\"te\\'st\\\"\\n')")[0]
  50     unescape_strings(parsed)
  51     escape_strings(parsed, 'sqlite')
  52     query = tlist2str(parsed)
  53     assert query == u"INSERT INTO test VALUES ('\"te''st\"\n')"
  54
  55
  56 def test_process():
  57     parsed = parse("select /*! test */ * from /* test */ `T`")[0]
  58     statement = next(process_statement(parsed))
  59     query = tlist2str(statement)
  60     assert query == u'SELECT * FROM /* test */ "T"'