public void ScriptInjectionFilterStream_BecomesPassthroughOnTimeout() { // Arrange MockSocketAdapter serverSocket = new MockSocketAdapter(); MockScriptInjectionFilterContext filterContext = new MockScriptInjectionFilterContext(); ScriptInjectionFilterStream filterStream = CreateFilterStream(serverSocket, filterContext); byte[] bytesToSend = Encoding.UTF8.GetBytes("<html><head></head><body></body></html>"); // Act I - Send some bytes Task writeTask = filterStream.WriteAsync(bytesToSend, 0, 20); // Assert TaskAssert.NotCompleted(writeTask, "Should be waiting for server to respond"); // Act II - Wait for the request to time out System.Threading.Thread.Sleep(1100); // Assert TaskAssert.Completed(writeTask, "Write should complete when server fails to respond"); Assert.True(serverSocket.IsClosed, "Should become passthrough"); Assert.Equal("<html><head></head><", filterContext.GetResponseBody(Encoding.UTF8)); AssertWithMessage.Equal(true, filterStream.ScriptInjectionTimedOut, "ScriptInjectionTimedOut"); }
public void ScriptInjectionFilterStream_ExceptionOnSubsequentWrite() { // Arrange MockSocketAdapter serverSocket = new MockSocketAdapter(); serverSocket.SendString("HTTP/1.1 200 OK\r\n", Encoding.ASCII); MockScriptInjectionFilterContext filterContext = new MockScriptInjectionFilterContext(); ScriptInjectionFilterStream filterStream = CreateFilterStream(serverSocket, filterContext); byte[] bytesToSend = Encoding.UTF8.GetBytes("<html><head></head><body></body></html>"); // Act I - Write some bytes filterStream.Write(bytesToSend, 0, 20); // Assert Assert.Contains("<html><head></head><", serverSocket.SentContent); AssertWithMessage.Equal(0, filterContext.GetResponseBody(Encoding.UTF8).Length, "No content should be sent to the response yet."); Assert.False(serverSocket.IsClosed, "Server connection should not have been closed."); // Act II - Attempt to write more bytes serverSocket.ThrowExceptionOnNextSendAsync(); Task result = filterStream.WriteAsync(bytesToSend, 20, bytesToSend.Length - 20); // Assert TaskAssert.Faulted(result, "Exception should have been re-thrown"); Assert.Equal("SendAsync after ThrowExceptionOnNextSendAsync was called.", result.Exception.InnerException.Message); }
public void ScriptInjectionFilterStream_CompleteFilterProcess() { // Arrange MockSocketAdapter serverSocket = new MockSocketAdapter(); MockScriptInjectionFilterContext filterContext = new MockScriptInjectionFilterContext(); ScriptInjectionFilterStream filterStream = CreateFilterStream(serverSocket, filterContext); byte[] bytesToSend = Encoding.UTF8.GetBytes("<html><head></head><body></body></html>"); // Act I - Write some bytes Task writeTask = filterStream.WriteAsync(bytesToSend, 0, 20); // Assert TaskAssert.NotCompleted(writeTask, "Write task should not complete until a successful response comes from the server."); // Act Ia - Response from the server serverSocket.SendString("HTTP/1.1 200 OK\r\n", Encoding.ASCII); // Assert TaskAssert.Completed(writeTask, "The write task should complete after we receive the response line from the server."); Assert.Contains("<html><head></head><", serverSocket.SentContent); AssertWithMessage.Equal(0, filterContext.GetResponseBody(Encoding.UTF8).Length, "No content should be sent to the response yet."); Assert.False(serverSocket.IsClosed, "Server connection should not have been closed."); // Act II - Write some more bytes filterStream.Write(bytesToSend, 20, bytesToSend.Length - 20); // Assert Assert.Contains("body></body></html>", serverSocket.SentContent); AssertWithMessage.Equal(0, filterContext.GetResponseBody(Encoding.UTF8).Length, "No content should be sent to the response yet."); Assert.False(serverSocket.IsClosed, "Server connection should not have been closed."); // Act III - Wait for complete Task completeTask = filterStream.FlushAsync(); // Assert TaskAssert.NotCompleted(completeTask, "Before response from server."); // Act IV - Send response from server serverSocket.SendString("Content-Length: 17\r\n\r\nXFiltered content", Encoding.ASCII); // Assert Assert.Equal("Filtered content", filterContext.GetResponseBody(Encoding.ASCII)); TaskAssert.Completed(completeTask, "After response from server."); Assert.False(filterStream.ScriptInjectionTimedOut); }
public void ScriptInjectionFilterStream_ExceptionOnFirstWrite() { // Arrange MockSocketAdapter serverSocket = new MockSocketAdapter(); MockScriptInjectionFilterContext filterContext = new MockScriptInjectionFilterContext(); ScriptInjectionFilterStream filterStream = CreateFilterStream(serverSocket, filterContext); byte[] bytesToSend = Encoding.UTF8.GetBytes("<html><head></head><body></body></html>"); // Act - Attempt to write some bytes serverSocket.ThrowExceptionOnNextSendAsync(); Task writeTask = filterStream.WriteAsync(bytesToSend, 0, 20); // Assert TaskAssert.NotFaulted(writeTask, "Task should not be faulted. Filter should switch to passthrough mode."); Assert.Equal("<html><head></head><", filterContext.GetResponseBody(Encoding.UTF8)); Assert.True(serverSocket.IsClosed, "Server connection should be closed."); }
public void ScriptInjectionFilterStream_ErrorResponseCodeFromServer() { // Arrange MockSocketAdapter serverSocket = new MockSocketAdapter(); MockScriptInjectionFilterContext filterContext = new MockScriptInjectionFilterContext(); ScriptInjectionFilterStream filterStream = CreateFilterStream(serverSocket, filterContext); byte[] bytesToSend = Encoding.UTF8.GetBytes("<html><head></head><body></body></html>"); // Act - Attempt to write some bytes Task writeTask = filterStream.WriteAsync(bytesToSend, 0, 20); // Assert TaskAssert.NotCompleted(writeTask, "Task should not complete until the response line is read."); // Act - Write an error response serverSocket.SendString("HTTP/1.1 404 Not Found\r\n", Encoding.ASCII); // Assert TaskAssert.Completed(writeTask, "Task should complete after the response line is received."); Assert.Equal("<html><head></head><", filterContext.GetResponseBody(Encoding.UTF8)); Assert.True(serverSocket.IsClosed, "Server connection should be closed."); }