Fix flakey goroutine test

99designs · Nov 26, 2018 · 8a5039d · 8a5039d
1 parent 4b08251
commit 8a5039d
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())
 		})