gluesql_core/ast_builder/
drop_table.rs

1use {
2    super::Build,
3    crate::{ast::Statement, result::Result},
4};
5
6#[derive(Clone, Debug)]
7pub struct DropTableNode {
8    table_name: String,
9    if_exists: bool,
10    cascade: bool,
11}
12
13impl DropTableNode {
14    pub fn new(table_name: String, exists: bool, cascade: bool) -> Self {
15        Self {
16            table_name,
17            if_exists: exists,
18            cascade,
19        }
20    }
21}
22
23impl Build for DropTableNode {
24    fn build(self) -> Result<Statement> {
25        let names = vec![self.table_name];
26        let if_exists = self.if_exists;
27        let cascade = self.cascade;
28
29        Ok(Statement::DropTable {
30            names,
31            if_exists,
32            cascade,
33        })
34    }
35}
36
37#[cfg(test)]
38mod tests {
39    use crate::ast_builder::{Build, table, test};
40
41    #[test]
42    fn drop_table() {
43        let actual = table("Foo").drop_table().build();
44        let expected = "DROP TABLE Foo";
45        test(actual, expected);
46
47        let actual = table("Foo").drop_table_if_exists().build();
48        let expected = "DROP TABLE IF EXISTS Foo";
49        test(actual, expected);
50
51        let actual = table("Foo").drop_table_cascade().build();
52        let expected = "DROP TABLE Foo CASCADE";
53        test(actual, expected);
54
55        let actual = table("Foo").drop_table_if_exists_cascade().build();
56        let expected = "DROP TABLE IF EXISTS Foo CASCADE";
57        test(actual, expected);
58    }
59}
gluesql_core/ast_builder/drop_table.rs

gluesql_core/ast_builder/
drop_table.rs
