Move all unit tests from src/ to tests/ directory and add 99 unit tests

- Moved all #[cfg(test)] blocks from src/ modules to corresponding tests/ files
- Created new test files: test_cli.rs and test_color_graph.rs
- Added comprehensive unit tests for helper functions across all modules
- Now have 99 total unit tests organized in tests/ directory
- Updated health.rs to make is_git_repo() function public with path parameter
- All unit tests properly test extracted helper functions for better coverage
- Improved code organization with clear separation between source and tests

Unit test breakdown by module:
- test_cli.rs: 14 tests (argument parsing)
- test_summary.rs: 13 tests (git log parsing, formatting)
- test_rename_branch.rs: 11 tests (branch operations)
- test_prune_branches.rs: 11 tests (branch protection, deletion)
- test_clean_branches.rs: 10 tests (merged branch cleanup)
- test_health.rs: 10 tests (repository health checks)
- test_what.rs: 8 tests (diff analysis, commit parsing)
- test_info.rs: 6 tests (repository information)
- test_color_graph.rs: 4 tests (colorized git log)
- test_graph.rs: 4 tests (git log formatting)
- test_since.rs: 4 tests (commit range analysis)
- test_undo.rs: 4 tests (git reset operations)

Author: simeg

src/clean_branches.rs

@@ -86,68 +86,3 @@ pub fn format_deletion_summary(count: usize, dry_run: bool) -> String {
     }
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn test_get_git_branch_args() {
-        assert_eq!(get_git_branch_args(), ["branch", "--merged"]);
-    }
-
-    #[test]
-    fn test_get_protected_branches() {
-        let protected = get_protected_branches();
-        assert_eq!(protected, vec!["main", "master", "develop"]);
-    }
-
-    #[test]
-    fn test_clean_branch_name() {
-        assert_eq!(clean_branch_name("  feature/test  "), "feature/test");
-        assert_eq!(clean_branch_name("* main"), "main");
-        assert_eq!(clean_branch_name("  * develop  "), "develop");
-        assert_eq!(clean_branch_name("bugfix/123"), "bugfix/123");
-    }
-
-    #[test]
-    fn test_is_protected_branch() {
-        assert!(is_protected_branch("main"));
-        assert!(is_protected_branch("master"));
-        assert!(is_protected_branch("develop"));
-        assert!(!is_protected_branch("feature/test"));
-        assert!(!is_protected_branch("hotfix/123"));
-    }
-
-    #[test]
-    fn test_get_git_delete_args() {
-        assert_eq!(
-            get_git_delete_args("feature"),
-            vec!["branch".to_string(), "-d".to_string(), "feature".to_string()]
-        );
-    }
-
-    #[test]
-    fn test_format_dry_run_message() {
-        assert_eq!(
-            format_dry_run_message("feature/test"),
-            "(dry run) Would delete: feature/test"
-        );
-    }
-
-    #[test]
-    fn test_format_no_branches_message() {
-        assert_eq!(format_no_branches_message(), "No merged branches to delete.");
-    }
-
-    #[test]
-    fn test_format_deletion_summary() {
-        assert_eq!(
-            format_deletion_summary(3, true),
-            "🧪 (dry run) 3 branches would be deleted:"
-        );
-        assert_eq!(
-            format_deletion_summary(2, false),
-            "🧹 Deleted 2 merged branches:"
-        );
-    }
-}

src/color_graph.rs

