Diffstat (limited to 'tests/protocol_visitor_recoverable.rs')
| -rw-r--r-- | tests/protocol_visitor_recoverable.rs | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/tests/protocol_visitor_recoverable.rs b/tests/protocol_visitor_recoverable.rs index 2b8e6a1..3979110 100644 --- a/tests/protocol_visitor_recoverable.rs +++ b/tests/protocol_visitor_recoverable.rs @@ -1,7 +1,6 @@ use common::protocol::recoverable::MockRecoverableVisitor; use treaty::{ any::OwnedStatic, - effect::blocking::Blocking, protocol::{ visitor::{Recoverable, ValueProto, VisitResult}, DynVisitor, @@ -9,7 +8,9 @@ use treaty::{ Flow, Status, }; -use crate::common::{builder::MockBuilder, protocol::recoverable::MockRecoverableScopeVisitor}; +use crate::common::{ + builder::MockBuilder, protocol::recoverable::MockRecoverableScopeVisitor, Blocking, +}; mod common; @@ -20,16 +21,16 @@ fn recoverable_can_be_visited() { // Expect a visit using the rescoverable protocol. mock.expect_visit().once().returning(|scope| { - let mut visitor = MockBuilder::<(), (), ()>::new(); + let mut visitor = MockBuilder::<(), (), (), Blocking>::new(); // 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); + assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).into_value(), Status::Ok); // Attempt to walk twice. - assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).value(), Status::Ok); + assert_eq!(scope.new_walk(DynVisitor(&mut visitor)).into_value(), Status::Ok); // We are done. VisitResult::Control(Flow::Done) @@ -51,7 +52,7 @@ fn recoverable_can_be_visited() { // Visit using the recoverable protocol. assert!(matches!( - visitor.visit(&mut scope).value(), + visitor.visit(&mut scope).into_value(), VisitResult::Control(Flow::Done) )); } |