Fix splitting of statements using CASE ... WHEN (fixes #580).

2 files changed, 14 insertions, 0 deletions

diff --git a/tests/files/casewhen_procedure.sql b/tests/files/casewhen_procedure.sql
new file mode 100644
index 0000000..e590d49
--- /dev/null
+++ b/tests/files/casewhen_procedure.sql
@@ -0,0 +1,8 @@
+create procedure procName()
+begin
+  select case when column = 'value' then column else 0 end;
+end;
+create procedure procName()
+begin
+  select 1;
+end;
diff --git a/tests/test_split.py b/tests/test_split.py
index f69e3d2..a9d7576 100644
--- a/tests/test_split.py
+++ b/tests/test_split.py
@@ -97,6 +97,12 @@ def test_split_casewhen():
     assert len(stmts) == 2
 
 
+def test_split_casewhen_procedure(load_file):
+    # see issue580
+    stmts = sqlparse.split(load_file('casewhen_procedure.sql'))
+    assert len(stmts) == 2
+
+
 def test_split_cursor_declare():
     sql = ('DECLARE CURSOR "foo" AS SELECT 1;\n'
            'SELECT 2;')