Function TestPreferenceManager_GetStats

pkg/memory/preference_test.go:418–460

(t *testing.T)

}

func TestPreferenceManager_GetStats(t *testing.T) {
	manager := NewPreferenceManager(DefaultPreferenceManagerConfig())

	// 添加多个偏好
	prefs := []*Preference{
		{UserID: "user-1", Category: CategoryUI, Key: "theme", Value: "dark", Strength: 0.9, Confidence: 0.9},
		{UserID: "user-1", Category: CategoryUI, Key: "font", Value: "mono", Strength: 0.5, Confidence: 0.8},
		{UserID: "user-1", Category: CategoryLanguage, Key: "lang", Value: "zh", Strength: 0.3, Confidence: 1.0},
	}

	for _, pref := range prefs {
		if err := manager.AddPreference(context.Background(), pref); err != nil {
			t.Fatalf("AddPreference failed: %v", err)
		}
	}

	stats := manager.GetStats("user-1")

	if stats.TotalPreferences != 3 {
		t.Errorf("TotalPreferences = %d, want 3", stats.TotalPreferences)
	}

	if stats.CategoryDistribution[CategoryUI] != 2 {
		t.Errorf("UI preferences = %d, want 2", stats.CategoryDistribution[CategoryUI])
	}

	if stats.HighStrengthCount != 1 {
		t.Errorf("HighStrengthCount = %d, want 1", stats.HighStrengthCount)
	}

	if stats.MediumStrengthCount != 1 {
		t.Errorf("MediumStrengthCount = %d, want 1", stats.MediumStrengthCount)
	}

	if stats.LowStrengthCount != 1 {
		t.Errorf("LowStrengthCount = %d, want 1", stats.LowStrengthCount)
	}

	expectedAverage := (0.9 + 0.5 + 0.3) / 3
	if stats.AverageStrength < expectedAverage-0.01 || stats.AverageStrength > expectedAverage+0.01 {
		t.Errorf("AverageStrength = %.2f, want ~%.2f", stats.AverageStrength, expectedAverage)
	}
}

func TestPreferenceExtractor_ExtractFromMessage(t *testing.T) {
	manager := NewPreferenceManager(DefaultPreferenceManagerConfig())