From 858323e6293dfff4a68bbefc1003e1e624481a25 Mon Sep 17 00:00:00 2001
From: prplV <maseeeeeeeed@gmail.com>
Date: Wed, 30 Oct 2024 16:37:12 +0300
Subject: [PATCH] services unit tests

---
 src/utils/services.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

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