From fe16a923190243dfde5db6ceff2ef0bcf9158926 Mon Sep 17 00:00:00 2001 From: Toby Vincent Date: Tue, 1 Oct 2024 13:15:24 -0500 Subject: feat: simplify service status type --- src/service/tcp.rs | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'src/service/tcp.rs') diff --git a/src/service/tcp.rs b/src/service/tcp.rs index 42791bc..7b79afd 100644 --- a/src/service/tcp.rs +++ b/src/service/tcp.rs @@ -32,7 +32,7 @@ impl ServiceSpawner for Tcp { Ok(conn) => { // TODO: figure out how to wait for connection to close conn.ready(Interest::READABLE).await?; - tx.send_if_modified(|s| s.update(Status::Pass)); + tx.send_if_modified(|s| s.update(Status::Ok)); } Err(err) => { tx.send_if_modified(|s| s.update(err.into())); @@ -50,18 +50,18 @@ mod tests { #[tracing_test::traced_test] #[ignore] async fn test_tcp_watch() { - let (tx, mut rx) = tokio::sync::watch::channel(Status::default()); + let (tx, mut rx) = tokio::sync::watch::channel(Status::Error(None)); let tests = tokio::spawn(async move { - assert!(matches!(*rx.borrow_and_update(), Status::Unknown)); + assert!(matches!(*rx.borrow_and_update(), Status::Error(None))); rx.changed().await.unwrap(); - assert!(matches!(*rx.borrow_and_update(), Status::Pass)); + assert!(matches!(*rx.borrow_and_update(), Status::Ok)); rx.changed().await.unwrap(); assert_eq!( *rx.borrow_and_update(), - Status::Fail(Some(String::from("Disconnected"))) + Status::Error(Some(String::from("Disconnected"))) ); }); -- cgit v1.2.3-70-g09d2