Add a test for print_tokens()

[sqlconvert.git] / tests / test_print_tokens.py

diff --git a/tests/test_print_tokens.py b/tests/test_print_tokens.py
new file mode 100644 (file)
index 0000000..af9c9ae
--- /dev/null
+++ b/tests/test_print_tokens.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+
+try:
+    from cStringIO import StringIO
+except ImportError:
+    try:
+        from StringIO import StringIO
+    except ImportError:
+        from io import StringIO
+
+from sqlparse import parse
+from sqlconvert.print_tokens import print_tokens
+
+
+def test_print_tokens():
+    sio = StringIO()
+    parsed = parse("select * from T")[0]
+    print_tokens(parsed, outfile=sio)
+    assert sio.getvalue() == u"SELECT * FROM T"