Diffstat (limited to 'tests/protocol_visitor_recoverable.rs')
| -rw-r--r-- | tests/protocol_visitor_recoverable.rs | 14 |
1 files changed, 4 insertions, 10 deletions
diff --git a/tests/protocol_visitor_recoverable.rs b/tests/protocol_visitor_recoverable.rs index 07ec092..6b70906 100644 --- a/tests/protocol_visitor_recoverable.rs +++ b/tests/protocol_visitor_recoverable.rs @@ -1,7 +1,7 @@ use common::protocol::recoverable::MockRecoverableVisitor; use treaty::{ any::OwnedStatic, - effect::Blocking, + effect::{Blocking, ReadyValue}, protocol::{ visitor::{Recoverable, ValueProto, VisitResult}, DynVisitor, @@ -30,16 +30,10 @@ fn recoverable_can_be_visited() { visitor.expect_traits().times(2).return_const(None); // Attempt to walk once. - assert_eq!( - scope.new_walk(DynVisitor(&mut visitor)).into_inner(), - Ok(()) - ); + assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Ok(())); // Attempt to walk twice. - assert_eq!( - scope.new_walk(DynVisitor(&mut visitor)).into_inner(), - Ok(()) - ); + assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Ok(())); // We are done. VisitResult::Control(Flow::Done) @@ -62,7 +56,7 @@ fn recoverable_can_be_visited() { // Visit using the recoverable protocol. assert!(matches!( - visitor.visit(&mut scope).into_inner(), + visitor.visit(&mut scope).value(), VisitResult::Control(Flow::Done) )); } |