fixup! Ensure upgraded streams can be paused, cancelled, and aborted …

…and add more tests
aspnet · Apr 18, 2017 · d8fc979 · d8fc979
1 parent f4eb808
commit d8fc979
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/src/Microsoft.AspNetCore.Server.Kestrel.Core/Internal/Http/Frame.cs b/src/Microsoft.AspNetCore.Server.Kestrel.Core/Internal/Http/Frame.cs
@@ -326,6 +326,12 @@ public void InitializeStreams(MessageBody messageBody)
             _frameStreams.Start(messageBody);
         }
 
+        public void PauseStreams() => _frameStreams.Pause();
+
+        public void ResumeStreams() => _frameStreams.Resume();
+
+        public void StopStreams() => _frameStreams.Stop();
+
         public void Reset()
         {
             FrameRequestHeaders?.Reset();
@@ -761,12 +767,6 @@ public Task InitializeResponse(int firstWriteByteCount)
             return TaskCache.CompletedTask;
         }
 
-        public void PauseStreams() => _frameStreams.Pause();
-
-        public void ResumeStreams() => _frameStreams.Resume();
-
-        public void StopStreams() => _frameStreams.Stop();
-
         private async Task InitializeResponseAwaited(int firstWriteByteCount)
         {
             await FireOnStarting();

diff --git a/test/shared/TestConnection.cs b/test/shared/TestConnection.cs
@@ -2,15 +2,13 @@
 // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
 
 using System;
-using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO;
 using System.Net;
 using System.Net.Sockets;
 using System.Text;
 using System.Threading.Tasks;
 using Xunit;
-using Xunit.Sdk;
 
 namespace Microsoft.AspNetCore.Testing
 {