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