@@ -53,59 +53,3 @@ pub fn format_color_git_error(stderr: &str) -> String {
     format!("❌ git log failed:\n{stderr}")
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
-    use std::process::{Output, ExitStatus};
-
-    #[test]
-    fn test_get_color_git_log_args() {
-        let args = get_color_git_log_args();
-        assert_eq!(args[0], "log");
-        assert_eq!(args[1], "--oneline");
-        assert_eq!(args[2], "--graph");
-        assert_eq!(args[3], "--decorate");
-        assert_eq!(args[4], "--all");
-        assert_eq!(args[5], "--color=always");
-        assert!(args[6].contains("--pretty=format:"));
-        assert!(args[6].contains("%C(auto)"));
-    }
-
-    #[test]
-    fn test_format_color_git_error() {
-        assert_eq!(
-            format_color_git_error("not a git repository"),
-            "❌ git log failed:\nnot a git repository"
-        );
-        assert_eq!(
-            format_color_git_error("permission denied"),
-            "❌ git log failed:\npermission denied"
-        );
-    }
-
-    #[test]
-    fn test_is_command_successful() {
-        use std::os::unix::process::ExitStatusExt;
-        
-        let success_output = Output {
-            status: ExitStatus::from_raw(0),
-            stdout: vec![],
-            stderr: vec![],
-        };
-        assert!(is_command_successful(&success_output));
-
-        let failure_output = Output {
-            status: ExitStatus::from_raw(256), // Exit code 1
-            stdout: vec![],
-            stderr: vec![],
-        };
-        assert!(!is_command_successful(&failure_output));
-    }
-
-    #[test]
-    fn test_convert_output_to_string() {
-        assert_eq!(convert_output_to_string(b"hello world"), "hello world");
-        assert_eq!(convert_output_to_string(b""), "");
-        assert_eq!(convert_output_to_string(b"git log output"), "git log output");
-    }
-}

src/graph.rs

@@ -25,27 +25,3 @@ pub fn format_git_error(stderr: &str) -> String {
     format!("❌ git log failed:\n{stderr}")
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn test_get_git_log_args() {
-        assert_eq!(
-            get_git_log_args(),
-            ["log", "--oneline", "--graph", "--decorate", "--all"]
-        );
-    }
-
-    #[test]
-    fn test_format_git_error() {
-        assert_eq!(
-            format_git_error("not a git repository"),
-            "❌ git log failed:\nnot a git repository"
-        );
-        assert_eq!(
-            format_git_error("unknown revision"),
-            "❌ git log failed:\nunknown revision"
-        );
-    }
-}

src/health.rs

