From a303346dcca9e5dd9a975ef522098c92034a487d Mon Sep 17 00:00:00 2001
From: Antoni Zawodny <zawodny@google.com>
Date: Wed, 6 May 2020 15:48:55 +0200
Subject: [PATCH] Add option to resize kubemark node objects size

---
 test/kubemark/start-kubemark.sh | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/test/kubemark/start-kubemark.sh b/test/kubemark/start-kubemark.sh
index 87bbbddb89a..be2bf1c2505 100755
--- a/test/kubemark/start-kubemark.sh
+++ b/test/kubemark/start-kubemark.sh
@@ -225,6 +225,24 @@ function start-hollow-nodes {
   wait-for-hollow-nodes-to-run-or-timeout
 }
 
+# Annotates the node objects in the kubemark cluster to make their size
+# similar to regular nodes.
+# TODO(#90833): Replace this with image preloading from ClusterLoader to better
+# reflect the reality in kubemark tests.
+function resize-node-objects {
+  if [[ -z "$KUBEMARK_NODE_OBJECT_SIZE_BYTES" ]]; then
+    return 0
+  fi
+
+  annotation_size_bytes="${KUBEMARK_NODE_OBJECT_SIZE_BYTES}"
+  echo "Annotating node objects with ${annotation_size_bytes} byte label"
+  label=$( (< /dev/urandom tr -dc 'a-zA-Z0-9' | fold -w "$annotation_size_bytes"; true) | head -n 1)
+  "${KUBECTL}" --kubeconfig="${LOCAL_KUBECONFIG}" get nodes -o name \
+    | xargs -n10 -P100 -r -I% "${KUBECTL}" --kubeconfig="${LOCAL_KUBECONFIG}" annotate --overwrite % label="$label"
+  echo "Annotating node objects completed"
+}
+
+
 detect-project &> /dev/null
 create-kubemark-master
 
@@ -237,6 +255,7 @@ fi
 MASTER_IP=$(grep server "${HOLLOWNODE_KUBECONFIG}" | awk -F "/" '{print $3}')
 
 start-hollow-nodes
+resize-node-objects
 
 echo ""
 echo "Master IP: ${MASTER_IP}"