1 files changed, 88 insertions, 0 deletions
diff --git a/tests/test_split.py b/tests/test_split.py
new file mode 100644
index 0000000..782b226
--- /dev/null
+++ b/tests/test_split.py
@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+
+# Tests splitting functions.
+
+import unittest
+
+from utils import load_file, TestCaseBase
+
+import sqlparse
+
+
+class SQLSplitTest(TestCaseBase):
+    """Tests sqlparse.sqlsplit()."""
+
+    _sql1 = 'select * from foo;'
+    _sql2 = 'select * from bar;'
+
+    def test_split_semicolon(self):
+        sql2 = 'select * from foo where bar = \'foo;bar\';'
+        stmts = sqlparse.parse(''.join([self._sql1, sql2]))
+        self.assertEqual(len(stmts), 2)
+        self.ndiffAssertEqual(unicode(stmts[0]), self._sql1)
+        self.ndiffAssertEqual(unicode(stmts[1]), sql2)
+
+    def test_create_function(self):
+        sql = load_file('function.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 1)
+        self.ndiffAssertEqual(unicode(stmts[0]), sql)
+
+    def test_create_function_psql(self):
+        sql = load_file('function_psql.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 1)
+        self.ndiffAssertEqual(unicode(stmts[0]), sql)
+
+    def test_create_function_psql3(self):
+        sql = load_file('function_psql3.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 1)
+        self.ndiffAssertEqual(unicode(stmts[0]), sql)
+
+    def test_create_function_psql2(self):
+        sql = load_file('function_psql2.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 1)
+        self.ndiffAssertEqual(unicode(stmts[0]), sql)
+
+    def test_dashcomments(self):
+        sql = load_file('dashcomment.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 3)
+        self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
+
+    def test_begintag(self):
+        sql = load_file('begintag.sql')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 3)
+        self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
+
+    def test_dropif(self):
+        sql = 'DROP TABLE IF EXISTS FOO;\n\nSELECT * FROM BAR;'
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 2)
+        self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
+
+    def test_comment_with_umlaut(self):
+        sql = (u'select * from foo;\n'
+               u'-- Testing an umlaut: ä\n'
+               u'select * from bar;')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 2)
+        self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
+
+    def test_comment_end_of_line(self):
+        sql = ('select * from foo; -- foo\n'
+               'select * from bar;')
+        stmts = sqlparse.parse(sql)
+        self.assertEqual(len(stmts), 2)
+        self.ndiffAssertEqual(''.join(unicode(q) for q in stmts), sql)
+        # make sure the comment belongs to first query
+        self.ndiffAssertEqual(unicode(stmts[0]), 'select * from foo; -- foo\n')
+
+    def test_casewhen(self):
+        sql = ('SELECT case when val = 1 then 2 else null end as foo;\n'
+               'comment on table actor is \'The actor table.\';')
+        stmts = sqlparse.split(sql)
+        self.assertEqual(len(stmts), 2)