157 lines
3 KiB
Go
157 lines
3 KiB
Go
package internal
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestEncodeNode(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
node *Node
|
|
expected map[string]string
|
|
}{
|
|
{
|
|
desc: "1 label",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.aaa": "bar",
|
|
},
|
|
},
|
|
{
|
|
desc: "2 labels",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
{Name: "bbb", Value: "bur"},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.aaa": "bar",
|
|
"traefik.bbb": "bur",
|
|
},
|
|
},
|
|
{
|
|
desc: "2 labels, 1 disabled",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
{Name: "bbb", Value: "bur", Disabled: true},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.aaa": "bar",
|
|
},
|
|
},
|
|
{
|
|
desc: "2 levels",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "foo", Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
}},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.foo.aaa": "bar",
|
|
},
|
|
},
|
|
{
|
|
desc: "3 levels",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "foo", Children: []*Node{
|
|
{Name: "bar", Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
}},
|
|
}},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.foo.bar.aaa": "bar",
|
|
},
|
|
},
|
|
{
|
|
desc: "2 levels, same root",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "foo", Children: []*Node{
|
|
{Name: "bar", Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
{Name: "bbb", Value: "bur"},
|
|
}},
|
|
}},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.foo.bar.aaa": "bar",
|
|
"traefik.foo.bar.bbb": "bur",
|
|
},
|
|
},
|
|
{
|
|
desc: "several levels, different root",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "bar", Children: []*Node{
|
|
{Name: "ccc", Value: "bir"},
|
|
}},
|
|
{Name: "foo", Children: []*Node{
|
|
{Name: "bar", Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
}},
|
|
}},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.foo.bar.aaa": "bar",
|
|
"traefik.bar.ccc": "bir",
|
|
},
|
|
},
|
|
{
|
|
desc: "multiple labels, multiple levels",
|
|
node: &Node{
|
|
Name: "traefik",
|
|
Children: []*Node{
|
|
{Name: "bar", Children: []*Node{
|
|
{Name: "ccc", Value: "bir"},
|
|
}},
|
|
{Name: "foo", Children: []*Node{
|
|
{Name: "bar", Children: []*Node{
|
|
{Name: "aaa", Value: "bar"},
|
|
{Name: "bbb", Value: "bur"},
|
|
}},
|
|
}},
|
|
},
|
|
},
|
|
expected: map[string]string{
|
|
"traefik.foo.bar.aaa": "bar",
|
|
"traefik.foo.bar.bbb": "bur",
|
|
"traefik.bar.ccc": "bir",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
labels := EncodeNode(test.node)
|
|
|
|
assert.Equal(t, test.expected, labels)
|
|
})
|
|
}
|
|
}
|