mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
4f0d95fa6e
Signed-off-by: Daniel Nephin <dnephin@docker.com>
96 lines
2.7 KiB
Go
96 lines
2.7 KiB
Go
package discovery // import "github.com/docker/docker/daemon/discovery"
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDiscoveryOptsErrors(t *testing.T) {
|
|
var testcases = []struct {
|
|
doc string
|
|
opts map[string]string
|
|
}{
|
|
{
|
|
doc: "discovery.ttl < discovery.heartbeat",
|
|
opts: map[string]string{"discovery.heartbeat": "10", "discovery.ttl": "5"},
|
|
},
|
|
{
|
|
doc: "discovery.ttl == discovery.heartbeat",
|
|
opts: map[string]string{"discovery.heartbeat": "10", "discovery.ttl": "10"},
|
|
},
|
|
{
|
|
doc: "negative discovery.heartbeat",
|
|
opts: map[string]string{"discovery.heartbeat": "-10", "discovery.ttl": "10"},
|
|
},
|
|
{
|
|
doc: "negative discovery.ttl",
|
|
opts: map[string]string{"discovery.heartbeat": "10", "discovery.ttl": "-10"},
|
|
},
|
|
{
|
|
doc: "invalid discovery.heartbeat",
|
|
opts: map[string]string{"discovery.heartbeat": "invalid"},
|
|
},
|
|
{
|
|
doc: "invalid discovery.ttl",
|
|
opts: map[string]string{"discovery.ttl": "invalid"},
|
|
},
|
|
}
|
|
|
|
for _, testcase := range testcases {
|
|
_, _, err := discoveryOpts(testcase.opts)
|
|
assert.Error(t, err, testcase.doc)
|
|
}
|
|
}
|
|
|
|
func TestDiscoveryOpts(t *testing.T) {
|
|
clusterOpts := map[string]string{"discovery.heartbeat": "10", "discovery.ttl": "20"}
|
|
heartbeat, ttl, err := discoveryOpts(clusterOpts)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 10*time.Second, heartbeat)
|
|
assert.Equal(t, 20*time.Second, ttl)
|
|
|
|
clusterOpts = map[string]string{"discovery.heartbeat": "10"}
|
|
heartbeat, ttl, err = discoveryOpts(clusterOpts)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 10*time.Second, heartbeat)
|
|
assert.Equal(t, 10*defaultDiscoveryTTLFactor*time.Second, ttl)
|
|
|
|
clusterOpts = map[string]string{"discovery.ttl": "30"}
|
|
heartbeat, ttl, err = discoveryOpts(clusterOpts)
|
|
require.NoError(t, err)
|
|
|
|
if ttl != 30*time.Second {
|
|
t.Fatalf("TTL - Expected : %v, Actual : %v", 30*time.Second, ttl)
|
|
}
|
|
|
|
expected := 30 * time.Second / defaultDiscoveryTTLFactor
|
|
if heartbeat != expected {
|
|
t.Fatalf("Heartbeat - Expected : %v, Actual : %v", expected, heartbeat)
|
|
}
|
|
|
|
discoveryTTL := fmt.Sprintf("%d", defaultDiscoveryTTLFactor-1)
|
|
clusterOpts = map[string]string{"discovery.ttl": discoveryTTL}
|
|
heartbeat, _, err = discoveryOpts(clusterOpts)
|
|
if err == nil && heartbeat == 0 {
|
|
t.Fatal("discovery.heartbeat must be positive")
|
|
}
|
|
|
|
clusterOpts = map[string]string{}
|
|
heartbeat, ttl, err = discoveryOpts(clusterOpts)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if heartbeat != defaultDiscoveryHeartbeat {
|
|
t.Fatalf("Heartbeat - Expected : %v, Actual : %v", defaultDiscoveryHeartbeat, heartbeat)
|
|
}
|
|
|
|
expected = defaultDiscoveryHeartbeat * defaultDiscoveryTTLFactor
|
|
if ttl != expected {
|
|
t.Fatalf("TTL - Expected : %v, Actual : %v", expected, ttl)
|
|
}
|
|
}
|