Tom Wilkie
8b061fd555
- Nameserver holds a sorted list of (hostname, peer, container id, ip) tuples. - Entries are gossiped using topology gossip / broadcast - only new/updated entries are propogated as expected. - Entries for a given peer are removed when the connection to the peer is lost. - Entries are tombstoned when containers die. Tombstones are kept for 10mins, as they do not need to survive partition, only gossip propagation delay.
114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package nameserver
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/weaveworks/weave/net/address"
|
|
"github.com/weaveworks/weave/router"
|
|
)
|
|
|
|
func TestAdd(t *testing.T) {
|
|
oldNow := now
|
|
defer func() { now = oldNow }()
|
|
now = func() int64 { return 1234 }
|
|
|
|
entries := Entries{}
|
|
entries.add("A", "", router.UnknownPeerName, address.Address(0))
|
|
expected := Entries{
|
|
Entry{Hostname: "A", Origin: router.UnknownPeerName, Addr: address.Address(0)},
|
|
}
|
|
require.Equal(t, entries, expected)
|
|
|
|
entries.tombstone(router.UnknownPeerName, func(e *Entry) bool { return e.Hostname == "A" })
|
|
expected = Entries{
|
|
Entry{Hostname: "A", Origin: router.UnknownPeerName, Addr: address.Address(0), Version: 1, Tombstone: 1234},
|
|
}
|
|
require.Equal(t, entries, expected)
|
|
|
|
entries.add("A", "", router.UnknownPeerName, address.Address(0))
|
|
expected = Entries{
|
|
Entry{Hostname: "A", Origin: router.UnknownPeerName, Addr: address.Address(0), Version: 2},
|
|
}
|
|
require.Equal(t, entries, expected)
|
|
}
|
|
|
|
func TestMerge(t *testing.T) {
|
|
e1 := Entries{
|
|
Entry{Hostname: "A"},
|
|
Entry{Hostname: "C"},
|
|
Entry{Hostname: "D"},
|
|
Entry{Hostname: "F"},
|
|
}
|
|
|
|
e2 := Entries{
|
|
Entry{Hostname: "B"},
|
|
Entry{Hostname: "E"},
|
|
Entry{Hostname: "F"},
|
|
}
|
|
|
|
e1.merge(e2)
|
|
expected := Entries{
|
|
Entry{Hostname: "A"},
|
|
Entry{Hostname: "B"},
|
|
Entry{Hostname: "C"},
|
|
Entry{Hostname: "D"},
|
|
Entry{Hostname: "E"},
|
|
Entry{Hostname: "F"},
|
|
}
|
|
|
|
require.Equal(t, e1, expected)
|
|
}
|
|
|
|
func TestTombstone(t *testing.T) {
|
|
oldNow := now
|
|
defer func() { now = oldNow }()
|
|
now = func() int64 { return 1234 }
|
|
|
|
es := Entries{
|
|
Entry{Hostname: "A"},
|
|
Entry{Hostname: "B"},
|
|
}
|
|
|
|
es.tombstone(router.UnknownPeerName, func(e *Entry) bool {
|
|
return e.Hostname == "B"
|
|
})
|
|
expected := Entries{
|
|
Entry{Hostname: "A"},
|
|
Entry{Hostname: "B", Version: 1, Tombstone: 1234},
|
|
}
|
|
require.Equal(t, es, expected)
|
|
}
|
|
|
|
func TestDelete(t *testing.T) {
|
|
es := Entries{
|
|
Entry{Hostname: "A"},
|
|
Entry{Hostname: "B"},
|
|
}
|
|
|
|
es.filter(func(e *Entry) bool {
|
|
return e.Hostname != "A"
|
|
})
|
|
expected := Entries{
|
|
Entry{Hostname: "B"},
|
|
}
|
|
require.Equal(t, es, expected)
|
|
}
|
|
|
|
func TestLookup(t *testing.T) {
|
|
es := Entries{
|
|
Entry{Hostname: "A"},
|
|
Entry{Hostname: "B", ContainerID: "foo"},
|
|
Entry{Hostname: "B", ContainerID: "bar"},
|
|
Entry{Hostname: "C"},
|
|
}
|
|
|
|
have := es.lookup("B")
|
|
want := Entries{
|
|
Entry{Hostname: "B", ContainerID: "foo"},
|
|
Entry{Hostname: "B", ContainerID: "bar"},
|
|
}
|
|
require.Equal(t, have, want)
|
|
}
|