github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-26 21:17:23 +00:00
Code Issues Projects Releases Wiki Activity

Service REST test: better IP and port alloc checks

Browse Source
This commit is contained in:
Tim Hockin 2021-07-01 23:01:36 -07:00
parent 43b13840db
commit 54b6a416fb
4 changed files with 46 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/api/service/testing/make.go
View File

@ -62,6 +62,7 @@ func SetTypeClusterIP(svc *api.Service) {
svc.Spec.Ports[i].NodePort = 0 svc.Spec.Ports[i].NodePort = 0
} }
svc.Spec.ExternalName = "" svc.Spec.ExternalName = ""
svc.Spec.ExternalTrafficPolicy = ""
} }
// SetTypeNodePort sets the service type to NodePort and clears other fields. // SetTypeNodePort sets the service type to NodePort and clears other fields.

2
pkg/registry/core/service/ipallocator/allocator.go
View File

@ -36,8 +36,6 @@ type Interface interface {
ForEach(func(net.IP)) ForEach(func(net.IP))
CIDR() net.IPNet CIDR() net.IPNet
IPFamily() api.IPFamily IPFamily() api.IPFamily
// For testing
Has(ip net.IP) bool Has(ip net.IP) bool
} }

2
pkg/registry/core/service/portallocator/allocator.go
View File

@ -34,8 +34,6 @@ type Interface interface {
AllocateNext() (int, error) AllocateNext() (int, error)
Release(int) error Release(int) error
ForEach(func(int)) ForEach(func(int))
// For testing
Has(int) bool Has(int) bool
} }

70
pkg/registry/core/service/storage/rest_test.go
View File

