Fix special case for andialbrecht#284 when statement starts with orphaned AS.

andialbrecht · andialbrecht · commit b68e4b781b8d · 2016-10-24T20:06:39.000+02:00
diff --git a/sqlparse/engine/grouping.py b/sqlparse/engine/grouping.py
@@ -380,7 +380,7 @@ def _group(tlist, cls, match,
 
         if match(token):
             nidx, next_ = tlist.token_next(tidx)
-            if valid_prev(prev_) and valid_next(next_):
+            if prev_ and valid_prev(prev_) and valid_next(next_):
                 from_idx, to_idx = post(tlist, pidx, tidx, nidx)
                 grp = tlist.group_tokens(cls, from_idx, to_idx, extend=extend)
 
diff --git a/tests/test_regressions.py b/tests/test_regressions.py
@@ -316,7 +316,10 @@ def test_token_next_doesnt_ignore_skip_cm():
     assert tok.value == 'select'
 
 
-def test_issue284_as_grouping():
-    sql = 'SELECT x AS'
-    p = sqlparse.parse(sql)[0]
-    assert sql == str(p)
+@pytest.mark.parametrize('s', [
+    'SELECT x AS',
+    'AS'
+])
+def test_issue284_as_grouping(s):
+    p = sqlparse.parse(s)[0]
+    assert s == str(p)
