We can now work with file-like objects.

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_split.py b/tests/test_split.py
index 29ee72e..c73d9d4 100644
--- a/tests/test_split.py
+++ b/tests/test_split.py
@@ -116,3 +116,18 @@ class SQLSplitTest(TestCaseBase):
                'SELECT t FROM two')
         stmts = sqlparse.split(sql)
         self.assertEqual(len(stmts), 2)
+
+    def test_split_stream(self):
+        import types
+        from cStringIO import StringIO
+
+        stream = StringIO("SELECT 1; SELECT 2;")
+        stmts = sqlparse.splitstream(stream)
+        self.assertEqual(type(stmts), types.GeneratorType)
+        self.assertEqual(len(list(stmts)), 2)
+
+    def test_encoding_splitstream(self):
+        from cStringIO import StringIO
+        stream = StringIO("SELECT 1; SELECT 2;")
+        stmts = list(sqlparse.splitstream(stream))
+        self.assertEqual(type(stmts[0].tokens[0].value), unicode)