mirror of https://github.com/status-im/consul.git
Adds tests for the string visitor.
This commit is contained in:
parent
799339acb5
commit
e3827923b8
|
@ -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)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue