Mole/tests/update_manager.bats

#!/usr/bin/env bats

setup_file() {
    PROJECT_ROOT="$(cd "${BATS_TEST_DIRNAME}/.." && pwd)"
    export PROJECT_ROOT

    ORIGINAL_HOME="${HOME:-}"
    export ORIGINAL_HOME

    HOME="$(mktemp -d "${BATS_TEST_DIRNAME}/tmp-update-manager.XXXXXX")"
    export HOME

    mkdir -p "$HOME"
}

teardown_file() {
    rm -rf "$HOME"
    if [[ -n "${ORIGINAL_HOME:-}" ]]; then
        export HOME="$ORIGINAL_HOME"
    fi
}

setup() {
    source "$PROJECT_ROOT/lib/common.sh"
    source "$PROJECT_ROOT/lib/update_manager.sh"
}

# Test brew_has_outdated function
@test "brew_has_outdated returns 1 when brew not installed" {
    function brew() {
        return 127  # Command not found
    }
    export -f brew

    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; brew_has_outdated"
    [ "$status" -eq 1 ]
}

@test "brew_has_outdated checks formula by default" {
    # Mock brew to simulate outdated formulas
    function brew() {
        if [[ "$1" == "outdated" && "$2" != "--cask" ]]; then
            echo "package1"
            echo "package2"
            return 0
        fi
        return 1
    }
    export -f brew

    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; brew_has_outdated"
    [ "$status" -eq 0 ]
}

@test "brew_has_outdated checks casks when specified" {
    # Mock brew to simulate outdated casks
    function brew() {
        if [[ "$1" == "outdated" && "$2" == "--cask" ]]; then
            echo "app1"
            return 0
        fi
        return 1
    }
    export -f brew

    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; brew_has_outdated cask"
    [ "$status" -eq 0 ]
}

# Test format_brew_update_label function
@test "format_brew_update_label returns empty when no updates" {
    result=$(BREW_OUTDATED_COUNT=0 bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; format_brew_update_label")
    [[ -z "$result" ]]
}

@test "format_brew_update_label formats with formula and cask counts" {
    result=$(BREW_OUTDATED_COUNT=5 BREW_FORMULA_OUTDATED_COUNT=3 BREW_CASK_OUTDATED_COUNT=2 bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; format_brew_update_label")
    [[ "$result" =~ "3 formula" ]]
    [[ "$result" =~ "2 cask" ]]
}

@test "format_brew_update_label shows total when breakdown unavailable" {
    result=$(BREW_OUTDATED_COUNT=5 bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; format_brew_update_label")
    [[ "$result" =~ "5 updates" ]]
}

# Test ask_for_updates function
@test "ask_for_updates returns 1 when no updates available" {
    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; ask_for_updates < /dev/null"
    [ "$status" -eq 1 ]
}

@test "ask_for_updates detects Homebrew updates" {
    # Mock environment with Homebrew updates
    export BREW_OUTDATED_COUNT=5
    export BREW_FORMULA_OUTDATED_COUNT=3
    export BREW_CASK_OUTDATED_COUNT=2

    # Use input redirection to simulate ESC (cancel)
    run bash -c "printf '\x1b' | source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; ask_for_updates"
    # Should show updates and ask for confirmation
    [ "$status" -eq 1 ]  # ESC cancels
}

@test "ask_for_updates detects App Store updates" {
    export APPSTORE_UPDATE_COUNT=3

    run bash -c "printf '\x1b' | source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; ask_for_updates"
    [ "$status" -eq 1 ]  # ESC cancels
}

@test "ask_for_updates detects macOS updates" {
    export MACOS_UPDATE_AVAILABLE=true

    run bash -c "printf '\x1b' | source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; ask_for_updates"
    [ "$status" -eq 1 ]  # ESC cancels
}

@test "ask_for_updates detects Mole updates" {
    export MOLE_UPDATE_AVAILABLE=true

    run bash -c "printf '\x1b' | source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; ask_for_updates"
    [ "$status" -eq 1 ]  # ESC cancels
}

# Test perform_updates function structure
@test "perform_updates handles brew formula updates" {
    # Mock brew to avoid actual updates
    function brew() {
        case "$1" in
            outdated)
                if [[ "${2:-}" == "--cask" ]]; then
                    return 1  # No cask updates
                else
                    echo "test-package"
                    return 0  # Has formula updates
                fi
                ;;
            upgrade)
                echo "Upgrading test-package..."
                return 0
                ;;
            *)
                return 1
                ;;
        esac
    }
    export -f brew
    export BREW_OUTDATED_COUNT=1
    export BREW_FORMULA_OUTDATED_COUNT=1
    export BREW_CASK_OUTDATED_COUNT=0

    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; perform_updates"
    [ "$status" -eq 0 ]
}

# Test update_homebrew function
@test "update_homebrew returns early when no updates" {
    function brew() {
        case "$1" in
            outdated)
                return 1  # No updates
                ;;
            *)
                return 1
                ;;
        esac
    }
    export -f brew

    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; update_homebrew"
    [ "$status" -eq 0 ]
}

@test "update_homebrew handles network failures gracefully" {
    function brew() {
        case "$1" in
            outdated)
                if [[ "${2:-}" == "--quiet" ]]; then
                    echo "test-package"
                    return 0
                fi
                ;;
            upgrade)
                return 1  # Simulate failure
                ;;
            *)
                return 1
                ;;
        esac
    }
    export -f brew

    run bash -c "source '$PROJECT_ROOT/lib/common.sh'; source '$PROJECT_ROOT/lib/update_manager.sh'; update_homebrew"
    # Should handle failure without crashing
    [ "$status" -eq 0 ] || [ "$status" -eq 1 ]
}