feat(elfo-test): improve test utils

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!(
+ "unexpected message: expected {}, got: {:?}",
+ extract_name_by_type::<T>(),
+ msg
+ ),
+ })
+}
+
+#[track_caller]
+pub fn expect_request<T: Request>(msg: Envelope) -> (T, ResponseToken<T>) {
+ msg!(match msg {
+ (msg @ T, token) => (msg, token),
+ msg => panic!(
+ "unexpected request: expected {}, got: {:?}",
+ extract_name_by_type::<T>(),
+ msg
+ ),
+ })
+}

elfo-test/src/lib.rs

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

elfo-test/src/proxy.rs

@@ -224,7 +224,14 @@ fn testers(tx: shared::OneshotSender<ProxyContext>) -> Blueprint {
  })
 }
 
-pub async fn proxy(blueprint: Blueprint, config: impl for<'de> Deserializer<'de>) -> Proxy {
+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 +251,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 +278,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::*;