SGA-13064: Query parsing failed due to casting to text(n)

src/dialect/databricks.rs

@@ -90,4 +90,9 @@ impl Dialect for DatabricksDialect {
     fn supports_optimize_table(&self) -> bool {
         true
     }
+
+    /// See <https://docs.databricks.com/aws/en/sql/language-manual/functions/bangsign>
+    fn supports_bang_not_operator(&self) -> bool {
+        true
+    }
 }

src/parser/mod.rs

@@ -11909,6 +11909,7 @@ impl<'a> Parser<'a> {
         self.advance_token();
         let next_token = self.get_current_token();
         let next_token_index = self.get_current_index();
+        let next_token_span = next_token.span;
 
         let mut trailing_bracket: MatchedTrailingBracket = false.into();
         let mut data = match &next_token.token {
@@ -12170,7 +12171,14 @@ impl<'a> Parser<'a> {
                     self.expect_token(&Token::RParen)?;
                     Ok(DataType::FixedString(character_length))
                 }
-                Keyword::TEXT => Ok(DataType::Text),
+                Keyword::TEXT => {
+                    let type_name = w.to_ident(next_token_span);
+                    if let Some(modifiers) = self.parse_optional_type_modifiers()? {
+                        Ok(DataType::Custom(type_name.into(), modifiers))
+                    } else {
+                        Ok(DataType::Text)
+                    }
+                }
                 Keyword::TINYTEXT => Ok(DataType::TinyText),
                 Keyword::MEDIUMTEXT => Ok(DataType::MediumText),
                 Keyword::LONGTEXT => Ok(DataType::LongText),

src/test_utils.rs

@@ -199,10 +199,15 @@ impl TestedDialects {
     /// Ensures that `sql` parses as a single [Statement], and that
     /// re-serializing the parse result produces the same `sql`
     /// string (is not modified after a serialization round-trip).
-    pub fn verified_stmt(&self, sql: &str) -> Statement {
+    pub fn verified_statement(&self, sql: &str) -> Statement {
         self.one_statement_parses_to(sql, sql)
     }
 
+    /// Shorthand for [`TestedDialects::verified_statement`].
+    pub fn verified_stmt(&self, sql: &str) -> Statement {
+        self.verified_statement(sql)
+    }
+
     /// Ensures that `sql` parses as a single [Query], and that
     /// re-serializing the parse result produces the same `sql`
     /// string (is not modified after a serialization round-trip).

tests/sqlparser_snowflake.rs

@@ -5010,3 +5010,37 @@ fn test_select_dollar_column_from_stage() {
     // With table function args, without alias
     snowflake().verified_stmt("SELECT $1, $2 FROM @mystage1(file_format => 'myformat')");
 }
+
+#[test]
+fn test_text_cast_with_length_modifier() {
+    let select = snowflake().verified_only_select("SELECT col::TEXT(16777216) AS col FROM t");
+
+    match &select.projection[0] {
+        SelectItem::ExprWithAlias {
+            expr: Expr::Cast { data_type, .. },
+            alias,
+        } => {
+            assert_eq!(alias, &Ident::new("col"));
+            assert_eq!(
+                data_type,
+                &DataType::Custom(
+                    ObjectName::from(vec![Ident::new("TEXT")]),
+                    vec!["16777216".to_string()],
+                )
+            );
+        }
+        _ => unreachable!(),
+    }
+
+    snowflake().verified_statement("SELECT col::TEXT(16777216) AS col FROM t");
+}
+
+#[test]
+fn test_plain_text_data_type_still_parses_as_text() {
+    match snowflake().verified_statement("CREATE TABLE t (c TEXT)") {
+        Statement::CreateTable(CreateTable { columns, .. }) => {
+            assert_eq!(columns[0].data_type, DataType::Text);
+        }
+        _ => unreachable!(),
+    }
+}