@ -259,7 +259,27 @@ func makeIPNet6(t *testing.T) *net.IPNet {
return net return net
} }
func ipIsAllocated(t *testing.T, alloc ipallocator.Interface, ipstr string) bool {
t.Helper()
ip := net.ParseIP(ipstr)
if ip == nil {
t.Errorf("error parsing IP %q", ipstr)
return false
}
return alloc.Has(ip)
}
func portIsAllocated(t *testing.T, alloc portallocator.Interface, port int32) bool {
t.Helper()
if port == 0 {
t.Errorf("port is 0")
return false
}
return alloc.Has(int(port))
}
func releaseServiceNodePorts(t *testing.T, ctx context.Context, svcName string, rest *REST) { func releaseServiceNodePorts(t *testing.T, ctx context.Context, svcName string, rest *REST) {
t.Helper()
obj, err := rest.Get(ctx, svcName, &metav1.GetOptions{}) obj, err := rest.Get(ctx, svcName, &metav1.GetOptions{})
if err != nil { if err != nil {
t.Fatalf("Unexpected error: %v", err) t.Fatalf("Unexpected error: %v", err)
@ -399,7 +419,7 @@ func TestServiceRegistryCreateDryRun(t *testing.T) {
for i, family := range tc.svc.Spec.IPFamilies { for i, family := range tc.svc.Spec.IPFamilies {
alloc := storage.serviceIPAllocatorsByFamily[family] alloc := storage.serviceIPAllocatorsByFamily[family]
if alloc.Has(net.ParseIP(tc.svc.Spec.ClusterIPs[i])) { if ipIsAllocated(t, alloc, tc.svc.Spec.ClusterIPs[i]) {
t.Errorf("unexpected side effect: ip allocated %v", tc.svc.Spec.ClusterIPs[i]) t.Errorf("unexpected side effect: ip allocated %v", tc.svc.Spec.ClusterIPs[i])
} }
} }
@ -428,7 +448,7 @@ func TestDryRunNodePort(t *testing.T) {
if createdSvc.Spec.Ports[0].NodePort == 0 { if createdSvc.Spec.Ports[0].NodePort == 0 {
t.Errorf("expected NodePort value assigned") t.Errorf("expected NodePort value assigned")
} }
if storage.serviceNodePorts.Has(int(createdSvc.Spec.Ports[0].NodePort)) { if portIsAllocated(t, storage.serviceNodePorts, createdSvc.Spec.Ports[0].NodePort) {
t.Errorf("unexpected side effect: NodePort allocated") t.Errorf("unexpected side effect: NodePort allocated")
} }
_, err = getService(storage, ctx, svc.Name, &metav1.GetOptions{}) _, err = getService(storage, ctx, svc.Name, &metav1.GetOptions{})
@ -455,7 +475,7 @@ func TestDryRunNodePort(t *testing.T) {
t.Errorf("Expected %v, but got %v", expectNodePorts, actualNodePorts) t.Errorf("Expected %v, but got %v", expectNodePorts, actualNodePorts)
} }
for i := range svc.Spec.Ports { for i := range svc.Spec.Ports {
if storage.serviceNodePorts.Has(int(svc.Spec.Ports[i].NodePort)) { if portIsAllocated(t, storage.serviceNodePorts, svc.Spec.Ports[i].NodePort) {
t.Errorf("unexpected side effect: NodePort allocated") t.Errorf("unexpected side effect: NodePort allocated")
} }
} }
@ -635,43 +655,45 @@ func TestServiceRegistryUpdateDryRun(t *testing.T) {
// Test dry run update request external name to node port // Test dry run update request external name to node port
new1 := svc.DeepCopy() new1 := svc.DeepCopy()
svctest.SetTypeNodePort(new1) svctest.SetTypeNodePort(new1)
updatedSvc, created, err := storage.Update(ctx, svc.Name, rest.DefaultUpdatedObjectInfo(new1), svctest.SetNodePorts(30001)(new1) // DryRun does not set port values yet
obj, created, err := storage.Update(ctx, new1.Name, rest.DefaultUpdatedObjectInfo(new1),
rest.ValidateAllObjectFunc, rest.ValidateAllObjectUpdateFunc, false, &metav1.UpdateOptions{DryRun: []string{metav1.DryRunAll}}) rest.ValidateAllObjectFunc, rest.ValidateAllObjectUpdateFunc, false, &metav1.UpdateOptions{DryRun: []string{metav1.DryRunAll}})
if err != nil { if err != nil {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
if updatedSvc == nil { if obj == nil {
t.Errorf("Expected non-nil object") t.Errorf("Expected non-nil object")
} }
if created { if created {
t.Errorf("expected not created") t.Errorf("expected not created")
} }
if storage.serviceNodePorts.Has(int(svc.Spec.Ports[0].NodePort)) { if portIsAllocated(t, storage.serviceNodePorts, new1.Spec.Ports[0].NodePort) {
t.Errorf("unexpected side effect: NodePort allocated") t.Errorf("unexpected side effect: NodePort allocated")
} }
// Test dry run update request external name to cluster ip // Test dry run update request external name to cluster ip
new2 := svc.DeepCopy() new2 := svc.DeepCopy()
svctest.SetTypeClusterIP(new2) svctest.SetTypeClusterIP(new2)
svctest.SetClusterIPs("1.2.3.4")(new2) // DryRun does not set IP values yet
_, _, err = storage.Update(ctx, svc.Name, rest.DefaultUpdatedObjectInfo(new2), _, _, err = storage.Update(ctx, svc.Name, rest.DefaultUpdatedObjectInfo(new2),
rest.ValidateAllObjectFunc, rest.ValidateAllObjectUpdateFunc, false, &metav1.UpdateOptions{DryRun: []string{metav1.DryRunAll}}) rest.ValidateAllObjectFunc, rest.ValidateAllObjectUpdateFunc, false, &metav1.UpdateOptions{DryRun: []string{metav1.DryRunAll}})
if err != nil { if err != nil {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
if storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].Has(net.ParseIP(svc.Spec.ClusterIP)) { if ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily], new2.Spec.ClusterIP) {
t.Errorf("unexpected side effect: ip allocated") t.Errorf("unexpected side effect: ip allocated")
} }
// Test dry run update request remove node port // Test dry run update request remove node port
obj, err = storage.Create(ctx, svctest.MakeService("foo2", svctest.SetTypeNodePort), rest.ValidateAllObjectFunc, &metav1.CreateOptions{}) obj, err = storage.Create(ctx, svctest.MakeService("foo2", svctest.SetTypeNodePort, svctest.SetNodePorts(30001)), rest.ValidateAllObjectFunc, &metav1.CreateOptions{})
if err != nil { if err != nil {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
svc = obj.(*api.Service) svc = obj.(*api.Service)
if !storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].Has(net.ParseIP(svc.Spec.ClusterIP)) { if !ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily], svc.Spec.ClusterIP) {
t.Errorf("expected IP to be allocated") t.Errorf("expected IP to be allocated")
} }
if !storage.serviceNodePorts.Has(int(svc.Spec.Ports[0].NodePort)) { if !portIsAllocated(t, storage.serviceNodePorts, svc.Spec.Ports[0].NodePort) {
t.Errorf("expected NodePort to be allocated") t.Errorf("expected NodePort to be allocated")
} }
@ -682,12 +704,12 @@ func TestServiceRegistryUpdateDryRun(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
if !storage.serviceNodePorts.Has(int(svc.Spec.Ports[0].NodePort)) { if !portIsAllocated(t, storage.serviceNodePorts, svc.Spec.Ports[0].NodePort) {
t.Errorf("unexpected side effect: NodePort unallocated") t.Errorf("unexpected side effect: NodePort unallocated")
} }
// Test dry run update request remove cluster ip // Test dry run update request remove cluster ip
obj, err = storage.Create(ctx, svctest.MakeService("foo3"), rest.ValidateAllObjectFunc, &metav1.CreateOptions{}) obj, err = storage.Create(ctx, svctest.MakeService("foo3", svctest.SetClusterIPs("1.2.3.4")), rest.ValidateAllObjectFunc, &metav1.CreateOptions{})
if err != nil { if err != nil {
t.Fatalf("expected no error: %v", err) t.Fatalf("expected no error: %v", err)
} }
@ -700,7 +722,7 @@ func TestServiceRegistryUpdateDryRun(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("expected no error: %v", err) t.Fatalf("expected no error: %v", err)
} }
if !storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].Has(net.ParseIP(svc.Spec.ClusterIP)) { if !ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily], svc.Spec.ClusterIP) {
t.Errorf("unexpected side effect: ip unallocated") t.Errorf("unexpected side effect: ip unallocated")
} }
} }
@ -862,14 +884,14 @@ func TestServiceRegistryDeleteDryRun(t *testing.T) {
if createdSvc.Spec.ClusterIP == "" { if createdSvc.Spec.ClusterIP == "" {
t.Fatalf("expected ClusterIP to be set") t.Fatalf("expected ClusterIP to be set")
} }
if !storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].Has(net.ParseIP(createdSvc.Spec.ClusterIP)) { if !ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily], createdSvc.Spec.ClusterIP) {
t.Errorf("expected ClusterIP to be allocated") t.Errorf("expected ClusterIP to be allocated")
} }
_, _, err = storage.Delete(ctx, svc.Name, rest.ValidateAllObjectFunc, &metav1.DeleteOptions{DryRun: []string{metav1.DryRunAll}}) _, _, err = storage.Delete(ctx, svc.Name, rest.ValidateAllObjectFunc, &metav1.DeleteOptions{DryRun: []string{metav1.DryRunAll}})
if err != nil { if err != nil {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
if !storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].Has(net.ParseIP(createdSvc.Spec.ClusterIP)) { if !ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily], createdSvc.Spec.ClusterIP) {
t.Errorf("unexpected side effect: ip unallocated") t.Errorf("unexpected side effect: ip unallocated")
} }
@ -883,7 +905,7 @@ func TestServiceRegistryDeleteDryRun(t *testing.T) {
if createdSvc.Spec.Ports[0].NodePort == 0 { if createdSvc.Spec.Ports[0].NodePort == 0 {
t.Fatalf("expected NodePort to be set") t.Fatalf("expected NodePort to be set")
} }
if !storage.serviceNodePorts.Has(int(createdSvc.Spec.Ports[0].NodePort)) { if !portIsAllocated(t, storage.serviceNodePorts, createdSvc.Spec.Ports[0].NodePort) {
t.Errorf("expected NodePort to be allocated") t.Errorf("expected NodePort to be allocated")
} }
@ -893,7 +915,7 @@ func TestServiceRegistryDeleteDryRun(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
if !storage.serviceNodePorts.Has(int(createdSvc.Spec.Ports[0].NodePort)) { if !portIsAllocated(t, storage.serviceNodePorts, createdSvc.Spec.Ports[0].NodePort) {
t.Errorf("unexpected side effect: NodePort unallocated") t.Errorf("unexpected side effect: NodePort unallocated")
} }
} }
@ -921,7 +943,7 @@ func TestDualStackServiceRegistryDeleteDryRun(t *testing.T) {
t.Fatalf("Expected no error: %v", err) t.Fatalf("Expected no error: %v", err)
} }
for i, family := range dualstack_svc.Spec.IPFamilies { for i, family := range dualstack_svc.Spec.IPFamilies {
if !dualstack_storage.serviceIPAllocatorsByFamily[family].Has(net.ParseIP(dualstack_svc.Spec.ClusterIPs[i])) { if !ipIsAllocated(t, dualstack_storage.serviceIPAllocatorsByFamily[family], dualstack_svc.Spec.ClusterIPs[i]) {
t.Errorf("unexpected side effect: ip unallocated %v", dualstack_svc.Spec.ClusterIPs[i]) t.Errorf("unexpected side effect: ip unallocated %v", dualstack_svc.Spec.ClusterIPs[i])
} }
} }
@ -1225,7 +1247,7 @@ func TestServiceRegistryIPAllocation(t *testing.T) {
testIPs := []string{"1.2.3.93", "1.2.3.94", "1.2.3.95", "1.2.3.96"} testIPs := []string{"1.2.3.93", "1.2.3.94", "1.2.3.95", "1.2.3.96"}
testIP := "not-an-ip" testIP := "not-an-ip"
for _, ip := range testIPs { for _, ip := range testIPs {
if !storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].(*ipallocator.Range).Has(net.ParseIP(ip)) { if !ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].(*ipallocator.Range), ip) {
testIP = ip testIP = ip
break break
} }
@ -1314,7 +1336,7 @@ func TestServiceRegistryIPUpdate(t *testing.T) {
testIPs := []string{"1.2.3.93", "1.2.3.94", "1.2.3.95", "1.2.3.96"} testIPs := []string{"1.2.3.93", "1.2.3.94", "1.2.3.95", "1.2.3.96"}
testIP := "" testIP := ""
for _, ip := range testIPs { for _, ip := range testIPs {
if !storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].(*ipallocator.Range).Has(net.ParseIP(ip)) { if !ipIsAllocated(t, storage.serviceIPAllocatorsByFamily[storage.defaultServiceIPFamily].(*ipallocator.Range), ip) {
testIP = ip testIP = ip
break break
} }
@ -1660,8 +1682,7 @@ func TestInitClusterIP(t *testing.T) {
family = api.IPv6Protocol family = api.IPv6Protocol
} }
allocator := storage.serviceIPAllocatorsByFamily[family] allocator := storage.serviceIPAllocatorsByFamily[family]
// has retruns true if it was allocated *sigh*.. if !ipIsAllocated(t, allocator, ip) {
if !allocator.Has(net.ParseIP(ip)) {
t.Fatalf("expected ip:%v to be allocated by %v allocator. it was not", ip, family) t.Fatalf("expected ip:%v to be allocated by %v allocator. it was not", ip, family)
} }
} }
@ -2107,8 +2128,7 @@ func TestServiceUpgrade(t *testing.T) {
for i, family := range updatedSvc.Spec.IPFamilies { for i, family := range updatedSvc.Spec.IPFamilies {
ip := updatedSvc.Spec.ClusterIPs[i] ip := updatedSvc.Spec.ClusterIPs[i]
allocator := storage.serviceIPAllocatorsByFamily[family] allocator := storage.serviceIPAllocatorsByFamily[family]
// has retruns true if it was allocated *sigh*.. if !ipIsAllocated(t, allocator, ip) {
if !allocator.Has(net.ParseIP(ip)) {
t.Fatalf("expected ip:%v to be allocated by %v allocator. it was not", ip, family) t.Fatalf("expected ip:%v to be allocated by %v allocator. it was not", ip, family)
} }
} }
@ -2241,7 +2261,7 @@ func TestServiceDowngrade(t *testing.T) {
releasedIPFamily := copySvc.Spec.IPFamilies[1] releasedIPFamily := copySvc.Spec.IPFamilies[1]
allocator := storage.serviceIPAllocatorsByFamily[releasedIPFamily] allocator := storage.serviceIPAllocatorsByFamily[releasedIPFamily]
if allocator.Has(net.ParseIP(releasedIP)) { if ipIsAllocated(t, allocator, releasedIP) {
t.Fatalf("expected ip:%v to be released by %v allocator. it was not", releasedIP, releasedIPFamily) t.Fatalf("expected ip:%v to be released by %v allocator. it was not", releasedIP, releasedIPFamily)
} }
} }