fixed datastore unit-test

Signed-off-by: Nir Rozenbaum <[email protected]>

Author: nirrozenbaum

pkg/epp/datastore/datastore_test.go

@@ -342,7 +342,7 @@ func TestMetrics(t *testing.T) {
 				WithScheme(scheme).
 				Build()
 			podInfoFactory := podinfo.NewPodInfoFactory(map[podinfo.Scraper]*podinfo.ScraperConfig{
-				&backendmetrics.FakeMetricsScraper{}: podinfo.NewScraperConfig(time.Millisecond, 5*time.Second),
+				test.fms: podinfo.NewScraperConfig(time.Millisecond, 5*time.Second),
 			})
 			ds := NewDatastore(ctx, podInfoFactory)
 			_ = ds.PoolSet(ctx, fakeClient, inferencePool)
@@ -352,8 +352,8 @@ func TestMetrics(t *testing.T) {
 			assert.EventuallyWithT(t, func(t *assert.CollectT) {
 				got := ds.PodGetAll()
 				metrics := []*backendmetrics.Metrics{}
-				for _, one := range got {
-					if podMetrics, ok := one.GetData(backendmetrics.MetricsDataKey); ok {
+				for _, podInfo := range got {
+					if podMetrics, ok := podInfo.GetData(backendmetrics.MetricsDataKey); ok {
 						metrics = append(metrics, podMetrics.(*backendmetrics.Metrics))
 					}
 				}