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 def test_incomplete():
   8     grouper = StatementGrouper()
   9     grouper.process_line("select * from `T`")
  10     assert not grouper.statements
  11     assert len(grouper.statements) == 0
  12     raises(ValueError, grouper.close)
  13
  14
  15 def test_statements():
  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