Added unit test for brotli handling

Author: bstordrup

Diff for: UnitTest/HttpSessionWatcherTest/HttpSessionWatcherTests.cs

@@ -4,21 +4,21 @@
  * GPLv3 licensed, see LICENSE for full text
  * Commercial licensing available
  */
-using System.Collections.Generic;
-using System.Reflection;
-using SharpPcap;
-using SharpPcap.LibPcap;
+using log4net;
 using PacketDotNet;
 using PacketDotNet.Connections;
 using PacketDotNet.Connections.Http;
-using log4net;
+using SharpPcap;
+using SharpPcap.LibPcap;
+using System.Collections.Generic;
+using System.Reflection;
 using Xunit;
 
 namespace Test
 {
     public class HttpSessionWatcherTests
     {
-        private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);        
+        private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);
 
         private class ExpectedMessageContainer
         {
@@ -63,51 +63,54 @@ private void CheckMessage(HttpMessage httpMessage, ExpectedMessageContainer.Mess
             {
                 log.DebugFormat("expectedMessageIndex {0}", expectedMessageIndex);
 
-                if(expectedMessageIndex > expectedMessages.Count)
+                if (expectedMessageIndex > expectedMessages.Count)
                 {
                     throw new System.InvalidOperationException("expectedMessageIndex "
                                                                + expectedMessageIndex
                                                                + " > expectedMessages.Count"
                                                                + expectedMessages.Count);
                 }
-    
+
                 ExpectedMessageContainer expectedMessage = expectedMessages[expectedMessageIndex];
-    
+
                 Assert.True(expectedMessage.messageType == theMessageType, "Message type does not match");
 
                 // perform comparisons that are generic to all HttpMessages
                 Assert.Equal(expectedMessage.message.HttpVersion, httpMessage.HttpVersion);
 
                 // if we expected a zero length body then we expect that
                 // the actual message has a null body since it was never set
-                if(expectedMessage.message.Body.Length == 0)
+                if (expectedMessage.message.Body.Length == 0)
                 {
                     Assert.True(httpMessage.Body == null);
-                } else
+                }
+                else
                 {
                     Assert.Equal(expectedMessage.message.Body.Length, httpMessage.Body.Length);
                 }
 
-                if(theMessageType == ExpectedMessageContainer.MessageTypes.HttpRequest)
+                if (theMessageType == ExpectedMessageContainer.MessageTypes.HttpRequest)
                 {
                     HttpRequest expected = (HttpRequest)expectedMessage.message;
                     HttpRequest request = (HttpRequest)httpMessage;
-    
+
                     Assert.Equal(expected.Url, request.Url);
                     Assert.Equal(expected.Method, request.Method);
-                } else
+                }
+                else
                 {
                     HttpStatus expected = (HttpStatus)expectedMessage.message;
                     HttpStatus status = (HttpStatus)httpMessage;
-    
+
                     Assert.Equal(expected.StatusCode, status.StatusCode);
                 }
-    
+
                 // move to the next message
                 expectedMessageIndex++;
-    
+
                 log.Debug("message matched");
-            } catch(System.Exception e)
+            }
+            catch (System.Exception e)
             {
                 // catch exceptions because the HttpSessionMonitor code will drop them
 
@@ -124,15 +127,15 @@ private void ProcessPackets(ICaptureDevice dev,
 
             GetPacketStatus status;
             PacketCapture e;
-            while((status = dev.GetNextPacket(out e)) == GetPacketStatus.PacketRead)
+            while ((status = dev.GetNextPacket(out e)) == GetPacketStatus.PacketRead)
             {
                 var rawCapture = e.GetPacket();
                 var p = Packet.ParsePacket(rawCapture.LinkLayerType, rawCapture.Data);
 
                 var tcpPacket = p.Extract<TcpPacket>();
 
                 // skip non-tcp packets, http is a tcp based protocol
-                if(p == null)
+                if (p == null)
                 {
                     continue;
                 }
@@ -206,13 +209,13 @@ public void TestHttpPostSession()
             status.StatusCode = HttpStatus.StatusCodes.OK_200;
             status.HttpVersion = HttpMessage.HttpVersions.Http11;
             status.Body = new byte[1132];
-            expectedMessage.message = status;            
+            expectedMessage.message = status;
             expectedMessages.Add(expectedMessage);
 
             ProcessPackets(dev, tcpConnectionManager);
         }
 
-        void HandleTcpConnectionManagerOnConnectionFound (TcpConnection c)
+        void HandleTcpConnectionManagerOnConnectionFound(TcpConnection c)
         {
             var httpSessionWatcher = new HttpSessionWatcher(c,
                                                             OnHttpRequestFound,
@@ -221,6 +224,54 @@ void HandleTcpConnectionManagerOnConnectionFound (TcpConnection c)
         }
 #endif
 
+#if true
+
+        [Fact]
+        public void TestHttpGetSession()
+        {
+            log4net.Config.XmlConfigurator.Configure(LogManager.GetRepository(Assembly.GetCallingAssembly()));
+
+            string captureFilename = "../../../captureFiles/http_get_request_and_response_brotli_compressed.pcap";
+            var tcpConnectionManager = new TcpConnectionManager();
+            tcpConnectionManager.OnConnectionFound += HandleTcpConnectionManagerOnConnectionFound;
+
+            // open the offline file
+            var dev = new CaptureFileReaderDevice(captureFilename);
+            dev.Open();
+            Assert.True(dev != null, "failed to open " + captureFilename);
+
+            // setup the expected events
+            ExpectedMessageContainer expectedMessage;
+            HttpRequest request;
+            HttpStatus status;
+            expectedMessages = new List<ExpectedMessageContainer>();
+
+            // GET /api/areaInformation/?requestAction=GetMapMunicipalityList
+            expectedMessage = new ExpectedMessageContainer();
+            expectedMessage.messageType = ExpectedMessageContainer.MessageTypes.HttpRequest;
+            request = new HttpRequest();
+            request.Url = "/api/areaInformation/?requestAction=GetMapMunicipalityList";
+            request.Method = HttpRequest.Methods.Get;
+            request.HttpVersion = HttpMessage.HttpVersions.Http11;
+            request.Body = new byte[0];
+            expectedMessage.message = request;
+            expectedMessages.Add(expectedMessage);
+
+            // HTTP 1.1 OK
+            expectedMessage = new ExpectedMessageContainer();
+            expectedMessage.messageType = ExpectedMessageContainer.MessageTypes.HttpStatus;
+            status = new HttpStatus();
+            status.StatusCode = HttpStatus.StatusCodes.OK_200;
+            status.HttpVersion = HttpMessage.HttpVersions.Http11;
+            status.Body = new byte[8269];
+            expectedMessage.message = status;
+            expectedMessages.Add(expectedMessage);
+
+            ProcessPackets(dev, tcpConnectionManager);
+        }
+
+#endif
+
 #if false
         [Fact]
         public void TestHttpGetWithGzippedContent()
@@ -295,7 +346,7 @@ public void TestHttpPipelining()
             status.StatusCode = HttpStatus.StatusCodes.OK_200;
             status.HttpVersion = HttpMessage.HttpVersions.Http11;
             status.Body = new byte[1123];
-            expectedMessage.message = status;            
+            expectedMessage.message = status;
             expectedMessages.Add(expectedMessage);
 
             // GET /img/menu-secondary.gif
@@ -316,7 +367,7 @@ public void TestHttpPipelining()
             status.StatusCode = HttpStatus.StatusCodes.OK_200;
             status.HttpVersion = HttpMessage.HttpVersions.Http11;
             status.Body = new byte[227];
-            expectedMessage.message = status;            
+            expectedMessage.message = status;
             expectedMessages.Add(expectedMessage);