X-Git-Url: https://git.phdru.name/?p=sqlconvert.git;a=blobdiff_plain;f=tests%2Ftest_process_tokens.py;fp=tests%2Ftest_process_tokens.py;h=2d30dc9b86561806d0c4b38f25e82171e3fc5724;hp=e561667d3c36de8a3d2a6ded7718743f9cee3d9e;hb=6f433a4531f0e301f3534de7f0278afbcf79d50e;hpb=8a3034769f6006936f2638715f457feeb27fba9d

diff --git a/tests/test_process_tokens.py b/tests/test_process_tokens.py
index e561667..2d30dc9 100644
--- 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"'