From 27b4a057a15d9b5b9b5d73fd03d3454cccc48508 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marie=20P=C3=ADchov=C3=A1?= <mapichov@microsoft.com>
Date: Thu, 13 Aug 2020 19:24:23 +0200
Subject: [PATCH] Fix timing issues with Http2_PingKeepAlive test.

---
 .../HttpClientHandlerTest.Http2.cs              | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/libraries/System.Net.Http/tests/FunctionalTests/HttpClientHandlerTest.Http2.cs b/src/libraries/System.Net.Http/tests/FunctionalTests/HttpClientHandlerTest.Http2.cs
index 837e079fe1257b..f96409b9fc2430 100644
--- a/src/libraries/System.Net.Http/tests/FunctionalTests/HttpClientHandlerTest.Http2.cs
+++ b/src/libraries/System.Net.Http/tests/FunctionalTests/HttpClientHandlerTest.Http2.cs
@@ -1503,7 +1503,7 @@ await Http2LoopbackServer.CreateClientAndServerAsync(
                     }
 
                     // Let connection live until server finishes.
-                    await serverFinished.Task.TimeoutAfter(pingTimeout * 2);
+                    await Task.WhenAny(serverFinished.Task, Task.Delay(pingTimeout * 3));
                 },
                 async server =>
                 {
@@ -1534,7 +1534,10 @@ await Http2LoopbackServer.CreateClientAndServerAsync(
                         {
                             ping = await connection.ReadPingAsync(pingTimeout);
                         }
-                        await Task.Delay(pongDelay);
+                        if (pongDelay > TimeSpan.Zero)
+                        {
+                            await Task.Delay(pongDelay);
+                        }
 
                         await connection.SendPingAckAsync(ping.Data);
                     }
@@ -1555,6 +1558,16 @@ await Http2LoopbackServer.CreateClientAndServerAsync(
                     }
                     else
                     {
+                        // If the pings were recently coming, just give the connection time to clear up streams
+                        // and still accept one stray ping.
+                        if (expectStreamPing)
+                        {
+                            try
+                            {
+                                await connection.ReadPingAsync(pingTimeout);
+                            }
+                            catch (OperationCanceledException) { } // if it failed once, it will fail again
+                        }
                         await Assert.ThrowsAsync<OperationCanceledException>(() => connection.ReadPingAsync(pingTimeout));
                     }
                     serverFinished.SetResult();