dim/query/query_test.go

53 lines
1.1 KiB
Go
Raw Normal View History

package query
import (
"dim/types"
"testing"
)
func TestFieldListToSelect(t *testing.T) {
tests := []struct {
names types.FieldList
mapping map[string]string
out string
}{
{types.NewFieldList("name"), map[string]string{"name": "name"}, "name as name"},
{
types.NewFieldList("foo", "bar", "baz"),
map[string]string{"foo": "c.foo", "bar": "b.bar"},
"b.bar as bar,attributes->'baz' as baz,c.foo as foo",
},
{
types.NewFieldList("subnets", "vlan", "name"),
map[string]string{"subnets": "s.subnets", "vlan": "vlan.vlan_id", "name": "p.name"},
"p.name as name,s.subnets as subnets,vlan.vlan_id as vlan",
},
}
for _, test := range tests {
out := FieldListToSelect(test.names, test.mapping)
if out != test.out {
t.Errorf("expected `%s`, got `%s`", test.out, out)
}
}
}
func TestNameToAttrPath(t *testing.T) {
tests := []struct {
in string
out string
}{
{"foo", `attributes->'foo'`},
{"foo.bar", `attributes->'foo'->'bar'`},
{"", `attributes`},
}
for _, test := range tests {
out := nameToAttrPath(test.in)
if test.out != out {
t.Errorf("expected `%s`, got `%s`", test.out, out)
}
}
}