From cc01ab2971981bdcbc3cdbb10b36b202b4655a12 Mon Sep 17 00:00:00 2001
From: Andi Albrecht <albrecht.andi@gmail.com>
Date: Wed, 23 Jun 2010 07:37:16 +0200
Subject: Improve statement detection for DECLARE statements outside
 function/procedure definitions.

The parser didn't recognized DECLARE statements outside
function/procedure definitions correctly and assumed that a
non-separating semicolon follows.
---
 tests/test_split.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tests/test_split.py')

diff --git a/tests/test_split.py b/tests/test_split.py
index 5146bcb..50b3a6b 100644
--- a/tests/test_split.py
+++ b/tests/test_split.py
@@ -1,3 +1,4 @@
+
 # -*- coding: utf-8 -*-
 
 # Tests splitting functions.
@@ -86,3 +87,9 @@ class SQLSplitTest(TestCaseBase):
                'comment on table actor is \'The actor table.\';')
         stmts = sqlparse.split(sql)
         self.assertEqual(len(stmts), 2)
+
+    def test_cursor_declare(self):
+        sql = ('DECLARE CURSOR "foo" AS SELECT 1;\n'
+               'SELECT 2;')
+        stmts = sqlparse.split(sql)
+        self.assertEqual(len(stmts), 2)
-- 
cgit v1.2.1