tests/test_tokens.py

   1 #! /usr/bin/env python
   2
   3
   4 import unittest
   5 from sqlparse import parse
   6
   7 from mysql2sql.process_tokens import requote_names
   8 from mysql2sql.print_tokens import get_tokens_str
   9 from tests import main
  10
  11
  12 class TestTokens(unittest.TestCase):
  13     def test_requote(self):
  14         parsed = parse("select * from `T`")[0]
  15         requote_names(parsed)
  16         query = get_tokens_str(parsed)
  17         self.assertEqual(query, 'SELECT * FROM "T";\n')
  18
  19
  20 if __name__ == "__main__":
  21     main()