open-consul/agent/structs/check_definition_test.go

119 lines
3.0 KiB
Go

package structs
import (
"reflect"
"testing"
"time"
fuzz "github.com/google/gofuzz"
"github.com/hashicorp/consul/api"
"github.com/mitchellh/reflectwalk"
"github.com/pascaldekloe/goe/verify"
)
func TestCheckDefinition_Defaults(t *testing.T) {
t.Parallel()
def := CheckDefinition{}
check := def.HealthCheck("node1")
// Health checks default to critical state
if check.Status != api.HealthCritical {
t.Fatalf("bad: %v", check.Status)
}
}
type walker struct {
fields map[string]reflect.Value
}
func (w *walker) Struct(reflect.Value) error {
return nil
}
func (w *walker) StructField(f reflect.StructField, v reflect.Value) error {
if !f.Anonymous {
w.fields[f.Name] = v
return nil
}
return reflectwalk.SkipEntry
}
func mapFields(t *testing.T, obj interface{}) map[string]reflect.Value {
w := &walker{make(map[string]reflect.Value)}
if err := reflectwalk.Walk(obj, w); err != nil {
t.Fatalf("failed to generate map fields for %+v - %v", obj, err)
}
return w.fields
}
func TestCheckDefinition_CheckType(t *testing.T) {
t.Parallel()
// Fuzz a definition to fill all its fields with data.
var def CheckDefinition
fuzz.New().Fuzz(&def)
orig := mapFields(t, def)
// Remap the ID field which changes name, and redact fields we don't
// expect in the copy.
orig["CheckID"] = orig["ID"]
delete(orig, "ID")
delete(orig, "ServiceID")
delete(orig, "Token")
// Now convert to a check type and ensure that all fields left match.
chk := def.CheckType()
copy := mapFields(t, chk)
for f, vo := range orig {
vc, ok := copy[f]
if !ok {
t.Fatalf("struct is missing field %q", f)
}
if !reflect.DeepEqual(vo.Interface(), vc.Interface()) {
t.Fatalf("copy skipped field %q", f)
}
}
}
func TestCheckDefinitionToCheckType(t *testing.T) {
t.Parallel()
got := &CheckDefinition{
ID: "id",
Name: "name",
Status: "green",
Notes: "notes",
ServiceID: "svcid",
Token: "tok",
ScriptArgs: []string{"/bin/foo"},
HTTP: "someurl",
TCP: "host:port",
Interval: 1 * time.Second,
DockerContainerID: "abc123",
Shell: "/bin/ksh",
TLSSkipVerify: true,
Timeout: 2 * time.Second,
TTL: 3 * time.Second,
DeregisterCriticalServiceAfter: 4 * time.Second,
}
want := &CheckType{
CheckID: "id",
Name: "name",
Status: "green",
Notes: "notes",
ScriptArgs: []string{"/bin/foo"},
HTTP: "someurl",
TCP: "host:port",
Interval: 1 * time.Second,
DockerContainerID: "abc123",
Shell: "/bin/ksh",
TLSSkipVerify: true,
Timeout: 2 * time.Second,
TTL: 3 * time.Second,
DeregisterCriticalServiceAfter: 4 * time.Second,
}
verify.Values(t, "", got.CheckType(), want)
}