testing for HAVING keyword

macbre · Sep 22, 2023 · beff146 · beff146
1 parent a4d35d3
commit beff146
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/test/test_getting_columns.py b/test/test_getting_columns.py
@@ -461,3 +461,20 @@ def test_aliases_switching_column_names():
     parsed = Parser(query)
     assert parsed.columns == ["a", "b"]
     assert parsed.columns_dict == {"select": ["a", "b"]}
+
+
+def test_having_columns():
+    query = """
+    SELECT Country
+    FROM Customers
+    GROUP BY Country
+    HAVING COUNT(CustomerID) > 5;
+    """
+    parsed = Parser(query)
+    assert parsed.columns == ["Country", "CustomerID"]
+    print(parsed.columns_dict)
+    assert parsed.columns_dict == {
+        "select": ["Country"],
+        "group_by": ["Country"],
+        "having": ["CustomerID"],
+    }