Diffstat (limited to 'tests/protocol_visitor_request_hint.rs')
-rw-r--r--tests/protocol_visitor_request_hint.rs5
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/protocol_visitor_request_hint.rs b/tests/protocol_visitor_request_hint.rs
index 234d64f..06f811f 100644
--- a/tests/protocol_visitor_request_hint.rs
+++ b/tests/protocol_visitor_request_hint.rs
@@ -14,6 +14,7 @@ use treaty::{
protocol::{
visitor::{RequestHint, RequestHintProto, ValueKnown, ValueProto, VisitResult},
walker::hint::HintProto,
+ DynWalker,
},
Flow,
};
@@ -28,7 +29,7 @@ fn hints_can_be_requested() {
// We will request a hint from the visitor.
mock.expect_request_hint().once().return_const(
- (|_, walker| {
+ (|_, mut walker| {
// Lookup the value protocol on the walker.
let obj = walker
.upcast_mut::<HintProto<ValueProto<OwnedStatic<i32>, Blocking>>>()
@@ -75,7 +76,7 @@ fn hints_can_be_requested() {
// Request a hint from the visitor.
assert!(matches!(
- visitor.request_hint(&mut mock).into_inner(),
+ visitor.request_hint(DynWalker(&mut mock)).into_inner(),
VisitResult::Control(Flow::Done)
));
}