github
/
consul
mirror of https://github.com/hashicorp/consul
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

router: make refreshServerRebalanceTimer test a lot more strict

pull/8970/head
Daniel Nephin 2020-10-15 11:42:40 -04:00
parent 81554b2edf
commit 8697cc2b45
1 changed files with 55 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
agent/router/manager_internal_test.go
View File

@ -8,9 +8,11 @@ import (
"testing" "testing"
"time" "time"
"github.com/hashicorp/consul/agent/metadata"
"github.com/hashicorp/go-hclog" "github.com/hashicorp/go-hclog"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"github.com/hashicorp/consul/agent/metadata"
) )
var ( var (
@ -262,55 +264,63 @@ func test_reconcileServerList(maxServers int) (bool, error) {
return true, nil return true, nil
} }
// func (l *serverList) refreshServerRebalanceTimer() { func TestManager_refreshServerRebalanceTimer(t *testing.T) {
func TestManagerInternal_refreshServerRebalanceTimer(t *testing.T) { type testCase struct {
type clusterSizes struct { clients int
numNodes int servers int
numServers int min time.Duration
minRebalance time.Duration max time.Duration
} expected time.Duration
clusters := []clusterSizes{
{0, 3, 2 * time.Minute},
{1, 0, 2 * time.Minute}, // partitioned cluster
{1, 3, 2 * time.Minute},
{2, 3, 2 * time.Minute},
{100, 0, 2 * time.Minute}, // partitioned
{100, 1, 2 * time.Minute}, // partitioned
{100, 3, 2 * time.Minute},
{1024, 1, 2 * time.Minute}, // partitioned
{1024, 3, 2 * time.Minute}, // partitioned
{1024, 5, 2 * time.Minute},
{16384, 1, 4 * time.Minute}, // partitioned
{16384, 2, 2 * time.Minute}, // partitioned
{16384, 3, 2 * time.Minute}, // partitioned
{16384, 5, 2 * time.Minute},
{65535, 0, 2 * time.Minute}, // partitioned
{65535, 1, 8 * time.Minute}, // partitioned
{65535, 2, 3 * time.Minute}, // partitioned
{65535, 3, 5 * time.Minute}, // partitioned
{65535, 5, 3 * time.Minute}, // partitioned
{65535, 7, 2 * time.Minute},
{1000000, 1, 4 * time.Hour}, // partitioned
{1000000, 2, 2 * time.Hour}, // partitioned
{1000000, 3, 80 * time.Minute}, // partitioned
{1000000, 5, 50 * time.Minute}, // partitioned
{1000000, 11, 20 * time.Minute}, // partitioned
{1000000, 19, 10 * time.Minute},
} }
logger := GetBufferedLogger() testCases := []testCase{
shutdownCh := make(chan struct{}) {servers: 3, clients: 0},
{servers: 0, clients: 1}, // partitioned cluster
{servers: 3, clients: 1},
{servers: 3, clients: 2},
{servers: 0, clients: 100}, // partitioned
{servers: 1, clients: 100}, // partitioned
{servers: 3, clients: 100},
{servers: 1, clients: 1024}, // partitioned
{servers: 3, clients: 1024}, // partitioned
{servers: 5, clients: 1024},
{servers: 1, clients: 16384, expected: 4*time.Minute + 16*time.Second}, // partitioned
{servers: 2, clients: 16384}, // partitioned
{servers: 3, clients: 16384}, // partitioned
{servers: 5, clients: 16384},
{servers: 0, clients: 65535}, // partitioned
{servers: 1, clients: 65535, expected: 17*time.Minute + 3984375000},
{servers: 2, clients: 65535, expected: 8*time.Minute + 31992187500}, // partitioned
{servers: 3, clients: 65535, expected: 5*time.Minute + 41328125000}, // partitioned
{servers: 5, clients: 65535, expected: 3*time.Minute + 24796875000}, // partitioned
{servers: 7, clients: 65535},
{servers: 1, clients: 1000000, expected: 4*time.Hour + 20*time.Minute + 25*time.Second}, // partitioned
{servers: 2, clients: 1000000, expected: 2*time.Hour + 10*time.Minute + 12500*time.Millisecond}, // partitioned
{servers: 3, clients: 1000000, expected: 86*time.Minute + 48333333333}, // partitioned
{servers: 5, clients: 1000000, expected: 52*time.Minute + 5*time.Second}, // partitioned
{servers: 11, clients: 1000000, expected: 23*time.Minute + 40454545454}, // partitioned
{servers: 19, clients: 1000000, expected: 13*time.Minute + 42368421052},
}
for _, s := range clusters { for _, tc := range testCases {
m := New(logger, shutdownCh, &fauxSerf{numNodes: s.numNodes}, &fauxConnPool{}, "", noopRebalancer) m := &Manager{clusterInfo: &fauxSerf{numNodes: tc.clients}, rebalanceTimer: time.NewTimer(0)}
for i := 0; i < s.numServers; i++ { m.saveServerList(serverList{servers: make([]*metadata.Server, tc.servers)})
nodeName := fmt.Sprintf("s%02d", i) delay := m.refreshServerRebalanceTimer()
m.AddServer(&metadata.Server{Name: nodeName})
if tc.expected != 0 {
assert.Equal(t, tc.expected, delay, "nodes=%d, servers=%d", tc.clients, tc.servers)
continue
} }
d := m.refreshServerRebalanceTimer() if tc.min == 0 {
if d < s.minRebalance { tc.min = 2 * time.Minute
t.Errorf("duration too short for cluster of size %d and %d servers (%s < %s)", s.numNodes, s.numServers, d, s.minRebalance) tc.max = 3 * time.Minute
}
if delay < tc.min {
t.Errorf("nodes=%d, servers=%d, expected >%v, actual %v", tc.clients, tc.servers, tc.min, delay)
}
if delay > tc.max {
t.Errorf("nodes=%d, servers=%d, expected <%v, actual %v", tc.clients, tc.servers, tc.max, delay)
} }
} }
} }