mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
0066225da5
* integrated hostdiscovery package with the new Docker Discovery * Integrated hostdiscovery package with libnetwork core * removed libnetwork_discovery tag * Introduced driver apis for discovery events * moved overlay driver to make use of the discovery events * Using Docker Discovery service. * Changed integration-tests to make use of the new discovery Signed-off-by: Madhu Venugopal <madhu@docker.com>
77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
package hostdiscovery
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
mapset "github.com/deckarep/golang-set"
|
|
_ "github.com/docker/libnetwork/testutils"
|
|
|
|
"github.com/docker/docker/pkg/discovery"
|
|
)
|
|
|
|
func TestDiff(t *testing.T) {
|
|
existing := mapset.NewSetFromSlice([]interface{}{"1.1.1.1", "2.2.2.2"})
|
|
addedIP := "3.3.3.3"
|
|
updated := existing.Clone()
|
|
updated.Add(addedIP)
|
|
|
|
added, removed := diff(existing, updated)
|
|
if len(added) != 1 {
|
|
t.Fatalf("Diff failed for an Add update. Expecting 1 element, but got %d elements", len(added))
|
|
}
|
|
if added[0].String() != addedIP {
|
|
t.Fatalf("Expecting : %v, Got : %v", addedIP, added[0])
|
|
}
|
|
if len(removed) > 0 {
|
|
t.Fatalf("Diff failed for remove use-case. Expecting 0 element, but got %d elements", len(removed))
|
|
}
|
|
|
|
updated = mapset.NewSetFromSlice([]interface{}{addedIP})
|
|
added, removed = diff(existing, updated)
|
|
if len(removed) != 2 {
|
|
t.Fatalf("Diff failed for an remove update. Expecting 2 element, but got %d elements", len(removed))
|
|
}
|
|
if len(added) != 1 {
|
|
t.Fatalf("Diff failed for add use-case. Expecting 1 element, but got %d elements", len(added))
|
|
}
|
|
}
|
|
|
|
func TestAddedCallback(t *testing.T) {
|
|
hd := hostDiscovery{}
|
|
hd.nodes = mapset.NewSetFromSlice([]interface{}{"1.1.1.1"})
|
|
update := []*discovery.Entry{&discovery.Entry{Host: "1.1.1.1", Port: "0"}, &discovery.Entry{Host: "2.2.2.2", Port: "0"}}
|
|
|
|
added := false
|
|
removed := false
|
|
hd.processCallback(update, func(hosts []net.IP) { added = true }, func(hosts []net.IP) { removed = true })
|
|
if !added {
|
|
t.Fatalf("Expecting a Added callback notification. But none received")
|
|
}
|
|
}
|
|
|
|
func TestRemovedCallback(t *testing.T) {
|
|
hd := hostDiscovery{}
|
|
hd.nodes = mapset.NewSetFromSlice([]interface{}{"1.1.1.1", "2.2.2.2"})
|
|
update := []*discovery.Entry{&discovery.Entry{Host: "1.1.1.1", Port: "0"}}
|
|
|
|
added := false
|
|
removed := false
|
|
hd.processCallback(update, func(hosts []net.IP) { added = true }, func(hosts []net.IP) { removed = true })
|
|
if !removed {
|
|
t.Fatalf("Expecting a Removed callback notification. But none received")
|
|
}
|
|
}
|
|
|
|
func TestNoCallback(t *testing.T) {
|
|
hd := hostDiscovery{}
|
|
hd.nodes = mapset.NewSetFromSlice([]interface{}{"1.1.1.1", "2.2.2.2"})
|
|
update := []*discovery.Entry{&discovery.Entry{Host: "1.1.1.1", Port: "0"}, &discovery.Entry{Host: "2.2.2.2", Port: "0"}}
|
|
|
|
added := false
|
|
removed := false
|
|
hd.processCallback(update, func(hosts []net.IP) { added = true }, func(hosts []net.IP) { removed = true })
|
|
if added || removed {
|
|
t.Fatalf("Not expecting any callback notification. But received a callback")
|
|
}
|
|
}
|