Fix parsing of UNION ALL (fixes andialbrecht#294).

andialbrecht · andialbrecht · commit ee2cbfea194f · 2016-09-26T13:03:42.000+02:00
diff --git a/CHANGELOG b/CHANGELOG
@@ -10,6 +10,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).
 
 Internal Changes
 
diff --git a/sqlparse/keywords.py b/sqlparse/keywords.py
@@ -73,6 +73,7 @@ def is_keyword(value):
          r'|(CROSS\s+|NATURAL\s+)?)?JOIN\b', tokens.Keyword),
         (r'END(\s+IF|\s+LOOP|\s+WHILE)?\b', tokens.Keyword),
         (r'NOT\s+NULL\b', tokens.Keyword),
+        (r'UNION\s+ALL\b', tokens.Keyword),
         (r'CREATE(\s+OR\s+REPLACE)?\b', tokens.Keyword.DDL),
         (r'DOUBLE\s+PRECISION\b', tokens.Name.Builtin),
 
diff --git a/tests/test_tokenize.py b/tests/test_tokenize.py
@@ -157,6 +157,12 @@ def test_parse_join(expr):
     assert p.tokens[0].ttype is T.Keyword
 
 
+def test_parse_union():  # issue294
+    p = sqlparse.parse('UNION ALL')[0]
+    assert len(p.tokens) == 1
+    assert p.tokens[0].ttype is T.Keyword
+
+
 @pytest.mark.parametrize('s', ['END IF', 'END   IF', 'END\t\nIF',
                                'END LOOP', 'END   LOOP', 'END\t\nLOOP'])
 def test_parse_endifloop(s):
