Make process_statement() a generator

[sqlconvert.git] / tests / test_process_tokens.py

diff --git a/tests/test_process_tokens.py b/tests/test_process_tokens.py
index e561667d3c36de8a3d2a6ded7718743f9cee3d9e..2d30dc9b86561806d0c4b38f25e82171e3fc5724 100644 (file)
--- a/tests/test_process_tokens.py
+++ b/tests/test_process_tokens.py
@@ -69,8 +69,8 @@ def test_escape_string_sqlite():
 
 def test_process():
     parsed = parse("select /*! test */ * from /* test */ `T`")[0]
-    process_statement(parsed)
-    query = tlist2str(parsed)
+    statement = next(process_statement(parsed))
+    query = tlist2str(statement)
     assert query == u'SELECT * FROM /* test */ "T"'