From 08441ebe6c4172241656b4792a542e60b411a89d Mon Sep 17 00:00:00 2001
From: Colin Delahunty <72827203+colin99d@users.noreply.github.com>
Date: Thu, 24 Nov 2022 21:04:03 +0100
Subject: [PATCH] Update app_protocol.rs

---
 src/enums/app_protocol.rs | 41 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

diff --git a/src/enums/app_protocol.rs b/src/enums/app_protocol.rs
index 017bfc49..db1e96ea 100644
--- a/src/enums/app_protocol.rs
+++ b/src/enums/app_protocol.rs
@@ -148,4 +148,43 @@ impl AppProtocol {
         AppProtocol::TFTP,
         AppProtocol::XMPP,
     ];
-}
\ No newline at end of file
+}
+
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn from_port_to_application_protocol_ftp() {
+        let result1 = from_port_to_application_protocol(20);
+        assert_eq!(AppProtocol::FTP, result1);
+        let result2 = from_port_to_application_protocol(21);
+        assert_eq!(AppProtocol::FTP, result2);
+    }
+
+    #[test]
+    fn from_port_to_application_protocol_ssh() {
+        let result = from_port_to_application_protocol(22);
+        assert_eq!(AppProtocol::SSH, result);
+    }
+
+    #[test]
+    fn from_port_to_application_protocol_other() {
+        let result = from_port_to_application_protocol(500);
+        assert_eq!(AppProtocol::Other, result);
+    }
+    // Please let me know if you want tests on every one, happy to do it so
+
+    #[test]
+    fn app_protocol_display_ftp() {
+        let test_str = AppProtocol::FTP.to_string();
+        assert_eq!(test_str, "FTP");
+    }
+
+    #[test]
+    fn app_protocol_display_other() {
+        let test_str = AppProtocol::Other.to_string();
+        assert_eq!(test_str, "All protocols");
+    }
+}