package ai

import (
	"testing"

	"github.com/rcourtman/pulse-go-rewrite/internal/alerts"
)

func TestAlertThresholdAdapter_Defaults(t *testing.T) {
	a := NewAlertThresholdAdapter(nil)
	if a.GetNodeCPUThreshold() != 80 {
		t.Fatalf("GetNodeCPUThreshold default = %v", a.GetNodeCPUThreshold())
	}
	if a.GetNodeMemoryThreshold() != 85 {
		t.Fatalf("GetNodeMemoryThreshold default = %v", a.GetNodeMemoryThreshold())
	}
	if a.GetGuestMemoryThreshold() != 85 {
		t.Fatalf("GetGuestMemoryThreshold default = %v", a.GetGuestMemoryThreshold())
	}
	if a.GetGuestDiskThreshold() != 90 {
		t.Fatalf("GetGuestDiskThreshold default = %v", a.GetGuestDiskThreshold())
	}
	if a.GetStorageThreshold() != 85 {
		t.Fatalf("GetStorageThreshold default = %v", a.GetStorageThreshold())
	}
}

func TestAlertThresholdAdapter_UsesAlertManagerConfig(t *testing.T) {
	mgr := alerts.NewManager()
	cfg := mgr.GetConfig()
	cfg.NodeDefaults.CPU = &alerts.HysteresisThreshold{Trigger: 70, Clear: 65}
	cfg.NodeDefaults.Memory = &alerts.HysteresisThreshold{Trigger: 75, Clear: 70}
	cfg.GuestDefaults.Memory = &alerts.HysteresisThreshold{Trigger: 72, Clear: 70}
	cfg.GuestDefaults.Disk = &alerts.HysteresisThreshold{Trigger: 91, Clear: 90}
	cfg.StorageDefault.Trigger = 77
	mgr.UpdateConfig(cfg)

	a := NewAlertThresholdAdapter(mgr)
	if a.GetNodeCPUThreshold() != 70 {
		t.Fatalf("GetNodeCPUThreshold = %v", a.GetNodeCPUThreshold())
	}
	if a.GetNodeMemoryThreshold() != 75 {
		t.Fatalf("GetNodeMemoryThreshold = %v", a.GetNodeMemoryThreshold())
	}
	if a.GetGuestMemoryThreshold() != 72 {
		t.Fatalf("GetGuestMemoryThreshold = %v", a.GetGuestMemoryThreshold())
	}
	if a.GetGuestDiskThreshold() != 91 {
		t.Fatalf("GetGuestDiskThreshold = %v", a.GetGuestDiskThreshold())
	}
	if a.GetStorageThreshold() != 77 {
		t.Fatalf("GetStorageThreshold = %v", a.GetStorageThreshold())
	}
}

func TestAlertThresholdAdapter_Fallbacks(t *testing.T) {
	mgr := alerts.NewManager()
	cfg := mgr.GetConfig()

	cfg.NodeDefaults.CPU = nil
	cfg.NodeDefaults.Memory = &alerts.HysteresisThreshold{Trigger: 0, Clear: 0}
	cfg.GuestDefaults.Memory = nil
	cfg.GuestDefaults.Disk = &alerts.HysteresisThreshold{Trigger: 0, Clear: 0}
	cfg.StorageDefault.Trigger = 0

	mgr.UpdateConfig(cfg)

	a := NewAlertThresholdAdapter(mgr)
	if a.GetNodeCPUThreshold() != 80 {
		t.Fatalf("GetNodeCPUThreshold default = %v", a.GetNodeCPUThreshold())
	}
	if a.GetNodeMemoryThreshold() != 85 {
		t.Fatalf("GetNodeMemoryThreshold default = %v", a.GetNodeMemoryThreshold())
	}
	if a.GetGuestMemoryThreshold() != 85 {
		t.Fatalf("GetGuestMemoryThreshold default = %v", a.GetGuestMemoryThreshold())
	}
	if a.GetGuestDiskThreshold() != 90 {
		t.Fatalf("GetGuestDiskThreshold default = %v", a.GetGuestDiskThreshold())
	}
	if a.GetStorageThreshold() != 85 {
		t.Fatalf("GetStorageThreshold default = %v", a.GetStorageThreshold())
	}
}