fix: send service ports on lb allocate

2026-08-01 07:25:02 +03:30
parent 10d741ceb6
commit 1ce335d3db
3 changed files with 83 additions and 0 deletions
@@ -0,0 +1,47 @@
+package controller
+
+import (
+	"testing"
+
+	corev1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+func TestServicePortsToAllocateRules(t *testing.T) {
+	svc := &corev1.Service{
+		ObjectMeta: metav1.ObjectMeta{Namespace: "default", Name: "web"},
+		Spec: corev1.ServiceSpec{Ports: []corev1.ServicePort{{Port: 80, Protocol: corev1.ProtocolTCP}, {Port: 53, Protocol: corev1.ProtocolUDP}}},
+	}
+
+	rules, err := servicePortsToAllocateRules(svc)
+	if err != nil {
+		t.Fatalf("servicePortsToAllocateRules: %v", err)
+	}
+	if len(rules) != 2 {
+		t.Fatalf("rules count = %d, want 2", len(rules))
+	}
+	if rules[0].Protocol != "tcp" || rules[0].PortFrom != 80 {
+		t.Fatalf("unexpected first rule: %+v", rules[0])
+	}
+	if rules[1].Protocol != "udp" || rules[1].PortFrom != 53 {
+		t.Fatalf("unexpected second rule: %+v", rules[1])
+	}
+}
+
+func TestServicePortsToAllocateRulesRejectsUnsupportedProtocol(t *testing.T) {
+	svc := &corev1.Service{
+		ObjectMeta: metav1.ObjectMeta{Namespace: "default", Name: "web"},
+		Spec: corev1.ServiceSpec{Ports: []corev1.ServicePort{{Port: 80, Protocol: corev1.ProtocolSCTP}}},
+	}
+
+	if _, err := servicePortsToAllocateRules(svc); err == nil {
+		t.Fatal("expected error for unsupported protocol")
+	}
+}
+
+func TestServicePortsToAllocateRulesRejectsNoPorts(t *testing.T) {
+	svc := &corev1.Service{ObjectMeta: metav1.ObjectMeta{Namespace: "default", Name: "web"}}
+	if _, err := servicePortsToAllocateRules(svc); err == nil {
+		t.Fatal("expected error for empty ports")
+	}
+}