Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions src/ast/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8774,6 +8774,8 @@ pub enum ObjectType {
User,
/// A stream.
Stream,
/// A Snowflake external volume.
ExternalVolume,
/// A warehouse.
Warehouse,
}
Expand All @@ -8794,6 +8796,7 @@ impl fmt::Display for ObjectType {
ObjectType::Type => "TYPE",
ObjectType::User => "USER",
ObjectType::Stream => "STREAM",
ObjectType::ExternalVolume => "EXTERNAL VOLUME",
ObjectType::Warehouse => "WAREHOUSE",
})
}
Expand Down
4 changes: 3 additions & 1 deletion src/parser/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8103,6 +8103,8 @@ impl<'a> Parser<'a> {
ObjectType::User
} else if self.parse_keyword(Keyword::STREAM) {
ObjectType::Stream
} else if self.parse_keywords(&[Keyword::EXTERNAL, Keyword::VOLUME]) {
ObjectType::ExternalVolume
} else if self.parse_keyword(Keyword::WAREHOUSE) {
ObjectType::Warehouse
} else if self.parse_keyword(Keyword::FUNCTION) {
Expand Down Expand Up @@ -8132,7 +8134,7 @@ impl<'a> Parser<'a> {
};
} else {
return self.expected_ref(
"COLLATION, CONNECTOR, DATABASE, EXTENSION, FUNCTION, INDEX, OPERATOR, POLICY, PROCEDURE, ROLE, SCHEMA, SECRET, SEQUENCE, STAGE, TABLE, TRIGGER, TYPE, VIEW, MATERIALIZED VIEW, USER or WAREHOUSE after DROP",
"COLLATION, CONNECTOR, DATABASE, EXTENSION, EXTERNAL VOLUME, FUNCTION, INDEX, OPERATOR, POLICY, PROCEDURE, ROLE, SCHEMA, SECRET, SEQUENCE, STAGE, TABLE, TRIGGER, TYPE, VIEW, MATERIALIZED VIEW, USER or WAREHOUSE after DROP",
self.peek_token_ref(),
);
};
Expand Down
37 changes: 37 additions & 0 deletions tests/sqlparser_snowflake.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5042,3 +5042,40 @@ fn test_stage_name_delimiters() {
ParserError::ParserError("Expected: stage name identifier, found: .".to_string()),
);
}

#[test]
fn test_drop_external_volume() {
match snowflake().verified_stmt("DROP EXTERNAL VOLUME my_vol") {
Statement::Drop {
object_type,
if_exists,
names,
..
} => {
assert_eq!(ObjectType::ExternalVolume, object_type);
assert!(!if_exists);
assert_eq!("my_vol", names[0].to_string());
}
_ => unreachable!(),
}
let err = snowflake()
.parse_sql_statements("DROP EXTERNAL my_vol")
.unwrap_err()
.to_string();
assert!(err.contains("EXTERNAL VOLUME"), "{err}");
}

#[test]
fn test_drop_external_volume_if_exists() {
match snowflake().verified_stmt("DROP EXTERNAL VOLUME IF EXISTS my_vol") {
Statement::Drop {
object_type,
if_exists,
..
} => {
assert_eq!(ObjectType::ExternalVolume, object_type);
assert!(if_exists);
}
_ => unreachable!(),
}
}
Loading