Use snowflake helper in TEXT(n) tests

Author: yoavcloud

src/dialect/snowflake.rs

@@ -1930,13 +1930,21 @@ fn parse_multi_table_insert_when_clauses(
 mod tests {
     use super::SnowflakeDialect;
     use crate::ast::{DataType, ObjectName};
-    use crate::parser::Parser;
+    use crate::test_utils::TestedDialects;
+    #[cfg(not(feature = "std"))]
+    use alloc::{boxed::Box, vec};
+    #[cfg(feature = "std")]
+    use std::{boxed::Box, vec};
+
+    fn snowflake() -> TestedDialects {
+        TestedDialects::new(vec![Box::new(SnowflakeDialect {})])
+    }
 
     #[test]
     fn parses_cast_to_text_with_size_in_snowflake() {
-        Parser::parse_sql(
-            &SnowflakeDialect,
-            r#"SELECT
+        snowflake()
+            .parse_sql_statements(
+                r#"SELECT
 _ID::TEXT(16777216) AS _ID,
 TO_TIMESTAMP_NTZ(CREATED_ON) AS CREATED_ON,
 DEPARTMENT::TEXT(16777216) AS DEPARTMENT,
@@ -1953,17 +1961,14 @@ USER_ID::TEXT(16777216) AS USER_ID,
 EMAIL::TEXT(16777216) AS EMAIL,
 JOB_TITLE::TEXT(16777216) AS JOB_TITLE
 FROM INCARE_ANALYTICS.USER_DETAILS;"#,
-        )
-        .unwrap();
+            )
+            .unwrap();
     }
 
     #[test]
     fn parses_text_with_size_as_custom_data_type() {
-        let data_type = Parser::new(&SnowflakeDialect)
-            .try_with_sql("TEXT(16777216)")
-            .unwrap()
-            .parse_data_type()
-            .unwrap();
+        let data_type = snowflake()
+            .run_parser_method("TEXT(16777216)", |parser| parser.parse_data_type().unwrap());
 
         assert_eq!(
             data_type,
@@ -1977,11 +1982,8 @@ FROM INCARE_ANALYTICS.USER_DETAILS;"#,
 
     #[test]
     fn preserves_bare_text_data_type() {
-        let data_type = Parser::new(&SnowflakeDialect)
-            .try_with_sql("TEXT")
-            .unwrap()
-            .parse_data_type()
-            .unwrap();
+        let data_type =
+            snowflake().run_parser_method("TEXT", |parser| parser.parse_data_type().unwrap());
 
         assert_eq!(data_type, DataType::Text);
     }