mirror of
https://github.com/containous/traefik.git
synced 2024-12-24 21:34:39 +03:00
352 lines
8.6 KiB
Go
352 lines
8.6 KiB
Go
package provider
|
|
|
|
import (
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/containous/traefik/types"
|
|
"github.com/mesosphere/mesos-dns/records/state"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestMesosTaskFilter(t *testing.T) {
|
|
|
|
cases := []struct {
|
|
mesosTask state.Task
|
|
expected bool
|
|
exposedByDefault bool
|
|
}{
|
|
{
|
|
mesosTask: state.Task{},
|
|
expected: false,
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(setState("TASK_RUNNING")))),
|
|
expected: false,
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "false"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // because label traefik.enable = false
|
|
exposedByDefault: false,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: true,
|
|
exposedByDefault: false,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: true,
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "false"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // because label traefik.enable = false (even wherek exposedByDefault = true)
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.portIndex", "1",
|
|
"traefik.port", "80"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // traefik.portIndex & traefik.port cannot be set both
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.portIndex", "1"),
|
|
discovery(setDiscoveryPorts("TCP", 80, "WEB HTTP", "TCP", 443, "WEB HTTPS")),
|
|
),
|
|
expected: true,
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true"),
|
|
discovery(setDiscoveryPorts("TCP", 80, "WEB HTTP", "TCP", 443, "WEB HTTPS")),
|
|
),
|
|
expected: false, // more than 1 discovery port but no traefik.port* label
|
|
exposedByDefault: true,
|
|
},
|
|
{
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.portIndex", "1"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // traefik.portIndex and discoveryPorts don't correspond
|
|
exposedByDefault: true,
|
|
}, {
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.portIndex", "0"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: true, // traefik.portIndex and discoveryPorts correspond
|
|
exposedByDefault: true,
|
|
}, {
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.port", "TRAEFIK"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // traefik.port is not an integer
|
|
exposedByDefault: true,
|
|
}, {
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.port", "443"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // traefik.port is not the same as discovery.port
|
|
exposedByDefault: true,
|
|
}, {
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(true))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.port", "80"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: true, // traefik.port is the same as discovery.port
|
|
exposedByDefault: true,
|
|
}, {
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.port", "80"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: true, // No healthCheck
|
|
exposedByDefault: true,
|
|
}, {
|
|
mesosTask: task(statuses(status(
|
|
setState("TASK_RUNNING"),
|
|
setHealthy(false))),
|
|
setLabels("traefik.enable", "true",
|
|
"traefik.port", "80"),
|
|
discovery(setDiscoveryPort("TCP", 80, "WEB")),
|
|
),
|
|
expected: false, // HealthCheck at false
|
|
exposedByDefault: true,
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
actual := mesosTaskFilter(c.mesosTask, c.exposedByDefault)
|
|
log.Errorf("Statuses : %v", c.mesosTask.Statuses)
|
|
log.Errorf("Label : %v", c.mesosTask.Labels)
|
|
log.Errorf("DiscoveryInfo : %v", c.mesosTask.DiscoveryInfo)
|
|
if actual != c.expected {
|
|
t.Fatalf("expected %v, got %v", c.expected, actual)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTaskRecords(t *testing.T) {
|
|
var task = state.Task{
|
|
SlaveID: "s_id",
|
|
State: "TASK_RUNNING",
|
|
}
|
|
var framework = state.Framework{
|
|
Tasks: []state.Task{task},
|
|
}
|
|
var slave = state.Slave{
|
|
ID: "s_id",
|
|
Hostname: "127.0.0.1",
|
|
}
|
|
var state = state.State{
|
|
Slaves: []state.Slave{slave},
|
|
Frameworks: []state.Framework{framework},
|
|
}
|
|
|
|
provider := &Mesos{
|
|
Domain: "docker.localhost",
|
|
ExposedByDefault: true,
|
|
}
|
|
var p = provider.taskRecords(state)
|
|
if len(p) == 0 {
|
|
t.Fatal("taskRecord should return at least one task")
|
|
}
|
|
if p[0].SlaveIP != slave.Hostname {
|
|
t.Fatalf("The SlaveIP (%s) should be set with the slave hostname (%s)", p[0].SlaveID, slave.Hostname)
|
|
}
|
|
}
|
|
|
|
func TestMesosLoadConfig(t *testing.T) {
|
|
cases := []struct {
|
|
applicationsError bool
|
|
tasksError bool
|
|
mesosTask state.Task
|
|
expected bool
|
|
exposedByDefault bool
|
|
expectedNil bool
|
|
expectedFrontends map[string]*types.Frontend
|
|
expectedBackends map[string]*types.Backend
|
|
}{}
|
|
for _, c := range cases {
|
|
provider := &Mesos{
|
|
Domain: "docker.localhost",
|
|
ExposedByDefault: true,
|
|
}
|
|
actualConfig := provider.loadMesosConfig()
|
|
if c.expectedNil {
|
|
if actualConfig != nil {
|
|
t.Fatalf("Should have been nil, got %v", actualConfig)
|
|
}
|
|
} else {
|
|
// Compare backends
|
|
if !reflect.DeepEqual(actualConfig.Backends, c.expectedBackends) {
|
|
t.Fatalf("expected %#v, got %#v", c.expectedBackends, actualConfig.Backends)
|
|
}
|
|
if !reflect.DeepEqual(actualConfig.Frontends, c.expectedFrontends) {
|
|
t.Fatalf("expected %#v, got %#v", c.expectedFrontends, actualConfig.Frontends)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// test helpers
|
|
|
|
type (
|
|
taskOpt func(*state.Task)
|
|
statusOpt func(*state.Status)
|
|
)
|
|
|
|
func task(opts ...taskOpt) state.Task {
|
|
var t state.Task
|
|
for _, opt := range opts {
|
|
opt(&t)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func statuses(st ...state.Status) taskOpt {
|
|
return func(t *state.Task) {
|
|
t.Statuses = append(t.Statuses, st...)
|
|
}
|
|
}
|
|
|
|
func discovery(dp state.DiscoveryInfo) taskOpt {
|
|
return func(t *state.Task) {
|
|
t.DiscoveryInfo = dp
|
|
}
|
|
}
|
|
|
|
func setLabels(kvs ...string) taskOpt {
|
|
return func(t *state.Task) {
|
|
if len(kvs)%2 != 0 {
|
|
panic("odd number")
|
|
}
|
|
|
|
for i := 0; i < len(kvs); i += 2 {
|
|
var label = state.Label{Key: kvs[i], Value: kvs[i+1]}
|
|
log.Errorf("Label1.1 : %v", label)
|
|
t.Labels = append(t.Labels, label)
|
|
log.Errorf("Label1.2 : %v", t.Labels)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func status(opts ...statusOpt) state.Status {
|
|
var s state.Status
|
|
for _, opt := range opts {
|
|
opt(&s)
|
|
}
|
|
return s
|
|
}
|
|
|
|
func setDiscoveryPort(proto string, port int, name string) state.DiscoveryInfo {
|
|
|
|
dp := state.DiscoveryPort{
|
|
Protocol: proto,
|
|
Number: port,
|
|
Name: name,
|
|
}
|
|
|
|
discoveryPorts := []state.DiscoveryPort{dp}
|
|
|
|
ports := state.Ports{
|
|
DiscoveryPorts: discoveryPorts,
|
|
}
|
|
|
|
return state.DiscoveryInfo{
|
|
Ports: ports,
|
|
}
|
|
}
|
|
|
|
func setDiscoveryPorts(proto1 string, port1 int, name1 string, proto2 string, port2 int, name2 string) state.DiscoveryInfo {
|
|
|
|
dp1 := state.DiscoveryPort{
|
|
Protocol: proto1,
|
|
Number: port1,
|
|
Name: name1,
|
|
}
|
|
|
|
dp2 := state.DiscoveryPort{
|
|
Protocol: proto2,
|
|
Number: port2,
|
|
Name: name2,
|
|
}
|
|
|
|
discoveryPorts := []state.DiscoveryPort{dp1, dp2}
|
|
|
|
ports := state.Ports{
|
|
DiscoveryPorts: discoveryPorts,
|
|
}
|
|
|
|
return state.DiscoveryInfo{
|
|
Ports: ports,
|
|
}
|
|
}
|
|
|
|
func setState(st string) statusOpt {
|
|
return func(s *state.Status) {
|
|
s.State = st
|
|
}
|
|
}
|
|
func setHealthy(b bool) statusOpt {
|
|
return func(s *state.Status) {
|
|
s.Healthy = &b
|
|
}
|
|
}
|