Skip to content

Commit 67d0654

Browse files
committed
fix ut
Signed-off-by: Kuromesi <[email protected]>
1 parent 3dee812 commit 67d0654

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

pkg/epp/backend/provider_test.go

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ import (
2626
"github.com/google/go-cmp/cmp/cmpopts"
2727
"github.com/stretchr/testify/assert"
2828
"k8s.io/apimachinery/pkg/types"
29+
"sigs.k8s.io/gateway-api-inference-extension/api/v1alpha2"
2930
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datastore"
3031
)
3132

@@ -68,6 +69,12 @@ var (
6869
},
6970
},
7071
}
72+
73+
inferencePool = &v1alpha2.InferencePool{
74+
Spec: v1alpha2.InferencePoolSpec{
75+
TargetPortNumber: 8000,
76+
},
77+
}
7178
)
7279

7380
func TestProvider(t *testing.T) {
@@ -127,7 +134,7 @@ func TestProvider(t *testing.T) {
127134

128135
for _, test := range tests {
129136
t.Run(test.name, func(t *testing.T) {
130-
ds := datastore.NewFakeDatastore(test.storePods, nil, nil)
137+
ds := datastore.NewFakeDatastore(test.storePods, nil, inferencePool)
131138
p := NewProvider(test.pmc, ds)
132139
ctx, cancel := context.WithCancel(context.Background())
133140
defer cancel()

0 commit comments

Comments
 (0)