feat(elfo-test): improve test utils

elfo-test/Cargo.toml

@@ -21,3 +21,4 @@ futures-intrusive = "0.5"
 futures = "0.3.12"
 tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 once_cell = { version = "1.8.0" }
+stability = "0.1.1"

elfo-test/src/expect.rs

@@ -0,0 +1,25 @@
+use elfo_core::{dumping::extract_name_by_type, msg, Envelope, Message, Request, ResponseToken};
+
+#[track_caller]
+pub fn expect_message<T: Message>(msg: Envelope) -> T {
+    msg!(match msg {
+        msg @ T => msg,
+        msg => panic!(
+            r#"unexpected message: expected "{}", got "{}""#,
+            extract_name_by_type::<T>(),
+            msg.message().name()
+        ),
+    })
+}
+
+#[track_caller]
+pub fn expect_request<T: Request>(msg: Envelope) -> (T, ResponseToken<T>) {
+    msg!(match msg {
+        (msg @ T, token) => (msg, token),
+        msg => panic!(
+            r#"unexpected request: expected "{}", got "{}""#,
+            extract_name_by_type::<T>(),
+            msg.message().name()
+        ),
+    })
+}

elfo-test/src/lib.rs

@@ -1,5 +1,10 @@
 #![warn(rust_2018_idioms, unreachable_pub)]
 
+pub use expect::{expect_message, expect_request};
 pub use proxy::{proxy, Proxy};
 
+#[cfg(feature = "unstable")]
+pub use proxy::proxy_with_route;
+
+mod expect;
 mod proxy;

elfo-test/src/proxy.rs

@@ -224,7 +224,16 @@ fn testers(tx: shared::OneshotSender<ProxyContext>) -> Blueprint {
         })
 }
 
-pub async fn proxy(blueprint: Blueprint, config: impl for<'de> Deserializer<'de>) -> Proxy {
+#[doc(hidden)]
+#[stability::unstable]
+pub async fn proxy_with_route<F>(
+    blueprint: Blueprint,
+    route_filter: F,
+    config: impl for<'de> Deserializer<'de>,
+) -> Proxy
+where
+    F: Fn(&Envelope) -> bool + Send + Sync + 'static,
+{
     let _ = tracing_subscriber::fmt()
         .with_target(false)
         .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
@@ -244,7 +253,7 @@ pub async fn proxy(blueprint: Blueprint, config: impl for<'de> Deserializer<'de>
     let subject_addr = subject.addr();
 
     testers.route_all_to(&subject);
-    subject.route_all_to(&testers);
+    subject.route_to(&testers, route_filter);
 
     // TODO: capture log messages.
     // TODO: capture metrics.
@@ -271,6 +280,10 @@ pub async fn proxy(blueprint: Blueprint, config: impl for<'de> Deserializer<'de>
     }
 }
 
+pub async fn proxy(blueprint: Blueprint, config: impl for<'de> Deserializer<'de>) -> Proxy {
+    proxy_with_route(blueprint, |_| true, config).await
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;