feat(scaletest/templates/scaletest-runner): scale provisioners up and down

johnstcn · johnstcn · commit aaf16d864a2e · 2023-10-10T13:17:57.000Z
diff --git a/scaletest/templates/scaletest-runner/scripts/cleanup.sh b/scaletest/templates/scaletest-runner/scripts/cleanup.sh
@@ -28,6 +28,12 @@ coder exp scaletest cleanup \
 	tee "${SCALETEST_RESULTS_DIR}/cleanup-${event}.txt"
 end_phase
 
+if [[ $event != prepare ]]; then
+	start_phase "Scaling down provisioners..."
+	maybedryrun "$DRY_RUN" kubectl scale deployment/coder-provisioner --replicas 1
+	maybedryrun "$DRY_RUN" kubectl rollout status deployment/coder-provisioner
+fi
+
 if [[ $event = manual ]]; then
 	echo 'Press any key to continue...'
 	read -s -r -n 1
diff --git a/scaletest/templates/scaletest-runner/scripts/prepare.sh b/scaletest/templates/scaletest-runner/scripts/prepare.sh
@@ -51,3 +51,9 @@ log "Cleaning up from previous runs (if applicable)..."
 "${SCRIPTS_DIR}/cleanup.sh" "prepare"
 
 log "Preparation complete!"
+
+log "Scaling up provisioners to ${SCALETEST_PARAM_CREATE_CONCURRENCY}..."
+maybedryrun "$DRY_RUN" kubectl scale deployment/coder-provisioner \
+	--replicas "${SCALETEST_PARAM_CREATE_CONCURRENCY}"
+log "Waiting for provisioners to scale up..."
+maybedryrun "$DRY_RUN" kubectl rollout status deployment/coder-provisioner
