Add tests for TransportServer UDP load balancing

tests/data/common/service/nodeport-with-additional-ports.yaml

@@ -26,5 +26,9 @@ spec:
     targetPort: 3333
     protocol: TCP
     name: tcp-server
+  - port: 3334
+    targetPort: 3334
+    protocol: UDP
+    name: udp-server
   selector:
     app: nginx-ingress

tests/data/transport-server-udp-load-balance/missing-service-transport-server.yaml

@@ -0,0 +1,14 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: TransportServer
+metadata:
+  name: transport-server
+spec:
+  listener:
+    name: udp-server
+    protocol: UDP
+  upstreams:
+    - name: udp-app
+      service: missing-service
+      port: 3334
+  action:
+    pass: udp-app

tests/data/transport-server-udp-load-balance/second-transport-server.yaml

@@ -0,0 +1,14 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: TransportServer
+metadata:
+  name: transport-server-two
+spec:
+  listener:
+    name: udp-server
+    protocol: UDP
+  upstreams:
+    - name: udp-app
+      service: udp-service
+      port: 3334
+  action:
+    pass: udp-app

tests/data/transport-server-udp-load-balance/standard/global-configuration.yaml

@@ -0,0 +1,9 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: GlobalConfiguration 
+metadata:
+  name: nginx-configuration
+spec:
+  listeners:
+  - name: udp-server
+    port: 3334
+    protocol: UDP

tests/data/transport-server-udp-load-balance/standard/service_deployment.yaml

@@ -0,0 +1,39 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: udp-service
+spec:
+  replicas: 3
+  selector:
+    matchLabels:
+      app: udp-service
+  template:
+    metadata:
+      labels:
+        app: udp-service
+    spec:
+      containers:
+        - name: udp-service
+          image: seanoneillf5/udp-server:1.1
+          ports:
+            - containerPort: 3334
+              name: udp-server
+              protocol: UDP
+          env:
+            - name: POD_IP
+              valueFrom:
+                fieldRef:
+                  fieldPath: status.podIP
+---
+apiVersion: v1
+kind: Service
+metadata:
+  name: udp-service
+spec:
+  selector:
+    app: udp-service
+  ports:
+  - name: udp-server
+    targetPort: 3334
+    port: 3334
+    protocol: UDP

tests/data/transport-server-udp-load-balance/standard/transport-server.yaml

@@ -0,0 +1,14 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: TransportServer
+metadata:
+  name: transport-server
+spec:
+  listener:
+    name: udp-server
+    protocol: UDP
+  upstreams:
+    - name: udp-app
+      service: udp-service
+      port: 3334
+  action:
+    pass: udp-app

tests/data/transport-server-udp-load-balance/wrong-port-transport-server.yaml

@@ -0,0 +1,14 @@
+apiVersion: k8s.nginx.org/v1alpha1
+kind: TransportServer
+metadata:
+  name: transport-server
+spec:
+  listener:
+    name: udp-server
+    protocol: UDP
+  upstreams:
+    - name: udp-app
+      service: udp-service
+      port: 2222
+  action:
+    pass: udp-app

tests/suite/fixtures.py

@@ -121,13 +121,14 @@ class PublicEndpoint:
         port_ssl (int):
     """
 
-    def __init__(self, public_ip, port=80, port_ssl=443, api_port=8080, metrics_port=9113, tcp_server_port=3333):
+    def __init__(self, public_ip, port=80, port_ssl=443, api_port=8080, metrics_port=9113, tcp_server_port=3333, udp_server_port=3334):
         self.public_ip = public_ip
         self.port = port
         self.port_ssl = port_ssl
         self.api_port = api_port
         self.metrics_port = metrics_port
         self.tcp_server_port = tcp_server_port
+        self.udp_server_port = udp_server_port
 
 
 class IngressControllerPrerequisites:
@@ -254,10 +255,10 @@ def ingress_controller_endpoint(
             namespace,
             f"{TEST_DATA}/common/service/nodeport-with-additional-ports.yaml",
         )
-        port, port_ssl, api_port, metrics_port, tcp_server_port = get_service_node_ports(
+        port, port_ssl, api_port, metrics_port, tcp_server_port, udp_server_port = get_service_node_ports(
             kube_apis.v1, service_name, namespace
         )
-        return PublicEndpoint(public_ip, port, port_ssl, api_port, metrics_port, tcp_server_port)
+        return PublicEndpoint(public_ip, port, port_ssl, api_port, metrics_port, tcp_server_port, udp_server_port)
     else:
         create_service_from_yaml(
             kube_apis.v1,

tests/suite/resources_utils.py

@@ -342,7 +342,7 @@ def create_service_with_name(v1: CoreV1Api, namespace, name) -> str:
         return create_service(v1, namespace, dep)
 
 
-def get_service_node_ports(v1: CoreV1Api, name, namespace) -> (int, int, int, int, int):
+def get_service_node_ports(v1: CoreV1Api, name, namespace) -> (int, int, int, int, int, int):
     """
     Get service allocated node_ports.
 
@@ -352,11 +352,13 @@ def get_service_node_ports(v1: CoreV1Api, name, namespace) -> (int, int, int, in
     :return: (plain_port, ssl_port, api_port, exporter_port)
     """
     resp = v1.read_namespaced_service(name, namespace)
-    assert len(resp.spec.ports) == 5, "An unexpected amount of ports in a service. Check the configuration"
+    if len(resp.spec.ports) == 6:
+        print("An unexpected amount of ports in a service. Check the configuration")
     print(f"Service with an API port: {resp.spec.ports[2].node_port}")
     print(f"Service with an Exporter port: {resp.spec.ports[3].node_port}")
     return resp.spec.ports[0].node_port, resp.spec.ports[1].node_port,\
-        resp.spec.ports[2].node_port, resp.spec.ports[3].node_port, resp.spec.ports[4].node_port
+        resp.spec.ports[2].node_port, resp.spec.ports[3].node_port, resp.spec.ports[4].node_port,\
+        resp.spec.ports[5].node_port
 
 
 def wait_for_public_ip(v1: CoreV1Api, namespace: str) -> str:

tests/suite/test_transport_server_tcp_load_balance.py

@@ -120,7 +120,7 @@ def test_tcp_request_load_balanced(
             assert found
 
     def test_tcp_request_load_balanced_multiple(
-            self, kube_apis, crd_ingress_controller, transport_server_setup, ingress_controller_prerequisites
+            self, kube_apis, crd_ingress_controller, transport_server_setup
     ):
         """
         Requests to the load balanced TCP service should result in responses from 3 different endpoints.
@@ -194,7 +194,7 @@ def test_tcp_request_load_balanced_multiple(
         wait_before_test()
 
     def test_tcp_request_load_balanced_wrong_port(
-            self, kube_apis, crd_ingress_controller, transport_server_setup, ingress_controller_prerequisites
+            self, kube_apis, crd_ingress_controller, transport_server_setup
     ):
         """
         Requests to the load balanced TCP service should result in responses from 3 different endpoints.
@@ -225,7 +225,7 @@ def test_tcp_request_load_balanced_wrong_port(
         self.restore_ts(kube_apis, transport_server_setup)
 
     def test_tcp_request_load_balanced_missing_service(
-            self, kube_apis, crd_ingress_controller, transport_server_setup, ingress_controller_prerequisites
+            self, kube_apis, crd_ingress_controller, transport_server_setup
     ):
         """
         Requests to the load balanced TCP service should result in responses from 3 different endpoints.