@@ -12,7 +12,7 @@ pub fn run() {
     println!();
 
     // Check if we're in a git repository
-    if !is_git_repo() {
+    if !is_git_repo(&std::env::current_dir().unwrap_or_else(|_| ".".into())) {
         println!("{} Not in a Git repository", red.apply_to("✗"));
         return;
     }
@@ -36,9 +36,10 @@ pub fn run() {
     println!("{}", bold.apply_to("Health check complete!"));
 }
 
-fn is_git_repo() -> bool {
+pub fn is_git_repo(path: &std::path::Path) -> bool {
     Command::new("git")
         .args(["rev-parse", "--git-dir"])
+        .current_dir(path)
         .output()
         .map(|output| output.status.success())
         .unwrap_or(false)
@@ -167,30 +168,4 @@ fn check_uncommitted_changes(green: &Style, yellow: &Style, _red: &Style) {
     }
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
 
-    #[test]
-    fn test_is_git_repo_returns_false_for_non_git_dir() {
-        // This test creates a temporary directory that's not a git repo
-        // and verifies that is_git_repo() correctly returns false
-        let temp_dir = std::env::temp_dir();
-        let original_dir = std::env::current_dir().unwrap();
-
-        // Change to temp directory (should not be a git repo)
-        std::env::set_current_dir(&temp_dir).unwrap();
-
-        // Test - this might fail if temp dir is somehow in a git repo
-        // So let's just test the basic functionality
-        let result = is_git_repo();
-
-        // Restore original directory
-        std::env::set_current_dir(original_dir).unwrap();
-
-        // The result depends on whether temp dir is in a git repo or not
-        // This test mainly ensures the function doesn't panic
-        // We don't assert a specific value since temp dir might be in git repo
-        let _ = result;
-    }
-}

src/info.rs

@@ -79,45 +79,4 @@ pub fn format_tracking_branch(tracking_raw: &str) -> String {
     }
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
 
-    #[test]
-    fn test_extract_repo_name() {
-        assert_eq!(extract_repo_name("/path/to/my-repo"), "my-repo");
-        assert_eq!(extract_repo_name("/home/user/git-x"), "git-x");
-        assert_eq!(extract_repo_name("relative-path"), "relative-path");
-        assert_eq!(extract_repo_name(""), "unknown");
-    }
-
-    #[test]
-    fn test_parse_ahead_behind_counts() {
-        assert_eq!(
-            parse_ahead_behind_counts("3 2"),
-            ("3".to_string(), "2".to_string())
-        );
-        assert_eq!(
-            parse_ahead_behind_counts("0 0"),
-            ("0".to_string(), "0".to_string())
-        );
-        assert_eq!(
-            parse_ahead_behind_counts("5"),
-            ("5".to_string(), "0".to_string())
-        );
-        assert_eq!(
-            parse_ahead_behind_counts(""),
-            ("0".to_string(), "0".to_string())
-        );
-    }
-
-    #[test]
-    fn test_format_tracking_branch() {
-        assert_eq!(format_tracking_branch("origin/main"), "origin/main");
-        assert_eq!(format_tracking_branch(""), "(no upstream)");
-        assert_eq!(
-            format_tracking_branch("upstream/develop"),
-            "upstream/develop"
-        );
-    }
-}

src/prune_branches.rs

@@ -115,81 +115,4 @@ pub fn format_no_branches_to_prune_message() -> &'static str {
     "✅ No merged branches to prune."
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
 
-    #[test]
-    fn test_get_default_protected_branches() {
-        assert_eq!(get_default_protected_branches(), vec!["main", "master", "develop"]);
-    }
-
-    #[test]
-    fn test_parse_except_branches() {
-        assert_eq!(
-            parse_except_branches("feature,hotfix"),
-            vec!["feature".to_string(), "hotfix".to_string()]
-        );
-        assert_eq!(
-            parse_except_branches("  branch1  ,  branch2  "),
-            vec!["branch1".to_string(), "branch2".to_string()]
-        );
-        assert_eq!(parse_except_branches("single"), vec!["single".to_string()]);
-        assert_eq!(parse_except_branches(""), Vec::<String>::new());
-    }
-
-    #[test]
-    fn test_get_all_protected_branches() {
-        let default_only = get_all_protected_branches(None);
-        assert_eq!(default_only, vec!["main", "master", "develop"]);
-        
-        let with_except = get_all_protected_branches(Some("feature,hotfix"));
-        assert_eq!(with_except, vec!["main", "master", "develop", "feature", "hotfix"]);
-    }
-
-    #[test]
-    fn test_clean_git_branch_name() {
-        assert_eq!(clean_git_branch_name("  feature/test  "), "feature/test");
-        assert_eq!(clean_git_branch_name("* main"), "main");
-        assert_eq!(clean_git_branch_name("develop"), "develop");
-    }
-
-    #[test]
-    fn test_is_branch_protected() {
-        let protected = vec!["main".to_string(), "develop".to_string()];
-        
-        assert!(is_branch_protected("main", "current", &protected));
-        assert!(is_branch_protected("develop", "current", &protected));
-        assert!(is_branch_protected("current", "current", &protected));
-        assert!(!is_branch_protected("feature", "current", &protected));
-    }
-
-    #[test]
-    fn test_get_git_branch_delete_args() {
-        assert_eq!(
-            get_git_branch_delete_args("feature"),
-            ["branch".to_string(), "-d".to_string(), "feature".to_string()]
-        );
-    }
-
-    #[test]
-    fn test_format_branch_deleted_message() {
-        assert_eq!(
-            format_branch_deleted_message("feature/test"),
-            "🧹 Deleted merged branch 'feature/test'"
-        );
-    }
-
-    #[test]
-    fn test_format_branch_delete_failed_message() {
-        assert_eq!(
-            format_branch_delete_failed_message("feature/test"),
-            "⚠️ Failed to delete branch 'feature/test'"
-        );
-    }
-
-    #[test]
-    fn test_format_no_branches_to_prune_message() {
-        assert_eq!(format_no_branches_to_prune_message(), "✅ No merged branches to prune.");
-    }
-}