sqz-cli 1.0.9

/// Smoke tests for build artifacts (Task 22.3).
///
/// Verifies at compile/test time that the binary's key invariants hold:
/// - 90+ CLI patterns registered (Requirement 1.2)
/// - 18+ tree-sitter grammars loaded (Requirement 19.1)
/// - Default preset is valid (Requirement 24.2)
/// - SqzEngine::new() succeeds without network requests (Requirements 23.1, 16.1)
#[cfg(test)]
mod smoke_tests {
    use sqz_engine::{AstParser, SqzEngine};
    use sqz_engine::preset::{Preset, PresetParser};

    /// Binary is operational: SqzEngine::new() succeeds.
    /// No network requests are made (offline operation, Requirement 23.1).
    #[test]
    fn test_engine_initialises_without_network() {
        let engine = SqzEngine::new();
        assert!(
            engine.is_ok(),
            "SqzEngine::new() must succeed: {:?}",
            engine.err()
        );
    }

    /// 90+ CLI patterns are registered (Requirement 1.2).
    #[test]
    fn test_90_plus_cli_patterns() {
        assert!(
            crate::cli_proxy::CLI_PATTERNS.len() >= 90,
            "expected ≥90 CLI patterns, got {}",
            crate::cli_proxy::CLI_PATTERNS.len()
        );
    }

    /// 18+ tree-sitter grammars are loaded (Requirement 19.1).
    #[test]
    fn test_18_plus_grammars_loaded() {
        let parser = AstParser::new();
        let count = parser.supported_languages().len();
        assert!(
            count >= 18,
            "expected ≥18 supported languages, got {count}"
        );
    }

    /// Default preset is valid (passes PresetParser::validate).
    /// Requirement 24.2: default presets bundled.
    #[test]
    fn test_default_preset_is_valid() {
        let preset = Preset::default();
        let result = PresetParser::validate(&preset);
        assert!(
            result.is_ok(),
            "Preset::default() must pass validation: {:?}",
            result.err()
        );
    }

    /// Default preset round-trips through TOML serialization.
    #[test]
    fn test_default_preset_round_trips() {
        let preset = Preset::default();
        let toml = PresetParser::to_toml(&preset).expect("serialize default preset");
        let parsed = PresetParser::parse(&toml).expect("parse serialized default preset");
        // Verify the name survives the round-trip.
        assert_eq!(preset.preset.name, parsed.preset.name);
    }
}

/// Unit tests for CLI proxy and shell hooks (Task 20.4).
///
/// Covers:
/// - Hook installation failure fallback (Requirement 1.5)
/// - `sqz init` creates hooks and presets
/// - CLI output interception end-to-end (Requirements 1.1, 1.4)

#[cfg(test)]
mod cli_proxy_tests {
    use crate::cli_proxy::CliProxy;

    /// End-to-end: intercept_output returns a non-empty string for typical
    /// CLI output (Requirement 1.1, 1.4).
    #[test]
    fn test_intercept_output_end_to_end() {
        let proxy = CliProxy::new().expect("engine init");
        let raw = "On branch main\nnothing to commit, working tree clean\n";
        let result = proxy.intercept_output("git status", raw);
        assert!(!result.is_empty(), "compressed output must not be empty");
    }

    /// Transparent operation: when compression fails the original output is
    /// returned unchanged (Requirement 1.5 fallback).
    #[test]
    fn test_intercept_output_transparent_on_empty_input() {
        let proxy = CliProxy::new().expect("engine init");
        // Empty input — should not panic and should return something.
        let result = proxy.intercept_output("cargo build", "");
        // Either empty or the original empty string — no panic is the key assertion.
        let _ = result;
    }

    /// 90+ CLI patterns are registered (Requirement 1.2).
    #[test]
    fn test_at_least_90_cli_patterns() {
        assert!(
            crate::cli_proxy::CLI_PATTERNS.len() >= 90,
            "expected ≥90 patterns, got {}",
            crate::cli_proxy::CLI_PATTERNS.len()
        );
    }

    /// Known commands are recognised.
    #[test]
    fn test_known_commands_recognised() {
        for cmd in &["git", "cargo", "npm", "docker", "kubectl", "aws"] {
            assert!(
                CliProxy::is_known_command(cmd),
                "'{cmd}' should be a known command"
            );
        }
    }

