Skip to content

Commit 2de4d90

Browse files
committed
Fix e2e test
1 parent e93541d commit 2de4d90

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

test/e2e/e2e_test.go

+9-5
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ var _ = ginkgo.Describe("InferencePool", func() {
6464
for _, m := range infModel.Spec.TargetModels {
6565
expected = append(expected, m.Name)
6666
}
67-
actual := []string{}
67+
actual := make(map[string]int)
6868
gomega.Eventually(func() error {
6969
resp, err := testutils.ExecCommandInPod(ctx, cfg, scheme, kubeCli, nsName, "curl", "curl", curlCmd)
7070
if err != nil {
@@ -75,12 +75,16 @@ var _ = ginkgo.Describe("InferencePool", func() {
7575
}
7676
for _, m := range expected {
7777
if strings.Contains(resp, m) {
78-
actual = append(actual, m)
78+
actual[m] = 0
7979
}
8080
}
81+
var got []string
82+
for m, _ := range actual {
83+
got = append(got, m)
84+
}
8185
// Compare ignoring order
82-
if !cmp.Equal(actual, expected, cmpopts.SortSlices(func(a, b string) bool { return a < b })) {
83-
return fmt.Errorf("actual (%v) != expected (%v); resp=%q", actual, expected, resp)
86+
if !cmp.Equal(got, expected, cmpopts.SortSlices(func(a, b string) bool { return a < b })) {
87+
return fmt.Errorf("actual (%v) != expected (%v); resp=%q", got, expected, resp)
8488
}
8589

8690
return nil
@@ -94,7 +98,7 @@ var _ = ginkgo.Describe("InferencePool", func() {
9498
func newInferenceModel(ns string) *infextv1a1.InferenceModel {
9599
targets := []infextv1a1.TargetModel{
96100
{
97-
Name: modelName + "%-0",
101+
Name: modelName + "-0",
98102
Weight: ptr.To(int32(50)),
99103
},
100104
{

0 commit comments

Comments
 (0)