2 from pytest import raises
3 from sqlconvert.print_tokens import tlist2str
4 from sqlconvert.process_tokens import StatementGrouper
7 class TestStGrouper(object):
8 def test_incomplete(self):
9 grouper = StatementGrouper()
10 grouper.process_line("select * from `T`")
11 assert not grouper.statements
12 assert len(grouper.statements) == 0
13 raises(ValueError, grouper.close)
15 def test_statements(self):
16 grouper = StatementGrouper()
17 grouper.process_line("select * from T;")
18 assert grouper.statements
19 assert len(grouper.statements) == 1
20 for statement in grouper.get_statements():
21 query = tlist2str(statement)
22 assert query == 'SELECT * FROM T;'
23 assert len(grouper.statements) == 0
24 assert grouper.close() is None