Add unit tests for request body

Author: rlakhtakia

pkg/epp/scheduling/plugins/filter/filter_test.go

@@ -54,7 +54,8 @@ func TestFilter(t *testing.T) {
 			ctx := types.NewSchedulingContext(context.Background(), test.req, test.input)
 			got := test.filter.Filter(ctx, test.input)
 
-			if diff := cmp.Diff(test.output, got); diff != "" {
+			opt := cmp.AllowUnexported(types.PodMetrics{})
+			if diff := cmp.Diff(test.output, got, opt); diff != "" {
 				t.Errorf("Unexpected output (-want +got): %v", diff)
 			}
 		})
@@ -189,7 +190,8 @@ func TestFilterFunc(t *testing.T) {
 			ctx := types.NewSchedulingContext(context.Background(), test.req, test.input)
 			got := test.f(ctx, test.input)
 
-			if diff := cmp.Diff(test.output, got); diff != "" {
+			opt := cmp.AllowUnexported(types.PodMetrics{})
+			if diff := cmp.Diff(test.output, got, opt); diff != "" {
 				t.Errorf("Unexpected output (-want +got): %v", diff)
 			}
 		})

pkg/epp/scheduling/scheduler.go

@@ -80,6 +80,8 @@ func NewSchedulerWithConfig(datastore Datastore, config *SchedulerConfig) *Sched
 		picker:              config.picker,
 		postSchedulePlugins: config.postSchedulePlugins,
 	}
+
+	return scheduler
 }
 
 type Scheduler struct {

pkg/epp/scheduling/scheduler_test.go

@@ -220,9 +220,17 @@ func TestSchedule(t *testing.T) {
 		},
 	}
 
+	schedConfig := &SchedulerConfig{
+		preSchedulePlugins:  []plugins.PreSchedule{},
+		scorers:             []plugins.Scorer{},
+		filters:             []plugins.Filter{defPlugin},
+		postSchedulePlugins: []plugins.PostSchedule{},
+		picker:              defPlugin,
+	}
+
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
-			scheduler := NewScheduler(&fakeDataStore{pods: test.input})
+			scheduler := NewSchedulerWithConfig(&fakeDataStore{pods: test.input}, schedConfig)
 			got, err := scheduler.Schedule(context.Background(), test.req)
 			if test.err != (err != nil) {
 				t.Errorf("Unexpected error, got %v, want %v", err, test.err)

pkg/epp/scheduling/types/types.go

@@ -43,6 +43,8 @@ func (r *LLMRequest) String() string {
 type Pod interface {
 	GetPod() *backendmetrics.Pod
 	GetMetrics() *backendmetrics.Metrics
+	SetScore(float64)
+	Score() float64
 	String() string
 }
 
@@ -74,7 +76,16 @@ func (pm *PodMetrics) GetMetrics() *backendmetrics.Metrics {
 	return pm.Metrics
 }
 
+func (pm *PodMetrics) SetScore(score float64) {
+	pm.score = score
+}
+
+func (pm *PodMetrics) Score() float64 {
+	return pm.score
+}
+
 type PodMetrics struct {
+	score float64
 	*backendmetrics.Pod
 	*backendmetrics.Metrics
 }