Merge pull request #13 from hex-inc/dscott/nova-3074-dependency-not-recognized-when-with-time-zone-is-present

fix null dereference from `timestamp with time zone`

Author: dylanscott

poetry.lock

@@ -469,7 +469,8 @@ def with_names(self) -> List[str]:
                         while token.next_token and not token.is_with_query_end:
                             token = token.next_token
                         is_end_of_with_block = (
-                            token.next_token_not_comment.normalized
+                            token.next_token_not_comment is not None
+                            and token.next_token_not_comment.normalized
                             in WITH_ENDING_KEYWORDS
                         )
                         if is_end_of_with_block:

sql_metadata/parser.py

@@ -158,3 +158,15 @@ def test_mega_join():
     for dataframe in dataframes:
         assert dataframe in parser.tables
     assert parser.query_type == "SELECT"
+
+
+def test_timestamp_with_time_zone():
+    parser = Parser(
+        """
+SELECT *,
+CAST(ORDER_DATE as TIMESTAMP WITH TIME ZONE) as ingested_at
+FROM dataframe
+        """
+    )
+    assert "dataframe" in parser.tables
+    assert parser.query_type == "SELECT"