Adds tests for the string visitor.

This commit is contained in:
James Phillips 2016-02-26 16:23:02 -08:00
parent 799339acb5
commit e3827923b8
1 changed files with 52 additions and 0 deletions

View File

@ -0,0 +1,52 @@
package prepared_query
import (
"fmt"
"reflect"
"testing"
"github.com/hashicorp/consul/consul/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.QueryDatacenterOptions{
Datacenters: []string{"dc1", "dc2"},
},
Tags: []string{"tag1", "tag2", "tag3"},
}
if err := walk(service, fn); err != nil {
t.Fatalf("err: %v", err)
}
expected := []string{
".Service:the-service",
".Failover.Datacenters[0]:dc1",
".Failover.Datacenters[1]:dc2",
".Tags[0]:tag1",
".Tags[1]:tag2",
".Tags[2]:tag3",
}
if !reflect.DeepEqual(actual, expected) {
t.Fatalf("bad: %#v", 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)
}
}