fix: Improved test coverage for scheduler_profile.go

Author: gyliu513

pkg/epp/scheduling/scheduler_profile_test.go

@@ -18,6 +18,7 @@ package scheduling
 
 import (
 	"context"
+	"strings"
 	"testing"
 
 	"github.com/google/go-cmp/cmp"
@@ -242,6 +243,188 @@ func (tp *testPlugin) reset() {
 	tp.NumOfPickerCandidates = 0
 }
 
+func TestAddPlugins(t *testing.T) {
+	tests := []struct {
+		name        string
+		plugins     []fwkplugin.Plugin
+		wantFilters int
+		wantScorers int
+		wantPicker  bool
+		wantErr     bool
+		errContains string
+	}{
+		{
+			name: "add WeightedScorer that also implements Filter and Picker",
+			plugins: []fwkplugin.Plugin{
+				NewWeightedScorer(&testPlugin{TypeRes: "multi"}, 1.0),
+			},
+			wantFilters: 1,
+			wantScorers: 1,
+			wantPicker:  true,
+		},
+		{
+			name: "add plugin that only implements Filter",
+			plugins: []fwkplugin.Plugin{
+				&filterOnlyPlugin{typedName: fwkplugin.TypedName{Name: "filter-only"}},
+			},
+			wantFilters: 1,
+			wantScorers: 0,
+			wantPicker:  false,
+		},
+		{
+			name: "error when adding Scorer without weight",
+			plugins: []fwkplugin.Plugin{
+				&testPlugin{TypeRes: "bare-scorer"},
+			},
+			wantErr:     true,
+			errContains: "without a weight",
+		},
+		{
+			name: "error when adding duplicate picker",
+			plugins: []fwkplugin.Plugin{
+				NewWeightedScorer(&testPlugin{TypeRes: "picker1"}, 1.0),
+				NewWeightedScorer(&testPlugin{TypeRes: "picker2"}, 1.0),
+			},
+			wantErr:     true,
+			errContains: "already have a registered picker",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			profile := NewSchedulerProfile()
+			err := profile.AddPlugins(test.plugins...)
+
+			if test.wantErr {
+				if err == nil {
+					t.Fatalf("expected error but got nil")
+				}
+				if test.errContains != "" && !strings.Contains(err.Error(), test.errContains) {
+					t.Errorf("error %q does not contain %q", err.Error(), test.errContains)
+				}
+				return
+			}
+			if err != nil {
+				t.Fatalf("unexpected error: %v", err)
+			}
+			if len(profile.filters) != test.wantFilters {
+				t.Errorf("got %d filters, want %d", len(profile.filters), test.wantFilters)
+			}
+			if len(profile.scorers) != test.wantScorers {
+				t.Errorf("got %d scorers, want %d", len(profile.scorers), test.wantScorers)
+			}
+			if test.wantPicker && profile.picker == nil {
+				t.Errorf("expected picker to be set")
+			}
+			if !test.wantPicker && profile.picker != nil {
+				t.Errorf("expected picker to be nil")
+			}
+		})
+	}
+}
+
+func TestSchedulerProfileString(t *testing.T) {
+	tp1 := &testPlugin{TypeRes: "test1"}
+	tp2 := &testPlugin{TypeRes: "test2"}
+	pickerPlugin := &testPlugin{TypeRes: "picker"}
+
+	profile := NewSchedulerProfile().
+		WithFilters(tp1, tp2).
+		WithScorers(NewWeightedScorer(tp1, 1.5), NewWeightedScorer(tp2, 2.0)).
+		WithPicker(pickerPlugin)
+
+	result := profile.String()
+
+	// Verify the string contains filter, scorer, and picker info
+	if !strings.Contains(result, "Filters:") {
+		t.Errorf("String() missing Filters section: %s", result)
+	}
+	if !strings.Contains(result, "Scorers:") {
+		t.Errorf("String() missing Scorers section: %s", result)
+	}
+	if !strings.Contains(result, "Picker:") {
+		t.Errorf("String() missing Picker section: %s", result)
+	}
+}
+
+func TestEnforceScoreRange(t *testing.T) {
+	tests := []struct {
+		name  string
+		score float64
+		want  float64
+	}{
+		{name: "negative score clamped to 0", score: -0.5, want: 0},
+		{name: "score above 1 clamped to 1", score: 1.5, want: 1},
+		{name: "score at 0 stays 0", score: 0, want: 0},
+		{name: "score at 1 stays 1", score: 1, want: 1},
+		{name: "score in range stays unchanged", score: 0.5, want: 0.5},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			got := enforceScoreRange(test.score)
+			if got != test.want {
+				t.Errorf("enforceScoreRange(%v) = %v, want %v", test.score, got, test.want)
+			}
+		})
+	}
+}
+
+func TestRunWithOutOfRangeScores(t *testing.T) {
+	// Scorer that returns negative score
+	negativeScorer := &testPlugin{
+		TypeRes:   "negative",
+		ScoreRes:  -0.5,
+		FilterRes: []k8stypes.NamespacedName{{Name: "pod1"}},
+	}
+	// Scorer that returns score > 1
+	overScorer := &testPlugin{
+		TypeRes:   "over",
+		ScoreRes:  1.5,
+		FilterRes: []k8stypes.NamespacedName{{Name: "pod1"}},
+	}
+	pickerPlugin := &testPlugin{
+		TypeRes: "picker",
+		PickRes: k8stypes.NamespacedName{Name: "pod1"},
+	}
+
+	profile := NewSchedulerProfile().
+		WithFilters().
+		WithScorers(NewWeightedScorer(negativeScorer, 1), NewWeightedScorer(overScorer, 1)).
+		WithPicker(pickerPlugin)
+
+	input := []fwksched.Endpoint{
+		fwksched.NewEndpoint(&fwkdl.EndpointMetadata{NamespacedName: k8stypes.NamespacedName{Name: "pod1"}}, nil, nil),
+	}
+
+	request := &fwksched.LLMRequest{
+		TargetModel: "test-model",
+		RequestId:   uuid.NewString(),
+	}
+
+	_, err := profile.Run(context.Background(), request, fwksched.NewCycleState(), input)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	// negative score clamped to 0, over score clamped to 1: total = 0*1 + 1*1 = 1.0
+	if pickerPlugin.WinnerEndpointScore != 1.0 {
+		t.Errorf("expected winner score 1.0, got %v", pickerPlugin.WinnerEndpointScore)
+	}
+}
+
+// filterOnlyPlugin implements only the Filter interface (not Scorer or Picker).
+type filterOnlyPlugin struct {
+	typedName fwkplugin.TypedName
+}
+
+func (p *filterOnlyPlugin) TypedName() fwkplugin.TypedName {
+	return p.typedName
+}
+
+func (p *filterOnlyPlugin) Filter(_ context.Context, _ *fwksched.CycleState, _ *fwksched.LLMRequest, endpoints []fwksched.Endpoint) []fwksched.Endpoint {
+	return endpoints
+}
+
 func findEndpoints(endpoints []fwksched.Endpoint, names ...k8stypes.NamespacedName) []fwksched.Endpoint {
 	res := []fwksched.Endpoint{}
 	for _, endpoint := range endpoints {