Removing delay, only using secondRequestReceived, adding default timeouts and using testUri

Author: MattyLeslie

src/SignalR/clients/csharp/Client/test/UnitTests/LongPollingTransportTests.cs

@@ -702,7 +702,6 @@ public async Task PollRequestsContainCorrectAcceptHeader()
         var responseTaskCompletionSource = new TaskCompletionSource<HttpResponseMessage>();
         var requestCount = 0;
         var allHeadersCorrect = true;
-        var firstRequestReceived = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
         var secondRequestReceived = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
 
         testHttpHandler.OnRequest(async (request, next, cancellationToken) =>
@@ -714,11 +713,7 @@ public async Task PollRequestsContainCorrectAcceptHeader()
 
             requestCount++;
 
-            if (requestCount == 1)
-            {
-                firstRequestReceived.SetResult();
-            }
-            else if (requestCount == 2)
+            if (requestCount == 2)
             {
                 secondRequestReceived.SetResult();
             }
@@ -743,28 +738,15 @@ public async Task PollRequestsContainCorrectAcceptHeader()
             var loggerFactory = NullLoggerFactory.Instance;
             var transport = new LongPollingTransport(httpClient, loggerFactory: loggerFactory);
 
-            var cts = new CancellationTokenSource();
-            var startTask = transport.StartAsync(new Uri("http://test.com"), TransferFormat.Text, cts.Token);
-
-            await firstRequestReceived.Task;
-
-            await Task.Delay(100);
-
-            cts.Cancel();
+            var startTask = transport.StartAsync(TestUri, TransferFormat.Text);
 
-            await secondRequestReceived.Task;
+            await secondRequestReceived.Task.DefaultTimeout();
 
             await transport.StopAsync();
 
-            if (!responseTaskCompletionSource.Task.IsCompleted)
-            {
-                responseTaskCompletionSource.TrySetResult(new HttpResponseMessage(HttpStatusCode.BadRequest));
-            }
-
             Assert.True(responseTaskCompletionSource.Task.IsCompleted);
             var response = await responseTaskCompletionSource.Task.DefaultTimeout();
             Assert.Equal(HttpStatusCode.OK, response.StatusCode);
         }
     }
-
 }