bendns' repos / treaty
Diffstat (limited to 'tests/common/protocol/tag.rs')
-rw-r--r--tests/common/protocol/tag.rs9
1 files changed, 3 insertions, 6 deletions
diff --git a/tests/common/protocol/tag.rs b/tests/common/protocol/tag.rs
index 2dcd805..890157e 100644
--- a/tests/common/protocol/tag.rs
+++ b/tests/common/protocol/tag.rs
@@ -8,7 +8,7 @@ use treaty::{
mock! {
pub TagVisitor<K: TagKind, E> {
- pub fn visit<'a, 'ctx>(&'a mut self, kind: K, walker: DynWalkerObjSafe<'a, 'ctx, E>) -> VisitResult<()>;
+ pub fn visit<'a, 'ctx>(&'a mut self, kind: K, walker: DynWalkerObjSafe<'a, 'ctx, E>) -> VisitResult;
}
}
@@ -25,10 +25,7 @@ impl<'ctx, K: TagKind, E: Effect> Tag<'ctx, K, E> for MockTagVisitor<K, E> {
&'a mut self,
kind: K,
walker: DynWalkerObjSafe<'b, 'ctx, E>,
- ) -> ErasedEffective<'c, VisitResult<DynWalkerObjSafe<'b, 'ctx, E>>, E> {
- E::ready(match self.visit(kind, walker) {
- VisitResult::Skipped(_) => VisitResult::Skipped(walker),
- VisitResult::Control(flow) => VisitResult::Control(flow),
- })
+ ) -> ErasedEffective<'c, VisitResult, E> {
+ E::ready(self.visit(kind, walker))
}
}