summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--sqlparse/sql.py2
-rw-r--r--tests/test_grouping.py8
2 files changed, 9 insertions, 1 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index 4b4627f..53c16be 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -529,7 +529,7 @@ class Where(TokenList):
"""A WHERE clause."""
M_OPEN = T.Keyword, 'WHERE'
M_CLOSE = T.Keyword, ('ORDER', 'GROUP', 'LIMIT', 'UNION', 'EXCEPT',
- 'HAVING')
+ 'HAVING', 'RETURNING')
class Case(TokenList):
diff --git a/tests/test_grouping.py b/tests/test_grouping.py
index 12d7310..be03110 100644
--- a/tests/test_grouping.py
+++ b/tests/test_grouping.py
@@ -188,6 +188,14 @@ def test_grouping_where():
assert isinstance(p.tokens[-1].tokens[0].tokens[-2], sql.Where)
+def test_returning_kw_ends_where_clause():
+ s = 'delete from foo where x > y returning z'
+ p = sqlparse.parse(s)[0]
+ assert isinstance(p.tokens[6], sql.Where)
+ assert p.tokens[7].ttype == T.Keyword
+ assert p.tokens[7].value == 'returning'
+
+
def test_grouping_typecast():
s = 'select foo::integer from bar'
p = sqlparse.parse(s)[0]