76 строки
2.1 KiB
Go
76 строки
2.1 KiB
Go
package network
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/Azure/azure-container-networking/netlink"
|
|
"github.com/Azure/azure-container-networking/ovsctl"
|
|
"github.com/Azure/azure-container-networking/platform"
|
|
)
|
|
|
|
const (
|
|
bridgeName = "testbridge"
|
|
hostIntf = "testintf"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func TestAddRoutes(t *testing.T) {
|
|
ovsctlClient := ovsctl.NewMockOvsctl(false, "", "")
|
|
ovsClient := NewOVSClient(bridgeName, hostIntf, ovsctlClient,
|
|
netlink.NewMockNetlink(false, ""), platform.NewMockExecClient(false))
|
|
|
|
if err := ovsClient.AddRoutes(nil, ""); err != nil {
|
|
t.Errorf("Add routes failed")
|
|
}
|
|
}
|
|
|
|
func TestDeleteBridge(t *testing.T) {
|
|
ovsctlClient := ovsctl.NewMockOvsctl(false, "", "")
|
|
|
|
ovsClient := NewOVSClient(bridgeName, hostIntf, ovsctlClient,
|
|
netlink.NewMockNetlink(false, ""), platform.NewMockExecClient(false))
|
|
if err := ovsClient.DeleteBridge(); err != nil {
|
|
t.Errorf("Error deleting the OVS bridge: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestAddL2Rules(t *testing.T) {
|
|
ovsctlClient := ovsctl.NewMockOvsctl(false, "", "")
|
|
extIf := externalInterface{
|
|
Name: hostIntf,
|
|
MacAddress: []byte("2C:54:91:88:C9:E3"),
|
|
}
|
|
|
|
ovsClient := NewOVSClient(bridgeName, hostIntf, ovsctlClient,
|
|
netlink.NewMockNetlink(false, ""), platform.NewMockExecClient(false))
|
|
if err := ovsClient.AddL2Rules(&extIf); err != nil {
|
|
t.Errorf("Unable to add L2 rules: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDeleteL2Rules(t *testing.T) {
|
|
ovsctlClient := ovsctl.NewMockOvsctl(false, "", "")
|
|
extIf := externalInterface{
|
|
Name: hostIntf,
|
|
MacAddress: []byte("2C:54:91:88:C9:E3"),
|
|
}
|
|
|
|
ovsClient := NewOVSClient(bridgeName, hostIntf, ovsctlClient,
|
|
netlink.NewMockNetlink(false, ""), platform.NewMockExecClient(false))
|
|
ovsClient.DeleteL2Rules(&extIf)
|
|
}
|
|
|
|
func TestSetBridgeMasterToHostInterface(t *testing.T) {
|
|
ovsctlClient := ovsctl.NewMockOvsctl(false, "", "")
|
|
|
|
ovsClient := NewOVSClient(bridgeName, hostIntf, ovsctlClient,
|
|
netlink.NewMockNetlink(false, ""), platform.NewMockExecClient(false))
|
|
if err := ovsClient.SetBridgeMasterToHostInterface(); err != nil {
|
|
t.Errorf("Unable to set bridge master to host intf: %v", err)
|
|
}
|
|
}
|