X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=tests%2Ftest_stgrouper.py;h=994ef4ef8eea3a600861d51d13f4e8da4b142b5f;hb=19e5f010328b314a5ead4468ba11b1fcb27fe42f;hp=88d0fcf2eefce1b6cc74821e9c92f0181028e657;hpb=d6d2d76d2b29e8384169c5b86e19e494fab1a464;p=sqlconvert.git diff --git a/tests/test_stgrouper.py b/tests/test_stgrouper.py index 88d0fcf..994ef4e 100644 --- a/tests/test_stgrouper.py +++ b/tests/test_stgrouper.py @@ -1,26 +1,24 @@ from pytest import raises from sqlconvert.print_tokens import tlist2str -from sqlconvert.process_mysql import requote_names from sqlconvert.process_tokens import StatementGrouper -class TestStGrouper(object): - def test_incomplete(self): - grouper = StatementGrouper() - grouper.process_line("select * from `T`") - assert not grouper.statements - assert len(grouper.statements) == 0 - raises(ValueError, grouper.close) +def test_incomplete(): + grouper = StatementGrouper() + grouper.process_line("select * from `T`") + assert not grouper.statements + assert len(grouper.statements) == 0 + raises(ValueError, grouper.close) - def test_statements(self): - grouper = StatementGrouper() - grouper.process_line("select * from `T`;") - assert grouper.statements - assert len(grouper.statements) == 1 - for statement in grouper.get_statements(): - requote_names(statement) - query = tlist2str(statement) - assert query == 'SELECT * FROM "T";' - assert len(grouper.statements) == 0 - assert grouper.close() is None + +def test_statements(): + grouper = StatementGrouper() + grouper.process_line("select * from T;") + assert grouper.statements + assert len(grouper.statements) == 1 + for statement in grouper.get_statements(): + query = tlist2str(statement) + assert query == 'SELECT * FROM T;' + assert len(grouper.statements) == 0 + assert grouper.close() is None