test: make hive serde DDL test working with sqlfluff

Author: reata

sqllineage/core/parser/sqlparse/handlers/cte.py

@@ -28,6 +28,6 @@ def _handle(self, token: Token, holder: SubQueryLineageHolder) -> None:
             cte = []
         for token in cte:
             sublist = list(token.get_sublists())
-            if sublist:
+            if sublist and not (isinstance(sublist[0], Function)):
                 # CTE: tbl AS (SELECT 1), tbl is alias and (SELECT 1) is subquery Parenthesis
                 holder.add_cte(SqlParseSubQuery.of(sublist[0], token.get_real_name()))

tests/test_others_dialect_specific.py

@@ -40,20 +40,19 @@ def test_create_table_as_with_postgres_dialect(dialect: str):
     )
 
 
-def test_create_using_serde():
+@pytest.mark.parametrize("dialect", ["databricks", "hive", "sparksql"])
+def test_create_using_serde(dialect: str):
     """
     https://cwiki.apache.org/confluence/display/Hive/LanguageManual+DDL#LanguageManualDDL-RowFormats&SerDe
     here with is not an indicator for CTE
-    FIXME: sqlfluff hive dialect doesn't support parsing this yet
     """
-    # Check
-    #
+    # use raw sql string to avoid Python recognize some character as escaping
     assert_table_lineage_equal(
-        """CREATE TABLE apachelog (
+        r"""CREATE TABLE apachelog (
   host STRING,
   identity STRING,
-  user STRING,
-  time STRING,
+  `user` STRING,
+  `time` STRING,
   request STRING,
   status STRING,
   size STRING,
@@ -66,7 +65,7 @@ def test_create_using_serde():
 STORED AS TEXTFILE""",  # noqa
         None,
         {"apachelog"},
-        test_sqlfluff=False,
+        dialect=dialect,
     )