64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package prepared_query
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
)
|
|
|
|
func TestWalk_ServiceQuery(t *testing.T) {
|
|
var actual []string
|
|
fn := func(path string, v reflect.Value) error {
|
|
actual = append(actual, fmt.Sprintf("%s:%s", path, v.String()))
|
|
return nil
|
|
}
|
|
|
|
service := &structs.ServiceQuery{
|
|
Service: "the-service",
|
|
Failover: structs.QueryFailoverOptions{
|
|
Datacenters: []string{"dc1", "dc2"},
|
|
},
|
|
Near: "_agent",
|
|
Tags: []string{"tag1", "tag2", "tag3"},
|
|
NodeMeta: map[string]string{"foo": "bar", "role": "server"},
|
|
EnterpriseMeta: *structs.DefaultEnterpriseMetaInDefaultPartition(),
|
|
}
|
|
if err := walk(service, fn); err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
expected := []string{
|
|
".Failover.Datacenters[0]:dc1",
|
|
".Failover.Datacenters[1]:dc2",
|
|
".Near:_agent",
|
|
".NodeMeta[foo]:bar",
|
|
".NodeMeta[role]:server",
|
|
".Service:the-service",
|
|
".Tags[0]:tag1",
|
|
".Tags[1]:tag2",
|
|
".Tags[2]:tag3",
|
|
".PeerName:",
|
|
}
|
|
expected = append(expected, entMetaWalkFields...)
|
|
sort.Strings(expected)
|
|
sort.Strings(actual)
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
|
|
func TestWalk_Visitor_Errors(t *testing.T) {
|
|
fn := func(path string, v reflect.Value) error {
|
|
return fmt.Errorf("bad")
|
|
}
|
|
|
|
service := &structs.ServiceQuery{}
|
|
err := walk(service, fn)
|
|
if err == nil || err.Error() != "bad" {
|
|
t.Fatalf("bad: %#v", err)
|
|
}
|
|
}
|