tests/test_stgrouper.py

   1 #! /usr/bin/env python
   2
   3
   4 import unittest
   5
   6 from sqlconvert.print_tokens import tlist2str
   7 from sqlconvert.process_mysql import requote_names
   8 from sqlconvert.process_tokens import StatementGrouper
   9 from tests import main
  10
  11
  12 class TestStGrouper(unittest.TestCase):
  13     def test_incomplete(self):
  14         grouper = StatementGrouper()
  15         grouper.process_line("select * from `T`")
  16         self.assertFalse(grouper.statements)
  17         self.assertEqual(len(grouper.statements), 0)
  18         self.assertRaises(ValueError, grouper.close)
  19
  20     def test_statements(self):
  21         grouper = StatementGrouper()
  22         grouper.process_line("select * from `T`;")
  23         self.assertTrue(grouper.statements)
  24         self.assertEqual(len(grouper.statements), 1)
  25         for statement in grouper.get_statements():
  26             requote_names(statement)
  27             query = tlist2str(statement)
  28             self.assertEqual(query, 'SELECT * FROM "T";')
  29         self.assertEqual(len(grouper.statements), 0)
  30         self.assertEqual(grouper.close(), None)
  31
  32 if __name__ == "__main__":
  33     main()