From 762bfa3d97e6ea7cb0b933741589712c6d099af4 Mon Sep 17 00:00:00 2001
From: Wojciech Tyczynski <wojtekt@google.com>
Date: Tue, 19 Apr 2016 14:28:21 +0200
Subject: [PATCH] Improve script for running scheduler benchmarks

---
 .../scheduler/perf/test-performance.sh        | 23 +++++++++++--------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/test/component/scheduler/perf/test-performance.sh b/test/component/scheduler/perf/test-performance.sh
index c8c8a4b39af..5c387a8f1ea 100755
--- a/test/component/scheduler/perf/test-performance.sh
+++ b/test/component/scheduler/perf/test-performance.sh
@@ -18,13 +18,16 @@ set -o errexit
 set -o nounset
 set -o pipefail
 
-pushd "../../../.."
-source "./hack/lib/util.sh"
-source "./cluster/lib/logging.sh"
-source "./hack/lib/etcd.sh"
-popd
+KUBE_ROOT=$(dirname "${BASH_SOURCE}")/../../../..
+source "${KUBE_ROOT}/hack/lib/init.sh"
+
+kube::golang::setup_env
+
+DIR_BASENAME=$(dirname "${BASH_SOURCE}")
+pushd ${DIR_BASENAME}
 
 cleanup() {
+  popd 2> /dev/null
   kube::etcd::cleanup
   kube::log::status "performance test cleanup complete"
 }
@@ -34,13 +37,13 @@ trap cleanup EXIT
 kube::etcd::start
 kube::log::status "performance test start"
 
-# TODO: set log-dir and prof output dir.
-DIR_BASENAME=$(basename `pwd`)
-go test -c -o "${DIR_BASENAME}.test"
 # We are using the benchmark suite to do profiling. Because it only runs a few pods and
 # theoretically it has less variance.
-"./${DIR_BASENAME}.test" -test.bench=. -test.run=xxxx -test.cpuprofile=prof.out -logtostderr=false
+if ${RUN_BENCHMARK:-false}; then
+  go test -c -o "perf.test"
+  "./perf.test" -test.bench=. -test.run=xxxx -test.cpuprofile=prof.out
+fi
 kube::log::status "benchmark tests finished"
 # Running density tests. It might take a long time.
-"./${DIR_BASENAME}.test" -test.run=. -test.timeout=60m
+go test -test.run=. -test.timeout=60m
 kube::log::status "density tests finished"