yoohoo

internal/config/preferences_test.go

@@ -0,0 +1,162 @@
+package config
+
+import (
+	"testing"
+
+	"git.gostacks.org/iwasforcedtobehere/WiseTLP/autotlp/pkg/types"
+)
+
+func TestValidatePreferences(t *testing.T) {
+	validPrefs := &types.UserPreferences{
+		PowerProfile:        types.PowerProfileBalanced,
+		UseCase:             types.UseCaseGeneral,
+		BatteryPriority:     types.BatteryPriorityBalanced,
+		PerformanceMode:     types.PerformanceModeAdaptive,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{},
+	}
+
+	if err := ValidatePreferences(validPrefs); err != nil {
+		t.Errorf("ValidatePreferences() with valid preferences failed: %v", err)
+	}
+
+	if err := ValidatePreferences(nil); err == nil {
+		t.Error("ValidatePreferences() with nil preferences should fail")
+	}
+
+	invalidPrefs := &types.UserPreferences{
+		PowerProfile:        "invalid_profile",
+		UseCase:             types.UseCaseGeneral,
+		BatteryPriority:     types.BatteryPriorityBalanced,
+		PerformanceMode:     types.PerformanceModeAdaptive,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{},
+	}
+
+	if err := ValidatePreferences(invalidPrefs); err == nil {
+		t.Error("ValidatePreferences() with invalid power profile should fail")
+	}
+
+	invalidPrefs2 := &types.UserPreferences{
+		PowerProfile:        types.PowerProfileBalanced,
+		UseCase:             "invalid_usecase",
+		BatteryPriority:     types.BatteryPriorityBalanced,
+		PerformanceMode:     types.PerformanceModeAdaptive,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{},
+	}
+
+	if err := ValidatePreferences(invalidPrefs2); err == nil {
+		t.Error("ValidatePreferences() with invalid use case should fail")
+	}
+
+	// Test conflicting preferences
+	conflictingPrefs := &types.UserPreferences{
+		PowerProfile:        types.PowerProfilePowerSaving,
+		UseCase:             types.UseCaseGeneral,
+		BatteryPriority:     types.BatteryPriorityBalanced,
+		PerformanceMode:     types.PerformanceModeMaximum,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{},
+	}
+
+	if err := ValidatePreferences(conflictingPrefs); err == nil {
+		t.Error("ValidatePreferences() with conflicting preferences should fail")
+	}
+}
+
+func TestGetRecommendedSettings(t *testing.T) {
+	// Test performance profile
+	perfPrefs := &types.UserPreferences{
+		PowerProfile:        types.PowerProfilePerformance,
+		UseCase:             types.UseCaseGaming,
+		BatteryPriority:     types.BatteryPriorityBalanced,
+		PerformanceMode:     types.PerformanceModeMaximum,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{},
+	}
+
+	settings := GetRecommendedSettings(perfPrefs)
+
+	// Should have TLP_ENABLE
+	if settings["TLP_ENABLE"] != "1" {
+		t.Error("TLP_ENABLE should be set to '1'")
+	}
+
+	// Performance profile should use performance governor
+	if settings["CPU_SCALING_GOVERNOR_ON_AC"] != "performance" {
+		t.Error("Performance profile should use performance governor on AC")
+	}
+
+	// Test power saving profile
+	powerSavePrefs := &types.UserPreferences{
+		PowerProfile:        types.PowerProfilePowerSaving,
+		UseCase:             types.UseCaseOffice,
+		BatteryPriority:     types.BatteryPriorityRuntime,
+		PerformanceMode:     types.PerformanceModeEfficient,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{},
+	}
+
+	settings2 := GetRecommendedSettings(powerSavePrefs)
+
+	// Power saving should use powersave governor
+	if settings2["CPU_SCALING_GOVERNOR_ON_AC"] != "powersave" {
+		t.Error("Power saving profile should use powersave governor on AC")
+	}
+
+	if settings2["CPU_SCALING_GOVERNOR_ON_BAT"] != "powersave" {
+		t.Error("Power saving profile should use powersave governor on battery")
+	}
+
+	// Test custom settings
+	customPrefs := &types.UserPreferences{
+		PowerProfile:    types.PowerProfileCustom,
+		UseCase:         types.UseCaseGeneral,
+		BatteryPriority: types.BatteryPriorityBalanced,
+		PerformanceMode: types.PerformanceModeAdaptive,
+		CustomSettings: map[string]interface{}{
+			"cpu_governor_ac":      "ondemand",
+			"cpu_governor_battery": "conservative",
+			"disk_apm_ac":          "200",
+			"disk_apm_battery":     "100",
+		},
+		SpecialRequirements: []string{},
+	}
+
+	settings3 := GetRecommendedSettings(customPrefs)
+
+	// Custom settings should override defaults
+	if settings3["CPU_SCALING_GOVERNOR_ON_AC"] != "ondemand" {
+		t.Error("Custom CPU governor AC setting should be applied")
+	}
+
+	if settings3["CPU_SCALING_GOVERNOR_ON_BAT"] != "conservative" {
+		t.Error("Custom CPU governor battery setting should be applied")
+	}
+
+	if settings3["DISK_APM_LEVEL_ON_AC"] != "200" {
+		t.Error("Custom disk APM AC setting should be applied")
+	}
+
+	if settings3["DISK_APM_LEVEL_ON_BAT"] != "100" {
+		t.Error("Custom disk APM battery setting should be applied")
+	}
+
+	// Test special requirements
+	wifiPrefs := &types.UserPreferences{
+		PowerProfile:        types.PowerProfileBalanced,
+		UseCase:             types.UseCaseGeneral,
+		BatteryPriority:     types.BatteryPriorityBalanced,
+		PerformanceMode:     types.PerformanceModeAdaptive,
+		CustomSettings:      make(map[string]interface{}),
+		SpecialRequirements: []string{"Maximum WiFi performance"},
+	}
+
+	settings4 := GetRecommendedSettings(wifiPrefs)
+
+	// WiFi performance requirement should disable power saving
+	if settings4["WIFI_PWR_ON_AC"] != "off" || settings4["WIFI_PWR_ON_BAT"] != "off" {
+		t.Error("Maximum WiFi performance should disable WiFi power saving")
+	}
+}