Diffstat (limited to 'tests/protocol_visitor_recoverable.rs')
| -rw-r--r-- | tests/protocol_visitor_recoverable.rs | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tests/protocol_visitor_recoverable.rs b/tests/protocol_visitor_recoverable.rs index 3243f2c..9c2e7d0 100644 --- a/tests/protocol_visitor_recoverable.rs +++ b/tests/protocol_visitor_recoverable.rs @@ -1,4 +1,5 @@ use common::protocol::recoverable::MockRecoverableVisitor; +use effectful::blocking::BlockingSpin; use treaty::{ any::OwnedStatic, protocol::{ @@ -9,7 +10,7 @@ use treaty::{ }; use crate::common::{ - builder::MockBuilder, protocol::recoverable::MockRecoverableScopeVisitor, Blocking, + builder::MockBuilder, protocol::recoverable::MockRecoverableScopeVisitor, }; mod common; @@ -17,11 +18,11 @@ mod common; /// Tests that the recoverable protocol allows multiple walks by the visitor. #[test] fn recoverable_can_be_visited() { - let mut mock = MockRecoverableVisitor::<Blocking>::new(); + let mut mock = MockRecoverableVisitor::<BlockingSpin>::new(); // Expect a visit using the rescoverable protocol. mock.expect_visit().once().returning(|scope| { - let mut visitor = MockBuilder::<(), (), (), Blocking>::new(); + let mut visitor = MockBuilder::<(), (), (), BlockingSpin>::new(); // Expect that the visitor gets used. visitor.expect_traits().times(2).return_const(None); @@ -42,7 +43,7 @@ fn recoverable_can_be_visited() { VisitResult::Control(Flow::Done) }); - let visitor: &mut dyn Recoverable<Blocking> = &mut mock; + let visitor: &mut dyn Recoverable<BlockingSpin> = &mut mock; let mut scope = MockRecoverableScopeVisitor::new(); @@ -50,7 +51,7 @@ fn recoverable_can_be_visited() { scope.expect_new_walk().times(2).returning(|visitor| { // Attempt to use the visitor. assert!(visitor - .upcast::<ValueProto<OwnedStatic<i32>, Blocking>>() + .upcast::<ValueProto<OwnedStatic<i32>, BlockingSpin>>() .is_none()); Status::Ok |