Added support for 'greater/less than or equal' on ANSI CASE statement (…

…sqlfluff#5728)
zapion · Apr 1, 2024 · f400d6d · f400d6d
1 parent ae0a45f
commit f400d6d
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 2 deletions.
diff --git a/src/sqlfluff/dialects/dialect_ansi.py b/src/sqlfluff/dialects/dialect_ansi.py
@@ -2000,7 +2000,11 @@ class CaseExpressionSegment(BaseSegment):
             Dedent,
             "END",
         ),
-        terminators=[Ref("CommaSegment"), Ref("BinaryOperatorGrammar")],
+        terminators=[
+            Ref("ComparisonOperatorGrammar"),
+            Ref("CommaSegment"),
+            Ref("BinaryOperatorGrammar"),
+        ],
     )
 
 

diff --git a/test/fixtures/dialects/ansi/select_case_c.sql b/test/fixtures/dialects/ansi/select_case_c.sql
@@ -24,3 +24,11 @@ select
         else col3
     end as text) as mycol
 from table1;
+
+SELECT
+    column1,
+    CASE
+        WHEN 'test' THEN 12
+        ELSE 0
+    END >= 0
+FROM boo;
diff --git a/test/fixtures/dialects/ansi/select_case_c.yml b/test/fixtures/dialects/ansi/select_case_c.yml
@@ -3,7 +3,7 @@
 # computed by SQLFluff when running the tests. Please run
 # `python test/generate_parse_fixture_yml.py`  to generate them after adding or
 # altering SQL files.
-_hash: ea171e2218df3a613023d2a683870954ca0b9e3d75cc3dc5b42592f7fcf71db0
+_hash: 87f657bcb3dc35a56ce876da187214ff201138c05fe2147c3cb79f78904d513e
 file:
 - statement:
     select_statement:
@@ -139,3 +139,39 @@ file:
               table_reference:
                 naked_identifier: table1
 - statement_terminator: ;
+- statement:
+    select_statement:
+      select_clause:
+      - keyword: SELECT
+      - select_clause_element:
+          column_reference:
+            naked_identifier: column1
+      - comma: ','
+      - select_clause_element:
+          expression:
+            case_expression:
+            - keyword: CASE
+            - when_clause:
+              - keyword: WHEN
+              - expression:
+                  quoted_literal: "'test'"
+              - keyword: THEN
+              - expression:
+                  numeric_literal: '12'
+            - else_clause:
+                keyword: ELSE
+                expression:
+                  numeric_literal: '0'
+            - keyword: END
+            comparison_operator:
+            - raw_comparison_operator: '>'
+            - raw_comparison_operator: '='
+            numeric_literal: '0'
+      from_clause:
+        keyword: FROM
+        from_expression:
+          from_expression_element:
+            table_expression:
+              table_reference:
+                naked_identifier: boo
+- statement_terminator: ;