110 lines
2.4 KiB
Go
110 lines
2.4 KiB
Go
package context
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
)
|
|
|
|
type args struct {
|
|
controllerName string
|
|
disabledByDefaultControllers []string
|
|
}
|
|
|
|
func TestContext_IsControllerEnabled(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
opts Options
|
|
args args
|
|
expected bool
|
|
}{
|
|
{
|
|
name: "on by name",
|
|
args: args{
|
|
controllerName: "bravo",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"alpha", "bravo", "-charlie"},
|
|
},
|
|
expected: true,
|
|
},
|
|
{
|
|
name: "off by name",
|
|
args: args{
|
|
controllerName: "charlie",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"alpha", "bravo", "-charlie"},
|
|
},
|
|
expected: false,
|
|
},
|
|
{
|
|
name: "on by default",
|
|
args: args{
|
|
controllerName: "alpha",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"*"},
|
|
},
|
|
expected: true,
|
|
},
|
|
{
|
|
name: "off by default",
|
|
args: args{
|
|
controllerName: "delta",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"*"},
|
|
},
|
|
expected: false,
|
|
},
|
|
{
|
|
name: "on by star, not off by name",
|
|
args: args{
|
|
controllerName: "alpha",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"*", "-charlie"},
|
|
},
|
|
expected: true,
|
|
},
|
|
{
|
|
name: "off by name with star",
|
|
args: args{
|
|
controllerName: "charlie",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"*", "-charlie"},
|
|
},
|
|
expected: false,
|
|
},
|
|
{
|
|
name: "off by default implicit, no star",
|
|
args: args{
|
|
controllerName: "foxtrot",
|
|
disabledByDefaultControllers: []string{"delta", "echo"},
|
|
},
|
|
opts: Options{
|
|
Controllers: []string{"alpha", "bravo", "-charlie"},
|
|
},
|
|
expected: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
c := Context{
|
|
Opts: tt.opts,
|
|
}
|
|
if got := c.IsControllerEnabled(tt.args.controllerName, sets.NewString(tt.args.disabledByDefaultControllers...)); got != tt.expected {
|
|
t.Errorf("IsControllerEnabled() = %v, want %v", got, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|