Diffstat (limited to 'tests/common/protocol/tag.rs')
| -rw-r--r-- | tests/common/protocol/tag.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/common/protocol/tag.rs b/tests/common/protocol/tag.rs index 7bc152a..581cced 100644 --- a/tests/common/protocol/tag.rs +++ b/tests/common/protocol/tag.rs @@ -3,12 +3,12 @@ use treaty::{ any::any_trait, effect::{Effect, Future}, protocol::visitor::{Tag, TagKind, TagProto, VisitResult}, - DynWalker, + DynWalkerObjSafe, }; mock! { pub TagVisitor<K: TagKind, E> { - pub fn visit<'a, 'ctx>(&'a mut self, kind: K, walker: DynWalker<'a, 'ctx, E>) -> VisitResult<()>; + pub fn visit<'a, 'ctx>(&'a mut self, kind: K, walker: DynWalkerObjSafe<'a, 'ctx, E>) -> VisitResult<()>; } } @@ -24,8 +24,8 @@ impl<'ctx, K: TagKind, E: Effect> Tag<'ctx, K, E> for MockTagVisitor<K, E> { fn visit<'a>( &'a mut self, kind: K, - walker: DynWalker<'a, 'ctx, E>, - ) -> Future<'a, VisitResult<DynWalker<'a, 'ctx, E>>, E> { + walker: DynWalkerObjSafe<'a, 'ctx, E>, + ) -> Future<'a, VisitResult<DynWalkerObjSafe<'a, 'ctx, E>>, E> { E::ready(match self.visit(kind, walker) { VisitResult::Skipped(_) => VisitResult::Skipped(walker), VisitResult::Control(flow) => VisitResult::Control(flow), |