add tests in ipvs/ipset_test.go

pkg/proxy/ipvs/ipset_test.go

@@ -18,6 +18,9 @@ package ipvs
 
 import (
 	"testing"
+
+	utilipset "k8s.io/kubernetes/pkg/util/ipset"
+	fakeipset "k8s.io/kubernetes/pkg/util/ipset/testing"
 )
 
 func TestCheckIPSetVersion(t *testing.T) {
@@ -47,3 +50,133 @@ func TestCheckIPSetVersion(t *testing.T) {
 		}
 	}
 }
+
+const testIPSetVersion = "v6.19"
+
+func TestSyncIPSetEntries(t *testing.T) {
+	testCases := []struct {
+		setName         string
+		setType         utilipset.Type
+		ipv6            bool
+		activeEntries   []string
+		currentEntries  []string
+		expectedEntries []string
+	}{
+		{ // case 0
+			setName:         "foo",
+			setType:         utilipset.HashIPPort,
+			ipv6:            false,
+			activeEntries:   []string{"172.17.0.4,tcp:80"},
+			currentEntries:  nil,
+			expectedEntries: []string{"172.17.0.4,tcp:80"},
+		},
+		{ // case 1
+			setName:         "abz",
+			setType:         utilipset.HashIPPort,
+			ipv6:            true,
+			activeEntries:   []string{"FE80::0202:B3FF:FE1E:8329,tcp:80"},
+			currentEntries:  []string{"FE80::0202:B3FF:FE1E:8329,tcp:80"},
+			expectedEntries: []string{"FE80::0202:B3FF:FE1E:8329,tcp:80"},
+		},
+		{ // case 2
+			setName:         "bca",
+			setType:         utilipset.HashIPPort,
+			ipv6:            false,
+			activeEntries:   []string{"172.17.0.4,tcp:80", "172.17.0.5,tcp:80"},
+			currentEntries:  []string{"172.17.0.5,udp:53"},
+			expectedEntries: []string{"172.17.0.4,tcp:80", "172.17.0.5,tcp:80"},
+		},
+		{ // case 3
+			setName:         "bar",
+			setType:         utilipset.HashIPPortIP,
+			ipv6:            false,
+			activeEntries:   []string{"172.17.0.4,tcp:80:172.17.0.4"},
+			currentEntries:  []string{"172.17.0.4,tcp:80:172.17.0.4"},
+			expectedEntries: []string{"172.17.0.4,tcp:80:172.17.0.4"},
+		},
+		{ // case 4
+			setName:         "baz",
+			setType:         utilipset.HashIPPortIP,
+			ipv6:            true,
+			activeEntries:   []string{"FE80:0000:0000:0000:0202:B3FF:FE1E:8329,tcp:8080:FE80:0000:0000:0000:0202:B3FF:FE1E:8329"},
+			currentEntries:  []string{"1111:0000:0000:0000:0202:B3FF:FE1E:8329,tcp:8081:1111:0000:0000:0000:0202:B3FF:FE1E:8329:8081"},
+			expectedEntries: []string{"FE80:0000:0000:0000:0202:B3FF:FE1E:8329,tcp:8080:FE80:0000:0000:0000:0202:B3FF:FE1E:8329"},
+		},
+		{ // case 5
+			setName:         "NOPE",
+			setType:         utilipset.HashIPPortIP,
+			ipv6:            false,
+			activeEntries:   []string{"172.17.0.4,tcp:80,172.17.0.9", "172.17.0.5,tcp:80,172.17.0.10"},
+			currentEntries:  nil,
+			expectedEntries: []string{"172.17.0.4,tcp:80,172.17.0.9", "172.17.0.5,tcp:80,172.17.0.10"},
+		},
+		{ // case 6
+			setName:         "ABC-DEF",
+			setType:         utilipset.HashIPPortNet,
+			ipv6:            false,
+			activeEntries:   []string{"172.17.0.4,tcp:80,172.17.0.0/16", "172.17.0.5,tcp:80,172.17.0.0/16"},
+			currentEntries:  nil,
+			expectedEntries: []string{"172.17.0.4,tcp:80,172.17.0.0/16", "172.17.0.5,tcp:80,172.17.0.0/16"},
+		},
+		{ // case 7
+			setName:         "zar",
+			setType:         utilipset.HashIPPortNet,
+			ipv6:            true,
+			activeEntries:   []string{"FE80::8329,tcp:8800,2001:db8::/32"},
+			currentEntries:  []string{"FE80::8329,tcp:8800,2001:db8::/32"},
+			expectedEntries: []string{"FE80::8329,tcp:8800,2001:db8::/32"},
+		},
+		{ // case 8
+			setName:         "bbb",
+			setType:         utilipset.HashIPPortNet,
+			ipv6:            true,
+			activeEntries:   nil,
+			currentEntries:  []string{"FE80::8329,udp:8801,2001:db8::/32"},
+			expectedEntries: nil,
+		},
+		{ // case 9
+			setName:         "AAA",
+			setType:         utilipset.BitmapPort,
+			activeEntries:   nil,
+			currentEntries:  []string{"80"},
+			expectedEntries: nil,
+		},
+		{ // case 10
+			setName:         "c-c-c",
+			setType:         utilipset.BitmapPort,
+			activeEntries:   []string{"8080", "9090"},
+			currentEntries:  []string{"80"},
+			expectedEntries: []string{"8080", "9090"},
+		},
+		{ // case 11
+			setName:         "NODE-PORT",
+			setType:         utilipset.BitmapPort,
+			activeEntries:   []string{"8080"},
+			currentEntries:  []string{"80", "9090", "8081", "8082"},
+			expectedEntries: []string{"8080"},
+		},
+	}
+
+	for i := range testCases {
+		set := NewIPSet(fakeipset.NewFake(testIPSetVersion), testCases[i].setName, testCases[i].setType, testCases[i].ipv6)
+
+		if err := set.handle.CreateSet(&set.IPSet, true); err != nil {
+			t.Errorf("Unexpected error: %v", err)
+		}
+		for _, entry := range testCases[i].expectedEntries {
+			set.handle.AddEntry(entry, testCases[i].setName, true)
+		}
+
+		set.activeEntries.Insert(testCases[i].activeEntries...)
+		set.syncIPSetEntries()
+		for _, entry := range testCases[i].expectedEntries {
+			found, err := set.handle.TestEntry(entry, testCases[i].setName)
+			if err != nil {
+				t.Errorf("Unexpected error: %v", err)
+			}
+			if !found {
+				t.Errorf("Unexpected entry 172.17.0.4,tcp:80 not found in set foo")
+			}
+		}
+	}
+}