From 16d8e6400e5b892055be8d981369bc685c94b376 Mon Sep 17 00:00:00 2001
From: gmarek <gmarek@google.com>
Date: Wed, 19 Oct 2016 12:37:17 +0200
Subject: [PATCH] Allow setting master root disk size in kubemark

---
 cluster/kubemark/config-default.sh | 1 +
 test/kubemark/start-kubemark.sh    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/cluster/kubemark/config-default.sh b/cluster/kubemark/config-default.sh
index ec9831fb66..4158d1fde5 100644
--- a/cluster/kubemark/config-default.sh
+++ b/cluster/kubemark/config-default.sh
@@ -28,6 +28,7 @@ NUM_NODES=${NUM_NODES:-100}
 MASTER_SIZE=${MASTER_SIZE:-n1-standard-$(get-master-size)}
 MASTER_DISK_TYPE=pd-ssd
 MASTER_DISK_SIZE=${MASTER_DISK_SIZE:-20GB}
+MASTER_ROOT_DISK_SIZE=${KUBEMARK_MASTER_ROOT_DISK_SIZE:-10GB}
 REGISTER_MASTER_KUBELET=${REGISTER_MASTER:-false}
 PREEMPTIBLE_NODE=${PREEMPTIBLE_NODE:-false}
 
diff --git a/test/kubemark/start-kubemark.sh b/test/kubemark/start-kubemark.sh
index 18f8b4df4b..046f714b72 100755
--- a/test/kubemark/start-kubemark.sh
+++ b/test/kubemark/start-kubemark.sh
@@ -97,6 +97,7 @@ run-gcloud-compute-with-retries instances create "${MASTER_NAME}" \
   --tags "${MASTER_TAG}" \
   --network "${NETWORK}" \
   --scopes "storage-ro,compute-rw,logging-write" \
+  --boot-disk-size "${MASTER_ROOT_DISK_SIZE}" \
   --disk "name=${MASTER_NAME}-pd,device-name=master-pd,mode=rw,boot=no,auto-delete=no"
 
 run-gcloud-compute-with-retries firewall-rules create "${INSTANCE_PREFIX}-kubemark-master-https" \