bendns' repos / treaty
Diffstat (limited to 'tests/protocol_visitor_request_hint.rs')
-rw-r--r--tests/protocol_visitor_request_hint.rs93
1 files changed, 93 insertions, 0 deletions
diff --git a/tests/protocol_visitor_request_hint.rs b/tests/protocol_visitor_request_hint.rs
new file mode 100644
index 0000000..234d64f
--- /dev/null
+++ b/tests/protocol_visitor_request_hint.rs
@@ -0,0 +1,93 @@
+use std::any::TypeId;
+
+use common::{
+ protocol::{
+ hint::MockHintWalker,
+ request_hint::{MockRequestHintVisitor, RequestHintFactory},
+ },
+ walker::MockWalker,
+};
+use mockall::predicate::eq;
+use treaty::{
+ any::{OwnedStatic, TypeNameId},
+ effect::Blocking,
+ protocol::{
+ visitor::{RequestHint, RequestHintProto, ValueKnown, ValueProto, VisitResult},
+ walker::hint::HintProto,
+ },
+ Flow,
+};
+
+use crate::common::protocol::hint::KnownFactory;
+
+mod common;
+
+#[test]
+fn hints_can_be_requested() {
+ let mut mock = MockRequestHintVisitor::<Blocking>::new();
+
+ // We will request a hint from the visitor.
+ mock.expect_request_hint().once().return_const(
+ (|_, walker| {
+ // Lookup the value protocol on the walker.
+ let obj = walker
+ .upcast_mut::<HintProto<ValueProto<OwnedStatic<i32>, Blocking>>>()
+ .unwrap();
+
+ // Get the known for the value protocol.
+ assert_eq!(
+ obj.known(&()).into_inner(),
+ Ok(ValueKnown {
+ preview: Some(&OwnedStatic(42))
+ })
+ );
+
+ // We are done.
+ VisitResult::Control(Flow::Done)
+ }) as RequestHintFactory,
+ );
+
+ // This mock becomes the visitor.
+ let visitor: &mut dyn RequestHint<Blocking> = &mut mock;
+
+ let mut mock = MockWalker::<(), ()>::new();
+
+ // The visitor will lookup the value protocol on the walker.
+ mock.expect_traits_mut()
+ .once()
+ .with(eq(TypeNameId::of::<
+ HintProto<ValueProto<OwnedStatic<i32>, Blocking>>,
+ >()))
+ .returning(|_id| {
+ let mut mock = MockHintWalker::<ValueProto<OwnedStatic<i32>, Blocking>>::new();
+
+ // Expect to give a known for the value protocol to the visitor.
+ mock.expect_known().once().return_const(
+ (|_, ()| {
+ Ok(ValueKnown {
+ preview: Some(&OwnedStatic(42)),
+ })
+ }) as KnownFactory<ValueProto<OwnedStatic<i32>, Blocking>>,
+ );
+
+ Some(Box::new(mock))
+ });
+
+ // Request a hint from the visitor.
+ assert!(matches!(
+ visitor.request_hint(&mut mock).into_inner(),
+ VisitResult::Control(Flow::Done)
+ ));
+}
+
+#[test]
+fn request_hint_proto() {
+ // The type id of the higher ranked type.
+ let id = TypeId::of::<RequestHintProto<Blocking>>();
+
+ // The type id for the lifetime containing value protocol trait object.
+ let name_id = TypeNameId::of_lower::<dyn RequestHint<Blocking> + Send + Sync>();
+
+ // They should be the same.
+ assert_eq!(id, name_id.into_type_id());
+}