github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-24 12:15:52 +00:00
Code Issues Projects Releases Wiki Activity

improve test coverage

Browse Source
This commit is contained in:
czm 2022-02-22 23:04:44 +08:00
parent d5263feb03
commit f31bf3ff12
1 changed files with 189 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

189
pkg/util/taints/taints_test.go
View File

@ -742,3 +742,192 @@ func TestValidateTaint(t *testing.T) {
}
}
}
func TestTaintSetDiff(t *testing.T) {
cases := []struct {
name string
t1 []v1.Taint
t2 []v1.Taint
expectedTaintsToAdd []*v1.Taint
expectedTaintsToRemove []*v1.Taint
}{
{
name: "two_taints_are_nil",
expectedTaintsToAdd: nil,
expectedTaintsToRemove: nil,
},
{
name: "one_taint_is_nil_and_the_other_is_not_nil",
t1: []v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
expectedTaintsToAdd: []*v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
expectedTaintsToRemove: nil,
},
{
name: "shared_taints_with_the_same_key_value_effect",
t1: []v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
t2: []v1.Taint{
{
Key: "foo_3",
Value: "bar_3",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
expectedTaintsToAdd: []*v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
},
expectedTaintsToRemove: []*v1.Taint{
{
Key: "foo_3",
Value: "bar_3",
Effect: v1.TaintEffectNoExecute,
},
},
},
{
name: "shared_taints_with_the_same_key_effect_different_value",
t1: []v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "different-value",
Effect: v1.TaintEffectNoSchedule,
},
},
t2: []v1.Taint{
{
Key: "foo_3",
Value: "bar_3",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
expectedTaintsToAdd: []*v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
},
expectedTaintsToRemove: []*v1.Taint{
{
Key: "foo_3",
Value: "bar_3",
Effect: v1.TaintEffectNoExecute,
},
},
},
{
name: "shared_taints_with_the_same_key_different_value_effect",
t1: []v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "different-value",
Effect: v1.TaintEffectNoExecute,
},
},
t2: []v1.Taint{
{
Key: "foo_3",
Value: "bar_3",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
expectedTaintsToAdd: []*v1.Taint{
{
Key: "foo_1",
Value: "bar_1",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "different-value",
Effect: v1.TaintEffectNoExecute,
},
},
expectedTaintsToRemove: []*v1.Taint{
{
Key: "foo_3",
Value: "bar_3",
Effect: v1.TaintEffectNoExecute,
},
{
Key: "foo_2",
Value: "bar_2",
Effect: v1.TaintEffectNoSchedule,
},
},
},
}
for _, tt := range cases {
t.Run(tt.name, func(t *testing.T) {
add, remove := TaintSetDiff(tt.t1, tt.t2)
if !reflect.DeepEqual(add, tt.expectedTaintsToAdd) {
t.Errorf("taintsToAdd: %v should equal %v, but get unexpected results", add, tt.expectedTaintsToAdd)
}
if !reflect.DeepEqual(remove, tt.expectedTaintsToRemove) {
t.Errorf("taintsToRemove: %v should equal %v, but get unexpected results", remove, tt.expectedTaintsToRemove)
}
})
}
}