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