[#142] having 절 테스트코드 추가

myyrakle · Jul 28, 2024 · 471cf6d · 471cf6d
1 parent 8d10d86
commit 471cf6d
Showing 1 changed file with 96 additions and 0 deletions.
diff --git a/src/parser/test/select.rs b/src/parser/test/select.rs
@@ -5,6 +5,7 @@ use crate::ast::dml::expressions::call::CallExpression;
 use crate::ast::dml::expressions::operators::BinaryOperator;
 use crate::ast::dml::parts::_where::WhereClause;
 use crate::ast::dml::parts::group_by::GroupByItem;
+use crate::ast::dml::parts::having::HavingClause;
 use crate::ast::dml::parts::join::{JoinClause, JoinType};
 use crate::ast::dml::parts::order_by::{OrderByItem, OrderByNulls, OrderByType};
 use crate::ast::dml::parts::select_item::{SelectItem, SelectWildCard};
@@ -1593,6 +1594,101 @@ fn test_select_query() {
             expected: Default::default(),
             want_error: true,
         },
+        TestCase {
+            name: r#"
+                SELECT 
+                    p.content as post
+                FROM post as p
+                GROUP BY p.content, p.user_id
+                HAVING p.user_id > 1
+            "#
+            .into(),
+            input: vec![
+                Token::Select,
+                Token::Identifier("p".into()),
+                Token::Period,
+                Token::Identifier("content".into()),
+                Token::As,
+                Token::Identifier("post".into()),
+                Token::From,
+                Token::Identifier("post".into()),
+                Token::As,
+                Token::Identifier("p".into()),
+                Token::Group,
+                Token::By,
+                Token::Identifier("p".into()),
+                Token::Period,
+                Token::Identifier("content".into()),
+                Token::Comma,
+                Token::Identifier("p".into()),
+                Token::Period,
+                Token::Identifier("user_id".into()),
+                Token::Having,
+                Token::Identifier("p".into()),
+                Token::Period,
+                Token::Identifier("user_id".into()),
+                Token::Operator(OperatorToken::Gt),
+                Token::Integer(1),
+            ],
+            expected: SelectQuery::builder()
+                .add_select_item(
+                    SelectItem::builder()
+                        .set_item(SelectColumn::new(Some("p".into()), "content".into()).into())
+                        .set_alias("post".into())
+                        .build(),
+                )
+                .set_from_table(TableName {
+                    database_name: None,
+                    table_name: "post".into(),
+                })
+                .set_from_alias("p".into())
+                .add_group_by(GroupByItem {
+                    item: SelectColumn::new(Some("p".into()), "content".into()),
+                })
+                .add_group_by(GroupByItem {
+                    item: SelectColumn::new(Some("p".into()), "user_id".into()),
+                })
+                .set_having(HavingClause {
+                    expression: BinaryOperatorExpression {
+                        operator: BinaryOperator::Gt,
+                        lhs: SelectColumn::new(Some("p".into()), "user_id".into()).into(),
+                        rhs: SQLExpression::Integer(1),
+                    }
+                    .into(),
+                })
+                .build(),
+            want_error: false,
+        },
+        TestCase {
+            name: r#"
+                실패: GROUP BY 없는 HAVING 절
+                SELECT 
+                    p.content as post
+                FROM post as p
+                HAVING p.user_id > 1
+            "#
+            .into(),
+            input: vec![
+                Token::Select,
+                Token::Identifier("p".into()),
+                Token::Period,
+                Token::Identifier("content".into()),
+                Token::As,
+                Token::Identifier("post".into()),
+                Token::From,
+                Token::Identifier("post".into()),
+                Token::As,
+                Token::Identifier("p".into()),
+                Token::Having,
+                Token::Identifier("p".into()),
+                Token::Period,
+                Token::Identifier("user_id".into()),
+                Token::Operator(OperatorToken::Gt),
+                Token::Integer(1),
+            ],
+            expected: Default::default(),
+            want_error: true,
+        },
     ];
 
     for t in test_cases {