Skip to content

Commit 2e30a00

Browse files
committed
Add additional multi line comment test
1 parent f010708 commit 2e30a00

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

tests/sqlparser_mssql.rs

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2914,8 +2914,8 @@ fn parse_mssql_go_keyword() {
29142914
assert_eq!(stmts.len(), 2);
29152915
assert_eq!(stmts[1], Statement::Go(GoStatement { count: None }));
29162916

2917-
let comment_following_go = "USE some_database;\nGO -- okay";
2918-
let stmts = ms().parse_sql_statements(comment_following_go).unwrap();
2917+
let single_line_comment_following_go = "USE some_database;\nGO -- okay";
2918+
let stmts = ms().parse_sql_statements(single_line_comment_following_go).unwrap();
29192919
assert_eq!(stmts.len(), 2);
29202920
assert_eq!(stmts[1], Statement::Go(GoStatement { count: None }));
29212921

@@ -2946,6 +2946,16 @@ fn parse_mssql_go_keyword() {
29462946
// _ => panic!("Expected Query statement"),
29472947
// }
29482948

2949+
let multi_line_comment_following = "USE some_database;\nGO/* okay */42";
2950+
let stmts = ms()
2951+
.parse_sql_statements(multi_line_comment_following)
2952+
.unwrap();
2953+
assert_eq!(stmts.len(), 2);
2954+
assert_eq!(stmts[1], Statement::Go(GoStatement { count: Some(42) }));
2955+
2956+
let actually_column_alias = "SELECT NULL GO";
2957+
let stmts = ms().parse_sql_statements(actually_column_alias).unwrap();
2958+
29492959
let invalid_go_position = "SELECT 1; GO";
29502960
let err = ms().parse_sql_statements(invalid_go_position);
29512961
assert_eq!(

0 commit comments

Comments
 (0)