6 from sqlconvert.print_tokens import tlist2str
7 from sqlconvert.process_tokens import requote_names, StatementGrouper
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)
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)
31 if __name__ == "__main__":