Move common test code to function

PiperOrigin-RevId: 214890335
Change-Id: I42743f0ce46a5a42834133bce2f32d187194fc87

1 files changed, 4 insertions, 2 deletions

diff --git a/runsc/container/container_test.go b/runsc/container/container_test.go
index de1e50a3f..c71bcc46d 100644
--- a/runsc/container/container_test.go
+++ b/runsc/container/container_test.go
@@ -62,7 +62,8 @@ func waitForProcessList(cont *Container, want []*control.Process) error {
 		}
 		return nil
 	}
-	return testutil.Poll(cb, 5*time.Second)
+	// Gives plenty of time as tests can run slow under --race.
+	return testutil.Poll(cb, 30*time.Second)
 }
 
 func waitForProcessCount(cont *Container, want int) error {
@@ -77,7 +78,8 @@ func waitForProcessCount(cont *Container, want int) error {
 		}
 		return nil
 	}
-	return testutil.Poll(cb, 5*time.Second)
+	// Gives plenty of time as tests can run slow under --race.
+	return testutil.Poll(cb, 30*time.Second)
 }
 
 // procListsEqual is used to check whether 2 Process lists are equal for all