tests/test_tokens.py

   1 #! /usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 import unittest
   5 from sqlparse import parse
   6
   7 from mysql2sql.process_tokens import requote_names
   8 from mysql2sql.print_tokens import tlist2str
   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 = tlist2str(parsed)
  17         self.assertEqual(query, 'SELECT * FROM "T"')
  18
  19     def test_encoding(self):
  20         parsed = parse("insert into test (1, 'тест')", 'utf-8')[0]
  21         query = tlist2str(parsed).encode('utf-8')
  22         self.assertEqual(query, "INSERT INTO test (1, 'тест')")
  23
  24     def test_unicode(self):
  25         parsed = parse(u"insert into test (1, 'тест')")[0]
  26         query = tlist2str(parsed)
  27         self.assertEqual(query, u"INSERT INTO test (1, 'тест')")
  28
  29
  30 if __name__ == "__main__":
  31     main()