mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-08-08 03:33:56 +00:00
proxy/ipvs: refactor TestClusterIP to use test tables
Signed-off-by: Andrew Sy Kim <kiman@vmware.com>
This commit is contained in:
Andrew Sy Kim
parent
75d5189612
commit
9af797c51e
@ -606,154 +606,161 @@ func TestNodePortNoEndpoint(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClusterIPNoEndpoint(t *testing.T) {
|
|
||||||
ipt := iptablestest.NewFake()
|
|
||||||
ipvs := ipvstest.NewFake()
|
|
||||||
ipset := ipsettest.NewFake(testIPSetVersion)
|
|
||||||
fp := NewFakeProxier(ipt, ipvs, ipset, nil, nil)
|
|
||||||
svcIP := "10.20.30.41"
|
|
||||||
svcPort := 80
|
|
||||||
svcPortName := proxy.ServicePortName{
|
|
||||||
NamespacedName: makeNSN("ns1", "svc1"),
|
|
||||||
Port: "p80",
|
|
||||||
}
|
|
||||||
|
|
||||||
makeServiceMap(fp,
|
|
||||||
makeTestService(svcPortName.Namespace, svcPortName.Namespace, func(svc *v1.Service) {
|
|
||||||
svc.Spec.ClusterIP = svcIP
|
|
||||||
svc.Spec.Ports = []v1.ServicePort{{
|
|
||||||
Name: svcPortName.Port,
|
|
||||||
Port: int32(svcPort),
|
|
||||||
Protocol: v1.ProtocolTCP,
|
|
||||||
}}
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
makeEndpointsMap(fp)
|
|
||||||
fp.syncProxyRules()
|
|
||||||
|
|
||||||
// check ipvs service and destinations
|
|
||||||
services, err := ipvs.GetVirtualServers()
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Failed to get ipvs services, err: %v", err)
|
|
||||||
}
|
|
||||||
if len(services) != 1 {
|
|
||||||
t.Errorf("Expect 1 ipvs services, got %d", len(services))
|
|
||||||
} else {
|
|
||||||
if services[0].Address.String() != svcIP || services[0].Port != uint16(svcPort) || services[0].Protocol != string(v1.ProtocolTCP) {
|
|
||||||
t.Errorf("Unexpected mismatch service")
|
|
||||||
} else {
|
|
||||||
destinations, _ := ipvs.GetRealServers(services[0])
|
|
||||||
if len(destinations) != 0 {
|
|
||||||
t.Errorf("Unexpected %d destinations, expect 0 destinations", len(destinations))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestClusterIP(t *testing.T) {
|
func TestClusterIP(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
services []*v1.Service
|
||||||
|
endpoints []*v1.Endpoints
|
||||||
|
expectedIPVS *ipvstest.FakeIPVS
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "2 services with Cluster IP, each with endpoints",
|
||||||
|
services: []*v1.Service{
|
||||||
|
makeTestService("ns1", "svc1", func(svc *v1.Service) {
|
||||||
|
svc.Spec.ClusterIP = "10.20.30.41"
|
||||||
|
svc.Spec.Ports = []v1.ServicePort{{
|
||||||
|
Name: "p80",
|
||||||
|
Port: int32(80),
|
||||||
|
Protocol: v1.ProtocolTCP,
|
||||||
|
}}
|
||||||
|
}),
|
||||||
|
makeTestService("ns2", "svc2", func(svc *v1.Service) {
|
||||||
|
svc.Spec.ClusterIP = "1002:ab8::2:1"
|
||||||
|
svc.Spec.Ports = []v1.ServicePort{{
|
||||||
|
Name: "p8080",
|
||||||
|
Port: int32(8080),
|
||||||
|
Protocol: v1.ProtocolTCP,
|
||||||
|
}}
|
||||||
|
}),
|
||||||
|
},
|
||||||
|
endpoints: []*v1.Endpoints{
|
||||||
|
makeTestEndpoints("ns1", "svc1", func(ept *v1.Endpoints) {
|
||||||
|
ept.Subsets = []v1.EndpointSubset{{
|
||||||
|
Addresses: []v1.EndpointAddress{{
|
||||||
|
IP: "10.180.0.1",
|
||||||
|
}},
|
||||||
|
Ports: []v1.EndpointPort{{
|
||||||
|
Name: "p80",
|
||||||
|
Port: int32(80),
|
||||||
|
}},
|
||||||
|
}}
|
||||||
|
}),
|
||||||
|
makeTestEndpoints("ns2", "svc2", func(ept *v1.Endpoints) {
|
||||||
|
ept.Subsets = []v1.EndpointSubset{{
|
||||||
|
Addresses: []v1.EndpointAddress{{
|
||||||
|
IP: "1009:ab8::5:6",
|
||||||
|
}},
|
||||||
|
Ports: []v1.EndpointPort{{
|
||||||
|
Name: "p8080",
|
||||||
|
Port: int32(8080),
|
||||||
|
}},
|
||||||
|
}}
|
||||||
|
}),
|
||||||
|
},
|
||||||
|
expectedIPVS: &ipvstest.FakeIPVS{
|
||||||
|
Services: map[ipvstest.ServiceKey]*utilipvs.VirtualServer{
|
||||||
|
{
|
||||||
|
IP: "10.20.30.41",
|
||||||
|
Port: 80,
|
||||||
|
Protocol: "TCP",
|
||||||
|
}: {
|
||||||
|
Address: net.ParseIP("10.20.30.41"),
|
||||||
|
Protocol: "TCP",
|
||||||
|
Port: uint16(80),
|
||||||
|
Scheduler: "rr",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
IP: "1002:ab8::2:1",
|
||||||
|
Port: 8080,
|
||||||
|
Protocol: "TCP",
|
||||||
|
}: {
|
||||||
|
Address: net.ParseIP("1002:ab8::2:1"),
|
||||||
|
Protocol: "TCP",
|
||||||
|
Port: uint16(8080),
|
||||||
|
Scheduler: "rr",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Destinations: map[ipvstest.ServiceKey][]*utilipvs.RealServer{
|
||||||
|
{
|
||||||
|
IP: "10.20.30.41",
|
||||||
|
Port: 80,
|
||||||
|
Protocol: "TCP",
|
||||||
|
}: {
|
||||||
|
{
|
||||||
|
Address: net.ParseIP("10.180.0.1"),
|
||||||
|
Port: uint16(80),
|
||||||
|
Weight: 1,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
IP: "1002:ab8::2:1",
|
||||||
|
Port: 8080,
|
||||||
|
Protocol: "TCP",
|
||||||
|
}: {
|
||||||
|
{
|
||||||
|
Address: net.ParseIP("1009:ab8::5:6"),
|
||||||
|
Port: uint16(8080),
|
||||||
|
Weight: 1,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "cluster IP service with no endpoints",
|
||||||
|
services: []*v1.Service{
|
||||||
|
makeTestService("ns1", "svc1", func(svc *v1.Service) {
|
||||||
|
svc.Spec.ClusterIP = "10.20.30.41"
|
||||||
|
svc.Spec.Ports = []v1.ServicePort{{
|
||||||
|
Name: "p80",
|
||||||
|
Port: int32(80),
|
||||||
|
Protocol: v1.ProtocolTCP,
|
||||||
|
}}
|
||||||
|
}),
|
||||||
|
},
|
||||||
|
endpoints: []*v1.Endpoints{},
|
||||||
|
expectedIPVS: &ipvstest.FakeIPVS{
|
||||||
|
Services: map[ipvstest.ServiceKey]*utilipvs.VirtualServer{
|
||||||
|
{
|
||||||
|
IP: "10.20.30.41",
|
||||||
|
Port: 80,
|
||||||
|
Protocol: "TCP",
|
||||||
|
}: {
|
||||||
|
Address: net.ParseIP("10.20.30.41"),
|
||||||
|
Protocol: "TCP",
|
||||||
|
Port: uint16(80),
|
||||||
|
Scheduler: "rr",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Destinations: map[ipvstest.ServiceKey][]*utilipvs.RealServer{
|
||||||
|
{
|
||||||
|
IP: "10.20.30.41",
|
||||||
|
Port: 80,
|
||||||
|
Protocol: "TCP",
|
||||||
|
}: {},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
ipt := iptablestest.NewFake()
|
ipt := iptablestest.NewFake()
|
||||||
ipvs := ipvstest.NewFake()
|
ipvs := ipvstest.NewFake()
|
||||||
ipset := ipsettest.NewFake(testIPSetVersion)
|
ipset := ipsettest.NewFake(testIPSetVersion)
|
||||||
fp := NewFakeProxier(ipt, ipvs, ipset, nil, nil)
|
fp := NewFakeProxier(ipt, ipvs, ipset, nil, nil)
|
||||||
|
|
||||||
svcIPv4 := "10.20.30.41"
|
makeServiceMap(fp, test.services...)
|
||||||
svcPortV4 := 80
|
makeEndpointsMap(fp, test.endpoints...)
|
||||||
svcPortNameV4 := proxy.ServicePortName{
|
|
||||||
NamespacedName: makeNSN("ns1", "svc1"),
|
|
||||||
Port: "p80",
|
|
||||||
}
|
|
||||||
svcIPv6 := "1002:ab8::2:1"
|
|
||||||
svcPortV6 := 8080
|
|
||||||
svcPortNameV6 := proxy.ServicePortName{
|
|
||||||
NamespacedName: makeNSN("ns2", "svc2"),
|
|
||||||
Port: "p8080",
|
|
||||||
}
|
|
||||||
makeServiceMap(fp,
|
|
||||||
makeTestService(svcPortNameV4.Namespace, svcPortNameV4.Name, func(svc *v1.Service) {
|
|
||||||
svc.Spec.ClusterIP = svcIPv4
|
|
||||||
svc.Spec.Ports = []v1.ServicePort{{
|
|
||||||
Name: svcPortNameV4.Port,
|
|
||||||
Port: int32(svcPortV4),
|
|
||||||
Protocol: v1.ProtocolTCP,
|
|
||||||
}}
|
|
||||||
}),
|
|
||||||
makeTestService(svcPortNameV6.Namespace, svcPortNameV6.Name, func(svc *v1.Service) {
|
|
||||||
svc.Spec.ClusterIP = svcIPv6
|
|
||||||
svc.Spec.Ports = []v1.ServicePort{{
|
|
||||||
Name: svcPortNameV6.Port,
|
|
||||||
Port: int32(svcPortV6),
|
|
||||||
Protocol: v1.ProtocolTCP,
|
|
||||||
}}
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
|
|
||||||
epIPv4 := "10.180.0.1"
|
|
||||||
epIPv6 := "1009:ab8::5:6"
|
|
||||||
makeEndpointsMap(fp,
|
|
||||||
makeTestEndpoints(svcPortNameV4.Namespace, svcPortNameV4.Name, func(ept *v1.Endpoints) {
|
|
||||||
ept.Subsets = []v1.EndpointSubset{{
|
|
||||||
Addresses: []v1.EndpointAddress{{
|
|
||||||
IP: epIPv4,
|
|
||||||
}},
|
|
||||||
Ports: []v1.EndpointPort{{
|
|
||||||
Name: svcPortNameV4.Port,
|
|
||||||
Port: int32(svcPortV4),
|
|
||||||
}},
|
|
||||||
}}
|
|
||||||
}),
|
|
||||||
makeTestEndpoints(svcPortNameV6.Namespace, svcPortNameV6.Name, func(ept *v1.Endpoints) {
|
|
||||||
ept.Subsets = []v1.EndpointSubset{{
|
|
||||||
Addresses: []v1.EndpointAddress{{
|
|
||||||
IP: epIPv6,
|
|
||||||
}},
|
|
||||||
Ports: []v1.EndpointPort{{
|
|
||||||
Name: svcPortNameV6.Port,
|
|
||||||
Port: int32(svcPortV6),
|
|
||||||
}},
|
|
||||||
}}
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
|
|
||||||
fp.syncProxyRules()
|
fp.syncProxyRules()
|
||||||
|
|
||||||
// check ipvs service and destinations
|
if !reflect.DeepEqual(ipvs, test.expectedIPVS) {
|
||||||
services, err := ipvs.GetVirtualServers()
|
t.Logf("actual ipvs state: %v", ipvs)
|
||||||
if err != nil {
|
t.Logf("expected ipvs state: %v", test.expectedIPVS)
|
||||||
t.Errorf("Failed to get ipvs services, err: %v", err)
|
t.Errorf("unexpected IPVS state")
|
||||||
}
|
|
||||||
if len(services) != 2 {
|
|
||||||
t.Errorf("Expect 2 ipvs services, got %d", len(services))
|
|
||||||
}
|
|
||||||
for i := range services {
|
|
||||||
// Check services
|
|
||||||
if services[i].Address.String() == svcIPv4 {
|
|
||||||
if services[i].Port != uint16(svcPortV4) || services[i].Protocol != string(v1.ProtocolTCP) {
|
|
||||||
t.Errorf("Unexpected mismatch service")
|
|
||||||
}
|
|
||||||
// Check destinations
|
|
||||||
destinations, _ := ipvs.GetRealServers(services[i])
|
|
||||||
if len(destinations) != 1 {
|
|
||||||
t.Errorf("Expected 1 destinations, got %d destinations", len(destinations))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if destinations[0].Address.String() != epIPv4 || destinations[0].Port != uint16(svcPortV4) {
|
|
||||||
t.Errorf("Unexpected mismatch destinations")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if services[i].Address.String() == svcIPv6 {
|
|
||||||
if services[i].Port != uint16(svcPortV6) || services[i].Protocol != string(v1.ProtocolTCP) {
|
|
||||||
t.Errorf("Unexpected mismatch service")
|
|
||||||
}
|
|
||||||
// Check destinations
|
|
||||||
destinations, _ := ipvs.GetRealServers(services[i])
|
|
||||||
if len(destinations) != 1 {
|
|
||||||
t.Errorf("Expected 1 destinations, got %d destinations", len(destinations))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if destinations[0].Address.String() != epIPv6 || destinations[0].Port != uint16(svcPortV6) {
|
|
||||||
t.Errorf("Unexpected mismatch destinations")
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user