use wafrift_types::{ConnectionBehavior, Signal};
#[must_use]
pub fn classify_connection(behavior: ConnectionBehavior) -> Signal {
Signal::ConnectionBehavior(behavior)
}
#[must_use]
pub fn tcp_reset() -> Signal {
Signal::ConnectionBehavior(ConnectionBehavior::TcpReset)
}
#[must_use]
pub fn ok_with_block_page() -> Signal {
Signal::ConnectionBehavior(ConnectionBehavior::OkWithBlockPage)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn tcp_reset_signal() {
let s = tcp_reset();
assert!(matches!(
s,
Signal::ConnectionBehavior(ConnectionBehavior::TcpReset)
));
}
#[test]
fn ok_block_page_signal() {
let s = ok_with_block_page();
assert!(matches!(
s,
Signal::ConnectionBehavior(ConnectionBehavior::OkWithBlockPage)
));
}
}