tests/test_stgrouper.py

   1
   2 from pytest import raises
   3 from sqlconvert.print_tokens import tlist2str
   4 from sqlconvert.process_tokens import StatementGrouper
   5
   6
   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)
  14
  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