From c597394a4ba15b523c7a952006b07f41c97103d3 Mon Sep 17 00:00:00 2001
From: myyrakle <sssang97@naver.com>
Date: Fri, 16 Aug 2024 01:15:59 +0900
Subject: [PATCH] =?UTF-8?q?[#153]=20ast=20=ED=85=8C=EC=8A=A4=ED=8A=B8?=
 =?UTF-8?q?=EC=BD=94=EB=93=9C=20=EA=B0=9C=EC=84=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/ast/dml/insert.rs | 6 +++---
 src/ast/dml/select.rs | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/ast/dml/insert.rs b/src/ast/dml/insert.rs
index 74ebad5..26a05da 100644
--- a/src/ast/dml/insert.rs
+++ b/src/ast/dml/insert.rs
@@ -80,14 +80,14 @@ mod tests {
             .build();
 
         assert_eq!(
-            insert_query,
-            InsertQuery {
+            SQLStatement::from(insert_query),
+            SQLStatement::DML(DMLStatement::InsertQuery(InsertQuery {
                 into_table: Some(TableName::new(None, "table".into())),
                 columns: vec!["a".into(), "b".into()],
                 data: InsertData::Values(vec![InsertValue {
                     list: vec![Some(SQLExpression::String("a".into()))],
                 }]),
-            }
+            }))
         );
     }
 }
diff --git a/src/ast/dml/select.rs b/src/ast/dml/select.rs
index 15f4121..06accff 100644
--- a/src/ast/dml/select.rs
+++ b/src/ast/dml/select.rs
@@ -299,8 +299,8 @@ mod tests {
             .build();
 
         assert_eq!(
-            select_query,
-            SelectQuery {
+            SQLStatement::from(select_query),
+            SQLStatement::DML(DMLStatement::SelectQuery(SelectQuery {
                 select_items: vec![SelectKind::SelectItem(SelectItem {
                     item: Some(SQLExpression::String("a".into())),
                     alias: None
@@ -314,7 +314,7 @@ mod tests {
                 limit: None,
                 offset: None,
                 has_aggregate: false,
-            }
+            }))
         );
     }
 }