From 0406c63fb480e10bc323ec40f22d7353deb376af Mon Sep 17 00:00:00 2001
From: camilamacedo86 <camilamacedo86@gmail.com>
Date: Sat, 3 Jun 2023 07:50:35 +0100
Subject: [PATCH] fix adding missing pod

---
 addons/flannel/0.21.5/install.sh        | 52 ++++++++++++++++++++++++-
 addons/flannel/template/base/install.sh | 52 ++++++++++++++++++++++++-
 2 files changed, 100 insertions(+), 4 deletions(-)

diff --git a/addons/flannel/0.21.5/install.sh b/addons/flannel/0.21.5/install.sh
index 1fc1cecb39..71f5c95d82 100644
--- a/addons/flannel/0.21.5/install.sh
+++ b/addons/flannel/0.21.5/install.sh
@@ -375,8 +375,8 @@ spec:
         topologyKey: "kubernetes.io/hostname"
 EOF
     else
-    echo "Create pod to do the test without use podAntiAffinity because has only one node"
-    kubectl apply -f - <<EOF
+      echo "Create pod to do the test without use podAntiAffinity because has only one node"
+      kubectl apply -f - <<EOF
 apiVersion: v1
 kind: Pod
 metadata:
@@ -392,6 +392,54 @@ spec:
 EOF
     fi
 
+    # Wait for udp-receiver to get Running
+    if ! spinner_until 120 $(kubectl get pod udp-receiver -o 'jsonpath={..status.phase}') != "Running" ]]; then
+      logWarn "Timeout waiting upd-receiver Pod to be Running"
+    fi
+
+    # Get the IP of udp-receiver
+    receiver_ip=$(kubectl get pod udp-receiver -o jsonpath="{.status.podIP}")
+
+    if [[ $num_nodes -gt 1 ]]; then
+      echo "Creating udp-sender with podAntiAffinity"
+      kubectl apply -f - <<EOF
+apiVersion: v1
+kind: Pod
+metadata:
+  name: udp-sender
+  labels:
+    app: udp-sender
+spec:
+  containers:
+  - name: udp-sender
+    image: busybox
+    command: ["sh", "-c", "echo 'Test message' | nc -u $receiver_ip 8472"]
+  affinity:
+    podAntiAffinity:
+      requiredDuringSchedulingIgnoredDuringExecution:
+      - labelSelector:
+          matchExpressions:
+          - key: app
+            operator: In
+            values:
+            - udp-receiver
+        topologyKey: "kubernetes.io/hostname"
+EOF
+    else
+      echo "Creating udp-sender without podAntiAffinity"
+      kubectl apply -f - <<EOF
+apiVersion: v1
+kind: Pod
+metadata:
+  name: udp-sender
+spec:
+  containers:
+  - name: udp-sender
+    image: busybox
+    command: ["sh", "-c", "echo 'Test message' | nc -u $receiver_ip 8472"]
+EOF
+    fi
+
     echo "Waiting up to 2 minutes to check pods created to perform the test are successfully running"
     if ! spinner_until 120 udp_pods_are_running; then
         logWarn "Pods to check UDP communication with Flannel did not become Ready within the expected time."
diff --git a/addons/flannel/template/base/install.sh b/addons/flannel/template/base/install.sh
index 1fc1cecb39..71f5c95d82 100644
--- a/addons/flannel/template/base/install.sh
+++ b/addons/flannel/template/base/install.sh
@@ -375,8 +375,8 @@ spec:
         topologyKey: "kubernetes.io/hostname"
 EOF
     else
-    echo "Create pod to do the test without use podAntiAffinity because has only one node"
-    kubectl apply -f - <<EOF
+      echo "Create pod to do the test without use podAntiAffinity because has only one node"
+      kubectl apply -f - <<EOF
 apiVersion: v1
 kind: Pod
 metadata:
@@ -392,6 +392,54 @@ spec:
 EOF
     fi
 
+    # Wait for udp-receiver to get Running
+    if ! spinner_until 120 $(kubectl get pod udp-receiver -o 'jsonpath={..status.phase}') != "Running" ]]; then
+      logWarn "Timeout waiting upd-receiver Pod to be Running"
+    fi
+
+    # Get the IP of udp-receiver
+    receiver_ip=$(kubectl get pod udp-receiver -o jsonpath="{.status.podIP}")
+
+    if [[ $num_nodes -gt 1 ]]; then
+      echo "Creating udp-sender with podAntiAffinity"
+      kubectl apply -f - <<EOF
+apiVersion: v1
+kind: Pod
+metadata:
+  name: udp-sender
+  labels:
+    app: udp-sender
+spec:
+  containers:
+  - name: udp-sender
+    image: busybox
+    command: ["sh", "-c", "echo 'Test message' | nc -u $receiver_ip 8472"]
+  affinity:
+    podAntiAffinity:
+      requiredDuringSchedulingIgnoredDuringExecution:
+      - labelSelector:
+          matchExpressions:
+          - key: app
+            operator: In
+            values:
+            - udp-receiver
+        topologyKey: "kubernetes.io/hostname"
+EOF
+    else
+      echo "Creating udp-sender without podAntiAffinity"
+      kubectl apply -f - <<EOF
+apiVersion: v1
+kind: Pod
+metadata:
+  name: udp-sender
+spec:
+  containers:
+  - name: udp-sender
+    image: busybox
+    command: ["sh", "-c", "echo 'Test message' | nc -u $receiver_ip 8472"]
+EOF
+    fi
+
     echo "Waiting up to 2 minutes to check pods created to perform the test are successfully running"
     if ! spinner_until 120 udp_pods_are_running; then
         logWarn "Pods to check UDP communication with Flannel did not become Ready within the expected time."