95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package bridge
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/docker/libnetwork/netutils"
|
|
"github.com/docker/libnetwork/testutils"
|
|
"github.com/vishvananda/netlink"
|
|
)
|
|
|
|
func TestSetupNewBridge(t *testing.T) {
|
|
defer testutils.SetupTestOSContext(t)()
|
|
|
|
nh, err := netlink.NewHandle()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer nh.Delete()
|
|
|
|
config := &networkConfiguration{BridgeName: DefaultBridgeName}
|
|
br := &bridgeInterface{nlh: nh}
|
|
|
|
if err := setupDevice(config, br); err != nil {
|
|
t.Fatalf("Bridge creation failed: %v", err)
|
|
}
|
|
if br.Link == nil {
|
|
t.Fatal("bridgeInterface link is nil (expected valid link)")
|
|
}
|
|
if _, err := nh.LinkByName(DefaultBridgeName); err != nil {
|
|
t.Fatalf("Failed to retrieve bridge device: %v", err)
|
|
}
|
|
if br.Link.Attrs().Flags&net.FlagUp == net.FlagUp {
|
|
t.Fatal("bridgeInterface should be created down")
|
|
}
|
|
}
|
|
|
|
func TestSetupNewNonDefaultBridge(t *testing.T) {
|
|
defer testutils.SetupTestOSContext(t)()
|
|
|
|
nh, err := netlink.NewHandle()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer nh.Delete()
|
|
|
|
config := &networkConfiguration{BridgeName: "test0", DefaultBridge: true}
|
|
br := &bridgeInterface{nlh: nh}
|
|
|
|
err = setupDevice(config, br)
|
|
if err == nil {
|
|
t.Fatal("Expected bridge creation failure with \"non default name\", succeeded")
|
|
}
|
|
|
|
if _, ok := err.(NonDefaultBridgeExistError); !ok {
|
|
t.Fatalf("Did not fail with expected error. Actual error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSetupDeviceUp(t *testing.T) {
|
|
defer testutils.SetupTestOSContext(t)()
|
|
|
|
nh, err := netlink.NewHandle()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer nh.Delete()
|
|
|
|
config := &networkConfiguration{BridgeName: DefaultBridgeName}
|
|
br := &bridgeInterface{nlh: nh}
|
|
|
|
if err := setupDevice(config, br); err != nil {
|
|
t.Fatalf("Bridge creation failed: %v", err)
|
|
}
|
|
if err := setupDeviceUp(config, br); err != nil {
|
|
t.Fatalf("Failed to up bridge device: %v", err)
|
|
}
|
|
|
|
lnk, _ := nh.LinkByName(DefaultBridgeName)
|
|
if lnk.Attrs().Flags&net.FlagUp != net.FlagUp {
|
|
t.Fatal("bridgeInterface should be up")
|
|
}
|
|
}
|
|
|
|
func TestGenerateRandomMAC(t *testing.T) {
|
|
defer testutils.SetupTestOSContext(t)()
|
|
|
|
mac1 := netutils.GenerateRandomMAC()
|
|
mac2 := netutils.GenerateRandomMAC()
|
|
if bytes.Compare(mac1, mac2) == 0 {
|
|
t.Fatalf("Generated twice the same MAC address %v", mac1)
|
|
}
|
|
}
|