Skip to content

Commit 437bfd1

Browse files
committed
fix: crd_resolver_test now use a consistent assertion style
1 parent 507e3e4 commit 437bfd1

File tree

1 file changed

+18
-44
lines changed

1 file changed

+18
-44
lines changed

listener/apischema/crd_resolver_test.go

Lines changed: 18 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,11 @@ package apischema
22

33
import (
44
"encoding/json"
5-
"errors"
65
"testing"
76

7+
"github.com/stretchr/testify/assert"
8+
"github.com/stretchr/testify/require"
9+
810
apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
911
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
1012
"k8s.io/client-go/openapi"
@@ -76,17 +78,9 @@ func TestGetCRDGroupKindVersions(t *testing.T) {
7678
for _, tc := range tests {
7779
t.Run(tc.name, func(t *testing.T) {
7880
gkv := getCRDGroupKindVersions(tc.spec)
79-
if gkv.Group != tc.wantG || gkv.Kind != tc.wantKind {
80-
t.Errorf("GroupKind mismatch: got %v/%v, want %v/%v", gkv.Group, gkv.Kind, tc.wantG, tc.wantKind)
81-
}
82-
if len(gkv.Versions) != len(tc.wantVers) {
83-
t.Fatalf("Versions length: got %d, want %d", len(gkv.Versions), len(tc.wantVers))
84-
}
85-
for i, v := range tc.wantVers {
86-
if gkv.Versions[i] != v {
87-
t.Errorf("Versions[%d]: got %q, want %q", i, gkv.Versions[i], v)
88-
}
89-
}
81+
assert.Equal(t, tc.wantG, gkv.Group, "Group mismatch")
82+
assert.Equal(t, tc.wantKind, gkv.Kind, "Kind mismatch")
83+
assert.Equal(t, tc.wantVers, gkv.Versions, "Versions mismatch")
9084
})
9185
}
9286
}
@@ -116,9 +110,7 @@ func TestIsCRDKindIncluded(t *testing.T) {
116110
for _, tc := range tests {
117111
t.Run(tc.name, func(t *testing.T) {
118112
got := isCRDKindIncluded(tc.gkv, tc.apiList)
119-
if got != tc.want {
120-
t.Errorf("expected %v, got %v", tc.want, got)
121-
}
113+
assert.Equal(t, tc.want, got, "result mismatch")
122114
})
123115
}
124116
}
@@ -162,22 +154,11 @@ func TestErrorIfCRDNotInPreferredApiGroups(t *testing.T) {
162154
t.Run(tc.name, func(t *testing.T) {
163155
groups, err := errorIfCRDNotInPreferredApiGroups(gkv, tc.lists)
164156
if tc.wantErr != nil {
165-
if !errors.Is(err, tc.wantErr) {
166-
t.Fatalf("expected error %v, got %v", tc.wantErr, err)
167-
}
157+
assert.ErrorIs(t, err, tc.wantErr)
168158
return
169159
}
170-
if err != nil {
171-
t.Fatalf("unexpected error: %v", err)
172-
}
173-
if len(groups) != len(tc.wantGroup) {
174-
t.Fatalf("group count: got %d, want %d", len(groups), len(tc.wantGroup))
175-
}
176-
for i := range groups {
177-
if groups[i] != tc.wantGroup[i] {
178-
t.Errorf("groups[%d]: got %q, want %q", i, groups[i], tc.wantGroup[i])
179-
}
180-
}
160+
require.NoError(t, err, "unexpected error")
161+
assert.Equal(t, tc.wantGroup, groups, "groups mismatch")
181162
})
182163
}
183164
}
@@ -236,17 +217,11 @@ func TestGetSchemaForPath(t *testing.T) {
236217
t.Run(tc.name, func(t *testing.T) {
237218
got, err := getSchemaForPath(tc.preferred, tc.path, tc.gv)
238219
if tc.wantErr != nil {
239-
if !errors.Is(err, tc.wantErr) {
240-
t.Fatalf("expected error %v, got %v", tc.wantErr, err)
241-
}
220+
assert.ErrorIs(t, err, tc.wantErr)
242221
return
243222
}
244-
if err != nil {
245-
t.Fatalf("unexpected error: %v", err)
246-
}
247-
if len(got) != tc.wantCount {
248-
t.Errorf("schema count: got %d, want %d", len(got), tc.wantCount)
249-
}
223+
require.NoError(t, err, "unexpected error")
224+
assert.Equal(t, tc.wantCount, len(got), "schema count mismatch")
250225
})
251226
}
252227
}
@@ -309,12 +284,11 @@ func TestResolveSchema(t *testing.T) {
309284
}
310285

311286
got, err := resolveSchema(resolver, resolver)
312-
if (err != nil) != tt.wantErr {
313-
t.Errorf("resolveSchema() error = %v, wantErr %v", err, tt.wantErr)
314-
return
315-
}
316-
if !tt.wantErr && got == nil {
317-
t.Error("resolveSchema() returned nil result when no error expected")
287+
if tt.wantErr {
288+
assert.Error(t, err, "expected an error")
289+
} else {
290+
assert.NoError(t, err, "unexpected error")
291+
assert.NotNil(t, got, "expected non-nil result")
318292
}
319293
})
320294
}

0 commit comments

Comments
 (0)