Test: add some unit-test

openyurtio · Jul 26, 2022 · ba0324d · ba0324d
1 parent 857d216
commit ba0324d
Show file tree

Hide file tree

Showing 3 changed files with 107 additions and 10 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -37,7 +37,7 @@ jobs:
             ${{ runner.os }}-buildx-
 
       - name: Unit test
-        run: go test -v -short ./pkg/... ./cmd/... -coverprofile cover.out
+        run: sudo go test -v -short ./pkg/... ./cmd/... -coverprofile cover.out
 
       - name: Publish Unit Test Coverage
         uses: codecov/codecov-action@v3
@@ -97,7 +97,7 @@ jobs:
             ${{ runner.os }}-buildx-
 
       - name: Unit test
-        run: go test ./...
+        run: sudo go test -v -short ./pkg/... ./cmd/...
 
       - name: Docker meta
         id: meta

diff --git a/pkg/networkengine/routedriver/vxlan/vxlan_test.go b/pkg/networkengine/routedriver/vxlan/vxlan_test.go
@@ -20,11 +20,13 @@ import (
 	"net"
 	"testing"
 
+	"github.com/openyurtio/raven-controller-manager/pkg/ravencontroller/apis/raven/v1alpha1"
 	"github.com/stretchr/testify/assert"
 	"github.com/vishvananda/netlink"
 
 	networkutil "github.com/openyurtio/raven/pkg/networkengine/util"
 	netlinkutil "github.com/openyurtio/raven/pkg/networkengine/util/netlink"
+	"github.com/openyurtio/raven/pkg/types"
 )
 
 func Test_applyRoutes(t *testing.T) {
@@ -319,3 +321,97 @@ func Test_applyRule(t *testing.T) {
 		})
 	}
 }
+
+func TestVxlan_Apply(t *testing.T) {
+	network := &types.Network{
+		LocalEndpoint: &types.Endpoint{
+			GatewayName: "gw-1",
+			NodeName:    "node-1",
+			Subnets: []string{
+				"10.10.1.0/24",
+			},
+			PrivateIP: "192.168.1.1",
+		},
+		LocalNodeInfo: map[types.NodeName]*v1alpha1.NodeInfo{
+			"node-1": {
+				NodeName:  "node-1",
+				PrivateIP: "192.168.1.1",
+				Subnets: []string{
+					"10.10.1.0/24",
+				},
+			},
+			"node-4": {
+				NodeName:  "node-4",
+				PrivateIP: "192.168.1.4",
+				Subnets: []string{
+					"10.10.4.0/24",
+				},
+			},
+		},
+		RemoteEndpoints: map[types.GatewayName]*types.Endpoint{
+			"gw-2": {
+				GatewayName: "gw-2",
+				NodeName:    "node-2",
+				Subnets: []string{
+					"10.10.2.0/24",
+				},
+				PrivateIP: "192.168.1.2",
+			},
+			"gw-3": {
+				GatewayName: "gw-3",
+				NodeName:    "remoteGwNode12",
+				Subnets: []string{
+					"10.10.3.0/24",
+				},
+				PrivateIP: "192.168.1.3",
+			},
+		},
+		RemoteNodeInfo: map[types.NodeName]*v1alpha1.NodeInfo{
+			"node-2": {
+				NodeName:  "node-2",
+				PrivateIP: "192.168.1.2",
+				Subnets: []string{
+					"10.10.2.0/24",
+				},
+			},
+			"node-3": {
+				NodeName:  "node-3",
+				PrivateIP: "192.168.1.3",
+				Subnets: []string{
+					"10.10.3.0/24",
+				},
+			},
+		},
+	}
+
+	testcases := []struct {
+		name     string
+		nodeName types.NodeName
+		network  *types.Network
+	}{
+		{
+			name:     "test-gateway-vxlan-apply",
+			nodeName: "node-1",
+			network:  network,
+		},
+		{
+			name:     "test-non-gateway-vxlan-apply",
+			nodeName: "node-4",
+			network:  network,
+		},
+	}
+
+	for _, v := range testcases {
+		t.Run(v.name, func(t *testing.T) {
+			vx := vxlan{
+				nodeName: v.nodeName,
+			}
+			a := assert.New(t)
+			a.NoError(vx.Init())
+			a.NoError(vx.Apply(v.network, func() (int, error) {
+				return 1500, nil
+			}))
+			a.NoError(vx.Cleanup())
+		})
+	}
+}
diff --git a/pkg/networkengine/vpndriver/libreswan/libreswan_test.go b/pkg/networkengine/vpndriver/libreswan/libreswan_test.go
@@ -121,6 +121,9 @@ func TestLibreswan_Apply(t *testing.T) {
 					},
 				},
 			},
+			findCentralGw: func(network *types.Network) *types.Endpoint {
+				return network.LocalEndpoint
+			},
 		}, {
 			name:     "all-NATed-gateway",
 			nodeName: "localGwNode",
@@ -155,6 +158,9 @@ func TestLibreswan_Apply(t *testing.T) {
 					},
 				},
 			},
+			findCentralGw: func(network *types.Network) *types.Endpoint {
+				return nil
+			},
 		}, {
 			name:     "NATed-gateway-connect-to-central-gateway",
 			nodeName: "localGwNode",
@@ -198,6 +204,9 @@ func TestLibreswan_Apply(t *testing.T) {
 					},
 				},
 			},
+			findCentralGw: func(network *types.Network) *types.Endpoint {
+				return network.RemoteEndpoints["centralGw"]
+			},
 		},
 		{
 			name:     "NATed-gateway-connect-to-central-gateway-and-not-NATed-gateway",
@@ -353,14 +362,6 @@ func TestLibreswan_Apply(t *testing.T) {
 	}
 	for _, v := range testcases {
 		t.Run(v.name, func(t *testing.T) {
-			if v.name != "central-gateway-connect-to-NATed-gateways-and-not-NATed-gateway" {
-				return
-			}
-			if v.findCentralGw != nil {
-				findCentralGw = func(network *types.Network) *types.Endpoint {
-					return network.LocalEndpoint
-				}
-			}
 			var cleanup bool
 			netlinkutil.XfrmPolicyFlush = func() error {
 				cleanup = true