From db0a3721f0805009e33ed8751828d136a4e64d04 Mon Sep 17 00:00:00 2001
From: Shyam Jeedigunta <shyamjvs@google.com>
Date: Thu, 8 Feb 2018 15:58:51 +0100
Subject: [PATCH] Fix bug with profile-gathering waitgroup in scale tests

---
 test/e2e/scalability/density.go | 1 +
 test/e2e/scalability/load.go    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/test/e2e/scalability/density.go b/test/e2e/scalability/density.go
index 618e776abfc..2942dc62eea 100644
--- a/test/e2e/scalability/density.go
+++ b/test/e2e/scalability/density.go
@@ -363,6 +363,7 @@ var _ = SIGDescribe("Density", func() {
 		// Stop apiserver CPU profile gatherer and gather memory allocations profile.
 		close(profileGathererStopCh)
 		wg := sync.WaitGroup{}
+		wg.Add(1)
 		framework.GatherApiserverMemoryProfile(&wg, "density")
 		wg.Wait()
 
diff --git a/test/e2e/scalability/load.go b/test/e2e/scalability/load.go
index 2825f3f3084..0d539438955 100644
--- a/test/e2e/scalability/load.go
+++ b/test/e2e/scalability/load.go
@@ -104,6 +104,7 @@ var _ = SIGDescribe("Load capacity", func() {
 		// Stop apiserver CPU profile gatherer and gather memory allocations profile.
 		close(profileGathererStopCh)
 		wg := sync.WaitGroup{}
+		wg.Add(1)
 		framework.GatherApiserverMemoryProfile(&wg, "load")
 		wg.Wait()