117 lines
3.1 KiB
Go
117 lines
3.1 KiB
Go
package xds
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/agent/proxycfg"
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFirstHealthyTarget(t *testing.T) {
|
|
passing := proxycfg.TestUpstreamNodesInStatus(t, "passing")
|
|
warning := proxycfg.TestUpstreamNodesInStatus(t, "warning")
|
|
critical := proxycfg.TestUpstreamNodesInStatus(t, "critical")
|
|
|
|
warnOnlyPassingTarget := structs.NewDiscoveryTarget("all-warn", "", "default", "dc1")
|
|
warnOnlyPassingTarget.Subset.OnlyPassing = true
|
|
failOnlyPassingTarget := structs.NewDiscoveryTarget("all-fail", "", "default", "dc1")
|
|
failOnlyPassingTarget.Subset.OnlyPassing = true
|
|
|
|
targets := map[string]*structs.DiscoveryTarget{
|
|
"all-ok.default.dc1": structs.NewDiscoveryTarget("all-ok", "", "default", "dc1"),
|
|
"all-warn.default.dc1": structs.NewDiscoveryTarget("all-warn", "", "default", "dc1"),
|
|
"all-fail.default.dc1": structs.NewDiscoveryTarget("all-fail", "", "default", "dc1"),
|
|
"all-warn-onlypassing.default.dc1": warnOnlyPassingTarget,
|
|
"all-fail-onlypassing.default.dc1": failOnlyPassingTarget,
|
|
}
|
|
targetHealth := map[string]structs.CheckServiceNodes{
|
|
"all-ok.default.dc1": passing,
|
|
"all-warn.default.dc1": warning,
|
|
"all-fail.default.dc1": critical,
|
|
"all-warn-onlypassing.default.dc1": warning,
|
|
"all-fail-onlypassing.default.dc1": critical,
|
|
}
|
|
|
|
cases := []struct {
|
|
primary string
|
|
secondary []string
|
|
expect string
|
|
}{
|
|
{
|
|
primary: "all-ok.default.dc1",
|
|
expect: "all-ok.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-warn.default.dc1",
|
|
expect: "all-warn.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-fail.default.dc1",
|
|
expect: "all-fail.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-warn-onlypassing.default.dc1",
|
|
expect: "all-warn-onlypassing.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-fail-onlypassing.default.dc1",
|
|
expect: "all-fail-onlypassing.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-ok.default.dc1",
|
|
secondary: []string{
|
|
"all-warn.default.dc1",
|
|
},
|
|
expect: "all-ok.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-warn.default.dc1",
|
|
secondary: []string{
|
|
"all-ok.default.dc1",
|
|
},
|
|
expect: "all-warn.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-warn-onlypassing.default.dc1",
|
|
secondary: []string{
|
|
"all-ok.default.dc1",
|
|
},
|
|
expect: "all-ok.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-fail.default.dc1",
|
|
secondary: []string{
|
|
"all-ok.default.dc1",
|
|
},
|
|
expect: "all-ok.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-fail-onlypassing.default.dc1",
|
|
secondary: []string{
|
|
"all-ok.default.dc1",
|
|
},
|
|
expect: "all-ok.default.dc1",
|
|
},
|
|
{
|
|
primary: "all-fail.default.dc1",
|
|
secondary: []string{
|
|
"all-warn-onlypassing.default.dc1",
|
|
"all-warn.default.dc1",
|
|
"all-ok.default.dc1",
|
|
},
|
|
expect: "all-warn.default.dc1",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
tc := tc
|
|
name := fmt.Sprintf("%s and %v", tc.primary, tc.secondary)
|
|
t.Run(name, func(t *testing.T) {
|
|
targetID := firstHealthyTarget(targets, targetHealth, tc.primary, tc.secondary)
|
|
require.Equal(t, tc.expect, targetID)
|
|
})
|
|
}
|
|
}
|