#!/usr/bin/env bats
# Tests for lib/detect.sh — hardware and config detection

load test_helper.sh

setup() {
    source_lib common.sh
    source_lib format.sh
    source_lib detect.sh
    setup_mock_sysfs
    # Override GPU_SYSFS AFTER sourcing detect.sh (which sets it at load time)
    GPU_SYSFS="$MOCK_SYSFS/class/drm/card0/device"
}

teardown() {
    teardown_mock_sysfs
}

# ── GPU sysfs reads ──────────────────────────────────────

@test "detect_vram_total: reads mocked value" {
    echo "34359738368" > "$GPU_SYSFS/mem_info_vram_total"
    run detect_vram_total
    assert_output "34359738368"
}

@test "detect_vram_total: returns 0 when sysfs file missing" {
    rm -f "$GPU_SYSFS/mem_info_vram_total"
    run detect_vram_total
    assert_output "0"
}

@test "detect_gtt_total: reads mocked value" {
    echo "64424509440" > "$GPU_SYSFS/mem_info_gtt_total"
    run detect_gtt_total
    assert_output "64424509440"
}

@test "detect_gpu_busy: reads percentage" {
    echo "42" > "$GPU_SYSFS/gpu_busy_percent"
    run detect_gpu_busy
    assert_output "42"
}

@test "detect_gpu_temp: reads millidegrees" {
    echo "55000" > "$GPU_SYSFS/hwmon/hwmon0/temp1_input"
    run detect_gpu_temp
    assert_output "55000"
}

@test "detect_gpu_power: reads microwatts" {
    echo "30000000" > "$GPU_SYSFS/hwmon/hwmon0/power1_average"
    run detect_gpu_power
    assert_output "30000000"
}

@test "detect_gpu_device_id: reads and strips 0x prefix" {
    echo "0x1586" > "$GPU_SYSFS/device"
    run detect_gpu_device_id
    assert_output "1586"
}

@test "detect_gpu_busy: returns 0 when sysfs file missing" {
    rm -f "$GPU_SYSFS/gpu_busy_percent"
    run detect_gpu_busy
    assert_output "0"
}

@test "detect_gpu_temp: returns 0 when hwmon missing" {
    rm -f "$GPU_SYSFS/hwmon/hwmon0/temp1_input"
    run detect_gpu_temp
    assert_output "0"
}

# ── Kernel param detection ───────────────────────────────
# These tests redefine detect_kernel_param inline to control /proc/cmdline content

_mock_kernel_param() {
    local param="$1" cmdline="$2"
    local pattern="${param//./\\.}"
    if [[ "$cmdline" =~ (^|[[:space:]])${pattern}=([^ ]+) ]]; then
        echo "${BASH_REMATCH[2]}"
    elif [[ "$cmdline" =~ (^|[[:space:]])${pattern}([[:space:]]|$) ]]; then
        echo "present"
    fi
}

@test "detect_kernel_param: extracts iommu=pt" {
    run _mock_kernel_param 'iommu' "BOOT_IMAGE=/boot/vmlinuz root=UUID=abc ro iommu=pt quiet"
    assert_output "pt"
}

@test "detect_kernel_param: extracts amdgpu.gttsize value" {
    run _mock_kernel_param 'amdgpu.gttsize' "BOOT_IMAGE=/boot/vmlinuz iommu=pt amdgpu.gttsize=61440 ttm.pages_limit=15728640 quiet"
    assert_output "61440"
}

@test "detect_kernel_param: returns empty when param missing" {
    run _mock_kernel_param 'iommu' "BOOT_IMAGE=/boot/vmlinuz root=UUID=abc ro quiet"
    assert_output ""
}

@test "detect_kernel_param: iommu does NOT match amd_iommu (word boundary)" {
    run _mock_kernel_param 'iommu' "BOOT_IMAGE=/boot/vmlinuz amd_iommu=off quiet"
    assert_output ""
}

@test "detect_kernel_param: amdgpu.gttsize does NOT match xamdgpu.gttsize" {
    run _mock_kernel_param 'amdgpu.gttsize' "BOOT_IMAGE=/boot/vmlinuz xamdgpu.gttsize=99999 quiet"
    assert_output ""
}

@test "detect_kernel_param: dot in param name is literal not wildcard" {
    run _mock_kernel_param 'amdgpu.gttsize' "BOOT_IMAGE=/boot/vmlinuz amdgpuXgttsize=99999 quiet"
    assert_output ""
}

@test "detect_kernel_param: param at start of cmdline (no leading space)" {
    run _mock_kernel_param 'iommu' "iommu=pt root=UUID=abc ro quiet"
    assert_output "pt"
}

