Test updates due to PodInfo rename

shmuelk · shmuelk · commit 6f41e8b8a516 · 2025-11-19T14:07:22.000+02:00
Signed-off-by: Shmuel Kallner &lt;kallner@il.ibm.com&gt;
diff --git a/pkg/epp/backend/metrics/fake.go b/pkg/epp/backend/metrics/fake.go
@@ -38,18 +38,18 @@ type FakePodMetrics struct {
 }
 
 func (fpm *FakePodMetrics) String() string {
-	return fmt.Sprintf("Pod: %v; Metrics: %v", fpm.GetPod(), fpm.GetMetrics())
+	return fmt.Sprintf("Pod: %v; Metrics: %v", fpm.GetMetadata(), fpm.GetMetrics())
 }
 
-func (fpm *FakePodMetrics) GetPod() *backend.Pod {
+func (fpm *FakePodMetrics) GetMetadata() *backend.Pod {
 	return fpm.Pod
 }
 
 func (fpm *FakePodMetrics) GetMetrics() *MetricsState {
 	return fpm.Metrics
 }
 
-func (fpm *FakePodMetrics) UpdatePod(pod *datalayer.PodInfo) {
+func (fpm *FakePodMetrics) UpdateMetadata(pod *datalayer.EndpointMetadata) {
 	fpm.Pod = pod
 }
 func (fpm *FakePodMetrics) GetAttributes() *datalayer.Attributes {
diff --git a/pkg/epp/backend/metrics/pod_metrics_test.go b/pkg/epp/backend/metrics/pod_metrics_test.go
@@ -30,7 +30,7 @@ import (
 )
 
 var (
-	pod1Info = &datalayer.PodInfo{
+	pod1Info = &datalayer.EndpointMetadata{
 		NamespacedName: types.NamespacedName{
 			Name:      "pod1-rank-0",
 			Namespace: "default",
diff --git a/pkg/epp/controller/inferencepool_reconciler_test.go b/pkg/epp/controller/inferencepool_reconciler_test.go
@@ -121,7 +121,7 @@ func TestInferencePoolReconciler(t *testing.T) {
 	if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {
 		t.Errorf("Unexpected InferencePool reconcile error: %v", err)
 	}
-	if diff := diffStore(ds, diffStoreParams{wantPool: pool1, wantPods: []string{"pod1-rank-0", "pod2-rank-0"}}); diff != "" {
+	if diff := diffStore(ds, diffStoreParams{wantPool: pool1, wantEndpoints: []string{"pod1-rank-0", "pod2-rank-0"}}); diff != "" {
 		t.Errorf("Unexpected diff (+got/-want): %s", diff)
 	}
 
@@ -139,7 +139,7 @@ func TestInferencePoolReconciler(t *testing.T) {
 	if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {
 		t.Errorf("Unexpected InferencePool reconcile error: %v", err)
 	}
-	if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantPods: []string{"pod5-rank-0"}}); diff != "" {
+	if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantEndpoints: []string{"pod5-rank-0"}}); diff != "" {
 		t.Errorf("Unexpected diff (+got/-want): %s", diff)
 	}
 
@@ -154,7 +154,7 @@ func TestInferencePoolReconciler(t *testing.T) {
 	if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {
 		t.Errorf("Unexpected InferencePool reconcile error: %v", err)
 	}
-	if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantPods: []string{"pod5-rank-0"}}); diff != "" {
+	if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantEndpoints: []string{"pod5-rank-0"}}); diff != "" {
 		t.Errorf("Unexpected diff (+got/-want): %s", diff)
 	}
 
@@ -168,14 +168,14 @@ func TestInferencePoolReconciler(t *testing.T) {
 	if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {
 		t.Errorf("Unexpected InferencePool reconcile error: %v", err)
 	}
-	if diff := diffStore(ds, diffStoreParams{wantPods: []string{}}); diff != "" {
+	if diff := diffStore(ds, diffStoreParams{wantEndpoints: []string{}}); diff != "" {
 		t.Errorf("Unexpected diff (+got/-want): %s", diff)
 	}
 }
 
 type diffStoreParams struct {
 	wantPool       *v1.InferencePool
-	wantPods       []string
+	wantEndpoints  []string
 	wantObjectives []*v1alpha2.InferenceObjective
 }
 
@@ -188,15 +188,15 @@ func diffStore(datastore datastore.Datastore, params diffStoreParams) string {
 	}
 
 	// Default wantPods if not set because PodGetAll returns an empty slice when empty.
-	if params.wantPods == nil {
-		params.wantPods = []string{}
+	if params.wantEndpoints == nil {
+		params.wantEndpoints = []string{}
 	}
-	gotPods := []string{}
-	for _, pm := range datastore.PodList(backendmetrics.AllPodsPredicate) {
-		gotPods = append(gotPods, pm.GetPod().NamespacedName.Name)
+	gotEndpoints := []string{}
+	for _, em := range datastore.PodList(backendmetrics.AllPodsPredicate) {
+		gotEndpoints = append(gotEndpoints, em.GetMetadata().NamespacedName.Name)
 	}
-	if diff := cmp.Diff(params.wantPods, gotPods, cmpopts.SortSlices(func(a, b string) bool { return a < b })); diff != "" {
-		return "pods:" + diff
+	if diff := cmp.Diff(params.wantEndpoints, gotEndpoints, cmpopts.SortSlices(func(a, b string) bool { return a < b })); diff != "" {
+		return "endpoints:" + diff
 	}
 
 	// Default wantModels if not set because ModelGetAll returns an empty slice when empty.
@@ -348,8 +348,8 @@ func xDiffStore(t *testing.T, datastore datastore.Datastore, params xDiffStorePa
 		params.wantPods = []string{}
 	}
 	gotPods := []string{}
-	for _, pm := range datastore.PodList(backendmetrics.AllPodsPredicate) {
-		gotPods = append(gotPods, pm.GetPod().NamespacedName.Name)
+	for _, em := range datastore.PodList(backendmetrics.AllPodsPredicate) {
+		gotPods = append(gotPods, em.GetMetadata().NamespacedName.Name)
 	}
 	if diff := cmp.Diff(params.wantPods, gotPods, cmpopts.SortSlices(func(a, b string) bool { return a < b })); diff != "" {
 		return "pods:" + diff
diff --git a/pkg/epp/controller/pod_reconciler_test.go b/pkg/epp/controller/pod_reconciler_test.go
@@ -212,8 +212,8 @@ func TestPodReconciler(t *testing.T) {
 			}
 
 			var gotPods []*corev1.Pod
-			for _, pm := range store.PodList(backendmetrics.AllPodsPredicate) {
-				pod := &corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: pm.GetPod().PodName, Namespace: pm.GetPod().NamespacedName.Namespace}, Status: corev1.PodStatus{PodIP: pm.GetPod().GetIPAddress()}}
+			for _, em := range store.PodList(backendmetrics.AllPodsPredicate) {
+				pod := &corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: em.GetMetadata().PodName, Namespace: em.GetMetadata().NamespacedName.Namespace}, Status: corev1.PodStatus{PodIP: em.GetMetadata().GetIPAddress()}}
 				gotPods = append(gotPods, pod)
 			}
 			if !cmp.Equal(gotPods, test.wantPods, cmpopts.SortSlices(func(a, b *corev1.Pod) bool { return a.Name < b.Name })) {
diff --git a/pkg/epp/datalayer/collector_test.go b/pkg/epp/datalayer/collector_test.go
@@ -44,14 +44,14 @@ func (d *DummySource) Collect(ctx context.Context, ep Endpoint) error {
 }
 
 func defaultEndpoint() Endpoint {
-	pod := &PodInfo{
+	meta := &EndpointMetadata{
 		NamespacedName: types.NamespacedName{
 			Name:      "pod-name",
 			Namespace: "default",
 		},
 		Address: "1.2.3.4:5678",
 	}
-	ms := NewEndpoint(pod, nil)
+	ms := NewEndpoint(meta, nil)
 	return ms
 }
 
diff --git a/pkg/epp/datastore/datastore_test.go b/pkg/epp/datastore/datastore_test.go
@@ -407,7 +407,7 @@ func TestPods(t *testing.T) {
 			test.op(ctx, ds)
 			var gotPods []*corev1.Pod
 			for _, pm := range ds.PodList(backendmetrics.AllPodsPredicate) {
-				pod := &corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: pm.GetPod().PodName, Namespace: pm.GetPod().NamespacedName.Namespace}, Status: corev1.PodStatus{PodIP: pm.GetPod().GetIPAddress()}}
+				pod := &corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: pm.GetMetadata().PodName, Namespace: pm.GetMetadata().NamespacedName.Namespace}, Status: corev1.PodStatus{PodIP: pm.GetMetadata().GetIPAddress()}}
 				gotPods = append(gotPods, pod)
 			}
 			if !cmp.Equal(gotPods, test.wantPods, cmpopts.SortSlices(func(a, b *corev1.Pod) bool { return a.Name < b.Name })) {
@@ -419,16 +419,16 @@ func TestPods(t *testing.T) {
 
 func TestPodInfo(t *testing.T) {
 	tests := []struct {
-		name         string
-		op           func(ctx context.Context, ds Datastore)
-		pool         *v1.InferencePool
-		existingPods []*corev1.Pod
-		wantPodInfos []*datalayer.PodInfo
+		name              string
+		op                func(ctx context.Context, ds Datastore)
+		pool              *v1.InferencePool
+		existingPods      []*corev1.Pod
+		wantEndpointMetas []*datalayer.EndpointMetadata
 	}{
 		{
 			name:         "Add new pod, no existing pods, should add",
 			existingPods: []*corev1.Pod{},
-			wantPodInfos: []*datalayer.PodInfo{
+			wantEndpointMetas: []*datalayer.EndpointMetadata{
 				{
 					NamespacedName: types.NamespacedName{
 						Name:      pod1.Name + "-rank-0",
@@ -450,7 +450,7 @@ func TestPodInfo(t *testing.T) {
 		{
 			name:         "Add new pod, no existing pods, should add, multiple target ports",
 			existingPods: []*corev1.Pod{},
-			wantPodInfos: []*datalayer.PodInfo{
+			wantEndpointMetas: []*datalayer.EndpointMetadata{
 				{
 					NamespacedName: types.NamespacedName{
 						Name:      pod1.Name + "-rank-0",
@@ -484,7 +484,7 @@ func TestPodInfo(t *testing.T) {
 		{
 			name:         "Add new pod, with existing pods, should add, multiple target ports",
 			existingPods: []*corev1.Pod{pod1},
-			wantPodInfos: []*datalayer.PodInfo{
+			wantEndpointMetas: []*datalayer.EndpointMetadata{
 				{
 					NamespacedName: types.NamespacedName{
 						Name:      pod1.Name + "-rank-0",
@@ -542,7 +542,7 @@ func TestPodInfo(t *testing.T) {
 		{
 			name:         "Delete the pod, multiple target ports",
 			existingPods: []*corev1.Pod{pod1, pod2},
-			wantPodInfos: []*datalayer.PodInfo{
+			wantEndpointMetas: []*datalayer.EndpointMetadata{
 				{
 					NamespacedName: types.NamespacedName{
 						Name:      pod1.Name + "-rank-0",
@@ -589,11 +589,11 @@ func TestPodInfo(t *testing.T) {
 			}
 
 			test.op(ctx, ds)
-			var gotPodInfos []*datalayer.PodInfo
+			var gotEndpointMetas []*datalayer.EndpointMetadata
 			for _, pm := range ds.PodList(backendmetrics.AllPodsPredicate) {
-				gotPodInfos = append(gotPodInfos, pm.GetPod())
+				gotEndpointMetas = append(gotEndpointMetas, pm.GetMetadata())
 			}
-			if diff := cmp.Diff(test.wantPodInfos, gotPodInfos, cmpopts.SortSlices(func(a, b *datalayer.PodInfo) bool { return a.NamespacedName.Name < b.NamespacedName.Name })); diff != "" {
+			if diff := cmp.Diff(test.wantEndpointMetas, gotEndpointMetas, cmpopts.SortSlices(func(a, b *datalayer.EndpointMetadata) bool { return a.NamespacedName.Name < b.NamespacedName.Name })); diff != "" {
 				t.Errorf("ConvertTo() mismatch (-want +got):\n%s", diff)
 			}
 		})
diff --git a/pkg/epp/requestcontrol/director_test.go b/pkg/epp/requestcontrol/director_test.go
@@ -698,7 +698,7 @@ func TestGetCandidatePodsForScheduling(t *testing.T) {
 			got := director.getCandidatePodsForScheduling(context.Background(), test.metadata)
 
 			diff := cmp.Diff(test.output, got, cmpopts.SortSlices(func(a, b backendmetrics.PodMetrics) bool {
-				return a.GetPod().NamespacedName.String() < b.GetPod().NamespacedName.String()
+				return a.GetMetadata().NamespacedName.String() < b.GetMetadata().NamespacedName.String()
 			}))
 			if diff != "" {
 				t.Errorf("Unexpected output (-want +got): %v", diff)

Original file line number	Diff line number	Diff line change
`@@ -38,18 +38,18 @@ type FakePodMetrics struct {`
`38`	`38`	`}`
`39`	`39`
`40`	`40`	`func (fpm *FakePodMetrics) String() string {`
`41`		`- return fmt.Sprintf("Pod: %v; Metrics: %v", fpm.GetPod(), fpm.GetMetrics())`
	`41`	`+ return fmt.Sprintf("Pod: %v; Metrics: %v", fpm.GetMetadata(), fpm.GetMetrics())`
`42`	`42`	`}`
`43`	`43`
`44`		`-func (fpm FakePodMetrics) GetPod() backend.Pod {`
	`44`	`+func (fpm FakePodMetrics) GetMetadata() backend.Pod {`
`45`	`45`	`return fpm.Pod`
`46`	`46`	`}`
`47`	`47`
`48`	`48`	`func (fpm FakePodMetrics) GetMetrics() MetricsState {`
`49`	`49`	`return fpm.Metrics`
`50`	`50`	`}`
`51`	`51`
`52`		`-func (fpm FakePodMetrics) UpdatePod(pod datalayer.PodInfo) {`
	`52`	`+func (fpm FakePodMetrics) UpdateMetadata(pod datalayer.EndpointMetadata) {`
`53`	`53`	`fpm.Pod = pod`
`54`	`54`	`}`
`55`	`55`	`func (fpm FakePodMetrics) GetAttributes() datalayer.Attributes {`
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ import (`
`30`	`30`	`)`
`31`	`31`
`32`	`32`	`var (`
`33`		`- pod1Info = &datalayer.PodInfo{`
	`33`	`+ pod1Info = &datalayer.EndpointMetadata{`
`34`	`34`	`NamespacedName: types.NamespacedName{`
`35`	`35`	`Name: "pod1-rank-0",`
`36`	`36`	`Namespace: "default",`
Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,7 @@ func TestInferencePoolReconciler(t *testing.T) {`
`121`	`121`	`if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {`
`122`	`122`	`t.Errorf("Unexpected InferencePool reconcile error: %v", err)`
`123`	`123`	`}`
`124`		`- if diff := diffStore(ds, diffStoreParams{wantPool: pool1, wantPods: []string{"pod1-rank-0", "pod2-rank-0"}}); diff != "" {`
	`124`	`+ if diff := diffStore(ds, diffStoreParams{wantPool: pool1, wantEndpoints: []string{"pod1-rank-0", "pod2-rank-0"}}); diff != "" {`
`125`	`125`	`t.Errorf("Unexpected diff (+got/-want): %s", diff)`
`126`	`126`	`}`
`127`	`127`
`@@ -139,7 +139,7 @@ func TestInferencePoolReconciler(t *testing.T) {`
`139`	`139`	`if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {`
`140`	`140`	`t.Errorf("Unexpected InferencePool reconcile error: %v", err)`
`141`	`141`	`}`
`142`		`- if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantPods: []string{"pod5-rank-0"}}); diff != "" {`
	`142`	`+ if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantEndpoints: []string{"pod5-rank-0"}}); diff != "" {`
`143`	`143`	`t.Errorf("Unexpected diff (+got/-want): %s", diff)`
`144`	`144`	`}`
`145`	`145`
`@@ -154,7 +154,7 @@ func TestInferencePoolReconciler(t *testing.T) {`
`154`	`154`	`if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {`
`155`	`155`	`t.Errorf("Unexpected InferencePool reconcile error: %v", err)`
`156`	`156`	`}`
`157`		`- if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantPods: []string{"pod5-rank-0"}}); diff != "" {`
	`157`	`+ if diff := diffStore(ds, diffStoreParams{wantPool: newPool1, wantEndpoints: []string{"pod5-rank-0"}}); diff != "" {`
`158`	`158`	`t.Errorf("Unexpected diff (+got/-want): %s", diff)`
`159`	`159`	`}`
`160`	`160`
`@@ -168,14 +168,14 @@ func TestInferencePoolReconciler(t *testing.T) {`
`168`	`168`	`if _, err := inferencePoolReconciler.Reconcile(ctx, req); err != nil {`
`169`	`169`	`t.Errorf("Unexpected InferencePool reconcile error: %v", err)`
`170`	`170`	`}`
`171`		`- if diff := diffStore(ds, diffStoreParams{wantPods: []string{}}); diff != "" {`
	`171`	`+ if diff := diffStore(ds, diffStoreParams{wantEndpoints: []string{}}); diff != "" {`
`172`	`172`	`t.Errorf("Unexpected diff (+got/-want): %s", diff)`
`173`	`173`	`}`
`174`	`174`	`}`
`175`	`175`
`176`	`176`	`type diffStoreParams struct {`
`177`	`177`	`wantPool *v1.InferencePool`
`178`		`- wantPods []string`
	`178`	`+ wantEndpoints []string`
`179`	`179`	`wantObjectives []*v1alpha2.InferenceObjective`
`180`	`180`	`}`
`181`	`181`
`@@ -188,15 +188,15 @@ func diffStore(datastore datastore.Datastore, params diffStoreParams) string {`
`188`	`188`	`}`
`189`	`189`
`190`	`190`	`// Default wantPods if not set because PodGetAll returns an empty slice when empty.`
`191`		`- if params.wantPods == nil {`
`192`		`- params.wantPods = []string{}`
	`191`	`+ if params.wantEndpoints == nil {`
	`192`	`+ params.wantEndpoints = []string{}`
`193`	`193`	`}`
`194`		`- gotPods := []string{}`
`195`		`- for _, pm := range datastore.PodList(backendmetrics.AllPodsPredicate) {`
`196`		`- gotPods = append(gotPods, pm.GetPod().NamespacedName.Name)`
	`194`	`+ gotEndpoints := []string{}`
	`195`	`+ for _, em := range datastore.PodList(backendmetrics.AllPodsPredicate) {`
	`196`	`+ gotEndpoints = append(gotEndpoints, em.GetMetadata().NamespacedName.Name)`
`197`	`197`	`}`
`198`		`- if diff := cmp.Diff(params.wantPods, gotPods, cmpopts.SortSlices(func(a, b string) bool { return a < b })); diff != "" {`
`199`		`- return "pods:" + diff`
	`198`	`+ if diff := cmp.Diff(params.wantEndpoints, gotEndpoints, cmpopts.SortSlices(func(a, b string) bool { return a < b })); diff != "" {`
	`199`	`+ return "endpoints:" + diff`
`200`	`200`	`}`
`201`	`201`
`202`	`202`	`// Default wantModels if not set because ModelGetAll returns an empty slice when empty.`
`@@ -348,8 +348,8 @@ func xDiffStore(t *testing.T, datastore datastore.Datastore, params xDiffStorePa`
`348`	`348`	`params.wantPods = []string{}`
`349`	`349`	`}`
`350`	`350`	`gotPods := []string{}`
`351`		`- for _, pm := range datastore.PodList(backendmetrics.AllPodsPredicate) {`
`352`		`- gotPods = append(gotPods, pm.GetPod().NamespacedName.Name)`
	`351`	`+ for _, em := range datastore.PodList(backendmetrics.AllPodsPredicate) {`
	`352`	`+ gotPods = append(gotPods, em.GetMetadata().NamespacedName.Name)`
`353`	`353`	`}`
`354`	`354`	`if diff := cmp.Diff(params.wantPods, gotPods, cmpopts.SortSlices(func(a, b string) bool { return a < b })); diff != "" {`
`355`	`355`	`return "pods:" + diff`
Original file line number	Diff line number	Diff line change
`@@ -212,8 +212,8 @@ func TestPodReconciler(t *testing.T) {`
`212`	`212`	`}`
`213`	`213`
`214`	`214`	`var gotPods []*corev1.Pod`
`215`		`- for _, pm := range store.PodList(backendmetrics.AllPodsPredicate) {`
`216`		`- pod := &corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: pm.GetPod().PodName, Namespace: pm.GetPod().NamespacedName.Namespace}, Status: corev1.PodStatus{PodIP: pm.GetPod().GetIPAddress()}}`
	`215`	`+ for _, em := range store.PodList(backendmetrics.AllPodsPredicate) {`
	`216`	`+ pod := &corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: em.GetMetadata().PodName, Namespace: em.GetMetadata().NamespacedName.Namespace}, Status: corev1.PodStatus{PodIP: em.GetMetadata().GetIPAddress()}}`
`217`	`217`	`gotPods = append(gotPods, pod)`
`218`	`218`	`}`
`219`	`219`	`if !cmp.Equal(gotPods, test.wantPods, cmpopts.SortSlices(func(a, b *corev1.Pod) bool { return a.Name < b.Name })) {`
Original file line number	Diff line number	Diff line change
`@@ -44,14 +44,14 @@ func (d *DummySource) Collect(ctx context.Context, ep Endpoint) error {`
`44`	`44`	`}`
`45`	`45`
`46`	`46`	`func defaultEndpoint() Endpoint {`
`47`		`- pod := &PodInfo{`
	`47`	`+ meta := &EndpointMetadata{`
`48`	`48`	`NamespacedName: types.NamespacedName{`
`49`	`49`	`Name: "pod-name",`
`50`	`50`	`Namespace: "default",`
`51`	`51`	`},`
`52`	`52`	`Address: "1.2.3.4:5678",`
`53`	`53`	`}`
`54`		`- ms := NewEndpoint(pod, nil)`
	`54`	`+ ms := NewEndpoint(meta, nil)`
`55`	`55`	`return ms`
`56`	`56`	`}`
`57`	`57`