File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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;\n GO -- okay" ;
2918- let stmts = ms ( ) . parse_sql_statements ( comment_following_go ) . unwrap ( ) ;
2917+ let single_line_comment_following_go = "USE some_database;\n GO -- 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;\n GO/* 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 ! (
You can’t perform that action at this time.
0 commit comments