From 621e1d84607a158491ce80c47814e51a8c366498 Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Sun, 26 Jul 2015 10:38:54 +0200 Subject: Don't treat DECLARE within BEGIN blocks as relevant to split level (fixes #193). --- tests/test_regressions.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) (limited to 'tests/test_regressions.py') diff --git a/tests/test_regressions.py b/tests/test_regressions.py index e30ddf0..6da94dc 100644 --- a/tests/test_regressions.py +++ b/tests/test_regressions.py @@ -243,4 +243,15 @@ def test_null_with_as(): ' NULL AS c2', 'FROM t1' ]) - assert formatted == tformatted \ No newline at end of file + assert formatted == tformatted + + +def test_issue193_splitting_function(): + sql = """CREATE FUNCTION a(x VARCHAR(20)) RETURNS VARCHAR(20) +BEGIN + DECLARE y VARCHAR(20); + RETURN x; +END; +SELECT * FROM a.b;""" + splitted = sqlparse.split(sql) + assert len(splitted) == 2 -- cgit v1.2.1 From fb6b59dece40a608a1c6cf741a4ac6ee88d85d2f Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Sun, 26 Jul 2015 11:20:49 +0200 Subject: Ignore comments at beginning of statement when calling Statement.get_type (fixes #186). --- tests/test_regressions.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'tests/test_regressions.py') diff --git a/tests/test_regressions.py b/tests/test_regressions.py index 6da94dc..ea8cd56 100644 --- a/tests/test_regressions.py +++ b/tests/test_regressions.py @@ -255,3 +255,9 @@ END; SELECT * FROM a.b;""" splitted = sqlparse.split(sql) assert len(splitted) == 2 + + +def test_issue186_get_type(): + sql = "-- comment\ninsert into foo" + p = sqlparse.parse(sql)[0] + assert p.get_type() == 'INSERT' -- cgit v1.2.1