Update keywords_lists.py (#428)

* Update keywords_lists.py

Add HAVING keyword for column sections

* testing for HAVING keyword

* Update test/test_getting_columns.py

---------

Co-authored-by: Andy Tang <atang@spotify.com>
Co-authored-by: Maciej Brencz <maciej.brencz@gmail.com>

sql_metadata/keywords_lists.py

@@ -9,6 +9,7 @@
 KEYWORDS_BEFORE_COLUMNS = {
     "SELECT",
     "WHERE",
+    "HAVING",
     "ORDERBY",
     "GROUPBY",
     "ON",
@@ -55,6 +56,7 @@
 COLUMNS_SECTIONS = {
     "SELECT": "select",
     "WHERE": "where",
+    "HAVING": "having",
     "ORDERBY": "order_by",
     "ON": "join",
     "USING": "join",

test/test_getting_columns.py

@@ -461,3 +461,19 @@ 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"]
+    assert parsed.columns_dict == {
+        "select": ["Country"],
+        "group_by": ["Country"],
+        "having": ["CustomerID"],
+    }