@@ -741,15 +741,16 @@ mod tests {
741741 }
742742 }
743743
744- fn do_visit < V : Visitor > ( sql : & str , visitor : & mut V ) -> Statement {
744+ fn do_visit < V : Visitor < Break = ( ) > > ( sql : & str , visitor : & mut V ) -> Statement {
745745 let dialect = GenericDialect { } ;
746746 let tokens = Tokenizer :: new ( & dialect, sql) . tokenize ( ) . unwrap ( ) ;
747747 let s = Parser :: new ( & dialect)
748748 . with_tokens ( tokens)
749749 . parse_statement ( )
750750 . unwrap ( ) ;
751751
752- s. visit ( visitor) ;
752+ let flow = s. visit ( visitor) ;
753+ assert_eq ! ( flow, ControlFlow :: Continue ( ( ) ) ) ;
753754 s
754755 }
755756
@@ -938,7 +939,8 @@ mod tests {
938939 . unwrap ( ) ;
939940
940941 let mut visitor = QuickVisitor { } ;
941- s. visit ( & mut visitor) ;
942+ let flow = s. visit ( & mut visitor) ;
943+ assert_eq ! ( flow, ControlFlow :: Continue ( ( ) ) ) ;
942944 }
943945}
944946
@@ -969,15 +971,16 @@ mod visit_mut_tests {
969971 }
970972 }
971973
972- fn do_visit_mut < V : VisitorMut > ( sql : & str , visitor : & mut V ) -> Statement {
974+ fn do_visit_mut < V : VisitorMut < Break = ( ) > > ( sql : & str , visitor : & mut V ) -> Statement {
973975 let dialect = GenericDialect { } ;
974976 let tokens = Tokenizer :: new ( & dialect, sql) . tokenize ( ) . unwrap ( ) ;
975977 let mut s = Parser :: new ( & dialect)
976978 . with_tokens ( tokens)
977979 . parse_statement ( )
978980 . unwrap ( ) ;
979981
980- s. visit ( visitor) ;
982+ let flow = s. visit ( visitor) ;
983+ assert_eq ! ( flow, ControlFlow :: Continue ( ( ) ) ) ;
981984 s
982985 }
983986
0 commit comments