refactor: add test case for OnSessionOpen in ReflectJobOrder plugin

Signed-off-by: Saurabh Kumar Singh <[email protected]>

Author: singh1203

pkg/scheduler/plugins/factory.go

@@ -26,7 +26,6 @@ import (
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/gpupack"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/gpusharingorder"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/gpuspread"
-	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/reflectjoborder"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/kubeflow"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/minruntime"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/nodeavailability"
@@ -37,6 +36,7 @@ import (
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/priority"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/proportion"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/ray"
+	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/reflectjoborder"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/resourcetype"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/snapshot"
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/plugins/subgrouporder"

pkg/scheduler/plugins/reflectjoborder/reflect_job_order_test.go

@@ -11,8 +11,29 @@ import (
 	"testing"
 
 	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/api/common_info"
+	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/api/podgroup_info"
+	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/conf"
+	"github.com/NVIDIA/KAI-scheduler/pkg/scheduler/framework"
 )
 
+func TestJobOrderPlugin_OnSessionOpen(t *testing.T) {
+	ssn := &framework.Session{
+		PodGroupInfos: map[common_info.PodGroupID]*podgroup_info.PodGroupInfo{
+			"pg1": {UID: "pg1", Priority: 5, Queue: "q1"},
+			"pg2": {UID: "pg2", Priority: 2, Queue: "q2"},
+		},
+		Config: &conf.SchedulerConfiguration{
+			QueueDepthPerAction: map[string]int{"Allocate": 10},
+		},
+	}
+	plugin := &JobOrderPlugin{}
+	plugin.OnSessionOpen(ssn)
+
+	if plugin.ReflectJobOrder == nil {
+		t.Fatalf("ReflectJobOrder should be initialized")
+	}
+}
+
 // Test serveJobs returns correct JSON and status when ReflectJobOrder is set
 func TestServeJobs_ReflectJobOrderReady(t *testing.T) {
 	plugin := &JobOrderPlugin{