aharpervc commented on code in PR #1831:
URL:
https://github.com/apache/datafusion-sqlparser-rs/pull/1831#discussion_r2067496065
##########
tests/sqlparser_mssql.rs:
##########
@@ -1393,6 +1394,85 @@ fn parse_mssql_declare() {
let _ = ms().verified_stmt(declare_cursor_for_select);
}
+#[test]
+fn test_mssql_cursor() {
+ let full_cursor_usage = "\
+ DECLARE Employee_Cursor CURSOR FOR \
+ SELECT LastName, FirstName \
+ FROM AdventureWorks2022.HumanResources.vEmployee \
+ WHERE LastName LIKE 'B%'; \
+ \
+ OPEN Employee_Cursor; \
+ \
+ FETCH NEXT FROM Employee_Cursor; \
+ \
+ WHILE @@FETCH_STATUS = 0 \
+ BEGIN \
+ FETCH NEXT FROM Employee_Cursor; \
+ END; \
+ \
+ CLOSE Employee_Cursor; \
+ DEALLOCATE Employee_Cursor\
+ ";
+ let _ = ms().statements_parse_to(full_cursor_usage, 6, "");
+}
+
+#[test]
+fn test_mssql_while_statement() {
+ let while_single_statement = "WHILE 1 = 0 PRINT 'Hello World';";
+ let stmt = ms().verified_stmt(while_single_statement);
+ assert_eq!(
+ stmt,
+ Statement::While(sqlparser::ast::WhileStatement {
+ while_block: ConditionalStatementBlock {
+ start_token: AttachedToken(TokenWithSpan {
+ token: Token::Word(Word {
+ value: "WHILE".to_string(),
+ quote_style: None,
+ keyword: Keyword::WHILE
+ }),
+ span: Span::empty()
+ }),
+ condition: Some(Expr::BinaryOp {
+ left: Box::new(Expr::Value(
+ (Value::Number("1".parse().unwrap(),
false)).with_empty_span()
+ )),
+ op: BinaryOperator::Eq,
+ right: Box::new(Expr::Value(
+ (Value::Number("0".parse().unwrap(),
false)).with_empty_span()
+ )),
+ }),
+ then_token: None,
+ conditional_statements: ConditionalStatements::Sequence {
+ statements: vec![Statement::Print(PrintStatement {
+ message: Box::new(Expr::Value(
+ (Value::SingleQuotedString("Hello
World".to_string()))
+ .with_empty_span()
+ )),
+ }),],
Review Comment:
Oops... I will fix this
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscr...@datafusion.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: github-unsubscr...@datafusion.apache.org
For additional commands, e-mail: github-h...@datafusion.apache.org
