Add error tests

apache · Jul 8, 2024 · e782bf5 · e782bf5
1 parent bb0dad8
commit e782bf5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/parser/mod.rs b/src/parser/mod.rs
@@ -4592,7 +4592,8 @@ impl<'a> Parser<'a> {
         let option = match self.parse_one_of_keywords(&[Keyword::CASCADE, Keyword::RESTRICT]) {
             Some(Keyword::CASCADE) => Some(ReferentialAction::Cascade),
             Some(Keyword::RESTRICT) => Some(ReferentialAction::Restrict),
-            _ => None,
+            Some(_) => unreachable!(), // parse_one_of_keywords does not return other keywords
+            None => None,
         };
         Ok(Statement::DropProcedure {
             if_exists,

diff --git a/tests/sqlparser_postgres.rs b/tests/sqlparser_postgres.rs
@@ -3717,6 +3717,18 @@ fn parse_drop_procedure() {
             option: None
         }
     );
+
+    let res = pg().parse_sql_statements("DROP PROCEDURE testproc DROP");
+    assert_eq!(
+        ParserError::ParserError("Expected: end of statement, found: DROP".to_string()),
+        res.unwrap_err()
+    );
+
+    let res = pg().parse_sql_statements("DROP PROCEDURE testproc SET NULL");
+    assert_eq!(
+        ParserError::ParserError("Expected: end of statement, found: SET".to_string()),
+        res.unwrap_err()
+    );
 }
 
 #[test]