Merge pull request 99designs#446 from 99designs/fix-flakey-test

Fix flakey goroutine test
cgxxv · Nov 26, 2018 · 856740b · 856740b
2 parents df1b571 + 5fb4c6d
commit 856740b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/codegen/testserver/generated_test.go b/codegen/testserver/generated_test.go
@@ -141,7 +141,10 @@ func TestGeneratedServer(t *testing.T) {
 			sub.Close()
 
 			// need a little bit of time for goroutines to settle
-			time.Sleep(200 * time.Millisecond)
+			start := time.Now()
+			for time.Since(start).Seconds() < 2 && initialGoroutineCount != runtime.NumGoroutine() {
+				time.Sleep(5 * time.Millisecond)
+			}
 
 			require.Equal(t, initialGoroutineCount, runtime.NumGoroutine())
 		})