|
| 1 | +package featuregates_test |
| 2 | + |
| 3 | +import ( |
| 4 | + "testing" |
| 5 | + |
| 6 | + "github.com/go-logr/logr" |
| 7 | + "github.com/stretchr/testify/require" |
| 8 | + "k8s.io/component-base/featuregate" |
| 9 | + |
| 10 | + "github.com/operator-framework/operator-controller/internal/shared/util/featuregates" |
| 11 | +) |
| 12 | + |
| 13 | +// fakeSink implements logr.LogSink, capturing Info calls for testing |
| 14 | +type fakeSink struct { |
| 15 | + level int |
| 16 | + msg string |
| 17 | + keysAndValues []interface{} |
| 18 | +} |
| 19 | + |
| 20 | +// Init is part of logr.LogSink |
| 21 | +func (f *fakeSink) Init(info logr.RuntimeInfo) {} |
| 22 | + |
| 23 | +// Enabled is part of logr.LogSink |
| 24 | +func (f *fakeSink) Enabled(level int) bool { return true } |
| 25 | + |
| 26 | +// Info captures the log level, message, and key/value pairs |
| 27 | +func (f *fakeSink) Info(level int, msg string, keysAndValues ...interface{}) { |
| 28 | + f.level = level |
| 29 | + f.msg = msg |
| 30 | + f.keysAndValues = append([]interface{}{}, keysAndValues...) |
| 31 | +} |
| 32 | + |
| 33 | +// Error is part of logr.LogSink; not used in this test |
| 34 | +func (f *fakeSink) Error(err error, msg string, keysAndValues ...interface{}) {} |
| 35 | + |
| 36 | +// WithValues returns a sink with additional values; for testing, return self |
| 37 | +func (f *fakeSink) WithValues(keysAndValues ...interface{}) logr.LogSink { return f } |
| 38 | + |
| 39 | +// WithName returns a sink with a new name; for testing, return self |
| 40 | +func (f *fakeSink) WithName(name string) logr.LogSink { return f } |
| 41 | + |
| 42 | +// TestLogFeatureGateStates verifies that LogFeatureGateStates logs features |
| 43 | +// sorted alphabetically with their enabled state |
| 44 | +func TestLogFeatureGateStates(t *testing.T) { |
| 45 | + // Define a set of feature specs with default states |
| 46 | + defs := map[featuregate.Feature]featuregate.FeatureSpec{ |
| 47 | + "AFeature": {Default: false}, |
| 48 | + "BFeature": {Default: true}, |
| 49 | + "CFeature": {Default: false}, |
| 50 | + } |
| 51 | + |
| 52 | + // create a mutable gate and register our definitions |
| 53 | + gate := featuregate.NewFeatureGate() |
| 54 | + require.NoError(t, gate.Add(defs)) |
| 55 | + |
| 56 | + // override CFeature to true. |
| 57 | + require.NoError(t, gate.SetFromMap(map[string]bool{ |
| 58 | + "CFeature": true, |
| 59 | + })) |
| 60 | + |
| 61 | + // prepare a fake sink and logger |
| 62 | + sink := &fakeSink{} |
| 63 | + logger := logr.New(sink) |
| 64 | + |
| 65 | + // log the feature states |
| 66 | + featuregates.LogFeatureGateStates(logger, "feature states", gate, defs) |
| 67 | + |
| 68 | + // verify the message |
| 69 | + require.Equal(t, "feature states", sink.msg) |
| 70 | + |
| 71 | + // Expect keys sorted: AFeature, BFeature, CFeature |
| 72 | + want := []interface{}{ |
| 73 | + featuregate.Feature("AFeature"), false, |
| 74 | + featuregate.Feature("BFeature"), true, |
| 75 | + featuregate.Feature("CFeature"), true, |
| 76 | + } |
| 77 | + require.Equal(t, want, sink.keysAndValues) |
| 78 | +} |
0 commit comments