modify snat client to accept interfaces for unit testing

Author: QxBytes

network/ovs_endpoint_snatroute_linux.go

@@ -33,6 +33,7 @@ func (client *OVSEndpointClient) NewSnatClient(snatBridgeIP, localIP string, epI
 			client.netlink,
 			client.plClient,
 			client.iptablesClient,
+			client.netioshim,
 		)
 	}
 }

network/snat/snat_linux.go

@@ -11,6 +11,7 @@ import (
 	"github.com/Azure/azure-container-networking/cni/log"
 	"github.com/Azure/azure-container-networking/ebtables"
 	"github.com/Azure/azure-container-networking/iptables"
+	"github.com/Azure/azure-container-networking/netio"
 	"github.com/Azure/azure-container-networking/netlink"
 	"github.com/Azure/azure-container-networking/network/networkutils"
 	"github.com/Azure/azure-container-networking/platform"
@@ -55,6 +56,7 @@ type Client struct {
 	netlink                netlink.NetlinkInterface
 	plClient               platform.ExecClient
 	ipTablesClient         ipTablesClient
+	netioClient            netio.NetIOInterface
 }
 
 func NewSnatClient(hostIfName string,
@@ -67,6 +69,7 @@ func NewSnatClient(hostIfName string,
 	nl netlink.NetlinkInterface,
 	plClient platform.ExecClient,
 	iptc ipTablesClient,
+	nio netio.NetIOInterface,
 ) Client {
 	snatClient := Client{
 		hostSnatVethName:       hostIfName,
@@ -78,6 +81,7 @@ func NewSnatClient(hostIfName string,
 		netlink:                nl,
 		plClient:               plClient,
 		ipTablesClient:         iptc,
+		netioClient:            nio,
 	}
 
 	snatClient.SkipAddressesFromBlock = append(snatClient.SkipAddressesFromBlock, skipAddressesFromBlock...)
@@ -223,7 +227,7 @@ func (client *Client) AllowInboundFromHostToNC() error {
 		return newErrorSnatClient(err.Error())
 	}
 
-	snatContainerVeth, err := net.InterfaceByName(client.containerSnatVethName)
+	snatContainerVeth, err := client.netioClient.GetNetworkInterfaceByName(client.containerSnatVethName)
 	if err != nil {
 		logger.Info("Could not find interface", zap.String("containerSnatVethName", client.containerSnatVethName))
 		return errors.Wrap(newErrorSnatClient(err.Error()), "could not find container snat veth name for allow host to nc")
@@ -323,7 +327,7 @@ func (client *Client) AllowInboundFromNCToHost() error {
 		return err
 	}
 
-	snatContainerVeth, err := net.InterfaceByName(client.containerSnatVethName)
+	snatContainerVeth, err := client.netioClient.GetNetworkInterfaceByName(client.containerSnatVethName)
 	if err != nil {
 		logger.Info("Could not find interface", zap.String("containerSnatVethName", client.containerSnatVethName))
 		return errors.Wrap(newErrorSnatClient(err.Error()), "could not find container snat veth name for allow nc to host")
@@ -424,7 +428,7 @@ func (client *Client) DropArpForSnatBridgeApipaRange(snatBridgeIP, azSnatVethIfN
 
 // This function creates linux bridge which will be used for outbound connectivity by NCs
 func (client *Client) createSnatBridge(snatBridgeIP, hostPrimaryMac string) error {
-	_, err := net.InterfaceByName(SnatBridgeName)
+	_, err := client.netioClient.GetNetworkInterfaceByName(SnatBridgeName)
 	if err == nil {
 		logger.Info("Snat Bridge already exists")
 	} else {

network/snat/snat_linux_test.go

@@ -4,12 +4,28 @@ import (
 	"os"
 	"testing"
 
-	"github.com/Azure/azure-container-networking/iptables"
+	"github.com/Azure/azure-container-networking/netio"
 	"github.com/Azure/azure-container-networking/netlink"
 )
 
 var anyInterface = "dummy"
 
+type mockIPTablesClient struct {
+}
+
+func (c mockIPTablesClient) InsertIptableRule(version, tableName, chainName, match, target string) error {
+	return nil
+}
+func (c mockIPTablesClient) AppendIptableRule(version, tableName, chainName, match, target string) error {
+	return nil
+}
+func (c mockIPTablesClient) DeleteIptableRule(version, tableName, chainName, match, target string) error {
+	return nil
+}
+func (c mockIPTablesClient) CreateChain(version, tableName, chainName string) error {
+	return nil
+}
+
 func TestMain(m *testing.M) {
 	exitCode := m.Run()
 
@@ -18,16 +34,22 @@ func TestMain(m *testing.M) {
 	os.Exit(exitCode)
 }
 
-func TestAllowInboundFromHostToNC(t *testing.T) {
-	nl := netlink.NewNetlink()
-	iptc := iptables.NewClient()
-	client := &Client{
+func GetTestClient(nl netlink.NetlinkInterface, iptc ipTablesClient, nio netio.NetIOInterface) *Client {
+	return &Client{
 		SnatBridgeIP:          "169.254.0.1/16",
 		localIP:               "169.254.0.4/16",
 		containerSnatVethName: anyInterface,
 		netlink:               nl,
 		ipTablesClient:        iptc,
+		netioClient:           nio,
 	}
+}
+
+func TestAllowInboundFromHostToNC(t *testing.T) {
+	nl := netlink.NewMockNetlink(false, "")
+	iptc := &mockIPTablesClient{}
+	nio := netio.NewMockNetIO(false, 0)
+	client := GetTestClient(nl, iptc, nio)
 
 	if err := nl.AddLink(&netlink.DummyLink{
 		LinkInfo: netlink.LinkInfo{
@@ -65,18 +87,18 @@ func TestAllowInboundFromHostToNC(t *testing.T) {
 	if err := nl.DeleteLink(SnatBridgeName); err != nil {
 		t.Errorf("Error removing snat bridge: %v", err)
 	}
+
+	client.netioClient = netio.NewMockNetIO(true, 1)
+	if err := client.AllowInboundFromHostToNC(); err == nil {
+		t.Errorf("Expected error when interface not found in allow host to nc but got nil")
+	}
 }
 
 func TestAllowInboundFromNCToHost(t *testing.T) {
-	nl := netlink.NewNetlink()
-	iptc := iptables.NewClient()
-	client := &Client{
-		SnatBridgeIP:          "169.254.0.1/16",
-		localIP:               "169.254.0.4/16",
-		containerSnatVethName: anyInterface,
-		netlink:               nl,
-		ipTablesClient:        iptc,
-	}
+	nl := netlink.NewMockNetlink(false, "")
+	iptc := &mockIPTablesClient{}
+	nio := netio.NewMockNetIO(false, 0)
+	client := GetTestClient(nl, iptc, nio)
 
 	if err := nl.AddLink(&netlink.DummyLink{
 		LinkInfo: netlink.LinkInfo{
@@ -114,4 +136,9 @@ func TestAllowInboundFromNCToHost(t *testing.T) {
 	if err := nl.DeleteLink(SnatBridgeName); err != nil {
 		t.Errorf("Error removing snat bridge: %v", err)
 	}
+
+	client.netioClient = netio.NewMockNetIO(true, 1)
+	if err := client.AllowInboundFromNCToHost(); err == nil {
+		t.Errorf("Expected error when interface not found in allow nc to host but got nil")
+	}
 }

network/transparent_vlan_endpoint_snatroute_linux.go

@@ -21,6 +21,7 @@ func (client *TransparentVlanEndpointClient) NewSnatClient(snatBridgeIP, localIP
 			client.netlink,
 			client.plClient,
 			client.iptablesClient,
+			client.netioshim,
 		)
 	}
 }