Function TestAddPod

pkg/scheduler/eventhandlers_test.go:830–883

(t *testing.T)

}

func TestAddPod(t *testing.T) {
	tests := []struct {
		name          string
		pod           *v1.Pod
		expectInQueue bool
		expectInCache bool
	}{
		{
			name:          "add unscheduled pod",
			pod:           st.MakePod().Name("pod1").Namespace("ns1").UID("pod1").SchedulerName("supported-scheduler").Obj(),
			expectInQueue: true,
		},
		{
			name: "add unscheduled pod with other scheduler name",
			pod:  st.MakePod().Name("pod1").Namespace("ns1").UID("pod1").SchedulerName("other-scheduler").Obj(),
		},
		{
			name:          "add scheduled pod",
			pod:           st.MakePod().Name("pod1").Namespace("ns1").UID("pod1").Node("node1").Obj(),
			expectInCache: true,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			logger, ctx := ktesting.NewTestContext(t)
			ctx, cancel := context.WithCancel(ctx)
			defer cancel()

			sched := &Scheduler{
				Cache:           internalcache.New(ctx, nil, false),
				SchedulingQueue: internalqueue.NewTestQueue(ctx, nil),
				logger:          logger,
				Profiles: profile.Map{
					"supported-scheduler": nil,
				},
			}

			sched.addPod(tt.pod)

			_, ok := sched.SchedulingQueue.GetPod(tt.pod.Name, tt.pod.Namespace)
			if tt.expectInQueue && !ok {
				t.Errorf("Expected pod to be in scheduling queue")
			} else if !tt.expectInQueue && ok {
				t.Errorf("Expected pod not to be in scheduling queue")
			}
			_, err := sched.Cache.GetPod(tt.pod)
			if tt.expectInCache && err != nil {
				t.Errorf("Expected pod to be in cache: %v", err)
			} else if !tt.expectInCache && err == nil {
				t.Errorf("Expected pod not to be in cache")
			}
		})
	}
}

func TestUpdatePod(t *testing.T) {
	pod := st.MakePod().Name("pod1").Namespace("ns1").UID("pod1").SchedulerName("supported-scheduler").Obj()