147 lines
3.9 KiB
Go
147 lines
3.9 KiB
Go
package testhelpers
|
|
|
|
import (
|
|
"github.com/containous/traefik/pkg/config"
|
|
)
|
|
|
|
// BuildConfiguration is a helper to create a configuration.
|
|
func BuildConfiguration(dynamicConfigBuilders ...func(*config.HTTPConfiguration)) *config.HTTPConfiguration {
|
|
conf := &config.HTTPConfiguration{}
|
|
for _, build := range dynamicConfigBuilders {
|
|
build(conf)
|
|
}
|
|
return conf
|
|
}
|
|
|
|
// WithRouters is a helper to create a configuration.
|
|
func WithRouters(opts ...func(*config.Router) string) func(*config.HTTPConfiguration) {
|
|
return func(c *config.HTTPConfiguration) {
|
|
c.Routers = make(map[string]*config.Router)
|
|
for _, opt := range opts {
|
|
b := &config.Router{}
|
|
name := opt(b)
|
|
c.Routers[name] = b
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithRouter is a helper to create a configuration.
|
|
func WithRouter(routerName string, opts ...func(*config.Router)) func(*config.Router) string {
|
|
return func(r *config.Router) string {
|
|
for _, opt := range opts {
|
|
opt(r)
|
|
}
|
|
return routerName
|
|
}
|
|
}
|
|
|
|
// WithRouterMiddlewares is a helper to create a configuration.
|
|
func WithRouterMiddlewares(middlewaresName ...string) func(*config.Router) {
|
|
return func(r *config.Router) {
|
|
r.Middlewares = middlewaresName
|
|
}
|
|
}
|
|
|
|
// WithServiceName is a helper to create a configuration.
|
|
func WithServiceName(serviceName string) func(*config.Router) {
|
|
return func(r *config.Router) {
|
|
r.Service = serviceName
|
|
}
|
|
}
|
|
|
|
// WithLoadBalancerServices is a helper to create a configuration.
|
|
func WithLoadBalancerServices(opts ...func(service *config.LoadBalancerService) string) func(*config.HTTPConfiguration) {
|
|
return func(c *config.HTTPConfiguration) {
|
|
c.Services = make(map[string]*config.Service)
|
|
for _, opt := range opts {
|
|
b := &config.LoadBalancerService{}
|
|
name := opt(b)
|
|
c.Services[name] = &config.Service{
|
|
LoadBalancer: b,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithService is a helper to create a configuration.
|
|
func WithService(name string, opts ...func(*config.LoadBalancerService)) func(*config.LoadBalancerService) string {
|
|
return func(r *config.LoadBalancerService) string {
|
|
for _, opt := range opts {
|
|
opt(r)
|
|
}
|
|
return name
|
|
}
|
|
}
|
|
|
|
// WithMiddlewares is a helper to create a configuration.
|
|
func WithMiddlewares(opts ...func(*config.Middleware) string) func(*config.HTTPConfiguration) {
|
|
return func(c *config.HTTPConfiguration) {
|
|
c.Middlewares = make(map[string]*config.Middleware)
|
|
for _, opt := range opts {
|
|
b := &config.Middleware{}
|
|
name := opt(b)
|
|
c.Middlewares[name] = b
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithMiddleware is a helper to create a configuration.
|
|
func WithMiddleware(name string, opts ...func(*config.Middleware)) func(*config.Middleware) string {
|
|
return func(r *config.Middleware) string {
|
|
for _, opt := range opts {
|
|
opt(r)
|
|
}
|
|
return name
|
|
}
|
|
}
|
|
|
|
// WithBasicAuth is a helper to create a configuration.
|
|
func WithBasicAuth(auth *config.BasicAuth) func(*config.Middleware) {
|
|
return func(r *config.Middleware) {
|
|
r.BasicAuth = auth
|
|
}
|
|
}
|
|
|
|
// WithEntryPoints is a helper to create a configuration.
|
|
func WithEntryPoints(eps ...string) func(*config.Router) {
|
|
return func(f *config.Router) {
|
|
f.EntryPoints = eps
|
|
}
|
|
}
|
|
|
|
// WithRule is a helper to create a configuration.
|
|
func WithRule(rule string) func(*config.Router) {
|
|
return func(f *config.Router) {
|
|
f.Rule = rule
|
|
}
|
|
}
|
|
|
|
// WithServers is a helper to create a configuration.
|
|
func WithServers(opts ...func(*config.Server)) func(*config.LoadBalancerService) {
|
|
return func(b *config.LoadBalancerService) {
|
|
for _, opt := range opts {
|
|
server := config.Server{}
|
|
opt(&server)
|
|
b.Servers = append(b.Servers, server)
|
|
}
|
|
}
|
|
}
|
|
|
|
// WithServer is a helper to create a configuration.
|
|
func WithServer(url string, opts ...func(*config.Server)) func(*config.Server) {
|
|
return func(s *config.Server) {
|
|
for _, opt := range opts {
|
|
opt(s)
|
|
}
|
|
s.URL = url
|
|
}
|
|
}
|
|
|
|
// WithStickiness is a helper to create a configuration.
|
|
func WithStickiness(cookieName string) func(*config.LoadBalancerService) {
|
|
return func(b *config.LoadBalancerService) {
|
|
b.Stickiness = &config.Stickiness{
|
|
CookieName: cookieName,
|
|
}
|
|
}
|
|
}
|