133 lines
4.5 KiB
Go
133 lines
4.5 KiB
Go
package kv
|
|
|
|
import (
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/docker/libkv/store"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type ByKey []*store.KVPair
|
|
|
|
func (a ByKey) Len() int { return len(a) }
|
|
func (a ByKey) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
|
func (a ByKey) Less(i, j int) bool { return a[i].Key < a[j].Key }
|
|
|
|
func filler(prefix string, opts ...func(string, map[string]*store.KVPair)) []*store.KVPair {
|
|
buf := make(map[string]*store.KVPair)
|
|
for _, opt := range opts {
|
|
opt(prefix, buf)
|
|
}
|
|
|
|
var result ByKey
|
|
for _, value := range buf {
|
|
result = append(result, value)
|
|
}
|
|
|
|
sort.Sort(result)
|
|
return result
|
|
}
|
|
|
|
func backend(name string, opts ...func(map[string]string)) func(string, map[string]*store.KVPair) {
|
|
return entry(pathBackends+name, opts...)
|
|
}
|
|
|
|
func frontend(name string, opts ...func(map[string]string)) func(string, map[string]*store.KVPair) {
|
|
return entry(pathFrontends+name, opts...)
|
|
}
|
|
|
|
func entry(root string, opts ...func(map[string]string)) func(string, map[string]*store.KVPair) {
|
|
return func(prefix string, pairs map[string]*store.KVPair) {
|
|
prefixedRoot := prefix + pathSeparator + strings.TrimPrefix(root, pathSeparator)
|
|
pairs[prefixedRoot] = &store.KVPair{Key: prefixedRoot, Value: []byte("")}
|
|
|
|
transit := make(map[string]string)
|
|
for _, opt := range opts {
|
|
opt(transit)
|
|
}
|
|
|
|
for key, value := range transit {
|
|
fill(pairs, prefixedRoot, key, value)
|
|
}
|
|
}
|
|
}
|
|
|
|
func fill(pairs map[string]*store.KVPair, previous string, current string, value string) {
|
|
clean := strings.TrimPrefix(current, pathSeparator)
|
|
|
|
i := strings.IndexRune(clean, '/')
|
|
if i > 0 {
|
|
key := previous + pathSeparator + clean[:i]
|
|
|
|
if _, ok := pairs[key]; !ok || len(pairs[key].Value) == 0 {
|
|
pairs[key] = &store.KVPair{Key: key, Value: []byte("")}
|
|
}
|
|
|
|
fill(pairs, key, clean[i:], value)
|
|
}
|
|
|
|
key := previous + pathSeparator + clean
|
|
pairs[key] = &store.KVPair{Key: key, Value: []byte(value)}
|
|
}
|
|
|
|
func withPair(key string, value string) func(map[string]string) {
|
|
return func(pairs map[string]string) {
|
|
if len(key) == 0 {
|
|
return
|
|
}
|
|
pairs[key] = value
|
|
}
|
|
}
|
|
|
|
func withErrorPage(name string, backend, query, status string) func(map[string]string) {
|
|
return func(pairs map[string]string) {
|
|
if len(name) == 0 {
|
|
return
|
|
}
|
|
|
|
withPair(pathFrontendErrorPages+name+pathFrontendErrorPagesBackend, backend)(pairs)
|
|
withPair(pathFrontendErrorPages+name+pathFrontendErrorPagesQuery, query)(pairs)
|
|
withPair(pathFrontendErrorPages+name+pathFrontendErrorPagesStatus, status)(pairs)
|
|
}
|
|
}
|
|
|
|
func TestFiller(t *testing.T) {
|
|
expected := []*store.KVPair{
|
|
{Key: "traefik/backends/backend.with.dot.too", Value: []byte("")},
|
|
{Key: "traefik/backends/backend.with.dot.too/servers", Value: []byte("")},
|
|
{Key: "traefik/backends/backend.with.dot.too/servers/server.with.dot", Value: []byte("")},
|
|
{Key: "traefik/backends/backend.with.dot.too/servers/server.with.dot.without.url", Value: []byte("")},
|
|
{Key: "traefik/backends/backend.with.dot.too/servers/server.with.dot.without.url/weight", Value: []byte("0")},
|
|
{Key: "traefik/backends/backend.with.dot.too/servers/server.with.dot/url", Value: []byte("http://172.17.0.2:80")},
|
|
{Key: "traefik/backends/backend.with.dot.too/servers/server.with.dot/weight", Value: []byte("0")},
|
|
{Key: "traefik/frontends/frontend.with.dot", Value: []byte("")},
|
|
{Key: "traefik/frontends/frontend.with.dot/backend", Value: []byte("backend.with.dot.too")},
|
|
{Key: "traefik/frontends/frontend.with.dot/routes", Value: []byte("")},
|
|
{Key: "traefik/frontends/frontend.with.dot/routes/route.with.dot", Value: []byte("")},
|
|
{Key: "traefik/frontends/frontend.with.dot/routes/route.with.dot/rule", Value: []byte("Host:test.localhost")},
|
|
}
|
|
|
|
pairs1 := filler("traefik",
|
|
frontend("frontend.with.dot",
|
|
withPair("backend", "backend.with.dot.too"),
|
|
withPair("routes/route.with.dot/rule", "Host:test.localhost")),
|
|
backend("backend.with.dot.too",
|
|
withPair("servers/server.with.dot/url", "http://172.17.0.2:80"),
|
|
withPair("servers/server.with.dot/weight", "0"),
|
|
withPair("servers/server.with.dot.without.url/weight", "0")),
|
|
)
|
|
assert.EqualValues(t, expected, pairs1)
|
|
|
|
pairs2 := filler("traefik",
|
|
entry("frontends/frontend.with.dot",
|
|
withPair("backend", "backend.with.dot.too"),
|
|
withPair("routes/route.with.dot/rule", "Host:test.localhost")),
|
|
entry("backends/backend.with.dot.too",
|
|
withPair("servers/server.with.dot/url", "http://172.17.0.2:80"),
|
|
withPair("servers/server.with.dot/weight", "0"),
|
|
withPair("servers/server.with.dot.without.url/weight", "0")),
|
|
)
|
|
assert.EqualValues(t, expected, pairs2)
|
|
}
|