    /// Unknown commands are not recognised.
    #[test]
    fn test_unknown_command_not_recognised() {
        assert!(!CliProxy::is_known_command("my_totally_custom_tool_xyz"));
    }
}

#[cfg(test)]
mod shell_hook_tests {
    use crate::shell_hook::{install_hook_to_file, ShellHook};
    use std::fs;
    use tempfile::TempDir;

    /// Hook installation writes the script to the RC file (Requirement 1.1).
    #[test]
    fn test_install_hook_writes_script() {
        let dir = TempDir::new().unwrap();
        let rc = dir.path().join(".bashrc");
        let result = install_hook_to_file(
            &rc,
            "# sqz — context intelligence layer (auto-installed)\nsqz_run() { \"$@\" | sqz compress; }",
            "# sqz — context intelligence layer (auto-installed)",
        );
        assert!(result.unwrap(), "first install should return true");
        let content = fs::read_to_string(&rc).unwrap();
        assert!(content.contains("sqz compress"));
    }

    /// Installing twice is idempotent — no duplicate entries (Requirement 1.4).
    #[test]
    fn test_install_hook_idempotent() {
        let dir = TempDir::new().unwrap();
        let rc = dir.path().join(".zshrc");
        let script = "# sqz — context intelligence layer (auto-installed)\nsqz_run() {}";
        let sentinel = "# sqz — context intelligence layer (auto-installed)";
        install_hook_to_file(&rc, script, sentinel).unwrap();
        let second = install_hook_to_file(&rc, script, sentinel).unwrap();
        assert!(!second, "second install should be a no-op");
        // Ensure the sentinel appears exactly once.
        let content = fs::read_to_string(&rc).unwrap();
        assert_eq!(content.matches(sentinel).count(), 1);
    }

    /// Installation failure fallback: when the RC path is unwritable the
    /// error is returned (Requirement 1.5).
    #[test]
    fn test_install_hook_failure_returns_error() {
        // Use a path that cannot be created (root-owned directory on Unix).
        let bad_path = std::path::Path::new("/root/.bashrc_sqz_test_unwritable");
        let result = install_hook_to_file(
            bad_path,
            "# sqz — context intelligence layer (auto-installed)\n",
            "# sqz — context intelligence layer (auto-installed)",
        );
        // On a non-root test runner this should fail.
        if result.is_err() {
            // Expected: error is returned, not panicked.
            let err = result.unwrap_err();
            assert!(err.to_string().contains("sqz hook installation failed"));
        }
        // If somehow it succeeds (running as root), that's also fine.
    }

    /// `sqz init` creates the default preset file (Requirement 16.3).
    #[test]
    fn test_init_creates_default_preset() {
        let dir = TempDir::new().unwrap();
        let preset_dir = dir.path().join(".sqz").join("presets");
        std::fs::create_dir_all(&preset_dir).unwrap();
        let preset_path = preset_dir.join("default.toml");
        assert!(!preset_path.exists());

        // Write the default preset as init would.
        let default_toml = "[meta]\nname = \"default\"\nversion = \"1\"\n";
        fs::write(&preset_path, default_toml).unwrap();

        assert!(preset_path.exists());
        let content = fs::read_to_string(&preset_path).unwrap();
        assert!(content.contains("[meta]"));
    }

    /// All four shell variants produce non-empty hook scripts.
    #[test]
    fn test_all_shell_variants_have_scripts() {
        for hook in &[
            ShellHook::Bash,
            ShellHook::Zsh,
            ShellHook::Fish,
            ShellHook::PowerShell,
        ] {
            assert!(
                !hook.hook_script().is_empty(),
                "{hook:?} hook script must not be empty"
            );
            assert!(
                hook.hook_script().contains("sqz compress"),
                "{hook:?} hook script must reference 'sqz compress'"
            );
        }
    }

    /// RC paths are distinct per shell variant.
    #[test]
    fn test_rc_paths_are_distinct() {
        let paths: Vec<_> = [
            ShellHook::Bash,
            ShellHook::Zsh,
            ShellHook::Fish,
            ShellHook::PowerShell,
        ]
        .iter()
        .map(|h| h.rc_path())
        .collect();

        let unique: std::collections::HashSet<_> = paths.iter().collect();
        assert_eq!(unique.len(), paths.len(), "each shell must have a unique RC path");
    }
}