unit tests for select, delete statements ending with semi colon

Author: virattara

tests/sqlparser_generic.rs

@@ -396,6 +396,33 @@ fn parse_parens() {
     , ast);
 }
 
+#[test]
+fn parse_select_with_semi_colon() {
+    let sql = String::from("SELECT id, fname, lname FROM customer WHERE id = 1;");
+    let ast = parse_sql(&sql);
+    match ast {
+        ASTNode::SQLSelect { projection, .. } => {
+            assert_eq!(3, projection.len());
+        }
+        _ => assert!(false),
+    }
+}
+
+#[test]
+fn parse_delete_with_semi_colon() {
+    let sql: &str = "DELETE FROM 'table';";
+
+    match parse_sql(&sql) {
+        ASTNode::SQLDelete { relation, .. } => {
+            assert_eq!(
+                Some(Box::new(ASTNode::SQLValue(Value::SingleQuotedString("table".to_string())))),
+                relation
+            );
+        }
+        _ => assert!(false),
+    }
+}
+
 fn parse_sql(sql: &str) -> ASTNode {
     let dialect = GenericSqlDialect {};
     let mut tokenizer = Tokenizer::new(&dialect,&sql, );