diff --git a/src/utils/services.rs b/src/utils/services.rs index 9a5a5be..0ea3f63 100644 --- a/src/utils/services.rs +++ b/src/utils/services.rs @@ -119,3 +119,17 @@ async fn check_service(hostname: &str, port: &u32) -> Result<(), CustomError> { Err(_) => Err(CustomError::Fatal), } } + + +#[cfg(test)] +mod service_unittests { + use super::check_service; + #[tokio::test] + async fn check_available_service() { + assert!(check_service("ya.ru", &443).await.is_ok()); + } + #[tokio::test] + async fn check_unavailable_service() { + assert!(check_service("unavailable.service", &1111).await.is_err()); + } +} \ No newline at end of file