Skip CREATE STAGE by DFParser (pass to dialect parser) (#11)

* DFParser should skip unsupported COPY INTO

* Add tests

* Add tests

* Add tests

* Add tests

* Fmt

* Fmt

* Fmt

Author: osipovartem

datafusion/sql/src/parser.rs

@@ -701,7 +701,11 @@ impl<'a> DFParser<'a> {
             self.parser.expect_keyword(Keyword::EXTERNAL)?;
             self.parse_create_external_table(true)
         } else {
-            Ok(Statement::Statement(Box::from(self.parser.parse_create()?)))
+            // Push back CREATE
+            self.parser.prev_token();
+            Ok(Statement::Statement(Box::from(
+                self.parser.parse_statement()?,
+            )))
         }
     }
 
@@ -1065,6 +1069,26 @@ mod tests {
         }
     }
 
+    #[test]
+    fn skip_create_stage_snowflake() -> Result<(), ParserError> {
+        let sql =
+            "CREATE OR REPLACE STAGE stage URL='s3://data.csv' FILE_FORMAT=(TYPE=csv)";
+        let dialect = Box::new(SnowflakeDialect);
+        let statements = DFParser::parse_sql_with_dialect(sql, dialect.as_ref())?;
+
+        assert_eq!(
+            statements.len(),
+            1,
+            "Expected to parse exactly one statement"
+        );
+        match &statements[0] {
+            Statement::Statement(stmt) => {
+                assert_eq!(stmt.to_string(), sql);
+            }
+            _ => panic!("Expected statement type"),
+        }
+        Ok(())
+    }
     #[test]
     fn create_external_table() -> Result<(), DataFusionError> {
         // positive case