diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/files/casewhen_procedure.sql | 8 | ||||
| -rw-r--r-- | tests/test_split.py | 6 |
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;') |