Skip to content

Commit ec4f3e3

Browse files
authored
Merge pull request kubernetes#86319 from Huang-Wei/fluffout-sched-tests
Fix up the sub-test style
2 parents a57a8f3 + 82afa2b commit ec4f3e3

File tree

2 files changed

+49
-39
lines changed

2 files changed

+49
-39
lines changed

pkg/scheduler/algorithm/priorities/image_locality_test.go

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -202,18 +202,21 @@ func TestImageLocalityPriority(t *testing.T) {
202202

203203
func TestNormalizedImageName(t *testing.T) {
204204
for _, testCase := range []struct {
205+
Name string
205206
Input string
206207
Output string
207208
}{
208-
{Input: "root", Output: "root:latest"},
209-
{Input: "root:tag", Output: "root:tag"},
210-
{Input: "gcr.io:5000/root", Output: "gcr.io:5000/root:latest"},
211-
{Input: "root@" + getImageFakeDigest("root"), Output: "root@" + getImageFakeDigest("root")},
209+
{Name: "add :latest postfix 1", Input: "root", Output: "root:latest"},
210+
{Name: "add :latest postfix 2", Input: "gcr.io:5000/root", Output: "gcr.io:5000/root:latest"},
211+
{Name: "keep it as is 1", Input: "root:tag", Output: "root:tag"},
212+
{Name: "keep it as is 2", Input: "root@" + getImageFakeDigest("root"), Output: "root@" + getImageFakeDigest("root")},
212213
} {
213-
image := normalizedImageName(testCase.Input)
214-
if image != testCase.Output {
215-
t.Errorf("expected image reference: %q, got %q", testCase.Output, image)
216-
}
214+
t.Run(testCase.Name, func(t *testing.T) {
215+
image := normalizedImageName(testCase.Input)
216+
if image != testCase.Output {
217+
t.Errorf("expected image reference: %q, got %q", testCase.Output, image)
218+
}
219+
})
217220
}
218221
}
219222

pkg/scheduler/eventhandlers_test.go

Lines changed: 38 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ import (
2828
)
2929

3030
func TestSkipPodUpdate(t *testing.T) {
31-
table := []struct {
31+
for _, test := range []struct {
3232
pod *v1.Pod
3333
isAssumedPodFunc func(*v1.Pod) bool
3434
getPodFunc func(*v1.Pod) *v1.Pod
@@ -102,8 +102,7 @@ func TestSkipPodUpdate(t *testing.T) {
102102
},
103103
expected: false,
104104
},
105-
}
106-
for _, test := range table {
105+
} {
107106
t.Run(test.name, func(t *testing.T) {
108107
c := &Scheduler{
109108
SchedulerCache: &fakecache.Cache{
@@ -123,7 +122,7 @@ func TestNodeAllocatableChanged(t *testing.T) {
123122
newQuantity := func(value int64) resource.Quantity {
124123
return *resource.NewQuantity(value, resource.BinarySI)
125124
}
126-
for _, c := range []struct {
125+
for _, test := range []struct {
127126
Name string
128127
Changed bool
129128
OldAllocatable v1.ResourceList
@@ -142,17 +141,19 @@ func TestNodeAllocatableChanged(t *testing.T) {
142141
NewAllocatable: v1.ResourceList{v1.ResourceMemory: newQuantity(1024), v1.ResourceStorage: newQuantity(1024)},
143142
},
144143
} {
145-
oldNode := &v1.Node{Status: v1.NodeStatus{Allocatable: c.OldAllocatable}}
146-
newNode := &v1.Node{Status: v1.NodeStatus{Allocatable: c.NewAllocatable}}
147-
changed := nodeAllocatableChanged(newNode, oldNode)
148-
if changed != c.Changed {
149-
t.Errorf("nodeAllocatableChanged should be %t, got %t", c.Changed, changed)
150-
}
144+
t.Run(test.Name, func(t *testing.T) {
145+
oldNode := &v1.Node{Status: v1.NodeStatus{Allocatable: test.OldAllocatable}}
146+
newNode := &v1.Node{Status: v1.NodeStatus{Allocatable: test.NewAllocatable}}
147+
changed := nodeAllocatableChanged(newNode, oldNode)
148+
if changed != test.Changed {
149+
t.Errorf("nodeAllocatableChanged should be %t, got %t", test.Changed, changed)
150+
}
151+
})
151152
}
152153
}
153154

154155
func TestNodeLabelsChanged(t *testing.T) {
155-
for _, c := range []struct {
156+
for _, test := range []struct {
156157
Name string
157158
Changed bool
158159
OldLabels map[string]string
@@ -172,17 +173,19 @@ func TestNodeLabelsChanged(t *testing.T) {
172173
NewLabels: map[string]string{"foo": "bar", "test": "value"},
173174
},
174175
} {
175-
oldNode := &v1.Node{ObjectMeta: metav1.ObjectMeta{Labels: c.OldLabels}}
176-
newNode := &v1.Node{ObjectMeta: metav1.ObjectMeta{Labels: c.NewLabels}}
177-
changed := nodeLabelsChanged(newNode, oldNode)
178-
if changed != c.Changed {
179-
t.Errorf("Test case %q failed: should be %t, got %t", c.Name, c.Changed, changed)
180-
}
176+
t.Run(test.Name, func(t *testing.T) {
177+
oldNode := &v1.Node{ObjectMeta: metav1.ObjectMeta{Labels: test.OldLabels}}
178+
newNode := &v1.Node{ObjectMeta: metav1.ObjectMeta{Labels: test.NewLabels}}
179+
changed := nodeLabelsChanged(newNode, oldNode)
180+
if changed != test.Changed {
181+
t.Errorf("Test case %q failed: should be %t, got %t", test.Name, test.Changed, changed)
182+
}
183+
})
181184
}
182185
}
183186

184187
func TestNodeTaintsChanged(t *testing.T) {
185-
for _, c := range []struct {
188+
for _, test := range []struct {
186189
Name string
187190
Changed bool
188191
OldTaints []v1.Taint
@@ -201,12 +204,14 @@ func TestNodeTaintsChanged(t *testing.T) {
201204
NewTaints: []v1.Taint{{Key: "key", Value: "value2"}},
202205
},
203206
} {
204-
oldNode := &v1.Node{Spec: v1.NodeSpec{Taints: c.OldTaints}}
205-
newNode := &v1.Node{Spec: v1.NodeSpec{Taints: c.NewTaints}}
206-
changed := nodeTaintsChanged(newNode, oldNode)
207-
if changed != c.Changed {
208-
t.Errorf("Test case %q failed: should be %t, not %t", c.Name, c.Changed, changed)
209-
}
207+
t.Run(test.Name, func(t *testing.T) {
208+
oldNode := &v1.Node{Spec: v1.NodeSpec{Taints: test.OldTaints}}
209+
newNode := &v1.Node{Spec: v1.NodeSpec{Taints: test.NewTaints}}
210+
changed := nodeTaintsChanged(newNode, oldNode)
211+
if changed != test.Changed {
212+
t.Errorf("Test case %q failed: should be %t, not %t", test.Name, test.Changed, changed)
213+
}
214+
})
210215
}
211216
}
212217

@@ -216,7 +221,7 @@ func TestNodeConditionsChanged(t *testing.T) {
216221
t.Errorf("NodeCondition type has changed. The nodeConditionsChanged() function must be reevaluated.")
217222
}
218223

219-
for _, c := range []struct {
224+
for _, test := range []struct {
220225
Name string
221226
Changed bool
222227
OldConditions []v1.NodeCondition
@@ -253,11 +258,13 @@ func TestNodeConditionsChanged(t *testing.T) {
253258
NewConditions: []v1.NodeCondition{{Type: v1.NodeReady, Status: v1.ConditionTrue}},
254259
},
255260
} {
256-
oldNode := &v1.Node{Status: v1.NodeStatus{Conditions: c.OldConditions}}
257-
newNode := &v1.Node{Status: v1.NodeStatus{Conditions: c.NewConditions}}
258-
changed := nodeConditionsChanged(newNode, oldNode)
259-
if changed != c.Changed {
260-
t.Errorf("Test case %q failed: should be %t, got %t", c.Name, c.Changed, changed)
261-
}
261+
t.Run(test.Name, func(t *testing.T) {
262+
oldNode := &v1.Node{Status: v1.NodeStatus{Conditions: test.OldConditions}}
263+
newNode := &v1.Node{Status: v1.NodeStatus{Conditions: test.NewConditions}}
264+
changed := nodeConditionsChanged(newNode, oldNode)
265+
if changed != test.Changed {
266+
t.Errorf("Test case %q failed: should be %t, got %t", test.Name, test.Changed, changed)
267+
}
268+
})
262269
}
263270
}

0 commit comments

Comments
 (0)