Handle operator grouping after identifying typecasts (fixes andialbrecht#297).

andialbrecht · andialbrecht · commit f0754a6b6540 · 2016-10-01T14:07:52.000+02:00
diff --git a/CHANGELOG b/CHANGELOG
@@ -11,6 +11,7 @@ Bug Fixes
 * Fix parsing of incomplete AS (issue284, by vmuriart).
 * Fix parsing of Oracle names containing dollars (issue291).
 * Fix parsing of UNION ALL (issue294).
+* Fix grouping of identifiers containing typecasts (issue297).
 
 Internal Changes
 
diff --git a/sqlparse/engine/grouping.py b/sqlparse/engine/grouping.py
@@ -343,9 +343,9 @@ def group(stmt):
         group_period,
         group_arrays,
         group_identifier,
-        group_operator,
         group_order,
         group_typecasts,
+        group_operator,
         group_as,
         group_aliased,
         group_assignment,
diff --git a/tests/test_grouping.py b/tests/test_grouping.py
@@ -64,6 +64,20 @@ def test_grouping_identifiers():
     assert identifiers[0].get_alias() == "col"
 
 
+@pytest.mark.parametrize('s', [
+    'foo, bar',
+    'sum(a), sum(b)',
+    'sum(a) as x, b as y',
+    'sum(a)::integer, b',
+    'sum(a)/count(b) as x, y',
+    'sum(a)::integer as x, y',
+    'sum(a)::integer/count(b) as x, y',  # issue297
+])
+def test_group_identifier_list(s):
+    parsed = sqlparse.parse(s)[0]
+    assert isinstance(parsed.tokens[0], sql.IdentifierList)
+
+
 def test_grouping_identifier_wildcard():
     p = sqlparse.parse('a.*, b.id')[0]
     assert isinstance(p.tokens[0], sql.IdentifierList)
