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_tokens import requote_names, StatementGrouper
   8 from tests import main
   9
  10
  11 class TestStGrouper(unittest.TestCase):
  12     def test_incomplete(self):
  13         grouper = StatementGrouper()
  14         grouper.process_line("select * from `T`")
  15         self.assertFalse(grouper.statements)
  16         self.assertEqual(len(grouper.statements), 0)
  17         self.assertRaises(ValueError, grouper.close)
  18
  19     def test_statements(self):
  20         grouper = StatementGrouper()
  21         grouper.process_line("select * from `T`;")
  22         self.assertTrue(grouper.statements)
  23         self.assertEqual(len(grouper.statements), 1)
  24         for statement in grouper.get_statements():
  25             requote_names(statement)
  26             query = tlist2str(statement)
  27             self.assertEqual(query, 'SELECT * FROM "T";')
  28         self.assertEqual(len(grouper.statements), 0)
  29         self.assertEqual(grouper.close(), None)
  30
  31 if __name__ == "__main__":
  32     main()