open-consul/agent/consul/config_test.go

132 lines
3.0 KiB
Go

package consul
import (
"reflect"
"testing"
"time"
fuzz "github.com/google/gofuzz"
"github.com/stretchr/testify/require"
)
func TestCloneSerfLANConfig(t *testing.T) {
config := DefaultConfig().SerfLANConfig
// NOTE: ALL fields on serf.Config and memberlist.Config MUST BE
// represented either here or in the CloneSerfLANConfig function body.
// Failure to add it to the clone or ignore sections will fail the test.
memberlistIgnoreFieldNames := []string{
"Alive",
"AwarenessMaxMultiplier",
"Conflict",
"Delegate",
"DelegateProtocolMax",
"DelegateProtocolMin",
"DelegateProtocolVersion",
"DisableTcpPings",
"DisableTcpPingsForNode",
"DNSConfigPath",
"EnableCompression",
"Events",
"GossipToTheDeadTime",
"HandoffQueueDepth",
"IndirectChecks",
"Label",
"Logger",
"LogOutput",
"Merge",
"Name",
"Ping",
"ProtocolVersion",
"PushPullInterval",
"RequireNodeNames",
"SkipInboundLabelCheck",
"SuspicionMaxTimeoutMult",
"TCPTimeout",
"Transport",
"UDPBufferSize",
}
serfIgnoreFieldNames := []string{
"BroadcastTimeout",
"CoalescePeriod",
"DisableCoordinates",
"EnableNameConflictResolution",
"EventBuffer",
"EventCh",
"FlapTimeout",
"LeavePropagateDelay",
"LogOutput",
"Logger",
"MaxQueueDepth",
"MemberlistConfig",
"Merge",
"MinQueueDepth",
"NodeName",
"ProtocolVersion",
"QueryBuffer",
"QueryResponseSizeLimit",
"QuerySizeLimit",
"QueryTimeoutMult",
"QueueCheckInterval",
"QueueDepthWarning",
"QuiescentPeriod",
"RecentIntentTimeout",
"ReconnectInterval",
"ReconnectTimeoutOverride",
"RejoinAfterLeave",
"SnapshotPath",
"Tags",
"UserCoalescePeriod",
"UserEventSizeLimit",
"UserQuiescentPeriod",
"ValidateNodeNames",
}
serfFuzzed := fuzzNonIgnoredFields(config, serfIgnoreFieldNames)
t.Logf("Fuzzing serf.Config fields: %v", serfFuzzed)
memberlistFuzzed := fuzzNonIgnoredFields(config.MemberlistConfig, memberlistIgnoreFieldNames)
t.Logf("Fuzzing memberlist.Config fields: %v", memberlistFuzzed)
clone := CloneSerfLANConfig(config)
require.Equal(t, config, clone)
}
func fuzzNonIgnoredFields(value interface{}, ignoredFields []string) []string {
ignored := make(map[string]struct{})
for _, field := range ignoredFields {
ignored[field] = struct{}{}
}
var fuzzed []string
// Walk the fields of our object to fuzz and selectively only fuzz the
// fields that were not ignored.
fuzzer := fuzz.NewWithSeed(time.Now().UnixNano())
v := reflect.ValueOf(value).Elem()
for i := 0; i < v.NumField(); i++ {
field := v.Field(i)
if !field.CanInterface() {
continue // skip unexported fields
}
fieldName := v.Type().Field(i).Name
if _, ok := ignored[fieldName]; ok {
continue
}
fuzzed = append(fuzzed, fieldName)
// copy the data somewhere mutable
tmp := reflect.New(field.Type())
tmp.Elem().Set(field)
// fuzz the copy
fuzzer.Fuzz(tmp.Interface())
// and set the fuzzed copy back to the original location
field.Set(reflect.Indirect(tmp))
}
return fuzzed
}