@test "detect_kernel_param: param at end of cmdline (no trailing space)" {
    run _mock_kernel_param 'iommu' "BOOT_IMAGE=/boot/vmlinuz iommu=pt"
    assert_output "pt"
}

@test "detect_kernel_param: boolean param without value" {
    run _mock_kernel_param 'quiet' "BOOT_IMAGE=/boot/vmlinuz iommu=pt quiet"
    assert_output "present"
}

@test "detect_kernel_param: param with equals in value" {
    run _mock_kernel_param 'root' "BOOT_IMAGE=/boot/vmlinuz root=UUID=abc-def-123"
    assert_output "UUID=abc-def-123"
}

# ── Recommended values ───────────────────────────────────

@test "recommended_gttsize_mib: 64 GiB system" {
    # Override detection functions AFTER source
    detect_system_ram_kb() { echo "33554432"; }  # 32 GiB
    detect_vram_total() { echo "34359738368"; }   # 32 GiB
    run recommended_gttsize_mib
    # total = (33554432 + 33554432) KB = 67108864 KB → 64 GiB → 60 GiB GTT → 61440 MiB
    assert_output "61440"
}

@test "recommended_gttsize_mib: 128 GiB system" {
    detect_system_ram_kb() { echo "130023424"; }  # ~124 GiB
    detect_vram_total() { echo "536870912"; }      # 0.5 GiB
    run recommended_gttsize_mib
    # ~124.5 GiB total → integer: 124 GiB → 120 GiB GTT → 122880 MiB
    assert_output "122880"
}

@test "recommended_gttsize_mib: tiny system (2 GiB) floors at 1 GiB" {
    detect_system_ram_kb() { echo "2097152"; }
    detect_vram_total() { echo "0"; }
    run recommended_gttsize_mib
    assert_output "1024"
}

@test "recommended_gttsize_mib: zero RAM floors at 1 GiB" {
    detect_system_ram_kb() { echo "0"; }
    detect_vram_total() { echo "0"; }
    run recommended_gttsize_mib
    assert_output "1024"
}

@test "recommended_gttsize_mib: exactly 4 GiB system floors at 1 GiB" {
    detect_system_ram_kb() { echo "4194304"; }  # 4 GiB
    detect_vram_total() { echo "0"; }
    run recommended_gttsize_mib
    # 4 - 4 = 0 → floored to 1 → 1024
    assert_output "1024"
}

@test "recommended_gttsize_mib: 5 GiB system → 1 GiB GTT" {
    detect_system_ram_kb() { echo "5242880"; }  # 5 GiB
    detect_vram_total() { echo "0"; }
    run recommended_gttsize_mib
    # 5 - 4 = 1 → 1024 MiB
    assert_output "1024"
}

@test "recommended_pages_limit: matches gttsize * 256" {
    detect_system_ram_kb() { echo "33554432"; }
    detect_vram_total() { echo "34359738368"; }
    local gttsize
    gttsize="$(recommended_gttsize_mib)"
    run recommended_pages_limit
    assert_output "$(( gttsize * 256 ))"
}

# ── Firmware detection ───────────────────────────────────

@test "detect_firmware_bad: returns true for known bad version" {
    detect_firmware_version() { echo "20251125-1"; }
    run detect_firmware_bad
    assert_success
}

@test "detect_firmware_bad: returns false for good version" {
    detect_firmware_version() { echo "20260309-1"; }
    run detect_firmware_bad
    assert_failure
}

@test "detect_firmware_bad: returns false for empty version" {
    detect_firmware_version() { echo "unknown"; }
    run detect_firmware_bad
    assert_failure
}

# ── Stack detection ──────────────────────────────────────

@test "detect_stack_ollama: reports missing when not installed" {
    is_cmd() { return 1; }
    run detect_stack_ollama
    assert_output "missing"
}

@test "detect_stack_ollama: reports installed when available" {
    is_cmd() { [[ "$1" == "ollama" ]] && return 0 || return 1; }
    run detect_stack_ollama
    assert_output "installed"
}

# ── detect_system_ram_kb ─────────────────────────────────

@test "detect_system_ram_kb: returns 0 on missing /proc/meminfo" {
    # Temporarily override to read from nonexistent file
    detect_system_ram_kb() {
        local kb
        kb="$(grep MemTotal /nonexistent/meminfo 2>/dev/null | awk '{print $2}')"
        echo "${kb:-0}"
    }
    run detect_system_ram_kb
    assert_output "0"
}