Diffstat (limited to 'tests/protocol_visitor_recoverable.rs')
| -rw-r--r-- | tests/protocol_visitor_recoverable.rs | 29 |
1 files changed, 12 insertions, 17 deletions
diff --git a/tests/protocol_visitor_recoverable.rs b/tests/protocol_visitor_recoverable.rs index cde5ca1..2b8e6a1 100644 --- a/tests/protocol_visitor_recoverable.rs +++ b/tests/protocol_visitor_recoverable.rs @@ -9,10 +9,7 @@ use treaty::{ Flow, Status, }; -use crate::common::{ - builder::MockBuilder, - protocol::recoverable::{MockRecoverableScopeVisitor, RecoverableScopeFactory}, -}; +use crate::common::{builder::MockBuilder, protocol::recoverable::MockRecoverableScopeVisitor}; mod common; @@ -22,23 +19,21 @@ fn recoverable_can_be_visited() { let mut mock = MockRecoverableVisitor::<Blocking>::new(); // Expect a visit using the rescoverable protocol. - mock.expect_visit().once().return_const( - (|(), scope| { - let mut visitor = MockBuilder::<(), (), ()>::new(); + mock.expect_visit().once().returning(|scope| { + let mut visitor = MockBuilder::<(), (), ()>::new(); - // Expect that the visitor gets used. - visitor.expect_traits().times(2).return_const(None); + // Expect that the visitor gets used. + visitor.expect_traits().times(2).return_const(None); - // Attempt to walk once. - assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Status::Ok); + // Attempt to walk once. + assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Status::Ok); - // Attempt to walk twice. - assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Status::Ok); + // Attempt to walk twice. + assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Status::Ok); - // We are done. - VisitResult::Control(Flow::Done) - }) as RecoverableScopeFactory<Blocking>, - ); + // We are done. + VisitResult::Control(Flow::Done) + }); let visitor: &mut dyn Recoverable<Blocking> = &mut mock; |