mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2024-09-20 07:47:31 -07:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

discovery manager - doesn't need sorting of the target groups so move it in the discovery manager tests as we only need it there.

Browse source 
discovery manager - refactor the discovery tests.

Signed-off-by: Krasi Georgiev <krasi.root@gmail.com>
This commit is contained in:
Krasi Georgiev 2018-01-04 21:41:54 +00:00
parent 638818a974
commit 135ea0f793
2 changed files with 196 additions and 135 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
discovery/manager.go
View file

@ -16,7 +16,6 @@ package discovery
import ( import (
"context" "context"
"fmt" "fmt"
"sort"
"github.com/go-kit/kit/log" "github.com/go-kit/kit/log"
"github.com/go-kit/kit/log/level" "github.com/go-kit/kit/log/level"
@ -59,14 +58,6 @@ type poolKey struct {
provider string provider string
} }
// byProvider implements sort.Interface for []poolKey based on the provider field.
// Sorting is needed so that we can have predictable tests.
type byProvider []poolKey
func (a byProvider) Len() int { return len(a) }
func (a byProvider) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a byProvider) Less(i, j int) bool { return a[i].provider < a[j].provider }
// NewManager is the Discovery Manager constructor // NewManager is the Discovery Manager constructor
func NewManager(logger log.Logger) *Manager { func NewManager(logger log.Logger) *Manager {
return &Manager{ return &Manager{
@ -182,18 +173,10 @@ func (m *Manager) allGroups() map[string][]*targetgroup.Group {
tSets := make(chan map[string][]*targetgroup.Group) tSets := make(chan map[string][]*targetgroup.Group)
m.actionCh <- func(ctx context.Context) { m.actionCh <- func(ctx context.Context) {
// Sorting by the poolKey is needed so that we can have predictable tests.
var pKeys []poolKey
for pk := range m.targets {
pKeys = append(pKeys, pk)
}
sort.Sort(byProvider(pKeys))
tSetsAll := map[string][]*targetgroup.Group{} tSetsAll := map[string][]*targetgroup.Group{}
for _, pk := range pKeys { for pkey, tsets := range m.targets {
for _, tg := range m.targets[pk] { for _, tg := range tsets {
tSetsAll[pk.setName] = append(tSetsAll[pk.setName], tg) tSetsAll[pkey.setName] = append(tSetsAll[pkey.setName], tg)
} }
} }
tSets <- tSetsAll tSets <- tSetsAll

240
discovery/manager_test.go
View file

@ -17,6 +17,7 @@ import (
"context" "context"
"fmt" "fmt"
"reflect" "reflect"
"sort"
"strconv" "strconv"
"testing" "testing"
"time" "time"
@ -103,11 +104,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}}, }},
}, },
@ -116,11 +117,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
@ -133,11 +134,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
@ -147,7 +148,7 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp2-initial1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
}, },
@ -158,24 +159,24 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, { }, {
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
{ {
Source: "tp2-initial1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
}, },
@ -188,18 +189,27 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
interval: 0, interval: 0,
}, },
{ {
targetGroups: []targetgroup.Group{}, targetGroups: []targetgroup.Group{
{
Source: "tp1_group1",
Targets: []model.LabelSet{},
},
{
Source: "tp1_group2",
Targets: []model.LabelSet{},
},
},
interval: 10, interval: 10,
}, },
}, },
@ -207,15 +217,24 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
{}, {
{
Source: "tp1_group1",
Targets: []model.LabelSet{},
},
{
Source: "tp1_group2",
Targets: []model.LabelSet{},
},
},
}, },
}, },
{ {
@ -225,11 +244,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
@ -238,13 +257,17 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
{
Source: "tp1_group3",
Targets: []model.LabelSet{{"__instance__": "1"}},
},
}, },
interval: 10, interval: 10,
}, },
@ -253,23 +276,27 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
{ {
{ {
Source: "update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
{
Source: "tp1_group3",
Targets: []model.LabelSet{{"__instance__": "1"}},
},
}, },
}, },
}, },
@ -280,11 +307,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
@ -293,11 +320,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp1-update1", Source: "tp1_group3",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "tp1-update2", Source: "tp1_group4",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
}, },
@ -308,11 +335,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp2-initial1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "5"}}, Targets: []model.LabelSet{{"__instance__": "5"}},
}, },
{ {
Source: "tp2-initial2", Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "6"}}, Targets: []model.LabelSet{{"__instance__": "6"}},
}, },
}, },
@ -321,11 +348,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp2-update1", Source: "tp2_group3",
Targets: []model.LabelSet{{"__instance__": "7"}}, Targets: []model.LabelSet{{"__instance__": "7"}},
}, },
{ {
Source: "tp2-update2", Source: "tp2_group4",
Targets: []model.LabelSet{{"__instance__": "8"}}, Targets: []model.LabelSet{{"__instance__": "8"}},
}, },
}, },
@ -336,82 +363,106 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
{ {
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
{ {
Source: "tp2-initial1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "5"}}, Targets: []model.LabelSet{{"__instance__": "5"}},
}, },
{ {
Source: "tp2-initial2", Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "6"}}, Targets: []model.LabelSet{{"__instance__": "6"}},
}, },
}, },
{ {
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
{ {
Source: "tp2-update1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "5"}},
},
{
Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "6"}},
},
{
Source: "tp2_group3",
Targets: []model.LabelSet{{"__instance__": "7"}}, Targets: []model.LabelSet{{"__instance__": "7"}},
}, },
{ {
Source: "tp2-update2", Source: "tp2_group4",
Targets: []model.LabelSet{{"__instance__": "8"}}, Targets: []model.LabelSet{{"__instance__": "8"}},
}, },
}, },
{ {
{ {
Source: "tp1-update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}},
},
{
Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}},
},
{
Source: "tp1_group3",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "tp1-update2", Source: "tp1_group4",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
{ {
Source: "tp2-update1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "5"}},
},
{
Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "6"}},
},
{
Source: "tp2_group3",
Targets: []model.LabelSet{{"__instance__": "7"}}, Targets: []model.LabelSet{{"__instance__": "7"}},
}, },
{ {
Source: "tp2-update2", Source: "tp2_group4",
Targets: []model.LabelSet{{"__instance__": "8"}}, Targets: []model.LabelSet{{"__instance__": "8"}},
}, },
}, },
}, },
}, },
{ {
title: "One tp initials arrive after other tp updates.", title: "One TP initials arrive after other TP updates.",
updates: map[string][]update{ updates: map[string][]update{
"tp1": { "tp1": {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
@ -420,11 +471,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp1-update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "tp1-update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
}, },
@ -435,11 +486,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp2-initial1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "5"}}, Targets: []model.LabelSet{{"__instance__": "5"}},
}, },
{ {
Source: "tp2-initial2", Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "6"}}, Targets: []model.LabelSet{{"__instance__": "6"}},
}, },
}, },
@ -448,11 +499,11 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "tp2-update1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "7"}}, Targets: []model.LabelSet{{"__instance__": "7"}},
}, },
{ {
Source: "tp2-update2", Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "8"}}, Targets: []model.LabelSet{{"__instance__": "8"}},
}, },
}, },
@ -463,57 +514,57 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "tp1-initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "tp1-initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
{ {
{ {
Source: "tp1-update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "tp1-update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
}, },
{ {
{ {
Source: "tp1-update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "tp1-update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
{ {
Source: "tp2-initial1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "5"}}, Targets: []model.LabelSet{{"__instance__": "5"}},
}, },
{ {
Source: "tp2-initial2", Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "6"}}, Targets: []model.LabelSet{{"__instance__": "6"}},
}, },
}, },
{ {
{ {
Source: "tp1-update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "tp1-update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
{ {
Source: "tp2-update1", Source: "tp2_group1",
Targets: []model.LabelSet{{"__instance__": "7"}}, Targets: []model.LabelSet{{"__instance__": "7"}},
}, },
{ {
Source: "tp2-update2", Source: "tp2_group2",
Targets: []model.LabelSet{{"__instance__": "8"}}, Targets: []model.LabelSet{{"__instance__": "8"}},
}, },
}, },
@ -521,34 +572,43 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
}, },
{ {
title: "Single TP Single provider empty update in between", title: "Single TP empty update in between",
updates: map[string][]update{ updates: map[string][]update{
"tp1": { "tp1": {
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
interval: 30, interval: 30,
}, },
{ {
targetGroups: []targetgroup.Group{}, targetGroups: []targetgroup.Group{
{
Source: "tp1_group1",
Targets: []model.LabelSet{},
},
{
Source: "tp1_group2",
Targets: []model.LabelSet{},
},
},
interval: 10, interval: 10,
}, },
{ {
targetGroups: []targetgroup.Group{ targetGroups: []targetgroup.Group{
{ {
Source: "update1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
}, },
@ -559,22 +619,31 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
expectedTargets: [][]*targetgroup.Group{ expectedTargets: [][]*targetgroup.Group{
{ {
{ {
Source: "initial1", Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "1"}}, Targets: []model.LabelSet{{"__instance__": "1"}},
}, },
{ {
Source: "initial2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "2"}}, Targets: []model.LabelSet{{"__instance__": "2"}},
}, },
}, },
{},
{ {
{ {
Source: "update1", Source: "tp1_group1",
Targets: []model.LabelSet{},
},
{
Source: "tp1_group2",
Targets: []model.LabelSet{},
},
},
{
{
Source: "tp1_group1",
Targets: []model.LabelSet{{"__instance__": "3"}}, Targets: []model.LabelSet{{"__instance__": "3"}},
}, },
{ {
Source: "update2", Source: "tp1_group2",
Targets: []model.LabelSet{{"__instance__": "4"}}, Targets: []model.LabelSet{{"__instance__": "4"}},
}, },
}, },
@ -606,6 +675,8 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
break Loop break Loop
case tsetMap := <-discoveryManager.SyncCh(): case tsetMap := <-discoveryManager.SyncCh():
for _, received := range tsetMap { for _, received := range tsetMap {
// Need to sort by the Groups source as the Discovery manager doesn't guarantee the order.
sort.Sort(byGroupSource(received))
if !reflect.DeepEqual(received, testCase.expectedTargets[x]) { if !reflect.DeepEqual(received, testCase.expectedTargets[x]) {
var receivedFormated string var receivedFormated string
for _, receivedTargets := range received { for _, receivedTargets := range received {
@ -628,7 +699,7 @@ func TestDiscoveryManagerSyncCalls(t *testing.T) {
} }
func TestTargetSetRecreatesTargetGroupsEveryRun(t *testing.T) { func TestTargetSetRecreatesTargetGroupsEveryRun(t *testing.T) {
verifyPresence := func(tSets map[poolKey][]*targetgroup.Group, poolKey poolKey, label string, present bool) { verifyPresence := func(tSets map[poolKey]map[string]*targetgroup.Group, poolKey poolKey, label string, present bool) {
if _, ok := tSets[poolKey]; !ok { if _, ok := tSets[poolKey]; !ok {
t.Fatalf("'%s' should be present in Pool keys: %v", poolKey, tSets) t.Fatalf("'%s' should be present in Pool keys: %v", poolKey, tSets)
return return
@ -729,3 +800,10 @@ func (tp mockdiscoveryProvider) sendUpdates() {
tp.up <- tgs tp.up <- tgs
} }
} }
// byGroupSource implements sort.Interface so we can sort by the Source field.
type byGroupSource []*targetgroup.Group
func (a byGroupSource) Len() int { return len(a) }
func (a byGroupSource) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
func (a byGroupSource) Less(i, j int) bool { return a[i].Source < a[j].Source }