From 1b8d85e1291fc9a7c47f0375e0906827ef245b1f Mon Sep 17 00:00:00 2001
From: Xevion <xevion@xevion.dev>
Date: Tue, 1 Oct 2024 17:38:32 -0500
Subject: [PATCH] Add tests

---
 src/main.rs | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index 8b4257a..a62f723 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -22,3 +22,58 @@ fn main() {
         username, hostname, os, arch, working_directory, execution_path
     );
 }
+
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_os_arch() {
+        let os = env::consts::OS;
+        let arch = env::consts::ARCH;
+        
+        // Make sure that OS and ARCH are not empty
+        assert!(!os.is_empty());
+        assert!(!arch.is_empty());
+    }
+
+    #[test]
+    fn test_username() {
+        let username = env::var("USER").unwrap_or_else(|_| "unknown".to_string());
+        
+        // In CI environments, USER might not be set, so allow "unknown"
+        assert!(!username.is_empty());
+    }
+
+    #[test]
+    fn test_hostname() {
+        let hostname = Command::new("hostname")
+            .output()
+            .map(|output| String::from_utf8_lossy(&output.stdout).trim().to_string())
+            .unwrap_or_else(|_| "unknown".to_string());
+        
+        // Since hostname might vary, just ensure it's not empty
+        assert!(!hostname.is_empty());
+    }
+
+    #[test]
+    fn test_execution_path() {
+        let execution_path = env::current_exe()
+            .map(|path| path.display().to_string())
+            .unwrap_or_else(|_| "unknown".to_string());
+
+        // Ensure execution path is not "unknown" when it's expected to resolve
+        assert!(execution_path != "unknown");
+    }
+
+    #[test]
+    fn test_working_directory() {
+        let working_directory = env::current_dir()
+            .map(|path| path.display().to_string())
+            .unwrap_or_else(|_| "unknown".to_string());
+        
+        // Ensure the working directory is a valid path
+        assert!(working_directory != "unknown");
+    }
+}
\ No newline at end of file