Addressed reviewer concerns and extended test coveraged checked with tarpaulin

Author: LucaCappelletti94

src/ast/mod.rs

@@ -73,13 +73,12 @@ pub use self::ddl::{
     CreatePolicyCommand, CreatePolicyType, CreateTable, CreateTextSearch, CreateTrigger,
     CreateView, Deduplicate, DeferrableInitial, DistStyle, DropBehavior, DropExtension,
     DropFunction, DropOperator, DropOperatorClass, DropOperatorFamily, DropOperatorSignature,
-    DropPolicy, DropTrigger, ForValues, FunctionReturnType, GeneratedAs,
-    GeneratedExpressionMode, IdentityParameters,
-    IdentityProperty, IdentityPropertyFormatKind, IdentityPropertyKind, IdentityPropertyOrder,
-    IndexColumn, IndexOption, IndexType, KeyOrIndexDisplay, Msck, NullsDistinctOption,
-    OperatorArgTypes, OperatorClassItem, OperatorFamilyDropItem, OperatorFamilyItem,
-    OperatorOption, OperatorPurpose, Owner, Partition, PartitionBoundValue, ProcedureParam,
-    ReferentialAction, RenameTableNameKind, ReplicaIdentity, TagsColumnOption,
+    DropPolicy, DropTrigger, ForValues, FunctionReturnType, GeneratedAs, GeneratedExpressionMode,
+    IdentityParameters, IdentityProperty, IdentityPropertyFormatKind, IdentityPropertyKind,
+    IdentityPropertyOrder, IndexColumn, IndexOption, IndexType, KeyOrIndexDisplay, Msck,
+    NullsDistinctOption, OperatorArgTypes, OperatorClassItem, OperatorFamilyDropItem,
+    OperatorFamilyItem, OperatorOption, OperatorPurpose, Owner, Partition, PartitionBoundValue,
+    ProcedureParam, ReferentialAction, RenameTableNameKind, ReplicaIdentity, TagsColumnOption,
     TextSearchObjectType, TriggerObjectKind, Truncate, UserDefinedTypeCompositeAttributeDef,
     UserDefinedTypeInternalLength, UserDefinedTypeRangeOption, UserDefinedTypeRepresentation,
     UserDefinedTypeSqlDefinitionOption, UserDefinedTypeStorage, ViewColumnDef,

src/parser/mod.rs

@@ -5113,12 +5113,7 @@ impl<'a> Parser<'a> {
         let create_view_params = self.parse_create_view_params()?;
         if self.peek_keywords(&[Keyword::SNAPSHOT, Keyword::TABLE]) {
             self.parse_create_snapshot_table().map(Into::into)
-        } else if self.parse_keywords(&[Keyword::TEXT, Keyword::SEARCH]) {
-            if or_replace || or_alter || temporary || global.is_some() || transient || persistent {
-                return Err(ParserError::ParserError(
-                    "CREATE TEXT SEARCH does not support CREATE modifiers".to_string(),
-                ));
-            }
+        } else if self.peek_keywords(&[Keyword::TEXT, Keyword::SEARCH]) {
             self.parse_create_text_search().map(Into::into)
         } else if self.parse_keyword(Keyword::TABLE) {
             self.parse_create_table(or_replace, temporary, global, transient)
@@ -5211,19 +5206,13 @@ impl<'a> Parser<'a> {
         }
     }
 
-    fn parse_text_search_option(&mut self) -> Result<SqlOption, ParserError> {
-        let key = self.parse_identifier()?;
-        self.expect_token(&Token::Eq)?;
-        let value = self.parse_expr()?;
-        Ok(SqlOption::KeyValue { key, value })
-    }
-
     /// Parse a PostgreSQL `CREATE TEXT SEARCH ...` statement.
     pub fn parse_create_text_search(&mut self) -> Result<CreateTextSearch, ParserError> {
+        self.expect_keywords(&[Keyword::TEXT, Keyword::SEARCH])?;
         let object_type = self.parse_text_search_object_type()?;
         let name = self.parse_object_name(false)?;
         self.expect_token(&Token::LParen)?;
-        let options = self.parse_comma_separated(Parser::parse_text_search_option)?;
+        let options = self.parse_comma_separated(Parser::parse_sql_option)?;
         self.expect_token(&Token::RParen)?;
         Ok(CreateTextSearch {
             object_type,
@@ -5246,6 +5235,7 @@ impl<'a> Parser<'a> {
 
     /// Parse a PostgreSQL `ALTER TEXT SEARCH ...` statement.
     pub fn parse_alter_text_search(&mut self) -> Result<AlterTextSearch, ParserError> {
+        self.expect_keywords(&[Keyword::TEXT, Keyword::SEARCH])?;
         let object_type = self.parse_text_search_object_type()?;
         let name = self.parse_object_name(false)?;
 
@@ -10723,7 +10713,7 @@ impl<'a> Parser<'a> {
 
     /// Parse an `ALTER <object>` statement and dispatch to the appropriate alter handler.
     pub fn parse_alter(&mut self) -> Result<Statement, ParserError> {
-        if self.parse_keywords(&[Keyword::TEXT, Keyword::SEARCH]) {
+        if self.peek_keywords(&[Keyword::TEXT, Keyword::SEARCH]) {
             return self.parse_alter_text_search().map(Into::into);
         }
 

tests/sqlparser_postgres.rs

@@ -8061,49 +8061,24 @@ fn parse_alter_operator_class() {
 }
 
 #[test]
-fn parse_create_and_alter_text_search_failure_cases() {
-    let sql_cases = [
-        "CREATE TEXT SEARCH DICTIONARY alt_ts_dict1 (template=simple)",
-        "CREATE TEXT SEARCH DICTIONARY alt_ts_dict2 (template=simple)",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict1 RENAME TO alt_ts_dict2",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict1 RENAME TO alt_ts_dict3",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict2 OWNER TO regress_alter_generic_user2",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict2 OWNER TO regress_alter_generic_user3",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict2 SET SCHEMA alt_nsp2",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict3 RENAME TO alt_ts_dict4",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict1 RENAME TO alt_ts_dict4",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict3 OWNER TO regress_alter_generic_user2",
-        "ALTER TEXT SEARCH DICTIONARY alt_ts_dict3 SET SCHEMA alt_nsp2",
-        "CREATE TEXT SEARCH CONFIGURATION alt_ts_conf1 (copy=english)",
-        "CREATE TEXT SEARCH CONFIGURATION alt_ts_conf2 (copy=english)",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf1 RENAME TO alt_ts_conf2",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf1 RENAME TO alt_ts_conf3",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf2 OWNER TO regress_alter_generic_user2",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf2 OWNER TO regress_alter_generic_user3",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf2 SET SCHEMA alt_nsp2",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf3 RENAME TO alt_ts_conf4",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf1 RENAME TO alt_ts_conf4",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf3 OWNER TO regress_alter_generic_user2",
-        "ALTER TEXT SEARCH CONFIGURATION alt_ts_conf3 SET SCHEMA alt_nsp2",
-        "CREATE TEXT SEARCH TEMPLATE alt_ts_temp1 (lexize=dsimple_lexize)",
-        "CREATE TEXT SEARCH TEMPLATE alt_ts_temp2 (lexize=dsimple_lexize)",
-        "ALTER TEXT SEARCH TEMPLATE alt_ts_temp1 RENAME TO alt_ts_temp2",
-        "ALTER TEXT SEARCH TEMPLATE alt_ts_temp1 RENAME TO alt_ts_temp3",
-        "ALTER TEXT SEARCH TEMPLATE alt_ts_temp2 SET SCHEMA alt_nsp2",
-        "CREATE TEXT SEARCH TEMPLATE tstemp_case (\"Init\" = init_function)",
-        "CREATE TEXT SEARCH PARSER alt_ts_prs1 (start = prsd_start, gettoken = prsd_nexttoken, end = prsd_end, lextypes = prsd_lextype)",
-        "CREATE TEXT SEARCH PARSER alt_ts_prs2 (start = prsd_start, gettoken = prsd_nexttoken, end = prsd_end, lextypes = prsd_lextype)",
-        "ALTER TEXT SEARCH PARSER alt_ts_prs1 RENAME TO alt_ts_prs2",
-        "ALTER TEXT SEARCH PARSER alt_ts_prs1 RENAME TO alt_ts_prs3",
-        "ALTER TEXT SEARCH PARSER alt_ts_prs2 SET SCHEMA alt_nsp2",
-        "CREATE TEXT SEARCH PARSER tspars_case (\"Start\" = start_function)",
-    ];
+fn parse_create_and_alter_text_search() {
+    // CREATE — one per object type
+    pg_and_generic().verified_stmt("CREATE TEXT SEARCH DICTIONARY d (template = simple)");
+    pg_and_generic().verified_stmt("CREATE TEXT SEARCH CONFIGURATION c (copy = english)");
+    pg_and_generic().verified_stmt("CREATE TEXT SEARCH TEMPLATE t (lexize = dsimple_lexize)");
+    pg_and_generic().verified_stmt(
+        "CREATE TEXT SEARCH PARSER p (start = prsd_start, gettoken = prsd_nexttoken, end = prsd_end, lextypes = prsd_lextype)",
+    );
 
-    for sql in sql_cases {
-        if let Err(err) = pg().parse_sql_statements(sql) {
-            panic!("Failed to parse `{sql}`: {err}");
-        }
-    }
+    // CREATE with quoted option key
+    pg_and_generic().verified_stmt("CREATE TEXT SEARCH TEMPLATE t (\"Init\" = init_function)");
+
+    // ALTER — one test per object type arm, one per operation kind
+    pg_and_generic().verified_stmt("ALTER TEXT SEARCH DICTIONARY d (opt = val)");
+    pg_and_generic().verified_stmt("ALTER TEXT SEARCH DICTIONARY d (opt)");
+    pg_and_generic().verified_stmt("ALTER TEXT SEARCH CONFIGURATION c OWNER TO some_user");
+    pg_and_generic().verified_stmt("ALTER TEXT SEARCH TEMPLATE t SET SCHEMA s");
+    pg_and_generic().verified_stmt("ALTER TEXT SEARCH PARSER p RENAME TO p2");
 
     // Object type must be an unquoted keyword-like token in this position.
     assert!(pg()
@@ -8114,17 +8089,6 @@ fn parse_create_and_alter_text_search_failure_cases() {
     assert!(pg()
         .parse_sql_statements("CREATE TEXT SEARCH DICTIONARY d (template)")
         .is_err());
-
-    // CREATE TEXT SEARCH does not support generic CREATE modifiers.
-    assert!(pg()
-        .parse_sql_statements("CREATE OR REPLACE TEXT SEARCH DICTIONARY d (template = simple)")
-        .is_err());
-    assert!(pg()
-        .parse_sql_statements("CREATE OR ALTER TEXT SEARCH DICTIONARY d (template = simple)")
-        .is_err());
-    assert!(pg()
-        .parse_sql_statements("CREATE TEMP TEXT SEARCH DICTIONARY d (template = simple)")
-        .is_err());
 }
 
 #[test]